37 lines
734 B
Python
37 lines
734 B
Python
|
# 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
|