app/models/policy/output_schema.py

"""Structured policy output schema."""

from __future__ import annotations

from pydantic import BaseModel, ConfigDict, Field

from app.common.enums import ActionType, DecisionMode, DoseBucket


class DecisionSchema(BaseModel):
    model_config = ConfigDict(extra="forbid")
    mode: DecisionMode
    action_type: ActionType
    target_drug: str | None
    replacement_drug: str | None
    dose_bucket: DoseBucket
    taper_days: int | None
    monitoring_plan: str | None
    evidence_query: str | None = None
    new_drug_name: str | None = None
    candidate_components: list[str] = Field(default_factory=list)
    candidate_id: str
    confidence: float
    rationale_brief: str | None = None