compose.yaml

# Environment variables shared by all agents
x-beeai-env: &beeai-env
  MCP_GATEWAY_URL: http://mcp-gateway:8000/sse
  UPSTREAM_SEARCH_API_URL: http://upstream-search.hosted.upshift.rdu2.redhat.com:80/v1
  REDIS_URL: redis://valkey:6379/0
  COLLECTOR_ENDPOINT: http://phoenix:6006/v1/traces
  GIT_REPO_BASEPATH: /git-repos
  MAX_RETRIES: 3
  DRY_RUN: ${DRY_RUN:-false}
  AUTO_CHAIN: ${AUTO_CHAIN:-true}
  REQUESTS_CA_BUNDLE: /etc/pki/tls/certs/ca-bundle.crt

# Base template for BeeAI agents
x-beeai-agent: &beeai-agent
  depends_on:
    - mcp-gateway
    - valkey
    - phoenix
  environment:
    <<: *beeai-env
  env_file:
    - .secrets/beeai-agent.env
  volumes:
    - ./ymir/agents:/home/beeai/ymir/agents:ro,z
    - ./ymir/tools:/home/beeai/ymir/tools:ro,z
    - ./ymir/common:/home/beeai/ymir/common:ro,z
    - git-repos:/git-repos:rw,U
    - .secrets/rhel-config.json:/home/beeai/rhel-config.json:ro,z,U
    - .secrets/jotnar-vertex-dev.json:/home/beeai/jotnar-vertex-dev.json:ro,z,U
  restart: unless-stopped

# C9S agent template
x-beeai-agent-c9s: &beeai-agent-c9s
  <<: *beeai-agent
  image: beeai-agent:c9s
  build:
    context: .
    dockerfile: Containerfile.c9s
  environment:
    <<: *beeai-env
    CONTAINER_VERSION: c9s

# C10S agent template
x-beeai-agent-c10s: &beeai-agent-c10s
  <<: *beeai-agent
  image: beeai-agent:c10s
  build:
    context: .
    dockerfile: Containerfile.c10s
  environment:
    <<: *beeai-env
    CONTAINER_VERSION: c10s

x-supervisor: &supervisor
  depends_on:
    - valkey
    - phoenix
  image: supervisor
  build:
    context: .
    dockerfile: Containerfile.supervisor
  environment:
    - REDIS_URL=redis://valkey:6379/0
    - COLLECTOR_ENDPOINT=http://phoenix:6006/v1/traces
    - GIT_REPO_BASEPATH=/git-repos
    - "KRB5CCNAME=FILE:/home/beeai/ccache/krb5cc"
  env_file:
    - .secrets/supervisor.env
  volumes:
    - ./ymir/agents:/home/beeai/ymir/agents:ro,z
    - ./ymir/tools:/home/beeai/ymir/tools:ro,z
    - ./ymir/common:/home/beeai/ymir/common:ro,z
    - ./ymir/supervisor:/home/beeai/ymir/supervisor:ro,z
    - .secrets/ccache:/home/beeai/ccache:ro,z,U
  restart: unless-stopped

