Spaces:
Sleeping
Sleeping
Update agent_langchain.py
Browse files- 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.
|
| 103 |
-
model_name="
|
|
|
|
| 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 |
|