ob-metaflow-extensions 1.4.12__tar.gz → 1.4.14rc0__tar.gz

{ob_metaflow_extensions-1.4.12 → ob_metaflow_extensions-1.4.14rc0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: ob_metaflow_extensions
-Version: 1.4.12
+Version: 1.4.14rc0
 Summary: Outerbounds Platform Extensions for Metaflow
 Author: Outerbounds, Inc.
 License: Commercial

{ob_metaflow_extensions-1.4.12 → ob_metaflow_extensions-1.4.14rc0}/metaflow_extensions/outerbounds/plugins/apps/app_deploy_decorator.py

@@ -106,7 +106,6 @@ class AppDeployDecorator(StepDecorator):
             code_package_key=package_sha,
             name_prefix=default_name,
             image=image,
-            max_entropy=self.MAX_ENTROPY,
             default_tags=[{k: str(v)} for k, v in default_tags.items()],
         )
         current._update_env(

{ob_metaflow_extensions-1.4.12 → ob_metaflow_extensions-1.4.14rc0}/metaflow_extensions/outerbounds/plugins/apps/core/__init__.py

@@ -1,6 +1,6 @@
 from . import app_cli
 from . import config
-from .deployer import AppDeployer, apps
+from .deployer import AppDeployer, apps, DeployedApp
 from .config.typed_configs import (
     ReplicaConfigDict,
     ResourceConfigDict,

{ob_metaflow_extensions-1.4.12 → ob_metaflow_extensions-1.4.14rc0}/metaflow_extensions/outerbounds/plugins/apps/core/deployer.py

@@ -3,8 +3,8 @@ from .perimeters import PerimeterExtractor
 from .capsule import CapsuleApi
 import json
 from ._state_machine import DEPLOYMENT_READY_CONDITIONS, LogLine
-from .app_config import AppConfig, AppConfigError
-from .capsule import CapsuleDeployer, list_and_filter_capsules
+from .app_config import AppConfig, AppConfigError, AuthType
+from .capsule import CapsuleDeployer, list_and_filter_capsules, _format_url_string
 from functools import partial
 import sys
 import uuid
@@ -37,6 +37,16 @@ class AppDeployer(TypedCoreConfig):
             self._app_config = AppConfig(self._config)
         return self._app_config
 
+    def _set_perimenter_and_api_server(self):
+        if self._state["perimeter"] is None:
+            perimeter, api_server = PerimeterExtractor.during_metaflow_execution()
+            if perimeter is None or api_server is None:
+                raise RuntimeError(
+                    "Perimeter not found in metaflow config or environment variables"
+                )
+            self._state["perimeter"] = perimeter
+            self._state["api_url"] = api_server
+
     # Things that need to be set before deploy
     @classmethod
     def _set_state(
@@ -47,7 +57,6 @@ class AppDeployer(TypedCoreConfig):
         code_package_key: str = None,
         name_prefix: str = None,
         image: str = None,
-        max_entropy: int = 4,
         default_tags: List[Dict[str, str]] = None,
         project: str = None,
         branch: str = None,
@@ -58,15 +67,10 @@ class AppDeployer(TypedCoreConfig):
         cls._state["code_package_key"] = code_package_key
         cls._state["name_prefix"] = name_prefix
         cls._state["image"] = image
-        cls._state["max_entropy"] = max_entropy
         cls._state["default_tags"] = default_tags
         cls._state["project"] = project
         cls._state["branch"] = branch
 
-        assert (
-            max_entropy > 0
-        ), "max_entropy must be greater than 0. Since AppDeployer's deploy fn can be called many time inside a step itself."
-
     def deploy(
         self,
         readiness_condition=DEPLOYMENT_READY_CONDITIONS.ATLEAST_ONE_RUNNING,
@@ -78,13 +82,7 @@ class AppDeployer(TypedCoreConfig):
         **kwargs,
     ) -> "DeployedApp":
 
-        # Name setting from top level if none is set in the code
-        if self._deploy_config._core_config.name is None:
-            name = self._state[
-                "name_prefix"
-            ]  # for now the name-prefix cannot be very large.
-            entropy = uuid.uuid4().hex[: self._state["max_entropy"]]
-            self._deploy_config._core_config.name = f"{name}-{entropy}"
+        self._set_perimenter_and_api_server()
 
         if len(self._state["default_tags"]) > 0:
             self._deploy_config._core_config.tags = (
@@ -174,10 +172,41 @@ class DeployedApp:
         self._deployed_version = deployed_version
         self._deployed_at = deployed_at
 
-    def _get_capsule_api(self) -> CapsuleApi:
+    @classmethod
+    def _get_capsule_api(cls) -> CapsuleApi:
         perimeter, api_server = PerimeterExtractor.during_metaflow_execution()
         return CapsuleApi(api_server, perimeter)
 
+    @classmethod
+    def from_capsule_id(cls, capsule_id: str) -> "DeployedApp":
+        capsule_api = cls._get_capsule_api()
+        capsule = capsule_api.get(capsule_id)
+        capsule_type = (
+            capsule.get("spec", {}).get("authConfig", {}).get("authType", None)
+        )
+        public_url = (
+            capsule.get("status", {}).get("accessInfo", {}).get("outOfClusterURL", None)
+        )
+        name = capsule.get("spec", {}).get(
+            "displayName",
+        )
+        deployed_version = capsule.get(
+            "version",
+        )
+        deployed_at = capsule.get("metadata", {}).get(
+            "createdAt",
+        )
+        if any(i is None for i in [capsule_type, public_url, name]):
+            raise ValueError(f"Invalid capsule id: {capsule_id}")
+        return cls(
+            capsule_id,
+            capsule_type,
+            _format_url_string(public_url),
+            name,
+            deployed_version,
+            deployed_at,
+        )
+
     def logs(self, previous=False) -> Dict[str, List[LogLine]]:
         """
         Returns a dictionary of worker_id to logs.
@@ -285,6 +314,15 @@ class DeployedApp:
             f"deployed_version='{self._deployed_version}')"
         )
 
+    def auth(self):
+        if self.auth_style != AuthType.API:
+            raise ValueError(
+                "Only API auth style is supported for accessing auth headers"
+            )
+        from metaflow.metaflow_config import SERVICE_HEADERS
+
+        return SERVICE_HEADERS
+
 
 class apps:
 

{ob_metaflow_extensions-1.4.12 → ob_metaflow_extensions-1.4.14rc0}/metaflow_extensions/outerbounds/toplevel/global_aliases_for_metaflow_package.py

@@ -108,4 +108,4 @@ from ..plugins.snowflake import Snowflake
 from ..plugins.checkpoint_datastores import nebius_checkpoints, coreweave_checkpoints
 from ..plugins.aws import assume_role
 from . import ob_internal
-from .ob_internal import AppDeployer
+from .ob_internal import AppDeployer, DeployedApp

{ob_metaflow_extensions-1.4.12 → ob_metaflow_extensions-1.4.14rc0}/metaflow_extensions/outerbounds/toplevel/ob_internal.py

@@ -1,4 +1,4 @@
 from ..plugins.kubernetes.pod_killer import PodKiller
 from ..plugins.fast_bakery.baker import bake_image
 from ..plugins.apps import core as app_core
-from ..plugins.apps.core import AppDeployer
+from ..plugins.apps.core import AppDeployer, DeployedApp

{ob_metaflow_extensions-1.4.12 → ob_metaflow_extensions-1.4.14rc0}/ob_metaflow_extensions.egg-info/PKG-INFO

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

ob_metaflow_extensions-1.4.14rc0/ob_metaflow_extensions.egg-info/requires.txt

@@ -0,0 +1,3 @@
+boto3
+kubernetes
+ob-metaflow==2.18.3.2

{ob_metaflow_extensions-1.4.12 → ob_metaflow_extensions-1.4.14rc0}/setup.py

@@ -2,7 +2,7 @@ from setuptools import setup, find_namespace_packages
 from pathlib import Path
 
 
-version = "1.4.12"
+version = "1.4.14rc0"
 this_directory = Path(__file__).parent
 long_description = (this_directory / "README.md").read_text()
 
@@ -19,6 +19,6 @@ setup(
     install_requires=[
         "boto3",
         "kubernetes",
-        "ob-metaflow == 2.18.2.1",
+        "ob-metaflow == 2.18.3.2",
     ],
 )

ob_metaflow_extensions-1.4.12/ob_metaflow_extensions.egg-info/requires.txt

@@ -1,3 +0,0 @@
-boto3
-kubernetes
-ob-metaflow==2.18.2.1