"""
=========================================
CV6 AI Trading OS
Risk Models
=========================================
"""

from typing import Optional
from pydantic import BaseModel


class RiskRequest(BaseModel):
    capital: float
    risk_percent: float

    entry_price: float
    stop_loss: float
    target_price: float


class RiskResponse(BaseModel):
    max_risk_amount: float

    quantity: int

    risk_per_share: float

    reward_per_share: float

    risk_reward_ratio: float

    valid_trade: bool

    message: str