From 3aad732faeab17efd73ce71c73ad7cd0ed5dd613 Mon Sep 17 00:00:00 2001
From: HuangHai <10402852@qq.com>
Date: Fri, 11 Jul 2025 16:42:29 +0800
Subject: [PATCH] 'commit'

---
 dsLightRag/Start.py               |  1 -
 dsLightRag/Util/PostgreSQLUtil.py | 27 +++++++++++++++++++++++++++
 2 files changed, 27 insertions(+), 1 deletion(-)
 create mode 100644 dsLightRag/Util/PostgreSQLUtil.py

diff --git a/dsLightRag/Start.py b/dsLightRag/Start.py
index ab167a52..fe180143 100644
--- a/dsLightRag/Start.py
+++ b/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(
diff --git a/dsLightRag/Util/PostgreSQLUtil.py b/dsLightRag/Util/PostgreSQLUtil.py
new file mode 100644
index 00000000..0e487427
--- /dev/null
+++ b/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)
\ No newline at end of file