ForgeBaselines/docker-compose.prod.yml at main · Par-t/ForgeBaselines · 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
version: '3.8'

services:
  nginx:
    image: nginx:alpine
    container_name: forgebaselines-nginx
    ports:
      - "80:80"
    volumes:
      - ./infra/nginx.conf:/etc/nginx/nginx.conf:ro
    networks:
      - forgebaselines
    depends_on:
      - orchestrator
    restart: always

  orchestrator:
    build:
      context: ./services/orchestrator
      dockerfile: Dockerfile
    container_name: forgebaselines-orchestrator
    environment:
      - STORAGE_BACKEND=${STORAGE_BACKEND:-s3}
      - S3_BUCKET=${S3_BUCKET:-forgebaselines-artifacts}
      - MLFLOW_TRACKING_URI=http://mlflow:5000
      - CLASSIFICATION_SERVICE_URL=http://classification:8001
      - IR_SERVICE_URL=http://ir:8002
      - ENV=production
      - AWS_ACCESS_KEY_ID=${AWS_ACCESS_KEY_ID}
      - AWS_SECRET_ACCESS_KEY=${AWS_SECRET_ACCESS_KEY}
      - AWS_DEFAULT_REGION=${AWS_DEFAULT_REGION:-us-east-2}
      - FIREBASE_SERVICE_ACCOUNT_JSON=${FIREBASE_SERVICE_ACCOUNT_JSON:-{}}
    volumes:
      - app-data:/app/data
    networks:
      - forgebaselines
    depends_on:
      - mlflow
      - classification
      - ir
    restart: always

  classification:
    build:
      context: ./services/classification
      dockerfile: Dockerfile
    container_name: forgebaselines-classification
    environment:
      - MLFLOW_TRACKING_URI=http://mlflow:5000
      - ENV=production
    volumes:
      - app-data:/app/data
    networks:
      - forgebaselines
    depends_on:
      - mlflow
    restart: always

  ir:
    build:
      context: ./services/ir
      dockerfile: Dockerfile
    container_name: forgebaselines-ir
    environment:
      - MLFLOW_TRACKING_URI=http://mlflow:5000
      - ENV=production
    volumes:
      - app-data:/app/data
    networks:
      - forgebaselines
    depends_on:
      - mlflow
    restart: always

  mlflow:
    build:
      context: ./services/mlflow
      dockerfile: Dockerfile
    container_name: forgebaselines-mlflow
    environment:
      - AWS_ACCESS_KEY_ID=${AWS_ACCESS_KEY_ID}
      - AWS_SECRET_ACCESS_KEY=${AWS_SECRET_ACCESS_KEY}
      - AWS_DEFAULT_REGION=${AWS_DEFAULT_REGION:-us-east-2}
      - MLFLOW_ARTIFACT_ROOT=${MLFLOW_ARTIFACT_ROOT:-/mlflow-artifacts}
    volumes:
      - mlflow-data:/mlflow
      - mlflow-artifacts:/mlflow-artifacts
    networks:
      - forgebaselines
    restart: always

networks:
  forgebaselines:
    driver: bridge

volumes:
  mlflow-data:
  mlflow-artifacts:
  app-data: