From ad786070e9494dfba4905d3bcbac8f81c7e7a8cd Mon Sep 17 00:00:00 2001
From: ahgraber <ahgraber@ninerealmlabs.com>
Date: Fri, 25 Oct 2024 21:22:02 -0400
Subject: [PATCH] feat: mlflow ai gateway

---
 .../apps/datasci/mlflow/app/helmrelease.yaml   | 18 ++++--------------
 1 file changed, 4 insertions(+), 14 deletions(-)

diff --git a/kubernetes/apps/datasci/mlflow/app/helmrelease.yaml b/kubernetes/apps/datasci/mlflow/app/helmrelease.yaml
index f62fd96cc..436af1c14 100755
--- a/kubernetes/apps/datasci/mlflow/app/helmrelease.yaml
+++ b/kubernetes/apps/datasci/mlflow/app/helmrelease.yaml
@@ -133,21 +133,11 @@ spec:
               # renovate: datasource=docker image=ghcr.io/...
               repository: ghcr.io/ninerealmlabs/mlflow-server
               tag: 2.17.0@sha256:37ea0b12fd5d06adad1e865475b49b68c081d9f2f9c0679b077b15b5ce1432a8
-            # -- Override the command(s) for the default container
-            # command: [/bin/bash, -c, mlflow gateway start]
-            # # -- Override the args for the default container
-            # args:
-            #   - "--config-path"
-            #   - "/config/config.yaml"
-            #   - "--host"
-            #   - "0.0.0.0"
-            #   - "--port"
-            #   - "5000"
             command: ["mlflow", "gateway", "start"]
             args:
               - "--config-path=/config/config.yaml"
               - "--host=0.0.0.0"
-              - "--port=5000"
+              - "--port=5001"
             env:
               TZ: ${TIMEZONE}
               MLFLOW_GATEWAY_CONFIG: "/config/config.yaml"
@@ -165,7 +155,7 @@ spec:
                   failureThreshold: 3
                   httpGet:
                     path: /health
-                    port: *port
+                    port: &gport 5001
               readiness: *probes
               startup:
                 enabled: true
@@ -177,7 +167,7 @@ spec:
                   failureThreshold: 30
                   httpGet:
                     path: /health
-                    port: *port
+                    port: *gport
             resources:
               requests:
                 cpu: 25m
@@ -193,7 +183,7 @@ spec:
         controller: mlflow
         ports:
           http:
-            port: *port
+            port: *gport
     ingress:
       main:
         className: internal