'commit'

dsRagAnything/T2_Query.py

@@ -6,32 +6,32 @@ from raganything import RAGAnything
 from lightrag import LightRAG
 from Util.RagUtil import create_llm_model_func, create_vision_model_func, create_embedding_func
 
-async def load_existing_lightrag():
-    lightrag_working_dir = "./rag_storage"
 
-    if os.path.exists(lightrag_working_dir) and os.listdir(lightrag_working_dir):
-        print("✅ 发现已存在的 LightRAG 实例，正在加载...")
-    else:
-        print("❌ 未找到已存在的 LightRAG 实例，将创建新实例")
+async def load_existing_lightrag():
+    # 索引位置
+    WORKING_DIR = "./rag_storage"
 
+    # 创建 LLM 模型自定义函数
     llm_model_func = create_llm_model_func()
+    # 创建可视模型自定义函数
     vision_model_func = create_vision_model_func(llm_model_func)
+    # 创建嵌入模型自定义函数
     embedding_func = create_embedding_func()
-
+    # 声明LightRAG实例
     lightrag_instance = LightRAG(
-        working_dir=lightrag_working_dir,
+        working_dir=WORKING_DIR,
         llm_model_func=llm_model_func,
         embedding_func=embedding_func
     )
-
+    # 初始化
     await lightrag_instance.initialize_storages()
     await initialize_pipeline_status()
-
+    # 创建RAGAnything实例，依托于LightRAG实例
     rag = RAGAnything(
         lightrag=lightrag_instance,
         vision_model_func=vision_model_func,
     )
-
+    # 查询
     result = await rag.aquery(
         "平台安全的保证方法有哪些？",
         mode="hybrid"