ai-workbench/docker-compose.yml at main · datastax/ai-workbench · 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
# AI Workbench — end-user quickstart.
#
#   docker compose up
#
# Opens http://localhost:8080. Data persists in the `workbench-data`
# named volume across `down` / `up` cycles; wipe with `down -v`. To
# enable Astra-backed workspaces or chat (OpenRouter), drop a `.env`
# file next to this one — it's auto-loaded when present.
#
# See docs/docker.md for the full guide (overrides, upgrading, etc.).

services:
  workbench:
    image: ghcr.io/datastax/ai-workbench:${AI_WORKBENCH_TAG:-latest}
    container_name: ai-workbench
    ports:
      # Override the host port with AI_WORKBENCH_PORT=9090 docker compose up.
      - "${AI_WORKBENCH_PORT:-8080}:8080"
    environment:
      # Bundled in the image — pins the control plane to file driver
      # rooted at /var/lib/workbench (mounted below).
      WORKBENCH_CONFIG: /app/examples/workbench.docker.yaml
      # Persistent volume root. Used by:
      #   - the file-driver control plane (above)
      #   - the setup wizard's managed env file at $WORKBENCH_DATA_DIR/.env
      #   - the `aiw` CLI's profile store at $WORKBENCH_DATA_DIR/cli/config.json
      WORKBENCH_DATA_DIR: /var/lib/workbench
      # Auto-load the wizard-written env file on every boot. When the
      # file doesn't exist yet (fresh install) the runtime still starts
      # — `WORKBENCH_ENV_FILE` is only fatal if explicitly set AND
      # missing-on-disk; the wizard creates it before /setup/restart.
      WORKBENCH_ENV_FILE: /var/lib/workbench/.env
      # Ollama default endpoint. Inside the container `localhost` is
      # the container itself, so the default points at the Docker host
      # via the gateway alias mapped under `extra_hosts` below.
      # Override (here or in `.env`) if Ollama listens elsewhere.
      OLLAMA_BASE_URL: ${OLLAMA_BASE_URL:-http://host.docker.internal:11434/v1}
    env_file: .env
      # Optional. `.env` carries ASTRA_DB_* / OPENROUTER_API_KEY /
      # other workspace credential refs. Compose v2 honors `required:
      # false` so the stack still boots without a `.env`.
    volumes:
      - workbench-data:/var/lib/workbench
    extra_hosts:
      # Make `host.docker.internal` resolve on Linux Engine too (it's
      # built in on Docker Desktop). `host-gateway` maps to the host's
      # gateway IP, which is how the container reaches services like
      # Ollama listening on the host.
      - "host.docker.internal:host-gateway"
    restart: unless-stopped
    healthcheck:
      test: ["CMD", "wget", "-qO-", "http://localhost:8080/healthz"]
      interval: 30s
      timeout: 5s
      retries: 5
      start_period: 20s

volumes:
  workbench-data: