qingdan/preprocess/saveEmbedding.py

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],}, ],

)