config.py

from pydantic_settings import BaseSettings


class Settings(BaseSettings):
    # ──────────────────────────────────────────────────────
    # PostgreSQL credentials — replace before running
    # ──────────────────────────────────────────────────────
    DB_USER: str = "YOUR_POSTGRES_USERNAME"      # ← replace
    DB_PASSWORD: str = "YOUR_POSTGRES_PASSWORD"  # ← replace
    DB_HOST: str = "localhost"
    DB_PORT: int = 5432
    DB_NAME: str = "sql_assistant"

    # ──────────────────────────────────────────────────────
    # AI Provider API keys — replace the ones you intend to use
    # ──────────────────────────────────────────────────────
    ANTHROPIC_API_KEY: str = "YOUR_ANTHROPIC_API_KEY"  # ← replace (Claude)
    OPENAI_API_KEY: str = "YOUR_OPENAI_API_KEY"        # ← replace (OpenAI)
    DEEPSEEK_API_KEY: str = "YOUR_DEEPSEEK_API_KEY"    # ← replace (Deepseek)

    # Ollama runs locally — no API key required
    OLLAMA_BASE_URL: str = "http://localhost:11434"

    # ──────────────────────────────────────────────────────
    # Default model per provider
    # ──────────────────────────────────────────────────────
    CLAUDE_MODEL: str = "claude-sonnet-4-6"
    OPENAI_MODEL: str = "gpt-4o"
    DEEPSEEK_MODEL: str = "deepseek-chat"
    OLLAMA_MODEL: str = "llama3.2"

    # ──────────────────────────────────────────────────────
    # Safety limits
    # ──────────────────────────────────────────────────────
    QUERY_ROW_LIMIT: int = 200
    QUERY_TIMEOUT_SECONDS: int = 10

    @property
    def database_url(self) -> str:
        return (
            f"postgresql+asyncpg://{self.DB_USER}:{self.DB_PASSWORD}"
            f"@{self.DB_HOST}:{self.DB_PORT}/{self.DB_NAME}"
        )

    class Config:
        env_file = ".env"
        env_file_encoding = "utf-8"


settings = Settings()