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