agents/shared/model_schema.py

from __future__ import annotations

from typing import Literal

from pydantic import BaseModel, ConfigDict


class StrictModel(BaseModel):
    model_config = ConfigDict(extra="forbid")


class ModelMessage(StrictModel):
    role: Literal["system", "user", "assistant"]
    content: str