add env vars to authenticate mlflow

digicatapult · Oct 9, 2024 · 36622ae · 36622ae
1 parent 75c2f47
commit 36622ae
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 10 deletions.
diff --git a/README.md b/README.md
@@ -14,16 +14,18 @@
 
 The following environment variables can be set:
 
-| Variable                       | Default Value                      | Description                                                |
-|--------------------------------|------------------------------------|------------------------------------------------------------|
-| MLFLOW_TRACKING_URI            | `http://localhost:8080`            | The URI for the MLflow tracking server                     |
-| DEPLOY_MODEL_NAME              | `house_price_prediction_prod`      | The name of the model to be deployed                       |
-| DEPLOY_MODEL_ALIAS             | `champion`                         | The alias for the deployed model                           |
-| DOCKER_REGISTRY                | `localhost:5000`                   | The Docker registry where images are stored                |
-| MLFLOW_BUILT_IMAGE_NAME        | `mlflow_model`                     | The name of the MLflow model Docker image                  |
-| MLFLOW_BUILT_IMAGE_TAG         | `latest`                           | The tag for the MLflow model Docker image                  |
-| MODEL_DOCKER_BUILD_CONTEXT_PVC | `model_docker_build_context_pvc`   | Name of the PVC allocated for this DAG                     |
-| MODEL_DOCKER_PUSH_SECRET       | `ecr-credentials`                  | Name of the secret to authenticate ECR access              | 
+| Variable                       | Default Value                    | Description                                   |
+|--------------------------------|----------------------------------|-----------------------------------------------|
+| MLFLOW_TRACKING_URI            | `http://localhost:8080`          | The URI for the MLflow tracking server        |
+| DEPLOY_MODEL_NAME              | `house_price_prediction_prod`    | The name of the model to be deployed          |
+| DEPLOY_MODEL_ALIAS             | `champion`                       | The alias for the deployed model              |
+| DOCKER_REGISTRY                | `localhost:5000`                 | The Docker registry where images are stored   |
+| MLFLOW_BUILT_IMAGE_NAME        | `mlflow_model`                   | The name of the MLflow model Docker image     |
+| MLFLOW_BUILT_IMAGE_TAG         | `latest`                         | The tag for the MLflow model Docker image     |
+| MODEL_DOCKER_BUILD_CONTEXT_PVC | `model_docker_build_context_pvc` | Name of the PVC allocated for this DAG        |
+| MODEL_DOCKER_PUSH_SECRET       | `ecr-credentials`                | Name of the secret to authenticate ECR access | 
+| MLFLOW_TRACKING_USERNAME       | None                             | MLFlow tracking username                      | 
+| MLFLOW_TRACKING_PASSWORD       | None                             | MLFlow tracking password                      | 
 
 
 

diff --git a/src/utils.py b/src/utils.py
@@ -21,6 +21,8 @@ class Envs(BaseSettings):
     mlflow_built_image_name: str = "mlflow_model"
     mlflow_built_image_tag: str = "latest"
     mlflow_docker_out_dir: str = "./mlflow-dockerfile"
+    mlflow_tracking_username: str | None = None
+    mlflow_tracking_password: str | None = None
 
     model_config = SettingsConfigDict(env_prefix="", case_sensitive=False)