UniqAI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2025-01-30 14:29:55 +00:00
Code Issues Actions 4 Packages Projects Releases Wiki Activity

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]