ob-metaflow-extensions 1.1.153__tar.gz → 1.1.155rc0__tar.gz

{ob-metaflow-extensions-1.1.153 → ob-metaflow-extensions-1.1.155rc0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: ob-metaflow-extensions
-Version: 1.1.153
+Version: 1.1.155rc0
 Summary: Outerbounds Platform Extensions for Metaflow
 Author: Outerbounds, Inc.
 License: Commercial

{ob-metaflow-extensions-1.1.153 → ob-metaflow-extensions-1.1.155rc0}/metaflow_extensions/outerbounds/__init__.py

@@ -1,6 +1,6 @@
 import metaflow.metaflow_config_funcs
 
-from metaflow_extensions.outerbounds.remote_config import init_config
+from metaflow_extensions.outerbounds.remote_config import init_config, reload_config
 
 # we want to overide OSS Metaflow's initialization behavior with our own to support remote configs
 # we're reassigning the METAFLOW_CONFIG variable because all downstream settings rely on it and

{ob-metaflow-extensions-1.1.153 → ob-metaflow-extensions-1.1.155rc0}/metaflow_extensions/outerbounds/plugins/secrets/secrets.py

@@ -187,18 +187,11 @@ class OuterboundsSecretsProvider(SecretsProvider):
             raise OuterboundsSecretsException(
                 f"Server error: {response.text}. Please reach out to your Outerbounds support team."
             )
-
-        body = response.json()
-        status_code = body.get("error", {}).get("statusCode", response.status_code)
+        status_code = response.status_code
         if status_code == 404:
-            raise OuterboundsSecretsException(f"Secret not found: {body}")
+            raise OuterboundsSecretsException(f"Secret not found: {response.text}")
 
         if status_code >= 400:
-            try:
-                raise OuterboundsSecretsException(
-                    f"status_code={status_code}\t*{body['error']['details']['kind']}*\n{body['error']['details']['message']}"
-                )
-            except KeyError:
-                raise OuterboundsSecretsException(
-                    f"status_code={status_code} Unexpected error: {body}"
-                )
+            raise OuterboundsSecretsException(
+                f"status_code={status_code}\t\n\t\t{response.text}"
+            )

{ob-metaflow-extensions-1.1.153 → ob-metaflow-extensions-1.1.155rc0}/metaflow_extensions/outerbounds/remote_config.py

@@ -112,3 +112,22 @@ def _init_debug(*args, **kwargs):
         init_str = "ob_extension_init:"
         kwargs["file"] = sys.stderr
         print(init_str, *args, **kwargs)
+
+
+def reload_config():
+    """
+    This function is used to reload the config. Currently its a best effort implementation
+    that will only reload auth token.
+    """
+    _init_debug("reloading config")
+    if CONFIG_READ_ONCE_KEY in os.environ:
+        del os.environ[CONFIG_READ_ONCE_KEY]
+    config = init_config()
+    import metaflow.metaflow_config
+
+    metaflow.metaflow_config.SERVICE_AUTH_KEY = config.get(AUTH_KEY)
+    metaflow.metaflow_config.SERVICE_HEADERS["x-api-key"] = config.get(AUTH_KEY)
+    if config:
+        _init_debug("reloaded config")
+    else:
+        _init_debug("no config to reload")

{ob-metaflow-extensions-1.1.153 → ob-metaflow-extensions-1.1.155rc0}/ob_metaflow_extensions.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: ob-metaflow-extensions
-Version: 1.1.153
+Version: 1.1.155rc0
 Summary: Outerbounds Platform Extensions for Metaflow
 Author: Outerbounds, Inc.
 License: Commercial

{ob-metaflow-extensions-1.1.153 → ob-metaflow-extensions-1.1.155rc0}/setup.py

@@ -2,7 +2,7 @@ from setuptools import setup, find_namespace_packages
 from pathlib import Path
 
 
-version = "1.1.153"
+version = "1.1.155rc0"
 this_directory = Path(__file__).parent
 long_description = (this_directory / "README.md").read_text()