from pydantic import BaseModel, Field


class ModelTestRequest(BaseModel):
    model_id: str = Field(..., description="已缓存模型的 ID")
    prompt: str
    max_new_tokens: int = 128
    temperature: float = 0.8
    top_p: float = 0.95


class ModelTestResponse(BaseModel):
    model_id: str
    prompt: str
    generated_text: str
    error: str | None = None