配置参数优化

zhaoqingang

2025-05-09 947cf0d625accd92edf1a5a832b12c2debddfd38

 app/config/config.py

@@ -1,3 +1,4 @@
import os
from pathlib import Path
import yaml

@@ -19,9 +20,19 @@
    basic_paper_url: str = ''
    dify_base_url: str = ''
    dify_api_token: str = ''
    dify_workflow_clean: str = ''
    dify_workflow_report: str = ''
    postgresql_database_url: str = ''
    dify_database_url: str = ''
    def __init__(self, **kwargs):
        # Check if all required fields are provided and set them
        # 替换配置中的IP地址
        host_ip = kwargs.get('host_ip', '127.0.0.1')
        kwargs['fwr_base_url'] = f"http://{host_ip}:11080"
        kwargs['database_url'] = f"mysql+pymysql://root:infini_rag_flow@{host_ip}:5455/rag_basic"
        kwargs['fwr_db_url'] = f"mysql+pymysql://root:infini_rag_flow@{host_ip}:15455/rag_flow"
        kwargs['dify_base_url'] = f"http://{host_ip}:13002"
        kwargs['postgresql_database_url'] = f"postgresql+asyncpg://kong:kongpass@{host_ip}:5432/kong"
        kwargs['dify_database_url'] = f"postgresql+psycopg2://postgres:difyai123456@192.168.20.119:15432/dify"
        for field in self.__annotations__.keys():
            if field not in kwargs:
                raise ValueError(f"Missing setting: {field}")
@@ -47,7 +58,9 @@


# Use pathlib to handle file paths
config_yaml_path = Path(__file__).parent / 'config.yaml'
# config_yaml_path = os.path.join(os.path.join(Path(__file__).parent, "env_conf"), "config.yaml")
config_yaml_path = Path(__file__).parent.joinpath("env_conf") / 'config.yaml'
print(config_yaml_path)
settings_data = load_yaml(config_yaml_path)

# Initialize settings object