point tracking at gateway

mlflow-gateway/config.yaml

@@ -1,6 +1,6 @@
 name: "mlflow-gateway"
 description: "MLflow Gateway for managing model providers"
-version: "0.0.8"
+version: "0.0.9"
 url: "https://github.com/akshaya-a/mlflow-hass/tree/main/mlflow-gateway"
 slug: "mlflow-gateway"
 init: false
@@ -13,6 +13,7 @@ panel_icon: mdi:flask
 panel_title: MLflow Gateway
 ingress: true
 ingress_port: 5001
+ingress_entry: /docs
 ports:
   5001/tcp: 5001
 environment:

mlflow-gateway/run.sh

@@ -1,8 +1,8 @@
 #!/bin/bash
 
 echo "Starting MLflow gateway"
+
 key=$(python /config.py)
-echo $key
 export OPENAI_API_KEY=$key
-export PYTHONPATH="/"
+
 mlflow gateway start --config-path /route-config.yaml --port 5001 --host 0.0.0.0

mlflow-tracking/config.yaml

@@ -1,6 +1,6 @@
 name: "mlflow-tracking"
 description: "MLflow Tracking Server for managing models + prompts"
-version: "0.0.8"
+version: "0.0.9"
 url: "https://github.com/akshaya-a/mlflow-hass/tree/main/mlflow-tracking"
 slug: "mlflow-tracking"
 init: false

mlflow-tracking/run.sh

@@ -1,6 +1,10 @@
 #!/bin/bash
 
 echo "Starting MLflow server"
+
+# TODO: convert this to Supervisor REPO_SLUG format via addon discovery
+export MLFLOW_GATEWAY_URI="http://localhost:5001"
+
 mlflow server \
   --backend-store-uri sqlite:///data/mydb.sqlite \
   --artifacts-destination /share/tracking/mlflow-hass \