docker-compose.yml

services:
  ai-service:
    build:
      context: ./ai-service
    environment:
      - KRONOS_DEVICE=cpu
      # Model Registry: HuggingFace model IDs (weights downloaded at startup)
      - KRONOS_MODEL_ID=NeoQuasar/Kronos-small
      - KRONOS_TOKENIZER_ID=NeoQuasar/Kronos-Tokenizer-base
      # Model Parameters
      - KRONOS_MAX_CONTEXT=512
      - KRONOS_CLIP=5
      # Optional: HuggingFace token for private models (mount as secret in production)
      # - HF_TOKEN=${HF_TOKEN}
    ports:
      - "5001:5001"
    # Optional: Mount HuggingFace cache to persist downloaded models
    # volumes:
    #   - ./cache/huggingface:/root/.cache/huggingface

  trading-service:
    build:
      context: .
    environment:
      - PREDICTION_SERVICE_BASE_URL=http://ai-service:5001
    ports:
      - "8080:8080"
    depends_on:
      - ai-service

  trading-ui:
    build:
      context: ./trading-ui
    environment:
      - NEXT_PUBLIC_API_BASE_URL=http://localhost:8080
    ports:
      - "3000:3000"
    depends_on:
      - trading-service