Update main.py

ffd0a5a2 · Timothy J. Baek · c68bb3b9 · ffd0a5a2
Commit ffd0a5a2 authored Jan 07, 2024 by Timothy J. Baek
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 3 deletions

backend/apps/rag/main.py backend/apps/rag/main.py +1 -3

No files found.
--- a/backend/apps/rag/main.py
+++ b/backend/apps/rag/main.py
@@ -64,9 +64,7 @@ def store_data_in_vector_db(data, collection_name) -> bool:
    metadatas = [doc.metadata for doc in docs]
    try:
-        collection = CHROMA_CLIENT.create_collection(
+        collection = CHROMA_CLIENT.create_collection(name=collection_name)
-            name=collection_name, embedding_function=EMBEDDING_FUNC
-        )
        collection.add(
            documents=texts, metadatas=metadatas, ids=[str(uuid.uuid1()) for _ in texts]