fallback to simple vs in windows system

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 = {}

crazy_functions/rag_fns/llama_index_worker.py

@@ -59,6 +59,11 @@ class SaveLoad():
     def create_new_vs(self):
         return GptacVectorStoreIndex.default_vector_store(embed_model=self.embed_model)
 
+    def purge(self):
+        import shutil
+        shutil.rmtree(self.checkpoint_dir, ignore_errors=True)
+        self.vs_index = self.create_new_vs()
+
 
 class LlamaIndexRagWorker(SaveLoad):
     def __init__(self, user_name, llm_kwargs, auto_load_checkpoint=True, checkpoint_dir=None) -> None: