vpp-ai-platform/skills-py/vpp_contracts/forecast_bundle.py

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

from __future__ import annotations

from enum import StrEnum
from typing import Literal

from pydantic import AwareDatetime, BaseModel, ConfigDict, Field, RootModel, constr


class Kind(StrEnum):
    LOAD = 'LOAD'
    PV = 'PV'
    PRICE = 'PRICE'


class Unit(StrEnum):
    mw = 'mw'
    yuan_per_mwh = 'yuan_per_mwh'


class Value(RootModel[constr(pattern=r'^-?\d+(\.\d+)?$')]):
    root: constr(pattern=r'^-?\d+(\.\d+)?$')


class P10(BaseModel):
    model_config = ConfigDict(
        extra='forbid',
    )
    interval_minutes: Literal[15]
    date: constr(pattern=r'^\d{4}-\d{2}-\d{2}$')
    values: list[Value] = Field(..., max_length=96, min_length=96)


class P50(BaseModel):
    model_config = ConfigDict(
        extra='forbid',
    )
    interval_minutes: Literal[15]
    date: constr(pattern=r'^\d{4}-\d{2}-\d{2}$')
    values: list[Value] = Field(..., max_length=96, min_length=96)


class P90(BaseModel):
    model_config = ConfigDict(
        extra='forbid',
    )
    interval_minutes: Literal[15]
    date: constr(pattern=r'^\d{4}-\d{2}-\d{2}$')
    values: list[Value] = Field(..., max_length=96, min_length=96)


class Quantiles(BaseModel):
    model_config = ConfigDict(
        extra='forbid',
    )
    p10: P10
    p50: P50
    p90: P90


class Model(BaseModel):
    model_config = ConfigDict(
        extra='forbid',
    )
    name: constr(min_length=1)
    version: constr(min_length=1)


class ForecastBundle(BaseModel):
    model_config = ConfigDict(
        extra='forbid',
    )
    id: constr(min_length=1)
    kind: Kind
    market_date: constr(pattern=r'^\d{4}-\d{2}-\d{2}$')
    unit: Unit
    quantiles: Quantiles
    model: Model
    features_snapshot_ref: constr(pattern=r'^[0-9a-f]{64}$')
    generated_at: AwareDatetime