docker-compose.yml

services:
  # I asked for "a very visible warning in "docker compose up" that we are running mock LLMs. FAir..
  mock-warning:
    image: alpine
    environment:
      - IS_INTEGRATION_TEST=${IS_INTEGRATION_TEST:-false}
    command: >
      sh -c "if [ \"$IS_INTEGRATION_TEST\" = \"true\" ]; then
        echo ' '
        echo '****************************************************'
        echo '*                                                  *'
        echo '*   WARNING: MOCK LLM MODE IS ACTIVE               *'
        echo '*   (IS_INTEGRATION_TEST=true)                      *'
        echo '*   Real reasoning traces will NOT be generated.   *'
        echo '*                                                  *'
        echo '****************************************************'
        echo ' '
      fi"

  controller:
    build:
      context: .
      dockerfile: controller/Dockerfile
    image: problemologist-ai-controller:latest
    ports:
      - "${CONTROLLER_HOST_PORT:-18000}:8000"
    environment:
      - POSTGRES_URL=postgresql+asyncpg://postgres:postgres@postgres:5432/postgres
      - S3_ENDPOINT=http://minio:9000
      - S3_ACCESS_KEY=minioadmin
      - S3_SECRET_KEY=minioadmin
      - OPENAI_API_KEY=${OPENAI_API_KEY}
      - OPENAI_API_BASE=${OPENAI_API_BASE}
      - IS_INTEGRATION_TEST=${IS_INTEGRATION_TEST:-false}
      - TEMPORAL_URL=temporal:7233
      - WORKER_URL=http://worker-light:8001
      - WORKER_LIGHT_TRANSPORT=ws
      - WORKER_HEAVY_URL=http://worker-heavy:8002
      - WORKER_RENDERER_URL=http://worker-renderer:8003
      - DATABASE_URL=postgresql+asyncpg://postgres:postgres@postgres:5432/postgres
      - ASSET_S3_BUCKET=problemologist
      - BACKUP_S3_BUCKET=problemologist-backup
      - LOG_LEVEL=${LOG_LEVEL:-INFO}
    networks:
      - default
    command:
      [
        "uvicorn",
        "controller.api.main:app",
        "--host",
        "0.0.0.0",
        "--port",
        "8000",
        "--reload",
        "--reload-dir",
        "controller",
        "--reload-dir",
        "shared",
      ]
    env_file:
      - .env
    depends_on:
      postgres:
        condition: service_healthy
      minio:
        condition: service_healthy
      temporal:
        condition: service_started
      migrations:
        condition: service_completed_successfully
    healthcheck:
      test: ["CMD", "curl", "-f", "http://localhost:8000/health"]
      interval: 5s
      timeout: 5s
      retries: 5

    # note to coding agents: bloody STOP putting hot reload servers into this config. They burn my CPU and crash the IDE with OOM.
    # use `docker compose up --build` instead - it's 3 seconds.

  migrations:
    image: problemologist-ai-controller:latest
    command: ["alembic", "-c", "alembic.ini", "upgrade", "head"]
    environment:
      - POSTGRES_URL=postgresql+psycopg://postgres:postgres@postgres:5432/postgres
    depends_on:
      postgres:
        condition: service_healthy

  controller-temporal-worker:
    image: problemologist-ai-controller:latest
    command: ["python", "controller/temporal_worker.py"]
    environment:
      - POSTGRES_URL=postgresql+asyncpg://postgres:postgres@postgres:5432/postgres
      - S3_ENDPOINT=http://minio:9000
      - S3_ACCESS_KEY=minioadmin
      - S3_SECRET_KEY=minioadmin
      - ASSET_S3_BUCKET=problemologist
      - OPENAI_API_KEY=${OPENAI_API_KEY}
      - OPENAI_API_BASE=${OPENAI_API_BASE}
      - IS_INTEGRATION_TEST=${IS_INTEGRATION_TEST:-false}
      - TEMPORAL_URL=temporal:7233
      - WORKER_URL=http://worker-light:8001
      - WORKER_LIGHT_TRANSPORT=ws
      - WORKER_HEAVY_URL=http://worker-heavy:8002
      - WORKER_RENDERER_URL=http://worker-renderer:8003
      - LOG_LEVEL=${LOG_LEVEL:-INFO}
    env_file:
      - .env
    depends_on:
      temporal:
        condition: service_started
      minio:
        condition: service_healthy
      postgres:
        condition: service_healthy

    # note to coding agents: bloody STOP putting hot reload servers into this config. They burn my CPU and crash the IDE with OOM.
    # use docker compose up --build instead - it's 3 seconds.

    # develop:
    #   watch:
    #     - action: sync+restart
    #       path: ./controller
    #       target: /app/controller
    #     - action: sync+restart
    #       path: ./shared
    #       target: /app/shared
    #     - action: sync+restart
    #       path: ./config
    #       target: /app/config

  worker-light:
    build:
      context: .
      dockerfile: worker_light/Dockerfile
    image: problemologist-ai-worker-light:latest
    ports:
      - "${WORKER_LIGHT_HOST_PORT:-18001}:8001"
    environment:
      - S3_ENDPOINT=http://minio:9000
      - AWS_ACCESS_KEY_ID=minioadmin
      - AWS_SECRET_ACCESS_KEY=minioadmin
      - LOG_LEVEL=${LOG_LEVEL:-INFO}
      - WORKER_TYPE=light
    networks:
      - default
    command:
      ["uvicorn", "worker_light.app:app", "--host", "0.0.0.0", "--port", "8001"]
    env_file:
      - .env
    depends_on:
      minio:
        condition: service_healthy
    volumes:
      - ./worker_light:/app/worker_light
      - ./shared:/app/shared

  worker-heavy:
    build:
      context: .
      dockerfile: worker_heavy/Dockerfile
    image: problemologist-ai-worker-heavy:latest
    ports:
      - "${WORKER_HEAVY_HOST_PORT:-18002}:8002"
    environment:
      - S3_ENDPOINT=http://minio:9000
      - AWS_ACCESS_KEY_ID=minioadmin
      - AWS_SECRET_ACCESS_KEY=minioadmin
      - EXTRA_DEBUG_LOG=/app/logs/worker_renderer_debug.log
      - EXTRA_ERROR_LOG=/app/logs/worker_renderer_errors.log
      - EXTRA_ERROR_JSON_LOG=/app/logs/json/worker_renderer_errors.json
      - MUJOCO_GL=osmesa
      - PYOPENGL_PLATFORM=osmesa
      - PYVISTA_OFF_SCREEN=true
      - VTK_DEFAULT_OPENGL_RENDERER=OSMesa
      - LOG_LEVEL=${LOG_LEVEL:-INFO}
      - WORKER_TYPE=heavy
      - TEMPORAL_URL=temporal:7233
      - CONTROLLER_URL=http://controller:8000
      - WORKER_RENDERER_URL=http://worker-renderer:8003
    networks:
      - default
    command:
      ["uvicorn", "worker_heavy.app:app", "--host", "0.0.0.0", "--port", "8002"]
    env_file:
      - .env
    depends_on:
      minio:
        condition: service_healthy
      temporal:
        condition: service_started
      worker-renderer:
        condition: service_healthy
    volumes:
      - ./worker_heavy:/app/worker_heavy
      - ./worker_light:/app/worker_light
      - ./shared:/app/shared

  worker-renderer:
    build:
      context: .
      dockerfile: worker_renderer/Dockerfile
    image: problemologist-ai-worker-renderer:latest
    ports:
      - "${WORKER_RENDERER_HOST_PORT:-18003}:8003"
    environment:
      - S3_ENDPOINT=http://minio:9000
      - AWS_ACCESS_KEY_ID=minioadmin
      - AWS_SECRET_ACCESS_KEY=minioadmin
      - LOG_LEVEL=${LOG_LEVEL:-INFO}
      - PROBLEMOLOGIST_RENDER_PARALLEL_MODALITIES=false
    networks:
      - default
    env_file:
      - .env
    depends_on:
      minio:
        condition: service_healthy
    volumes:
      - ./worker_renderer:/app/worker_renderer
      - ./shared:/app/shared
      - ./config:/app/config
      - ${WORKER_RENDERER_LOG_DIR:-./logs}:/app/logs

  worker-heavy-temporal:
    image: problemologist-ai-worker-heavy:latest
    environment:
      - TEMPORAL_URL=temporal:7233
      - CONTROLLER_URL=http://controller:8000
      - LOG_LEVEL=${LOG_LEVEL:-INFO}
      - WORKER_RENDERER_URL=http://worker-renderer:8003
    networks:
      - default
    command: ["python", "-m", "worker_heavy.temporal_worker"]
    env_file:
      - .env
    depends_on:
      temporal:
        condition: service_started
      worker-heavy:
        condition: service_started
      worker-renderer:
        condition: service_healthy
    volumes:
      - ./worker_heavy:/app/worker_heavy
      - ./worker_light:/app/worker_light
      - ./shared:/app/shared
    # develop:
    #   watch:
    #     - action: sync
    #       path: ./worker
    #       target: /app/worker
    #     - action: sync
    #       path: ./shared
    #       target: /app/shared
    #     - action: rebuild
    #       path: pyproject.toml

  postgres:
    image: postgres:15-alpine
    environment:
      - POSTGRES_USER=postgres
      - POSTGRES_PASSWORD=postgres
      - POSTGRES_DB=postgres
    ports:
      - "${POSTGRES_HOST_PORT:-15432}:5432"
    healthcheck:
      test: ["CMD-SHELL", "pg_isready -U postgres"]
      interval: 5s
      timeout: 5s
      retries: 5
    volumes:
      - postgres_data:/var/lib/postgresql/data
  minio:
    image: minio/minio
    ports:
      - "${MINIO_HOST_PORT:-19000}:9000"
      - "${MINIO_CONSOLE_HOST_PORT:-19001}:9001"
    environment:
      - MINIO_ROOT_USER=minioadmin
      - MINIO_ROOT_PASSWORD=minioadmin
    command: server /data --console-address ":9001"
    volumes:
      - minio_data:/data
    healthcheck:
      test: ["CMD", "curl", "-f", "http://localhost:9000/minio/health/live"]
      interval: 5s
      timeout: 5s
      retries: 5

    # ulimits:
    #   nofile:
    #     soft: 262144
    #     hard: 262144

  temporal:
    image: temporalio/auto-setup:1.22.0
    ports:
      - "${TEMPORAL_HOST_PORT:-17233}:7233"
    environment:
      - DB=postgresql
      - DB_PORT=5432
      - POSTGRES_USER=postgres
      - POSTGRES_PWD=postgres
      - POSTGRES_SEEDS=postgres
    depends_on:
      postgres:
        condition: service_healthy

  temporal-ui:
    image: temporalio/ui:2.21.3
    ports:
      - "${TEMPORAL_UI_HOST_PORT:-18081}:8080"
    environment:
      - TEMPORAL_ADDRESS=temporal:7233
    depends_on:
      temporal:
        condition: service_started

  createbuckets:
    image: minio/mc
    depends_on:
      minio:
        condition: service_healthy
    entrypoint: >
      /bin/sh -c "
      until /usr/bin/mc alias set myminio http://minio:9000 minioadmin minioadmin; do echo 'Waiting for Minio...'; sleep 1; done;
      /usr/bin/mc mb myminio/problemologist || true;
      /usr/bin/mc mb myminio/problemologist-backup || true;
      exit 0;
      "

volumes:
  postgres_data:
  minio_data: