Update agent_langchain.py

agent_langchain.py

@@ -49,12 +49,13 @@ clf_model = AutoModelForSequenceClassification.from_pretrained(clf_model_name)
 # -------------------------------
 # Initialize ChromaDB Client for KB
 # -------------------------------
-chroma_client = chromadb.Client(Settings(
-    chroma_db_impl="duckdb+parquet",
-    persist_directory="/data/chroma_db"
-))
+# ✅ Use new API — persistent on Hugging Face writable directory
+chroma_client = chromadb.PersistentClient(path="/tmp/chroma")
 
-COLLECTION_NAME = "kb_collection"
+# ✅ Create or get your KB collection
+kb_collection = chroma_client.get_or_create_collection("KB")
+
+COLLECTION_NAME = "KB"
 try:
     kb_collection = chroma_client.get_collection(COLLECTION_NAME)
 except: