from typing import Literal

from pydantic import BaseModel


class AgentRequest(BaseModel):
    input: str


class TextChunk(BaseModel):
    type: Literal["text", "tool_call", "tool_result", "error"] = "text"
    content: str = ""
    tool_name: str | None = None
    tool_args: dict | None = None
    phase: str | None = None