MLOps/docker-compose.yml at main · imEag/MLOps · 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
122
123
124
services:
  postgres:
    image: postgres:15
    ports:
      - "${POSTGRES_PORT:-5432}:5432"
    environment:
      - POSTGRES_USER=${POSTGRES_USER:-postgres}
      - POSTGRES_PASSWORD=${POSTGRES_PASSWORD:-postgres}
      - POSTGRES_DB=${POSTGRES_DB:-mlops}
    volumes:
      - postgres_data:/var/lib/postgresql/data
      - ./init-db:/docker-entrypoint-initdb.d
    networks:
      - mlops-network
    healthcheck:
      test: ["CMD-SHELL", "pg_isready -U ${POSTGRES_USER:-postgres}"]
      interval: 5s
      timeout: 5s
      retries: 5

  mlflow:
    image: imeag/mlops-mlflow:latest
    build:
      context: ./backend
      dockerfile: Dockerfile.mlflow
    ports:
      - "${MLFLOW_PORT:-5001}:${MLFLOW_PORT:-5001}"
    volumes:
      - ./backend/mlruns:/mlflow/mlruns
      - ./backend/artifacts:/mlflow/artifacts
    environment:
      - MLFLOW_TRACKING_URI=http://localhost:${MLFLOW_PORT:-5001}
      - MLFLOW_PORT=${MLFLOW_PORT:-5001}
      - POSTGRES_HOST=postgres
      - POSTGRES_PORT=${POSTGRES_PORT:-5432}
      - POSTGRES_USER=${POSTGRES_USER:-postgres}
      - POSTGRES_PASSWORD=${POSTGRES_PASSWORD:-postgres}
      - POSTGRES_DB=${POSTGRES_DB:-mlops}
      - GIT_PYTHON_REFRESH=quiet
    depends_on:
      postgres:
        condition: service_healthy
    networks:
      - mlops-network

  prefect:
    image: imeag/mlops-prefect:latest
    build:
      context: ./backend
      dockerfile: Dockerfile
    ports:
      - "${PREFECT_PORT:-4200}:4200"
    volumes:
      - ./backend:/app
      - ./backend/artifacts:/mlflow/artifacts
    environment:
      - PREFECT_API_URL=http://0.0.0.0:4200/api
      - MLFLOW_TRACKING_URI=http://mlflow:${MLFLOW_PORT:-5001}
      - PREFECT_API_DATABASE_CONNECTION_URL=postgresql+asyncpg://${POSTGRES_USER:-postgres}:${POSTGRES_PASSWORD:-postgres}@postgres:${POSTGRES_PORT:-5432}/prefect_db
      - PREFECT_SERVER_HOST=0.0.0.0
      - PREFECT_SERVER_PORT=4200
      - GIT_PYTHON_REFRESH=quiet
    depends_on:
      mlflow:
        condition: service_started
      postgres:
        condition: service_healthy
    networks:
      - mlops-network

  fastapi:
    image: imeag/mlops-fastapi:latest
    build:
      context: ./backend
      dockerfile: Dockerfile.fastapi
    ports:
      - "${FASTAPI_PORT:-8000}:8000"
    volumes:
      - ./backend:/app
      - ./backend/artifacts:/mlflow/artifacts
    environment:
      - MLFLOW_TRACKING_URI=http://mlflow:${MLFLOW_PORT:-5001}
      - MLFLOW_PORT=${MLFLOW_PORT:-5001}
      - POSTGRES_HOST=postgres
      - POSTGRES_PORT=${POSTGRES_PORT:-5432}
      - POSTGRES_USER=${POSTGRES_USER:-postgres}
      - POSTGRES_PASSWORD=${POSTGRES_PASSWORD:-postgres}
      - POSTGRES_DB=${POSTGRES_DB:-mlops}
      - PREFECT_API_URL=http://prefect:4200/api
      - PREFECT_API_DATABASE_CONNECTION_URL=postgresql+asyncpg://${POSTGRES_USER:-postgres}:${POSTGRES_PASSWORD:-postgres}@postgres:${POSTGRES_PORT:-5432}/prefect_db
      - GIT_PYTHON_REFRESH=quiet
    depends_on:
      mlflow:
        condition: service_started
      postgres:
        condition: service_healthy
      prefect:
        condition: service_started
    networks:
      - mlops-network

  frontend:
    image: imeag/mlops-frontend:latest
    build:
      context: ./frontend
      dockerfile: Dockerfile
    ports:
      - "${FRONTEND_PORT:-3000}:3000"
    volumes:
      - ./frontend:/app
      - /app/node_modules
    environment:
      - VITE_API_URL=http://localhost:${FASTAPI_PORT:-8000}
    depends_on:
      - fastapi
    networks:
      - mlops-network

volumes:
  postgres_data:

networks:
  mlops-network:
    driver: bridge