from typing import Any

from pydantic import BaseModel, ConfigDict, Field


class AgentRunCreateRequest(BaseModel):
    model_config = ConfigDict(extra="forbid")

    agent_type: str
    message: str = Field(min_length=1)
    session_id: str
    idempotency_key: str = Field(min_length=16, max_length=128)


class AgentRunAcceptedResponse(BaseModel):
    run_id: str
    trace_id: str
    status: str
    status_url: str
    events_url: str


class AgentRunStatusResponse(BaseModel):
    run_id: str
    trace_id: str
    status: str
    agent_type: str
    session_id: str
    result: dict[str, Any] | None = None
    error_code: str | None = None
    created_at: str
    completed_at: str | None = None