from typing import Literal

from pydantic import BaseModel, Field


class RobotPlan(BaseModel):
    steps: list[str] = Field(
        description="Последовательность высокоуровневых действий робота",
        min_length=1
    )


class Message(BaseModel):
    role: Literal["system",
                  "user",
                  "assistant"]

    content: str


class SFTFormat(BaseModel):
    message: list[Message]


class DatasetRecord(BaseModel):
    task: str

    system_prompt: str
    instructions: str
    user_prompt: str

    model_response: str
    assistant_content: str

    provider: str
    model: str

    temperature: float
    max_tokens: int