'commit'

dsLightRag/Config/Config.py

@@ -31,3 +31,10 @@ NEO4J_URI = "bolt://localhost:7687"
 NEO4J_USERNAME = "neo4j"
 NEO4J_PASSWORD = "DsideaL147258369"
 NEO4J_AUTH = (NEO4J_USERNAME, NEO4J_PASSWORD)
+
+# POSTGRESQL配置信息
+POSTGRES_HOST = "10.10.14.208"
+POSTGRES_PORT = 5432
+POSTGRES_USER = "postgres"
+POSTGRES_PASSWORD = "postgres"
+POSTGRES_DATABASE = "rag"

dsLightRag/Start.py

@@ -14,6 +14,7 @@ from starlette.responses import StreamingResponse
 from starlette.staticfiles import StaticFiles
 
 from Util.LightRagUtil import *
+from Util.PostgreSQLUtil import init_postgres_pool
 
 # 更详细地控制日志输出
 logger = logging.getLogger('lightrag')

dsLightRag/Util/PostgreSQLUtil.py

@@ -0,0 +1,22 @@
+"""
+pip install asyncpg
+"""
+import asyncpg
+
+from Config.Config import *
+
+# PostgreSQL 配置
+POSTGRES_CONFIG = {
+    "host": POSTGRES_HOST,
+    "port": POSTGRES_PORT,
+    "user": POSTGRES_USER,
+    "password": POSTGRES_PASSWORD,
+    "database": POSTGRES_DATABASE,
+    "min_size": 1,  # 设置为0表示不保留空闲连接
+    "max_size": 20,
+    "command_timeout": 60
+}
+
+# 初始化 PostgreSQL 连接池
+async def init_postgres_pool():
+    return await asyncpg.create_pool(**POSTGRES_CONFIG)