"""
Pydantic schemas for FastAPI request/response models
"""
from pydantic import BaseModel


class CtaBacktestRequest(BaseModel):
    """CTA backtest request schema"""
    symbol: str
    strategy: str
    params: dict = {}
    start: str
    end: str


class OptimizeRequest(BaseModel):
    """Optimization request schema"""
    symbol: str
    strategy: str
    grid: dict
    start: str
    end: str
    max_workers: int = 2


class FactorAnalysisRequest(BaseModel):
    """Factor analysis request schema"""
    symbols: list[str]
    factor_names: list[str]
    start: str
    end: str