'commit'

dsLightRag/Start.py

@@ -14,7 +14,6 @@ from starlette.responses import StreamingResponse
 from starlette.staticfiles import StaticFiles
 
 from Util.LightRagUtil import *
-from Util.MySQLUtil import init_mysql_pool
 
 # 在程序开始时添加以下配置
 logging.basicConfig(

dsLightRag/Util/PostgreSQLUtil.py

@@ -0,0 +1,27 @@
+"""
+pip install asyncpg
+"""
+import logging
+import asyncpg
+
+from Config.Config import *
+
+# 配置日志
+logging.basicConfig(level=logging.INFO, format="%(asctime)s - %(levelname)s - %(message)s")
+logger = logging.getLogger(__name__)
+
+# PostgreSQL 配置
+POSTGRES_CONFIG = {
+    "host": POSTGRES_HOST,
+    "port": POSTGRES_PORT,
+    "user": POSTGRES_USER,
+    "password": POSTGRES_PASSWORD,
+    "database": POSTGRES_DATABASE,
+    "min_size": 1,
+    "max_size": 20,
+    "command_timeout": 60
+}
+
+# 初始化 PostgreSQL 连接池
+async def init_postgres_pool():
+    return await asyncpg.create_pool(**POSTGRES_CONFIG)