"""Request context models produced by the Context Builder."""

from __future__ import annotations

from pydantic import BaseModel, Field


class ConversationTurn(BaseModel):
    role: str  # "user" | "assistant"
    content: str


class RetrievedItem(BaseModel):
    source: str
    content: str
    score: float


class RequestContext(BaseModel):
    user_query: str
    session_id: str | None
    trace_id: str
    conversation_history: list[ConversationTurn] = Field(default_factory=list)
    retrieved_context: list[RetrievedItem] = Field(default_factory=list)
    system_prompt: str | None = None