Update RAG.py

RAG.py

@@ -48,14 +48,22 @@ def load_models_llm():
 #     vector_store = Chroma.from_documents(documents, embedding=embedding, client=client)
 #     return vector_store
 
+# def create_database(embedding, documents):
+#     # Use a local directory for Chroma database storage
+#     vector_store = Chroma.from_documents(
+#         documents,
+#         embedding=embedding)
+#     return vector_store
+
 def create_database(embedding, documents):
-    # Use a local directory for Chroma database storage
-    vector_store = Chroma.from_documents(
-        documents,
-        embedding=embedding)
+    settings = Settings(
+        chroma_db_impl="duckdb+parquet",
+        persist_directory=None
+    )
+    # Explicitly clean up and reinitialize Chroma client
+    vector_store = Chroma.from_documents(documents, embedding=embedding, client_settings=settings)
     return vector_store
 
-
 # def create_database(embedding, documents):
 #     vector_store = Chroma.from_documents(documents, embedding=embedding)
 #     return vector_store