"""
=========================================
CV6 AI Trading OS
Portfolio Models
=========================================
"""

from pydantic import BaseModel


class PortfolioRequest(BaseModel):
    capital: float
    used_margin: float
    pnl: float
    open_positions: int


class PortfolioResponse(BaseModel):
    available_capital: float
    margin_used_percent: float
    portfolio_health: str