model-registry/docker-compose.yaml at main · syntaxsdev/model-registry · 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
125
126
127
128
# Local development demo/example compose file
# Composes:
# - a MySQL database,
# - a model-registry server connected to the MySQL database.
# - a model-catalog server
# - a model-registry-ui (standalone mode)
# or
# - a PostgreSQL database,
# - a model-registry server connected to the PostgreSQL database.
# - a model-catalog server
# - a model-registry-ui (standalone mode)
version: '3.8'
services:
  model-registry-ui:
    image: ghcr.io/kubeflow/model-registry/ui-standalone:latest
    pull_policy: always
    command:
      - --port=9000
      - --mock-k8s-client=true
      - --mock-mr-client=false
      - --mock-mr-catalog-client=false
      - --dev-mode=true
      - --deployment-mode=standalone
      - --log-level=DEBUG
    container_name: model-registry-ui
    ports:
      - "9000:9000"
    depends_on:
      - model-registry
      - model-catalog
    profiles:
      - postgres
    extra_hosts:
      - "localhost:host-gateway"

  model-catalog:
    image: ghcr.io/kubeflow/model-registry/server:latest
    pull_policy: always
    command: ["catalog", "--listen", "0.0.0.0:8081", "--catalogs-path", "/testdata/test-catalog-sources.yaml"]
    container_name: model-catalog
    ports:
      - "8081:8081"
    volumes:
      - ./catalog/internal/catalog/testdata:/testdata
    depends_on:
      - postgres
    profiles:
      - postgres
    environment:
      - PGHOST=postgres
      - PGDATABASE=model_catalog
      - PGUSER=postgres
      - PGPASSWORD=demo

  model-registry:
    image: ghcr.io/kubeflow/model-registry/server:latest
    pull_policy: always
    entrypoint: ["/bin/sh"]
    command:
      - -c
      - |
        if [ "$$DB_TYPE" = "postgres" ]; then
          exec /model-registry proxy --hostname 0.0.0.0 --datastore-type embedmd --embedmd-database-type postgres --embedmd-database-dsn "host=postgres port=5432 user=postgres password=demo dbname=model_registry sslmode=disable"
        else
          exec /model-registry proxy --hostname 0.0.0.0 --datastore-type embedmd --embedmd-database-dsn "root:demo@tcp(mysql:3306)/model_registry?charset=utf8mb4"
        fi
    container_name: model-registry
    ports:
      - "8080:8080"
    depends_on:
      - ${DB_TYPE:-mysql}
    environment:
      - DB_TYPE=${DB_TYPE:-mysql}

  mysql:
    image: mysql:8.3
    container_name: mysql
    command:
      - --datadir=/var/lib/mysql/datadir
      - --default-authentication-plugin=mysql_native_password
    environment:
      - MYSQL_ROOT_PASSWORD=demo
      - MYSQL_DATABASE=model_registry
    ports:
      - "3306:3306"
    volumes:
      - mysql_data:/var/lib/mysql
    healthcheck:
      test: ["CMD-SHELL", "mysql -D $$MYSQL_DATABASE -uroot -p$$MYSQL_ROOT_PASSWORD -e 'SELECT 1'"]
      interval: 10s
      timeout: 5s
      retries: 5
      start_period: 20s
    profiles:
      - mysql

  postgres:
    image: postgres:16
    container_name: postgres
    environment:
      - POSTGRES_DB=model_registry
      - POSTGRES_USER=postgres
      - POSTGRES_PASSWORD=demo
    ports:
      - "5432:5432"
    volumes:
      - postgres_data:/var/lib/postgresql/data
    healthcheck:
      test: ["CMD-SHELL", "pg_isready -U postgres -d model_registry"]
      interval: 10s
      timeout: 5s
      retries: 5
      start_period: 20s
    profiles:
      - postgres
    configs:
      - source: pginit
        target: /docker-entrypoint-initdb.d/pginit.sql

volumes:
  mysql_data:
  postgres_data:

configs:
  pginit:
    content: |
      CREATE DATABASE model_catalog;
      GRANT ALL PRIVILEGES ON model_catalog TO postgres;