From 26bc46f9c4d9c1c59daf622c612ee69e662758cc Mon Sep 17 00:00:00 2001 From: AK Date: Mon, 30 Oct 2023 18:11:38 -0700 Subject: [PATCH] point tracking at gateway --- mlflow-gateway/config.yaml | 3 ++- mlflow-gateway/run.sh | 4 ++-- mlflow-tracking/config.yaml | 2 +- mlflow-tracking/run.sh | 4 ++++ 4 files changed, 9 insertions(+), 4 deletions(-) diff --git a/mlflow-gateway/config.yaml b/mlflow-gateway/config.yaml index 65fbd23..e33c930 100644 --- a/mlflow-gateway/config.yaml +++ b/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: diff --git a/mlflow-gateway/run.sh b/mlflow-gateway/run.sh index b4d46fc..4749d51 100644 --- a/mlflow-gateway/run.sh +++ b/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 diff --git a/mlflow-tracking/config.yaml b/mlflow-tracking/config.yaml index 39ff788..23ced48 100644 --- a/mlflow-tracking/config.yaml +++ b/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 diff --git a/mlflow-tracking/run.sh b/mlflow-tracking/run.sh index 32e640a..878e3b5 100644 --- a/mlflow-tracking/run.sh +++ b/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 \