# generated by datamodel-codegen:
#   filename:  router_decision.json

from __future__ import annotations

from enum import StrEnum

from pydantic import BaseModel, ConfigDict, constr


class WorkflowId(StrEnum):
    day_ahead_situation = 'day-ahead-situation'
    day_ahead_bid = 'day-ahead-bid'
    adhoc_analysis = 'adhoc-analysis'


class Params(BaseModel):
    model_config = ConfigDict(
        extra='forbid',
    )
    market_date: constr(pattern=r'^\d{4}-\d{2}-\d{2}$')


class Confidence(StrEnum):
    HIGH = 'HIGH'
    MEDIUM = 'MEDIUM'
    LOW = 'LOW'


class RouterDecision(BaseModel):
    model_config = ConfigDict(
        extra='forbid',
    )
    workflow_id: WorkflowId
    params: Params
    confidence: Confidence