from typing import Literal
from pydantic import BaseModel, Field


class QueryClassification(BaseModel):
    """Classification result for routing requests in the orchestrated workflow."""
    classification: Literal["CLASSIC", "SUMMARIZE", "UNKNOWN"] = Field(
        description="Classification of the user request",
        examples=["CLASSIC", "SUMMARIZE", "UNKNOWN"],
    )
    reasoning: str = Field(
        description="Short explanation of the classification decision",
        default="",
    )