diff --git a/MindSpider/config.py b/MindSpider/config.py new file mode 100644 index 0000000..05d5206 --- /dev/null +++ b/MindSpider/config.py @@ -0,0 +1,35 @@ +# -*- coding: utf-8 -*- +""" +存储数据库连接信息和API密钥 +""" + +from pydantic_settings import BaseSettings +from typing import Optional +from pydantic import Field +from pathlib import Path + +# 计算 .env 优先级:优先当前工作目录,其次项目根目录(MindSpider 的上级目录) +PROJECT_ROOT: Path = Path(__file__).resolve().parents[1] +CWD_ENV: Path = Path.cwd() / ".env" +ENV_FILE: str = str(CWD_ENV if CWD_ENV.exists() else (PROJECT_ROOT / ".env")) + +class Settings(BaseSettings): + """全局配置管理,优先从环境变量和.env加载。支持MySQL/PostgreSQL统一数据库参数命名。""" + DB_DIALECT: str = Field("mysql", description="数据库类型,支持'mysql'或'postgresql'") + DB_HOST: str = Field("your_host", description="数据库主机名或IP地址") + DB_PORT: int = Field(3306, description="数据库端口号") + DB_USER: str = Field("your_username", description="数据库用户名") + DB_PASSWORD: str = Field("your_password", description="数据库密码") + DB_NAME: str = Field("mindspider", description="数据库名称") + DB_CHARSET: str = Field("utf8mb4", description="数据库字符集") + MINDSPIDER_API_KEY: Optional[str] = Field(None, description="MINDSPIDER API密钥") + MINDSPIDER_BASE_URL: Optional[str] = Field("https://api.deepseek.com", description="MINDSPIDER API基础URL,推荐deepseek-chat模型使用https://api.deepseek.com") + MINDSPIDER_MODEL_NAME: Optional[str] = Field("deepseek-chat", description="MINDSPIDER API模型名称, 推荐deepseek-chat") + + class Config: + env_file = ENV_FILE + env_prefix = "" + case_sensitive = False + extra = "allow" + +settings = Settings()