Update retriever.py

retriever.py

@@ -1,8 +1,14 @@
 import chromadb
 from sentence_transformers import SentenceTransformer
+import os
+
+# Ensure the /tmp/data directory exists
+os.makedirs("/tmp/data", exist_ok=True)
 
 # Initialize ChromaDB
-chroma_client = chromadb.PersistentClient(path="data/chroma_db")
+# chroma_client = chromadb.PersistentClient(path="data/chroma_db")
+# Use /tmp/data as the database path
+chroma_client = chromadb.PersistentClient(path="/tmp/data/chroma_db")
 
 # Load embedding model
 embed_model = SentenceTransformer("sentence-transformers/all-MiniLM-L6-v2")