"""Request models for KGraph-MCP API."""

from pydantic import BaseModel, Field


class TaskRequest(BaseModel):
    """Task creation request model."""

    title: str = Field(description="Task title")
    description: str = Field(description="Task description")
    dependencies: list[int] = Field(
        default_factory=list, description="Task dependencies"
    )


class ToolSuggestionRequest(BaseModel):
    """Tool suggestion request model."""

    query: str = Field(description="User query for tool suggestion")
    top_k: int = Field(default=3, description="Number of tools to suggest", ge=1, le=10)


class PlanRequest(BaseModel):
    """Plan generation request model for tool+prompt combinations."""

    query: str = Field(description="User query for plan generation")
    top_k: int = Field(
        default=3, description="Number of planned steps to return", ge=1, le=10
    )