# ============================================================================= # 企微IT智能服务台 — 配置管理模块 # ============================================================================= # 说明:使用 pydantic-settings 从环境变量读取所有配置项 # 优先级:环境变量 > .env 文件 > 默认值 # 所有配置项集中管理,避免散落在代码各处 # ============================================================================= from typing import List import redis.asyncio as aioredis from pydantic_settings import BaseSettings, SettingsConfigDict class Settings(BaseSettings): """应用配置类。 使用 pydantic-settings 自动从环境变量读取配置值。 支持 .env 文件自动加载(开发环境便利)。 Attributes: wecom_corp_id: 企业微信企业ID wecom_agent_id: 企业微信应用AgentId wecom_secret: 企业微信应用Secret wecom_token: 企业微信回调Token wecom_encoding_aes_key: 企业微信回调EncodingAESKey(43位) database_url: PostgreSQL 数据库连接地址 redis_url: Redis 连接地址 backend_host: 后端监听地址 backend_port: 后端监听端口 cors_origins: CORS 允许的源地址(逗号分隔) """ # ---------------------------------------------------------------------- # 企微自建应用配置 # ---------------------------------------------------------------------- # 企业ID(在企微管理后台 > 我的企业 > 企业信息 中查看) wecom_corp_id: str = "ww1234567890abcdef" # 应用AgentId(在企微管理后台 > 应用管理 > 自建应用 中查看) wecom_agent_id: str = "1000002" # 应用Secret(在企微管理后台 > 应用管理 > 自建应用 中查看) wecom_secret: str = "your-agent-secret" # 回调Token(在企微管理后台 > 应用管理 > 接收消息 中设置) wecom_token: str = "your-callback-token" # 回调EncodingAESKey(43位字符串,用于消息加解密) wecom_encoding_aes_key: str = "your-aes-key-43-characters-long-encoding-key" # ---------------------------------------------------------------------- # 数据库配置 # ---------------------------------------------------------------------- # PostgreSQL 连接地址 # Docker 环境使用容器名 postgres,本地开发使用 localhost database_url: str = "postgresql://wecom:wecom_secret@localhost:5432/wecom_it_desk" # ---------------------------------------------------------------------- # Redis 配置 # ---------------------------------------------------------------------- # Redis 连接地址 # Docker 环境使用容器名 redis,本地开发使用 localhost redis_url: str = "redis://localhost:6379/0" # ---------------------------------------------------------------------- # 服务配置 # ---------------------------------------------------------------------- # 后端监听地址(0.0.0.0 表示监听所有网卡) backend_host: str = "0.0.0.0" # 后端监听端口 backend_port: int = 8000 # CORS 允许的源地址(逗号分隔的字符串) cors_origins: str = "http://localhost:5173,http://localhost:5174,http://localhost:5175" # ---------------------------------------------------------------------- # AI 服务配置(Dify) # ---------------------------------------------------------------------- # Dify API 端点(兼容 OpenAI Chat Completions 格式) # 必须通过环境变量 DIFY_API_URL 配置,不设置默认值(防止凭据泄露) dify_api_url: str = "" # Dify API Key(格式:base_url|app_id|app_name) # 必须通过环境变量 DIFY_API_KEY 配置,不设置默认值(防止凭据泄露) dify_api_key: str = "" # Dify API 请求超时(秒),在网络慢时可调大 dify_timeout: int = 30 # ---------------------------------------------------------------------- # AI Wingman 服务配置(Dify Agent 2 — 坐席端辅助) # ---------------------------------------------------------------------- # 坐席端 Wingman 专用 Dify API 端点(与员工端 Agent 分开) # 留空则禁用 Wingman 功能(不影响主流程) dify_wingman_api_url: str = "" # 坐席端 Wingman Dify API Key(需要新建 Agent 后填入,留空则禁用) # 格式:base_url|app_id|app_name(与 dify_api_key 相同格式) dify_wingman_api_key: str = "" # Wingman API 请求超时(秒) dify_wingman_timeout: int = 30 # ---------------------------------------------------------------------- # Mock 登录配置(测试阶段使用,跳过企微 OAuth2) # ---------------------------------------------------------------------- # 是否启用 Mock 登录(默认 false,生产环境必须关闭) mock_login_enabled: bool = False # ---------------------------------------------------------------------- # Pydantic-settings 配置 # ---------------------------------------------------------------------- model_config = SettingsConfigDict( # 自动从 .env 文件加载环境变量 env_file=".env", # .env 文件编码 env_file_encoding="utf-8", # 环境变量名大小写不敏感 case_sensitive=False, # 额外字段不允许(防止拼写错误的配置被忽略) extra="ignore", ) @property def cors_origins_list(self) -> List[str]: """将 CORS 源地址字符串解析为列表。 将逗号分隔的字符串(如 "http://a,http://b") 转换为列表(如 ["http://a", "http://b"]), 方便 FastAPI 的 CORSMiddleware 使用。 Returns: List[str]: CORS 允许的源地址列表 """ # 去除每项的前后空格,过滤空字符串 return [origin.strip() for origin in self.cors_origins.split(",") if origin.strip()] def create_redis_client(self) -> aioredis.Redis: """创建 Redis 异步客户端实例。 自动附加 protocol=2 参数,强制使用 RESP2 协议。 原因:Windows 版 Redis 3.x 不支持 RESP3 协议(HELLO 命令), 而 redis-py 8.0+ 默认使用 RESP3,会导致连接失败。 全项目统一使用此方法创建 Redis 客户端,避免协议不匹配。 Returns: aioredis.Redis: 配置好的 Redis 异步客户端 """ return aioredis.from_url(self.redis_url, protocol=2) # 创建全局配置实例 # 整个应用通过 from app.config import settings 使用同一个实例 settings = Settings()