"""
=========================================
CV6 AI Trading OS
Broker Schemas
=========================================
"""

from pydantic import BaseModel
from typing import Optional


class BrokerConnect(BaseModel):
    """
    Connect Broker Request
    """

    broker_name: str
    api_key: str
    client_id: str
    access_token: Optional[str] = None


class BrokerResponse(BaseModel):
    """
    Broker Response
    """

    broker_name: str
    status: str
    message: str