From 350a4eaf05dc5ebfcb806d05c6bbc0b3055982c0 Mon Sep 17 00:00:00 2001
From: zeryx <1892175+zeryx@users.noreply.github.com>
Date: Fri, 22 Jul 2022 11:29:31 -0300
Subject: [PATCH] enable safe replacement of mlops.json by setting environment
 variables

---
 adk/mlops.py | 16 ++++++++++------
 1 file changed, 10 insertions(+), 6 deletions(-)

diff --git a/adk/mlops.py b/adk/mlops.py
index cbde73a..34d44e9 100644
--- a/adk/mlops.py
+++ b/adk/mlops.py
@@ -15,14 +15,18 @@ def __init__(self, api_token, path):
         if os.path.exists(path):
             with open(path) as f:
                 mlops_config = json.load(f)
-        else:
-            raise Exception("'mlops.json' file does not exist, but mlops was requested.")
+            self.endpoint = mlops_config['datarobot_mlops_service_url']
+            self.model_id = mlops_config['model_id']
+            self.deployment_id = mlops_config['deployment_id']
+            self.mlops_name = mlops_config.get('mlops_dir_name', 'datarobot_mlops_package-8.1.2')
+        if "MLOPS_SERVICE_URL" in os.environ:
+            self.endpoint = os.environ['MLOPS_SERVICE_URL']
+        if "MODEL_ID" in os.environ:
+            self.model_id = os.environ['MODEL_ID']
+        if "DEPLOYMENT_ID" in os.environ:
+            self.deployment_id = os.environ['DEPLOYMENT_ID']
         if not os.path.exists(self.agent_dir):
             raise Exception("environment is not configured for mlops.\nPlease select a valid mlops enabled environment.")
-        self.endpoint = mlops_config['datarobot_mlops_service_url']
-        self.model_id = mlops_config['model_id']
-        self.deployment_id = mlops_config['deployment_id']
-        self.mlops_name = mlops_config.get('mlops_dir_name', 'datarobot_mlops_package-8.1.2')
 
     def init(self):
         os.environ['MLOPS_DEPLOYMENT_ID'] = self.deployment_id