multi-agent-researcher/docker-compose.yml at main · nikhilll30/multi-agent-researcher · 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
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
version: "3.11"

# ---------------------------------------------------------------------------
# Multi-Agent Research Assistant — Docker Compose
#
# Start everything:
#   docker-compose up --build
#
# Environment variables are loaded from .env (see .env.example).
# ---------------------------------------------------------------------------

services:

  # -------------------------------------------------------------------------
  # Web Search Agent — A2A server wrapping Tavily live-web search
  # -------------------------------------------------------------------------
  web_search_agent:
    build:
      context: .
      dockerfile: Dockerfile
    command: python agents/web_search/agent.py
    ports:
      - "8001:8001"
    env_file: .env
    environment:
      - TAVILY_API_KEY=${TAVILY_API_KEY}
    restart: unless-stopped
    healthcheck:
      test: ["CMD", "python", "-c", "import urllib.request; urllib.request.urlopen('http://localhost:8001/health')"]
      interval: 15s
      timeout: 5s
      retries: 5
      start_period: 10s

  # -------------------------------------------------------------------------
  # RAG Agent — A2A server backed by ChromaDB via MCP
  # -------------------------------------------------------------------------
  rag_agent:
    build:
      context: .
      dockerfile: Dockerfile
    command: python agents/rag_agent/agent.py
    ports:
      - "8002:8002"
    env_file: .env
    environment:
      - ANTHROPIC_API_KEY=${ANTHROPIC_API_KEY}
    volumes:
      - ../rag-doc-qa:/rag-doc-qa
    restart: unless-stopped
    healthcheck:
      test: ["CMD", "python", "-c", "import urllib.request; urllib.request.urlopen('http://localhost:8002/health')"]
      interval: 15s
      timeout: 5s
      retries: 5
      start_period: 10s

  # -------------------------------------------------------------------------
  # Synthesis Agent — A2A server that writes the final report with Claude
  # -------------------------------------------------------------------------
  synthesis_agent:
    build:
      context: .
      dockerfile: Dockerfile
    command: python agents/synthesis/agent.py
    ports:
      - "8003:8003"
    env_file: .env
    environment:
      - ANTHROPIC_API_KEY=${ANTHROPIC_API_KEY}
    restart: unless-stopped
    healthcheck:
      test: ["CMD", "python", "-c", "import urllib.request; urllib.request.urlopen('http://localhost:8003/health')"]
      interval: 15s
      timeout: 5s
      retries: 5
      start_period: 10s

  # -------------------------------------------------------------------------
  # Orchestrator — LangGraph state machine exposed as a FastAPI REST API
  # -------------------------------------------------------------------------
  orchestrator:
    build:
      context: .
      dockerfile: Dockerfile
    command: uvicorn api.main:app --host 0.0.0.0 --port 8000
    ports:
      - "8000:8000"
    env_file: .env
    environment:
      - ANTHROPIC_API_KEY=${ANTHROPIC_API_KEY}
    depends_on:
      web_search_agent:
        condition: service_healthy
      rag_agent:
        condition: service_healthy
      synthesis_agent:
        condition: service_healthy
    restart: unless-stopped
    healthcheck:
      test: ["CMD", "python", "-c", "import urllib.request; urllib.request.urlopen('http://localhost:8000/health')"]
      interval: 15s
      timeout: 5s
      retries: 5
      start_period: 15s

  # -------------------------------------------------------------------------
  # Streamlit UI — interactive front-end for end users
  # -------------------------------------------------------------------------
  ui:
    build:
      context: .
      dockerfile: Dockerfile
    command: streamlit run ui/app.py --server.port 8501 --server.address 0.0.0.0
    ports:
      - "8501:8501"
    env_file: .env
    depends_on:
      orchestrator:
        condition: service_healthy
    restart: unless-stopped