'commit'

dsLightRag/Config/Config.py

@@ -29,4 +29,12 @@ MYSQL_HOST = "127.0.0.1"
 MYSQL_PORT = 22066
 MYSQL_USER = "root"
 MYSQL_PASSWORD = "DsideaL147258369"
-MYSQL_DB_NAME = "base_db"
+MYSQL_DB_NAME = "base_db"
+
+# POSTGRESQL配置信息
+AGE_GRAPH_NAME = "dickens"
+POSTGRES_HOST = "10.10.14.208"
+POSTGRES_PORT = 5432
+POSTGRES_USER = "postgres"
+POSTGRES_PASSWORD = "DsideaL147258369"
+POSTGRES_DATABASE = "rag"

dsLightRag/Util/LightRagUtil.py

@@ -142,16 +142,13 @@ def create_embedding_func():
         ),
     )
 
-
-
 # AGE
-os.environ["AGE_GRAPH_NAME"] = "dickens"
+os.environ["AGE_GRAPH_NAME"] = AGE_GRAPH_NAME
-os.environ["POSTGRES_HOST"] = "10.10.14.208"
+os.environ["POSTGRES_HOST"] = POSTGRES_HOST
-os.environ["POSTGRES_PORT"] = "5432"
+os.environ["POSTGRES_PORT"] = str(POSTGRES_PORT)
-os.environ["POSTGRES_USER"] = "postgres"
+os.environ["POSTGRES_USER"] = POSTGRES_USER
-os.environ["POSTGRES_PASSWORD"] = "postgres"
+os.environ["POSTGRES_PASSWORD"] = POSTGRES_PASSWORD
-os.environ["POSTGRES_DATABASE"] = "rag"
+os.environ["POSTGRES_DATABASE"] = POSTGRES_DATABASE
-
 
 async def initialize_pg_rag(WORKING_DIR):
     rag = LightRAG(