Composer Trade MCP

from pydantic import BaseModel, Field
from typing import Dict, List, Optional, Union
from datetime import date

class LegendEntry(BaseModel):
    """Schema for a legend entry that maps a ticker/symbol ID to a display name."""
    name: str = Field(..., description="Display name for the ticker/symbol")

DvmCapitalEntry = Dict[int, float]
DvmCapital = Dict[str, DvmCapitalEntry]
Legend = Dict[str, LegendEntry]

class ParsedDailyValue(BaseModel):
    """Schema for a single parsed daily value row."""
    date: str = Field(..., description="Date in YYYY-MM-DD format")
    # Additional fields will be dynamically added based on the tickers in the data

class ParsedDailyValues(BaseModel):
    """Schema for the parsed daily values returned by parse_dvm_capital."""
    values: List[ParsedDailyValue] = Field(..., description="List of daily value rows")

class BacktestResponse(BaseModel):
    """Schema for the response from the backtest API."""
    data_warnings: Optional[Dict[str, List[Dict[str, str]]]] = Field(None, description="List of data warnings")
    first_day: Optional[int] = Field(None, description="First day of the backtest")
    capital: Optional[float] = Field(None, description="Initial capital of the backtest")
    last_market_day: Optional[int] = Field(None, description="Last market day of the backtest")
    last_market_days_holdings: Optional[Dict[str, float]] = Field(None, description="Last market days shares of the backtest")
    last_market_days_value: Optional[float] = Field(None, description="Last market days value of the backtest")
    stats: Optional[Dict] = Field(None, description="Stats of the backtest")
    dvm_capital: Optional[DvmCapital] = Field(None, description="DVM capital of the backtest")
    legend: Optional[Legend] = Field(None, description="Legend of the backtest")