import pandas as pd import os import copy import chromadb import numpy as np client = chromadb.HttpClient(host='localhost', port=8000) collection = client.get_or_create_collection(name="qingdan") a = np.load("qd_embedding.npy") content = np.load("qd_content.npy") for i in range(len(a)): print(i) collection.add( ids=[content[i][0]], embeddings=[a[i]], documents=[content[i][1]], metadatas=[{"parent": content[i][2],}, ], )