o cj@sddlmZmZmZmZmZmZddlmZm Z m Z ddl m Z m Z ddl mZGdddeeZGdddeeZGd d d eZGd d d eZGd ddeZGdddeZGdddeZGdddeZGdddeZGdddeZGdddeZdS)) BaseModelFieldfield_validatorEmailStrValidationInfomodel_validator)OptionalListLiteral)datetimedate)Enumc@seZdZdZdZdZdS) SortDirectionzEnumeration for sort directionsascdescN)__name__ __module__ __qualname____doc__ASCDESCrr9/var/www/html/oms-prod-mcp/app/schemas/request_schemas.pyrsrc@seZdZdZdZdZdZdS) DatabaseTypezEnumeration for database typesmysqlmongodb cassandraN)rrrrMYSQLMONGODB CASSANDRArrrrrs rc@sNeZdZUdZedddZeed<edddZe ed<e de dd Z d S) TimeSeriesDataPointz*Schema for a single time series data point.z@Date/timestamp in ISO format (YYYY-MM-DD or YYYY-MM-DD HH:MM:SS) descriptiondsz!Numeric value for the time seriesycCs0z t|dd|WStytdw)zValidate date formatZz+00:00z>Date must be in ISO format (YYYY-MM-DD or YYYY-MM-DD HH:MM:SS))r fromisoformatreplace ValueError)clsvrrr validate_dates  z!TimeSeriesDataPoint.validate_dateN) rrrrrr#str__annotations__r$floatr classmethodr+rrrrr s r c@seZdZUdZedddZeeded<edddZ ee ed <edd dZ ee ed <ed d dZ ee ed<edddddZ eeed<edddddZeeed<edddddZeeed<edddddZeeed <edddd!dZeeed"<d#S)$ ProphetConfigz&Schema for Prophet model configurationadditivezSeasonality moder!)r1multiplicativeseasonality_modeTzEnable yearly seasonalityyearly_seasonalityzEnable weekly seasonalityweekly_seasonalityFzEnable daily seasonalitydaily_seasonalityg?gMbP?2zChangepoint prior scalegeler"changepoint_prior_scaleg$@g{Gz?dzSeasonality prior scaleseasonality_prior_scalezHolidays prior scaleholidays_prior_scaleg?gg?zChangepoint rangechangepoint_rangeu!Confidence interval width (0–1)interval_widthN)rrrrrr3rr r-r4boolr5r6r;r.r=r>r?r@rrrrr0)s. r0c@s`eZdZUdZeddddZeeed<eddd d d Z e e ed <edd dZ e e ed<dS)ProphetTrainRequestz#Schema for training a Prophet model.zTime series data points min_lengthr"dataNr<zOptional model identifier)rE max_lengthr"model_idzProphet model configurationr!config)rrrrrrFr r r-rIrr,rJr0rrrrrBCs rBc@sbeZdZUdZedddddZeeed<edd d Z ee d ed <ed dd Z ee ed<dS)ProphetForecastRequestzSchema for generating forecastsrGmNumber of periods to forecastr8periodsDForecast frequencyr!rPHWMQYfreqTz#Include historical data in responseinclude_historyN) rrrrrrOrintr-rXr rYrArrrrrKKs rKc@seZdZUdZedddddZeeed<edd d Z ee d ed <ed ddddZ eeed<edddddZ eeed<dS)ProphetPlotRequestz$Schema for generating forecast plotsrLrGrMrNr8rOrPrQr!rRrXi iizPlot width in pixelswidthiXi,izPlot height in pixelsheightN) rrrrrrOrrZr-rXr r\r]rrrrr[Ss  r[c@s,eZdZUdZeddddZeeed<dS)ProphetValidateRequestz&Schema for validating time series data.rGz#Time series data points to validaterDrFN) rrrrrrFr r r-rrrrr^\s r^c@s@eZdZUdZeddddZeed<eddd Ze e ed <dS) RegressionTrainRequestz8Schema for training a regression or classification model.rz!Company ID to scope training datagtr" company_idNz)Optional XGBoost hyperparameter overridesr!rJ) rrrrrrbrZr-rJrdictrrrrr_ds r_c@sDeZdZUdZeddddZeed<eddd Ze e eed <dS) CollectionRiskScoreRequestz)Schema for scoring client collection risk.r Company IDr`rbNz*Specific client IDs to score; omit for allr! client_ids) rrrrrrbrZr-rfrr rrrrrdjs  rdc@sveZdZUdZeddddZeed<eddddZeed<ed d d Z e e eed <ed d d Z e e eed<d S)SalesTargetPredictRequestz-Schema for predicting sales target attainment.rrer`rbzAccount year to predict foraccount_year_idNzFilter by product head IDsr!product_head_idszFilter by district IDs district_ids) rrrrrrbrZr-rhrirr rjrrrrrgps  rgN)pydanticrrrrrrtypingrr r r r enumr r,rrr r0rBrKr[r^r_rdrgrrrrs