From bbb8f11cfbc8164225329e85d41bc173ecedf545 Mon Sep 17 00:00:00 2001 From: HuangHai <10402852@qq.com> Date: Fri, 11 Jul 2025 16:12:27 +0800 Subject: [PATCH] 'commit' --- dsLightRag/Test/T1_WriteToPg.py | 4 ++-- dsLightRag/Test/T2_ReadFromPg.py | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/dsLightRag/Test/T1_WriteToPg.py b/dsLightRag/Test/T1_WriteToPg.py index 458dbcdd..2598a18c 100644 --- a/dsLightRag/Test/T1_WriteToPg.py +++ b/dsLightRag/Test/T1_WriteToPg.py @@ -39,7 +39,7 @@ os.environ["POSTGRES_PASSWORD"] = "postgres" os.environ["POSTGRES_DATABASE"] = "rag" -async def initialize_rag(): +async def initialize_pg_rag(): rag = LightRAG( working_dir=WORKING_DIR, llm_model_func=llm_model_func, @@ -66,7 +66,7 @@ async def initialize_rag(): async def main(): - rag = await initialize_rag() + rag = await initialize_pg_rag() with open(f"../Txt/sushi.txt", "r", encoding="utf-8") as f: await rag.ainsert(f.read()) diff --git a/dsLightRag/Test/T2_ReadFromPg.py b/dsLightRag/Test/T2_ReadFromPg.py index 31806422..377d0b14 100644 --- a/dsLightRag/Test/T2_ReadFromPg.py +++ b/dsLightRag/Test/T2_ReadFromPg.py @@ -40,7 +40,7 @@ os.environ["POSTGRES_PASSWORD"] = "postgres" os.environ["POSTGRES_DATABASE"] = "rag" -async def initialize_rag(): +async def initialize_pg_rag(): rag = LightRAG( working_dir=WORKING_DIR, llm_model_func=llm_model_func, @@ -67,7 +67,7 @@ async def initialize_rag(): async def main(): try: - rag = await initialize_rag() + rag = await initialize_pg_rag() resp = await rag.aquery( "What are the top themes in this story", param=QueryParam(mode="hybrid", stream=True),