Update agent_langchain.py

agent_langchain.py

@@ -99,10 +99,11 @@ CHROMA_PATH = "/tmp/chroma"  # Persistent path in Space
 COLLECTION_NAME = "knowledge_base"
 
 kb_collection = None  # Global placeholder, initialized via /setup
-embedding_func = embedding_functions.OnnxMiniLM_L6_V2(
-    model_name="all-MiniLM-L6-v2",
-    download_path="/tmp/onnx_models"  # <-- writable path
+embedding_func = embedding_functions.SentenceTransformerEmbeddingFunction(
+    model_name="sentence-transformers/all-MiniLM-L6-v2"
+
 )
+
 kb_lock = threading.Lock()  # Ensure thread-safe updates
 
 def initialize_kb(json_texts: list, metadatas: list):