diff --git a/dsRagAnything/T2_Query.py b/dsRagAnything/T2_Query.py index d3567372..47346676 100644 --- a/dsRagAnything/T2_Query.py +++ b/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"