AI-Trader/docs/config.yaml

# AI Trader Frontend Configuration
# This file contains all agent configurations and display settings

# Market configurations - Easy to extend for new markets (e.g., crypto)
markets:
  us:
    name: "US Market (Nasdaq-100)"
    subtitle: "Track how different AI models perform in Nasdaq-100 stock trading"
    data_dir: "agent_data"
    benchmark_file: "Adaily_prices_QQQ.json"
    benchmark_name: "QQQ"
    benchmark_display_name: "QQQ Invesco"
    currency: "USD"
    icon: "🇺🇸"
    price_data_type: "individual"  # individual JSON files per stock
    time_granularity: "hourly"  # hourly timestamps with hour component
    enabled: true
    agents:  # US market specific agents
      - folder: "gemini-2.5-flash"
        display_name: "Gemini 2.5 Flash"
        icon: "./figs/google.svg"
        color: "#00d4ff"
      - folder: "qwen3-max"
        display_name: "Qwen3 Max"
        icon: "./figs/qwen.svg"
        color: "#00ffcc"
      - folder: "deepseek-chat-v3.1"
        display_name: "DeepSeek Chat v3.1"
        icon: "./figs/deepseek.svg"
        color: "#ff006e"
      - folder: "gpt-5"
        display_name: "GPT-5"
        icon: "./figs/openai.svg"
        color: "#ffbe0b"
      - folder: "claude-3.7-sonnet"
        display_name: "Claude 3.7 Sonnet"
        icon: "./figs/claude-color.svg"
        color: "#8338ec"
      - folder: "MiniMax-M2"
        display_name: "MiniMax M2"
        icon: "./figs/minimax.svg"
        color: "#3a86ff"

  cn:
    name: "A-Shares (SSE 50)"
    subtitle: "Track how different AI models perform in SSE 50 A-share stock trading"
    data_dir: "agent_data_astock"
    benchmark_file: "A_stock/index_daily_sse_50.json"
    benchmark_name: "SSE 50"
    benchmark_display_name: "SSE 50 Index"
    currency: "CNY"
    icon: "🇨🇳"
    price_data_type: "merged"  # merged.jsonl file
    price_data_file: "A_stock/merged.jsonl"
    time_granularity: "daily"  # daily dates without hour component
    enabled: true
    agents:  # CN market specific agents
      - folder: "gemini-2.5-flash"
        display_name: "Gemini 2.5 Flash"
        icon: "./figs/google.svg"
        color: "#00d4ff"
      - folder: "qwen3-max"
        display_name: "Qwen3 Max"
        icon: "./figs/qwen.svg"
        color: "#00ffcc"
      - folder: "deepseek-chat-v3.1"
        display_name: "DeepSeek Chat v3.1"
        icon: "./figs/deepseek.svg"
        color: "#ff006e"
      - folder: "gpt-5"
        display_name: "GPT-5"
        icon: "./figs/openai.svg"
        color: "#ffbe0b"
      - folder: "claude-3.7-sonnet"
        display_name: "Claude 3.7 Sonnet"
        icon: "./figs/claude-color.svg"
        color: "#8338ec"
      - folder: "MiniMax-M2"
        display_name: "MiniMax M2"
        icon: "./figs/minimax.svg"
        color: "#3a86ff"
      - folder: "glm-4.6"
        display_name: "GLM 4.6"
        icon: "./figs/zhipu-color.svg"
        color: "#6610f2"

  cn_hour:
    name: "A-Shares (Hourly)"
    subtitle: "Track how different AI models perform in SSE 50 A-share stock trading (Hourly)"
    data_dir: "agent_data_astock_hour"
    benchmark_file: "A_stock/index_daily_sse_50.json"
    benchmark_name: "SSE 50"
    benchmark_display_name: "SSE 50 Index"
    currency: "CNY"
    icon: "🇨🇳"
    price_data_type: "merged"  # merged.jsonl file
    price_data_file: "A_stock/merged_hourly.jsonl"
    time_granularity: "hourly"  # hourly timestamps
    enabled: false # Hidden from main selector, toggled via JS
    agents:  # CN market specific agents (hourly)
      - folder: "gemini-2.5-flash-astock-hour"
        display_name: "Gemini 2.5 Flash"
        icon: "./figs/google.svg"
        color: "#00d4ff"
      - folder: "qwen3-max-astock-hour"
        display_name: "Qwen3 Max"
        icon: "./figs/qwen.svg"
        color: "#00ffcc"
      - folder: "deepseek-chat-v3.1-astock-hour"
        display_name: "DeepSeek Chat v3.1"
        icon: "./figs/deepseek.svg"
        color: "#ff006e"
      - folder: "gpt-5-astock-hour"
        display_name: "GPT-5"
        icon: "./figs/openai.svg"
        color: "#ffbe0b"
      - folder: "claude-3.7-sonnet-astock-hour"
        display_name: "Claude 3.7 Sonnet"
        icon: "./figs/claude-color.svg"
        color: "#8338ec"
      - folder: "MiniMax-M2-astock-hour"
        display_name: "MiniMax M2"
        icon: "./figs/minimax.svg"
        color: "#3a86ff"
      - folder: "glm-4.6-astock-hour"
        display_name: "GLM 4.6"
        icon: "./figs/zhipu-color.svg"
        color: "#6610f2"

  # crypto:  # Future crypto market support
  #   name: "Crypto Market"
  #   subtitle: "Track how different AI models perform in cryptocurrency trading"
  #   data_dir: "agent_data_crypto"
  #   benchmark_file: "crypto/btc.json"
  #   benchmark_name: "BTC"
  #   benchmark_display_name: "Bitcoin"
  #   currency: "USD"
  #   icon: "₿"
  #   price_data_type: "individual"
  #   enabled: false

# Data paths configuration (legacy support - will use market-specific paths when available)
data:
  base_path: "./data"
  price_file_prefix: "daily_prices_"
  benchmark_file: "Adaily_prices_QQQ.json"

# Benchmark configuration
benchmark:
  folder: "QQQ"
  display_name: "QQQ Invesco"
  icon: "./figs/stock.svg"
  color: "#ff6b00"
  enabled: true

# Chart display settings
chart:
  default_scale: "linear"  # or "logarithmic"
  max_ticks: 15
  point_radius: 0
  point_hover_radius: 7
  border_width: 3
  tension: 0.42

# UI settings
ui:
  initial_value: 10000  # Starting cash for all agents
  max_recent_trades: 20  # Number of trades to show in portfolio view
  date_formats:
    hourly: "MM/DD HH:mm"
    daily: "YYYY-MM-DD"