"""
Request and response schemas for agent API.
"""
from src.shared.base import BaseSchema


class AgentRunRequest(BaseSchema):
    """Request to run an agent."""
    prompt: str
    working_dir: str = "."
    agent_type: str = "explore"


class AgentRunResponse(BaseSchema):
    """Response from agent execution."""
    response: str
    agent_type: str
    success: bool = True
    error: str | None = None


class AgentInfo(BaseSchema):
    """Information about an agent."""
    name: str
    description: str


class AgentListResponse(BaseSchema):
    """List of available agents."""
    agents: list[AgentInfo]