from pydantic import BaseModel


class PromptRequest(BaseModel):
    prompt: str


class PredictionResponse(BaseModel):
    prompt: str
    command: str | None
    raw_output: str
    latency_seconds: float