def main(): parser = argparse.ArgumentParser() parser.add_argument('dataset_path', type=str) args = parser.parse_args() index_client = milvus_client() conn = connect_mysql() cursor = conn.cursor() init_table(index_client, conn, cursor) insert_data(index_client, conn, cursor, args.dataset_path)
def init_conn(): conn = connect_mysql() cursor = conn.cursor() index_client = milvus_client() return index_client, conn, cursor