vakra/docker-compose.yml at main · IBM/vakra · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
name: benchmark

# Alternative to `make start` / `python benchmark_setup.py --start-containers`.
# Requires the image to be built or pulled first:
#   make build          (build locally)
#   make pull           (pull from Docker Hub)
#
# Usage:
#   docker compose up -d                                        # start all capability containers
#   docker compose down                                         # stop and remove containers
#   docker compose logs -f                                      # tail all logs
#   docker compose up -d capability_4_multiturn                 # start a single capability
#
# The benchmark runner and MCP clients connect via docker exec (stdio),
# not via exposed ports. No port mapping is needed here.

x-base: &base
  image: benchmark_environ
  environment:
    MCP_VERIFY_CHECKSUMS: "1"
  volumes:
    - ./data/databases:/app/db:ro
    - ./environment/configs:/app/environment/configs:ro

services:
  capability_1_bi_apis:
    <<: *base
    container_name: capability_1_bi_apis

  capability_2_dashboard_apis:
    <<: *base
    container_name: capability_2_dashboard_apis

  capability_3_multihop_reasoning:
    <<: *base
    container_name: capability_3_multihop_reasoning

  capability_4_multiturn:
    <<: *base
    container_name: capability_4_multiturn
    mem_limit: 8g
    volumes:
      - ./data/databases:/app/db:ro
      - ./environment/configs:/app/environment/configs:ro
      - ./data/indexed_documents:/app/retrievers/chroma_data
      - ./data/queries:/app/retrievers/queries:ro

  # ---------------------------------------------------------------------------
  # Phoenix / Arize — optional observability UI for LLM traces
  # Start with:  docker compose --profile phoenix up -d
  # UI:          http://localhost:6006
  # OTLP HTTP:   http://localhost:6006/v1/traces  (used by benchmark_runner.py)
  # ---------------------------------------------------------------------------
  phoenix:
    image: arizephoenix/phoenix:latest
    profiles: [phoenix]
    container_name: m3_phoenix
    ports:
      - "6006:6006"
    environment:
      PHOENIX_WORKING_DIR: /phoenix
    volumes:
      - phoenix_data:/phoenix
    restart: unless-stopped

volumes:
  phoenix_data: