From bd18c9c8ad4ffeda6ff1e509147a49c61448dfef Mon Sep 17 00:00:00 2001
From: yangdx <yangdx@znipower.com>
Date: Mon, 21 Apr 2025 01:22:04 +0800
Subject: [PATCH] Update sample code in README.md

---
 README-zh.md                     | 48 ++++++++++++++++----------------
 README.md                        | 48 ++++++++++++++++----------------
 examples/lightrag_openai_demo.py |  2 --
 3 files changed, 48 insertions(+), 50 deletions(-)

diff --git a/README-zh.md b/README-zh.md
index 78ee9850..afadb941 100644
--- a/README-zh.md
+++ b/README-zh.md
@@ -107,42 +107,42 @@ from lightrag.utils import setup_logger
 
 setup_logger("lightrag", level="INFO")
 
+if not os.path.exists(WORKING_DIR):
+    os.mkdir(WORKING_DIR)
+
 async def initialize_rag():
     rag = LightRAG(
-        working_dir="your/path",
+        working_dir=WORKING_DIR,
         embedding_func=openai_embed,
-        llm_model_func=gpt_4o_mini_complete
+        llm_model_func=gpt_4o_mini_complete,
     )
-
     await rag.initialize_storages()
     await initialize_pipeline_status()
-
     return rag
 
 def main():
-    # 初始化RAG实例
-    rag = asyncio.run(initialize_rag())
-    # 插入文本
-    rag.insert("Your text")
+    try:
+        # Initialize RAG instance
+        rag = await initialize_rag()
+		    rag.insert("Your text")
+  
+        # Perform hybrid search
+        mode="hybrid"
+        print(
+          await rag.query(
+              "What are the top themes in this story?",
+              param=QueryParam(mode=mode)
+          )
+        )
 
-    # 执行朴素搜索
-    mode="naive"
-    # 执行本地搜索
-    mode="local"
-    # 执行全局搜索
-    mode="global"
-    # 执行混合搜索
-    mode="hybrid"
-    # 混合模式集成知识图谱和向量检索
-    mode="mix"
-
-    rag.query(
-        "这个故事的主要主题是什么？",
-        param=QueryParam(mode=mode)
-    )
+    except Exception as e:
+        print(f"An error occurred: {e}")
+    finally:
+        if rag:
+            await rag.finalize_storages()
 
 if __name__ == "__main__":
-    main()
+    asyncio.run(main())
 ```
 
 ### 查询参数
diff --git a/README.md b/README.md
index fdc69cf6..39bdc269 100644
--- a/README.md
+++ b/README.md
@@ -143,42 +143,42 @@ from lightrag.utils import setup_logger
 
 setup_logger("lightrag", level="INFO")
 
+if not os.path.exists(WORKING_DIR):
+    os.mkdir(WORKING_DIR)
+
 async def initialize_rag():
     rag = LightRAG(
-        working_dir="your/path",
+        working_dir=WORKING_DIR,
         embedding_func=openai_embed,
-        llm_model_func=gpt_4o_mini_complete
+        llm_model_func=gpt_4o_mini_complete,
     )
-
     await rag.initialize_storages()
     await initialize_pipeline_status()
-
     return rag
 
 def main():
-    # Initialize RAG instance
-    rag = asyncio.run(initialize_rag())
-    # Insert text
-    rag.insert("Your text")
+    try:
+        # Initialize RAG instance
+        rag = await initialize_rag()
+		    rag.insert("Your text")
+  
+        # Perform hybrid search
+        mode="hybrid"
+        print(
+          await rag.query(
+              "What are the top themes in this story?",
+              param=QueryParam(mode=mode)
+          )
+        )
 
-    # Perform naive search
-    mode="naive"
-    # Perform local search
-    mode="local"
-    # Perform global search
-    mode="global"
-    # Perform hybrid search
-    mode="hybrid"
-    # Mix mode Integrates knowledge graph and vector retrieval.
-    mode="mix"
-
-    rag.query(
-        "What are the top themes in this story?",
-        param=QueryParam(mode=mode)
-    )
+    except Exception as e:
+        print(f"An error occurred: {e}")
+    finally:
+        if rag:
+            await rag.finalize_storages()
 
 if __name__ == "__main__":
-    main()
+    asyncio.run(main())
 ```
 
 ### Query Param
diff --git a/examples/lightrag_openai_demo.py b/examples/lightrag_openai_demo.py
index 8790fdc6..42a37d27 100644
--- a/examples/lightrag_openai_demo.py
+++ b/examples/lightrag_openai_demo.py
@@ -82,7 +82,6 @@ async def initialize_rag():
         working_dir=WORKING_DIR,
         embedding_func=openai_embed,
         llm_model_func=gpt_4o_mini_complete,
-        # llm_model_func=gpt_4o_complete
     )
 
     await rag.initialize_storages()
@@ -92,7 +91,6 @@ async def initialize_rag():
 
 
 async def main():
-    rag = None
     try:
         # Initialize RAG instance
         rag = await initialize_rag()