diff --git a/dsLightRag/Config/Config.py b/dsLightRag/Config/Config.py index 19c944ca..bcd7d590 100644 --- a/dsLightRag/Config/Config.py +++ b/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" diff --git a/dsLightRag/Start.py b/dsLightRag/Start.py index 6c8d760e..0079a9b4 100644 --- a/dsLightRag/Start.py +++ b/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') diff --git a/dsLightRag/Util/PostgreSQLUtil.py b/dsLightRag/Util/PostgreSQLUtil.py new file mode 100644 index 00000000..dfc47ddd --- /dev/null +++ b/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) \ No newline at end of file