Doiiars
36 lines
1.7 KiB
Python
36 lines
1.7 KiB
Python
# -*- 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()
|