'commit'

dsRagAnything/T2_Query.py

@@ -3,10 +3,10 @@ import inspect
 from Util.LightRagUtil import configure_logging, initialize_rag, print_stream
 from lightrag import QueryParam
 
-async def main():
+async def query():
     try:
         user_prompt = "简洁回复。"
-        rag = await initialize_rag('Output')
+        rag = await initialize_rag('./Topic/HuangWanQiao')
         resp = await rag.aquery(
             "文档的主要内容是什么",
             param=QueryParam(mode="hybrid", stream=True, user_prompt=user_prompt),
@@ -24,4 +24,4 @@ async def main():
 
 if __name__ == "__main__":
     configure_logging()
-    asyncio.run(main())
+    asyncio.run(query())