GPT-Docker / app /load_test.py
heikowagner's picture
upload
1f84a9a
raw
history blame
No virus
604 Bytes
# %%
# %%
import os
import pathlib
from load_model import load_embedding
from utils import get_chroma_client
from load_vectors import load_from_web, create_and_add, load_and_split
collection="axaterms"
client = get_chroma_client()
# Load collection to get metadata
loaded_collection = client.get_collection(collection)
# %%
model_name = loaded_collection.metadata['model_name']
# %%
print( loaded_collection.json() )
# %%
client.get_collection(collection).json() #add documents destroys the metadata... maybe :)
# %%
#loaded_collection.modify(metadata={"Test":99})
# %%
loaded_collection.json()