PyRIT/docker/docker-compose.yaml at 88dc9d166f689a937f2a07b2d305b2452a8c8604 · microsoft/PyRIT · 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
# NOTE: You must build the devcontainer base image first:
#   docker build -f .devcontainer/Dockerfile -t pyrit-devcontainer .devcontainer
# Or use: python docker/build_pyrit_docker.py --source local
#
# Environment files: ~/.pyrit/.env and ~/.pyrit/.env.local are mounted to the container.

services:
  pyrit-jupyter:
    build:
      context: ..
      dockerfile: docker/Dockerfile
      args:
        BASE_IMAGE: pyrit-devcontainer
        PYRIT_SOURCE: local
    image: pyrit:latest
    container_name: pyrit-jupyter
    ports:
      - "8888:8888"
    volumes:
      - notebooks:/app/notebooks
      - data:/app/data
      - ../assets:/app/assets
      - ~/.pyrit/.env:/home/vscode/.pyrit/.env:ro
      - ~/.pyrit/.env.local:/home/vscode/.pyrit/.env.local:ro
    environment:
      - PYRIT_MODE=jupyter
    restart: unless-stopped
    healthcheck:
      test: ["CMD-SHELL", "curl -sf http://localhost:8888 || exit 1"]
      interval: 30s
      timeout: 10s
      retries: 3
      start_period: 40s
    profiles:
      - jupyter

  pyrit-gui:
    build:
      context: ..
      dockerfile: docker/Dockerfile
      args:
        BASE_IMAGE: pyrit-devcontainer
        PYRIT_SOURCE: local
    image: pyrit:latest
    container_name: pyrit-gui
    ports:
      - "8000:8000"
    volumes:
      - ../assets:/app/assets
      - ~/.pyrit/.env:/home/vscode/.pyrit/.env:ro
      - ~/.pyrit/.env.local:/home/vscode/.pyrit/.env.local:ro
      - ~/.pyrit/.pyrit_conf:/home/vscode/.pyrit/.pyrit_conf:ro
      - ~/.azure:/home/vscode/.azure
    environment:
      - PYRIT_MODE=gui
    restart: unless-stopped
    healthcheck:
      test: ["CMD-SHELL", "curl -sf http://localhost:8000/api/health || exit 1"]
      interval: 30s
      timeout: 10s
      retries: 3
      start_period: 40s
    profiles:
      - gui

volumes:
  notebooks:
  data: