From 4704d694bc0061df4bd50eed60b5557e6ba0bc75 Mon Sep 17 00:00:00 2001 From: HuangHai <10402852@qq.com> Date: Fri, 11 Jul 2025 20:20:15 +0800 Subject: [PATCH] 'commit' --- dsLightRag/Doc/8、使用PG数据库.md | 1 + dsLightRag/Test/T1_WriteToPg.py | 2 +- dsLightRag/Util/LightRagUtil.py | 7 +++++-- 3 files changed, 7 insertions(+), 3 deletions(-) diff --git a/dsLightRag/Doc/8、使用PG数据库.md b/dsLightRag/Doc/8、使用PG数据库.md index 058ddbc7..b318ef4d 100644 --- a/dsLightRag/Doc/8、使用PG数据库.md +++ b/dsLightRag/Doc/8、使用PG数据库.md @@ -6,5 +6,6 @@ docker tag swr.cn-north-4.myhuaweicloud.com/ddn-k8s/docker.io/shangor/postgres- - 启动 ``` +docker rm if -f postgres-LightRag docker run -p 5432:5432 -d --name postgres-LightRag shangor/postgres-for-rag sh -c "service postgresql start && sleep infinity" ``` \ No newline at end of file diff --git a/dsLightRag/Test/T1_WriteToPg.py b/dsLightRag/Test/T1_WriteToPg.py index 0eb4bcde..45e596b7 100644 --- a/dsLightRag/Test/T1_WriteToPg.py +++ b/dsLightRag/Test/T1_WriteToPg.py @@ -21,7 +21,7 @@ logging.basicConfig(format="%(levelname)s:%(message)s", level=logging.INFO) async def main(): try: - rag = await initialize_pg_rag(WORKING_DIR) + rag = await initialize_pg_rag(WORKING_DIR=WORKING_DIR,namespace_prefix='huanghai') with open(f"../Txt/sushi.txt", "r", encoding="utf-8") as f: await rag.ainsert(f.read()) finally: diff --git a/dsLightRag/Util/LightRagUtil.py b/dsLightRag/Util/LightRagUtil.py index f0c88c47..0a37a96b 100644 --- a/dsLightRag/Util/LightRagUtil.py +++ b/dsLightRag/Util/LightRagUtil.py @@ -142,6 +142,7 @@ def create_embedding_func(): ), ) + # AGE os.environ["AGE_GRAPH_NAME"] = AGE_GRAPH_NAME os.environ["POSTGRES_HOST"] = POSTGRES_HOST @@ -150,7 +151,8 @@ os.environ["POSTGRES_USER"] = POSTGRES_USER os.environ["POSTGRES_PASSWORD"] = POSTGRES_PASSWORD os.environ["POSTGRES_DATABASE"] = POSTGRES_DATABASE -async def initialize_pg_rag(WORKING_DIR): + +async def initialize_pg_rag(WORKING_DIR, namespace_prefix): rag = LightRAG( working_dir=WORKING_DIR, llm_model_func=llm_model_func, @@ -168,9 +170,10 @@ async def initialize_pg_rag(WORKING_DIR): graph_storage="PGGraphStorage", vector_storage="PGVectorStorage", auto_manage_storages_states=False, + namespace_prefix=namespace_prefix, ) await rag.initialize_storages() await initialize_pipeline_status() - return rag \ No newline at end of file + return rag