From ffd0a5a2a06ffe392b8461c368232b46e5955e4a Mon Sep 17 00:00:00 2001
From: "Timothy J. Baek" <timothyjrbeck@gmail.com>
Date: Sun, 7 Jan 2024 08:34:05 -0800
Subject: [PATCH] Update main.py

---
 backend/apps/rag/main.py | 4 +---
 1 file changed, 1 insertion(+), 3 deletions(-)

diff --git a/backend/apps/rag/main.py b/backend/apps/rag/main.py
index f9553a60..5bf74b09 100644
--- 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(
-            name=collection_name, embedding_function=EMBEDDING_FUNC
-        )
+        collection = CHROMA_CLIENT.create_collection(name=collection_name)
 
         collection.add(
             documents=texts, metadatas=metadatas, ids=[str(uuid.uuid1()) for _ in texts]