fallback to simple vs in windows system

2024-09-09 00:27:02 +08:00
parent ab32c314ab
commit 7ef39770c7
3 changed files with 12 additions and 5 deletions
--- a/crazy_functions/Rag_Interface.py
+++ b/crazy_functions/Rag_Interface.py
@@ -4,10 +4,14 @@ from crazy_functions.crazy_utils import request_gpt_model_in_new_thread_with_ui_

 VECTOR_STORE_TYPE = "Milvus"

+if VECTOR_STORE_TYPE == "Milvus":
+    try:
+        from crazy_functions.rag_fns.milvus_worker import MilvusRagWorker as LlamaIndexRagWorker
+    except:
+        VECTOR_STORE_TYPE = "Simple"
+
 if VECTOR_STORE_TYPE == "Simple":
    from crazy_functions.rag_fns.llama_index_worker import LlamaIndexRagWorker
-if VECTOR_STORE_TYPE == "Milvus":
-    from crazy_functions.rag_fns.milvus_worker import MilvusRagWorker as LlamaIndexRagWorker


 RAG_WORKER_REGISTER = {}