docker-compose.yaml

version: '3.7'

volumes:
  postgres:
  mlflow:
  prefect:

networks:
  backend:
  frontend:

services:
  tracking-server-db:
    image: postgres
    environment:
      - POSTGRES_USER=${POSTGRES_USER}
      - POSTGRES_PASSWORD=${POSTGRES_PASSWORD}
      - POSTGRES_DB=${POSTGRES_DB}
      - POSTGRES_HOST=${POSTGRES_HOST}
    volumes:
      - postgres:/var/lib/postgresql/data
    networks:
      - backend
    expose:
      - 5432
    restart: always

  mlflow:
    image: mlflow_tracking_server
    build:
      context: ./monitoring/mlflow
      dockerfile: Dockerfile
    volumes:
      - mlflow:/var/lib/mlflow/data
    networks:
      - backend
      - frontend
    ports:
      - 5000:5000
    depends_on:
      - tracking-server-db
    restart: always
    command: mlflow server --backend-store-uri postgresql+psycopg2://${POSTGRES_USER}:${POSTGRES_PASSWORD}@tracking-server-db:5432/${POSTGRES_DB} --default-artifact-root file:///var/lib/mlflow/data --host 0.0.0.0

  prefect:
    image: prefect_server
    build:
      context: ./monitoring/prefect
      dockerfile: Dockerfile
    volumes:
      - prefect:/root/.prefect
    environment:
      - PREFECT_UI_URL=http://127.0.0.0:4200/api
      - PREFECT_API_URL=http://127.0.0.1:4200/api
      - PREFECT_SERVER_API_HOST=0.0.0.0
    networks:
      - backend
      - frontend
    ports:
      - 4200:4200
    restart: always
    entrypoint: ["prefect", "server", "start"]