GPT-Docker / app /exploration.py
heikowagner's picture
upload
1f84a9a
raw
history blame
524 Bytes
# %%
from utils import retrieve_collections, get_chroma_client
from load_model import load_embedding
#retrieve_collections()
client = get_chroma_client()
# %%
client.reset()
# %%
collections = tuple( [collection.name for collection in client.list_collections()] ) ##Keine Embedding function in der Collection angelegt...
ef = load_embedding("hkunlp/instructor-large")
collection="heikostest2"
client.create_collection(collection, embedding_function=ef, metadata={"loaded_docs":[]})
# %%
client.list_collections()