fix: update chromaDB persist to resolve deprecation warning

crux-bphc · Jan 18, 2025 · 0b816bf · 0b816bf
1 parent 373b541
commit 0b816bf
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/src/campus_rag/create_database.py b/src/campus_rag/create_database.py
@@ -56,8 +56,7 @@ def save_to_chroma(chunks: list[Document]):
     if os.path.exists(CHROMA_PATH):
         shutil.rmtree(CHROMA_PATH)
 
-    db = Chroma.from_documents(chunks, embeddings, persist_directory=CHROMA_PATH)
-    db.persist()
+    Chroma.from_documents(chunks, embeddings, persist_directory=CHROMA_PATH)
     print(f"Saved {len(chunks)} chunks to {CHROMA_PATH}.")