"""
=========================================
CV6 AI Trading OS
Strategy Models
=========================================
"""

from typing import Optional

from pydantic import BaseModel


class StrategyRequest(BaseModel):
    ema: float
    rsi: float
    macd: float
    supertrend: float
    atr: float
    vwap: float


class StrategyResponse(BaseModel):
    strategy: str
    buy_score: int
    sell_score: int
    confidence: float

    ema: Optional[float] = None
    rsi: Optional[float] = None
    macd: Optional[float] = None
    supertrend: Optional[float] = None
    atr: Optional[float] = None
    vwap: Optional[float] = None