tdpeuter/open-webui
tdpeuter/open-webui
1
Fork 0
forked from open-webui/open-webui
Code Issues Pull requests 1 Releases 2 Packages Activity Actions

Update main.py

Browse source
This commit is contained in:
Timothy J. Baek 2024-01-07 08:34:05 -08:00
parent c68bb3b950
commit ffd0a5a2a0
1 changed files with 1 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
backend/apps/rag/main.py
View file

@ -64,9 +64,7 @@ def store_data_in_vector_db(data, collection_name) -> bool:
metadatas = [doc.metadata for doc in docs] metadatas = [doc.metadata for doc in docs]
try: try:
collection = CHROMA_CLIENT.create_collection( collection = CHROMA_CLIENT.create_collection(name=collection_name)
name=collection_name, embedding_function=EMBEDDING_FUNC
)
collection.add( collection.add(
documents=texts, metadatas=metadatas, ids=[str(uuid.uuid1()) for _ in texts] documents=texts, metadatas=metadatas, ids=[str(uuid.uuid1()) for _ in texts]