Simon
147 lines
6.6 KiB
Python
147 lines
6.6 KiB
Python
# =============================================================================
|
||
# 企微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()
|