Update ingest.py

ingest.py

@@ -1,16 +1,15 @@
 import os
 import logging
-import faiss
 from langchain.document_loaders import PDFMinerLoader
 from langchain.text_splitter import RecursiveCharacterTextSplitter
 from langchain.embeddings import HuggingFaceEmbeddings
-from langchain.vectorstores import FAISS
+from langchain.vectorstores import Chroma
 
 # Configure logging
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
-def create_faiss_index():
+def create_vector_store():
     documents = []
     docs_dir = "docs"
     if not os.path.exists(docs_dir):
@@ -55,11 +54,11 @@ def create_faiss_index():
         return
 
     try:
-        faiss_index = FAISS.from_documents(texts, embeddings)
-        faiss_index.save_local("faiss_index")
-        logger.info(f"Created FAISS index with {len(texts)} vectors.")
+        vector_store = Chroma.from_documents(texts, embeddings, persist_directory="./chroma_db")
+        vector_store.persist()
+        logger.info(f"Created Chroma vector store with {len(texts)} vectors.")
     except Exception as e:
-        logger.error(f"Failed to create FAISS index: {e}")
+        logger.error(f"Failed to create Chroma vector store: {e}")
 
 if __name__ == "__main__":
-    create_faiss_index()
+    create_vector_store()