src/rag/tool_schemas.py

from pydantic import BaseModel, Field


class RetrieveContextInput(BaseModel):
    query: str = Field(
        description=(
            "Search query for current programme context. Use the same language as "
            "the requested language when possible."
        )
    )
    program: str = Field(
        description=(
            "Programme identifier to retrieve context for. Expected values are "
            "'emba', 'iemba', or 'emba x'."
        )
    )
    language: str | None = Field(
        default=None,
        description="Optional language code for retrieval, either 'de' or 'en'.",
    )


class ProgrammeAgentInput(BaseModel):
    query: str = Field(
        description=(
            "Question or user context to pass to the programme-specific support "
            "agent. Include relevant profile details when available."
        )
    )