代码提交

.env

@@ -1,8 +1,9 @@
-POSTGRESQL_HOST = localhost
-POSTGRESQL_DATABASE = kg
+POSTGRESQL_HOST = 173.18.12.203
+POSTGRESQL_DATABASE = medkg
 POSTGRESQL_DATABASE_AGENT = agent
-POSTGRESQL_USER = postgres
-POSTGRESQL_PASSWORD = difyai123456
+POSTGRESQL_USER = knowledge
+POSTGRESQL_PASSWORD = qwer1234.
+POSTGRESQL_PORT = 5432
 
 #indexing
 ELASTICSEARCH_HOST=https://localhost:9200

config/site.py

@@ -39,6 +39,7 @@ class SiteConfig:
             'POSTGRESQL_DATABASE':  os.getenv("POSTGRESQL_DATABASE","kg"),
             'POSTGRESQL_USER':  os.getenv("POSTGRESQL_USER","dify"),
             'POSTGRESQL_PASSWORD':  os.getenv("POSTGRESQL_PASSWORD",quote("difyai123456")),
+            'POSTGRESQL_PORT': os.getenv("POSTGRESQL_PORT", "5432")
         }
     def get_config(self, config_name): 
         config_name = config_name.upper()       

db/session.py

@@ -1,5 +1,7 @@
 from sqlalchemy import create_engine
 from sqlalchemy.orm import sessionmaker, scoped_session
+
+from config.site import SiteConfig
 from .base_class import Base
 import os
 from pgvector.sqlalchemy import Vector
@@ -7,11 +9,12 @@ from pgvector.sqlalchemy import Vector
 # 数据库配置
 # 远程PostgreSQL数据库连接配置
 # 从环境变量获取数据库连接信息，如果未设置则使用默认值
-DB_HOST = os.getenv("DB_HOST", "173.18.12.203")
-DB_PORT = os.getenv("DB_PORT", "5432")
-DB_USER = os.getenv("DB_USER", "knowledge")
-DB_PASS = os.getenv("DB_PASSWORD", "qwer1234.")
-DB_NAME = os.getenv("DB_NAME", "postgres")
+config = SiteConfig()
+DB_HOST = config.get_config('POSTGRESQL_HOST')
+DB_PORT = config.get_config('POSTGRESQL_PORT')
+DB_USER = config.get_config('POSTGRESQL_USER')
+DB_PASS = config.get_config('POSTGRESQL_PASSWORD')
+DB_NAME = config.get_config('POSTGRESQL_DATABASE')
 
 DATABASE_URL = f"postgresql://{DB_USER}:{DB_PASS}@{DB_HOST}:{DB_PORT}/{DB_NAME}"