from pydantic_settings import BaseSettings, SettingsConfigDict


class Settings(BaseSettings):
    model_config = SettingsConfigDict(env_file=".env", extra="ignore")

    # 应用
    app_env: str = "development"
    log_level: str = "INFO"
    api_secret_key: str = "change_this_key"

    # 数据库
    database_url: str = "postgresql+asyncpg://compliance:compliance123@postgres:5432/compliance_db"
    redis_url: str = "redis://:redis123@redis:6379/0"

    # Milvus
    milvus_host: str = "milvus"
    milvus_port: int = 19530

    # Neo4j
    neo4j_uri: str = "bolt://neo4j:7687"
    neo4j_user: str = "neo4j"
    neo4j_password: str = "neo4j123"

    # AI 服务
    embedding_service_url: str = "http://embedding-service:8010"
    mcp_server_url: str = "http://mcp-server:8011"

    # LLM
    llm_provider: str = "deepseek"   # deepseek / qwen
    deepseek_api_key: str = ""
    deepseek_model: str = "deepseek-chat"
    dashscope_api_key: str = ""
    qwen_model: str = "qwen-plus"


settings = Settings()