from pydantic import BaseModel, ConfigDict, Field


class AiSettingBase(BaseModel):
    provider: str = Field(default="nvidia")
    api_base: str = Field(default="https://integrate.api.nvidia.com/v1")
    api_key: str = Field(default="")
    model_name: str = Field(default="nvidia/llama-3.1-nemotron-70b-instruct")
    temperature: float = Field(default=0.7, ge=0, le=2)
    max_tokens: int = Field(default=2048, ge=1)
    crawl_enabled: bool = Field(default=False)
    crawl_keywords: str = Field(default="")
    max_pages_per_run: int = Field(default=3, ge=1)


class AiSettingCreate(AiSettingBase):
    pass