Pulastya0 commited on
Commit
e3cb604
·
verified ·
1 Parent(s): 1a82358

Update agent_langchain.py

Browse files
Files changed (1) hide show
  1. agent_langchain.py +3 -2
agent_langchain.py CHANGED
@@ -99,8 +99,9 @@ CHROMA_PATH = "/tmp/chroma" # Persistent path in Space
99
  COLLECTION_NAME = "knowledge_base"
100
 
101
  kb_collection = None # Global placeholder, initialized via /setup
102
- embedding_func = embedding_functions.SentenceTransformerEmbeddingFunction(
103
- model_name="sentence-transformers/all-MiniLM-L6-v2"
 
104
  )
105
  kb_lock = threading.Lock() # Ensure thread-safe updates
106
 
 
99
  COLLECTION_NAME = "knowledge_base"
100
 
101
  kb_collection = None # Global placeholder, initialized via /setup
102
+ embedding_func = embedding_functions.OnnxMiniLM_L6_V2(
103
+ model_name="all-MiniLM-L6-v2",
104
+ download_path="/tmp/onnx_models" # <-- writable path
105
  )
106
  kb_lock = threading.Lock() # Ensure thread-safe updates
107