from typing import Optional

from pydantic import BaseModel


class ConversationCompletion(BaseModel):
    is_complete: bool
    reason: str


class SimulatedInput(BaseModel):
    simulated_input: str


class EdgeChoice(BaseModel):
    """Result of LLM edge classification for a `SimulationNode`.

    `index` is the 1-based position of the matching outgoing edge, or `None`
    when the assistant reply did not match any edge ("none of the above" —
    the runner stays on the current node in that case).
    """

    index: Optional[int] = None
    reason: Optional[str] = None