from solar_eval.models.base import CustomBaseModel


class PromptCreate(CustomBaseModel):
    project_id: str
    task: str
    system_prompt: str
    model: str = "solar-pro2"
    temperature: float = 0.0
    max_tokens: int = 8000
    reasoning_effort: str = "high"


class PromptUpdate(CustomBaseModel):
    system_prompt: str | None = None
    model: str | None = None
    temperature: float | None = None
    max_tokens: int | None = None
    reasoning_effort: str | None = None


class PromptResponse(CustomBaseModel):
    id: str
    project_id: str
    task: str
    version: int
    system_prompt: str
    model: str
    temperature: float
    max_tokens: int
    reasoning_effort: str
    created_at: str