'commit'

dsLightRag/.idea/misc.xml

@@ -1,4 +1,7 @@
 <?xml version="1.0" encoding="UTF-8"?>
 <project version="4">
+  <component name="Black">
+    <option name="sdkName" value="D:\anaconda3\envs\lightrag" />
+  </component>
   <component name="ProjectRootManager" version="2" project-jdk-name="D:\anaconda3\envs\lightrag" project-jdk-type="Python SDK" />
 </project>

dsLightRag/Test/T1_Train_index.py

@@ -0,0 +1,41 @@
+import asyncio
+from Util.LightRagUtil import configure_logging, initialize_rag
+
+
+async def main():
+    # 注释掉或删除以下清理代码
+    # files_to_delete = [
+    #     "graph_chunk_entity_relation.graphml",
+    #     "kv_store_doc_status.json",
+    #     "kv_store_full_docs.json",
+    #     "kv_store_text_chunks.json",
+    #     "vdb_chunks.json",
+    #     "vdb_entities.json",
+    #     "vdb_relationships.json",
+    # ]
+    # for file in files_to_delete:
+    #     file_path = os.path.join(WORKING_DIR, file)
+    #     if os.path.exists(file_path):
+    #         os.remove(file_path)
+    #         print(f"Deleting old file:: {file_path}")
+
+    # 数学
+    SOURCE_DIR = r"D:\dsWork\dsProject\dsRagAnything\Tools\output\小学数学教学中的若干问题_MATH_1\auto"
+    TXT_FILE = "小学数学教学中的若干问题_MATH_1.md"
+    WORKING_DIR="./Math"
+
+    try:
+        rag = await initialize_rag(WORKING_DIR)
+        with open(SOURCE_DIR +"\\"+ TXT_FILE, "r", encoding="utf-8") as f:
+            await rag.ainsert(f.read())
+        print("\nIndexing completed successfully!")
+    except Exception as e:
+        print(f"An error occurred: {e}")
+    finally:
+        if rag:
+            await rag.finalize_storages()
+
+
+if __name__ == "__main__":
+    configure_logging()
+    asyncio.run(main())

dsLightRag/Test/T2_Query.py

@@ -0,0 +1,35 @@
+import asyncio
+import inspect
+from Util.LightRagUtil import configure_logging, initialize_rag, print_stream
+from lightrag import QueryParam
+
+# 数学
+WORKING_DIR = "./Topic/Math"
+TXT_FILE = "小学数学教学中的若干问题.txt"
+
+# 苏轼
+#WORKING_DIR = "./Topic/Chinese"
+#TXT_FILE = "sushi.txt"
+
+async def main():
+    try:
+        rag = await initialize_rag(WORKING_DIR)
+        resp = await rag.aquery(
+            #"苏轼有几个妻子？",
+            "小学数学有哪些模型",
+            param=QueryParam(mode="hybrid", stream=True),
+        )
+        if inspect.isasyncgen(resp):
+            await print_stream(resp)
+        else:
+            print(resp)
+    except Exception as e:
+        print(f"An error occurred: {e}")
+    finally:
+        if rag:
+            await rag.finalize_storages()
+
+
+if __name__ == "__main__":
+    configure_logging()
+    asyncio.run(main())