services:
  mcp-gateway:
    image: mcp-gateway
    build:
      context: .
      dockerfile: Containerfile.mcp
    environment:
      - SSE_PORT=8000
      - KEYTAB_FILE=/home/mcp/keytab
      # default cache location is a keyring
      - KRB5CCNAME=FILE:/tmp/krb5cc
      - DRY_RUN=${DRY_RUN:-false}
      - MOCK_JIRA=${MOCK_JIRA:-false}
      - JIRA_DRY_RUN=${JIRA_DRY_RUN:-false}
      - GIT_REPO_BASEPATH=/git-repos
      - JIRA_MOCK_FILES=ymir/tools/privileged/tests/data
    env_file:
      - .secrets/mcp-gateway.env
    volumes:
      - ./ymir/tools/privileged:/home/mcp/ymir/tools/privileged:ro,z
      - ./ymir/tools/privileged/tests/data:/home/mcp/ymir/tools/privileged/tests/data:rw,z
      - ./ymir/common:/home/mcp/ymir/common:ro,z
      - .secrets/keytab:/home/mcp/keytab:ro,z,U
      - git-repos:/git-repos
      - .secrets/rhel-config.json:/home/mcp/rhel-config.json:ro,z,U
    command: ["python", "-m", "ymir.tools.privileged.gateway"]
    restart: unless-stopped
    profiles: ["agents", "supervisor", "e2e-test"]

  valkey:
    image: docker.io/valkey/valkey:8
    volumes:
      - valkey-data:/data
    restart: unless-stopped
    profiles: ["agents", "supervisor", "e2e-test"]

  phoenix:
    image: docker.io/arizephoenix/phoenix:version-11.6.2
    ports:
      - "0.0.0.0:6006:6006"
    environment:
      - PHOENIX_WORKING_DIR=/mnt/data
    volumes:
      - phoenix-data:/mnt/data
    restart: unless-stopped
    profiles: ["agents", "supervisor", "e2e-test"]

  redis-commander:
    image: ghcr.io/joeferner/redis-commander:0.9.0
    environment:
      - REDIS_HOSTS=local:valkey:6379
    ports:
      - "8081:8081"
    depends_on:
      - valkey
    restart: unless-stopped
    profiles: ["agents", "supervisor"]

  triage-agent:
    <<: *beeai-agent-c10s
    environment:
      <<: *beeai-env
    command: ["python", "-m", "ymir.agents.triage_agent"]
    profiles: ["agents"]

  triage-agent-e2e-tests:
    <<: *beeai-agent-c10s
    environment:
      <<: *beeai-env
    # the option about default loop is here because of litellm issue
    # https://github.com/BerriAI/litellm/issues/14521
    command: ["pytest", "ymir/agents/tests/e2e/test_triage.py", "-o", "asyncio_default_test_loop_scope=session"]
    profiles: ["e2e-test"]

  backport-agent-c9s:
    <<: *beeai-agent-c9s
    command: ["python", "-m", "ymir.agents.backport_agent"]
    profiles: ["agents"]

  backport-agent-c10s:
    <<: *beeai-agent-c10s
    command: ["python", "-m", "ymir.agents.backport_agent"]
    profiles: ["agents"]

  rebase-agent-c9s:
    <<: *beeai-agent-c9s
    command: ["python", "-m", "ymir.agents.rebase_agent"]
    profiles: ["agents"]

  rebase-agent-c10s:
    <<: *beeai-agent-c10s
    command: ["python", "-m", "ymir.agents.rebase_agent"]
    profiles: ["agents"]

  rebuild-agent-c9s:
    <<: *beeai-agent-c9s
    command: ["python", "agents/rebuild_agent.py"]
    profiles: ["agents"]

  rebuild-agent-c10s:
    <<: *beeai-agent-c10s
    command: ["python", "agents/rebuild_agent.py"]
    profiles: ["agents"]

  mr-agent-c9s:
    <<: *beeai-agent-c9s
    command: ["python", "-m", "ymir.agents.merge_request_agent"]
    profiles: ["agents"]

  mr-agent-c10s:
    <<: *beeai-agent-c10s
    command: ["python", "-m", "ymir.agents.merge_request_agent"]
    profiles: ["agents"]

  # This is here so we can use it for '$(COMPOSE) run'
  supervisor:
    <<: *supervisor
    command: ["sh", "-c", "echo 'This service isn't run directly' && exit 1"]
    profiles: []
    restart: no


  jira-issue-fetcher:
    # No hard dependencies - will connect to existing services or fail gracefully
    # otherwise it will kill agents already running
    image: jira-issue-fetcher
    build:
      context: .
      dockerfile: Containerfile.jira-issue-fetcher
    environment:
      - REDIS_URL=redis://valkey:6379/0
    env_file:
      - .secrets/jira-issue-fetcher.env
    restart: "no"
    profiles: ["manual"]  # Don't start by default, run manually for testing

volumes:
  valkey-data:
  phoenix-data:
  git-repos: