소스 검색

Update main.py

Timothy J. Baek 1 년 전
부모
커밋
ffd0a5a2a0
1개의 변경된 파일1개의 추가작업 그리고 3개의 파일을 삭제
  1. 1 3
      backend/apps/rag/main.py

+ 1 - 3
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]