zenml-nightly 0.72.0.dev20250115__py3-none-any.whl → 0.72.0.dev20250117__py3-none-any.whl

zenml/zen_server/utils.py

@@ -20,6 +20,7 @@ from typing import (
     TYPE_CHECKING,
     Any,
     Callable,
+    Dict,
     List,
     Optional,
     Tuple,
@@ -31,6 +32,7 @@ from urllib.parse import urlparse
 
 from pydantic import BaseModel, ValidationError
 
+from zenml import __version__ as zenml_version
 from zenml.config.global_config import GlobalConfiguration
 from zenml.config.server_config import ServerConfiguration
 from zenml.constants import (
@@ -571,3 +573,65 @@ def is_user_request(request: "Request") -> bool:
 
     # If none of the above conditions are met, consider it a user request
     return True
+
+
+def is_same_or_subdomain(source_domain: str, target_domain: str) -> bool:
+    """Check if the source domain is the same or a subdomain of the target domain.
+
+    Examples:
+        is_same_or_subdomain("example.com", "example.com") -> True
+        is_same_or_subdomain("alpha.example.com", "example.com") -> True
+        is_same_or_subdomain("alpha.example.com", ".example.com") -> True
+        is_same_or_subdomain("example.com", "alpha.example.com") -> False
+        is_same_or_subdomain("alpha.beta.example.com", "beta.example.com") -> True
+        is_same_or_subdomain("alpha.beta.example.com", "alpha.example.com") -> False
+        is_same_or_subdomain("alphabeta.gamma.example", "beta.gamma.example") -> False
+
+    Args:
+        source_domain: The source domain to check.
+        target_domain: The target domain to compare against.
+
+    Returns:
+        True if the source domain is the same or a subdomain of the target
+        domain, False otherwise.
+    """
+    import tldextract
+
+    # Extract the registered domain and suffix for both
+    src_parts = tldextract.extract(source_domain)
+    tgt_parts = tldextract.extract(target_domain)
+
+    if src_parts == tgt_parts:
+        return True  # Same domain
+
+    # Reconstruct the base domains (e.g., example.com)
+    src_base_domain = f"{src_parts.domain}.{src_parts.suffix}"
+    tgt_base_domain = f"{tgt_parts.domain}.{tgt_parts.suffix}"
+
+    if src_base_domain != tgt_base_domain:
+        return False  # Different base domains
+
+    if tgt_parts.subdomain == "":
+        return True  # Subdomain
+
+    if src_parts.subdomain.endswith(f".{tgt_parts.subdomain.lstrip('.')}"):
+        return True  # Subdomain of subdomain
+
+    return False
+
+
+def get_zenml_headers() -> Dict[str, str]:
+    """Get the ZenML specific headers to be included in requests made by the server.
+
+    Returns:
+        The ZenML specific headers.
+    """
+    config = server_config()
+    headers = {
+        "zenml-server-id": str(config.get_external_server_id()),
+        "zenml-server-version": zenml_version,
+    }
+    if config.server_url:
+        headers["zenml-server-url"] = config.server_url
+
+    return headers

zenml/zen_server/zen_server_api.py

@@ -54,6 +54,7 @@ from zenml.constants import (
 )
 from zenml.enums import AuthScheme, SourceContextTypes
 from zenml.models import ServerDeploymentType
+from zenml.zen_server.cloud_utils import send_pro_tenant_status_update
 from zenml.zen_server.exceptions import error_detail
 from zenml.zen_server.routers import (
     actions_endpoints,
@@ -389,6 +390,10 @@ def initialize() -> None:
     initialize_plugins()
     initialize_secure_headers()
     initialize_memcache(cfg.memcache_max_capacity, cfg.memcache_default_expiry)
+    if cfg.deployment_type == ServerDeploymentType.CLOUD:
+        # Send a tenant status update to the Cloud API to indicate that the
+        # ZenML server is running or to update the version and server URL.
+        send_pro_tenant_status_update()
 
 
 DASHBOARD_REDIRECT_URL = None

zenml/zen_stores/base_zen_store.py

@@ -47,6 +47,7 @@ from zenml.exceptions import IllegalOperationError
 from zenml.logger import get_logger
 from zenml.models import (
     ServerDatabaseType,
+    ServerDeploymentType,
     ServerModel,
     StackFilter,
     StackResponse,
@@ -390,7 +391,7 @@ class BaseZenStore(
         secrets_store_type = SecretsStoreType.NONE
         if isinstance(self, SqlZenStore) and self.config.secrets_store:
             secrets_store_type = self.config.secrets_store.type
-        return ServerModel(
+        store_info = ServerModel(
             id=GlobalConfiguration().user_id,
             active=True,
             version=zenml.__version__,
@@ -405,6 +406,23 @@ class BaseZenStore(
             metadata=metadata,
         )
 
+        # Add ZenML Pro specific store information to the server model, if available.
+        if store_info.deployment_type == ServerDeploymentType.CLOUD:
+            from zenml.config.server_config import ServerProConfiguration
+
+            pro_config = ServerProConfiguration.get_server_config()
+
+            store_info.pro_api_url = pro_config.api_url
+            store_info.pro_dashboard_url = pro_config.dashboard_url
+            store_info.pro_organization_id = pro_config.organization_id
+            store_info.pro_tenant_id = pro_config.tenant_id
+            if pro_config.tenant_name:
+                store_info.pro_tenant_name = pro_config.tenant_name
+            if pro_config.organization_name:
+                store_info.pro_organization_name = pro_config.organization_name
+
+        return store_info
+
     def is_local_store(self) -> bool:
         """Check if the store is local or connected to a local ZenML server.
 

zenml/zen_stores/rest_zen_store.py

@@ -124,9 +124,9 @@ from zenml.io import fileio
 from zenml.logger import get_logger
 from zenml.login.credentials import APIToken
 from zenml.login.credentials_store import get_credentials_store
+from zenml.login.pro.constants import ZENML_PRO_API_URL
 from zenml.login.pro.utils import (
     get_troubleshooting_instructions,
-    is_zenml_pro_server_url,
 )
 from zenml.models import (
     ActionFilter,
@@ -450,6 +450,7 @@ class RestZenStore(BaseZenStore):
     CONFIG_TYPE: ClassVar[Type[StoreConfiguration]] = RestZenStoreConfiguration
     _api_token: Optional[APIToken] = None
     _session: Optional[requests.Session] = None
+    _server_info: Optional[ServerModel] = None
 
     # ====================================
     # ZenML Store interface implementation
@@ -469,7 +470,7 @@ class RestZenStore(BaseZenStore):
         """
         try:
             client_version = zenml.__version__
-            server_version = self.get_store_info().version
+            server_version = self.server_info.version
 
         # Handle cases where the ZenML server is not available
         except ConnectionError as e:
@@ -500,7 +501,7 @@ class RestZenStore(BaseZenStore):
 
         except Exception as e:
             zenml_pro_extra = ""
-            if is_zenml_pro_server_url(self.url):
+            if ".zenml.io" in self.url:
                 zenml_pro_extra = (
                     "\nHINT: " + get_troubleshooting_instructions(self.url)
                 )
@@ -522,6 +523,17 @@ class RestZenStore(BaseZenStore):
                 ENV_ZENML_DISABLE_CLIENT_SERVER_MISMATCH_WARNING,
             )
 
+    @property
+    def server_info(self) -> ServerModel:
+        """Get cached information about the server.
+
+        Returns:
+            Cached information about the server.
+        """
+        if self._server_info is None:
+            return self.get_store_info()
+        return self._server_info
+
     def get_store_info(self) -> ServerModel:
         """Get information about the server.
 
@@ -529,7 +541,8 @@ class RestZenStore(BaseZenStore):
             Information about the server.
         """
         body = self.get(INFO)
-        return ServerModel.model_validate(body)
+        self._server_info = ServerModel.model_validate(body)
+        return self._server_info
 
     def get_deployment_id(self) -> UUID:
         """Get the ID of the deployment.
@@ -537,7 +550,7 @@ class RestZenStore(BaseZenStore):
         Returns:
             The ID of the deployment.
         """
-        return self.get_store_info().id
+        return self.server_info.id
 
     # -------------------- Server Settings --------------------
 
@@ -4028,19 +4041,6 @@ class RestZenStore(BaseZenStore):
             token = credentials.api_token if credentials else None
             if credentials and token and not token.expired:
                 self._api_token = token
-
-                # Populate the server info in the credentials store if it is
-                # not already present
-                if not credentials.server_id:
-                    try:
-                        server_info = self.get_store_info()
-                    except Exception as e:
-                        logger.warning(f"Failed to get server info: {e}.")
-                    else:
-                        credentials_store.update_server_info(
-                            self.url, server_info
-                        )
-
                 return self._api_token.access_token
 
             # Token is expired or not found in the cache. Time to get a new one.
@@ -4089,13 +4089,23 @@ class RestZenStore(BaseZenStore):
                     "username": username,
                     "password": password,
                 }
-            elif is_zenml_pro_server_url(self.url):
+            elif self.server_info.is_pro_server():
                 # ZenML Pro tenants use a proprietary authorization grant
                 # where the ZenML Pro API session token is exchanged for a
                 # regular ZenML server access token.
 
                 # Get the ZenML Pro API session token, if cached and valid
-                pro_token = credentials_store.get_pro_token(allow_expired=True)
+
+                # We need to determine the right ZenML Pro API URL to use
+                pro_api_url = self.server_info.pro_api_url
+                if not pro_api_url and credentials and credentials.pro_api_url:
+                    pro_api_url = credentials.pro_api_url
+                if not pro_api_url:
+                    pro_api_url = ZENML_PRO_API_URL
+
+                pro_token = credentials_store.get_pro_token(
+                    pro_api_url, allow_expired=True
+                )
                 if not pro_token:
                     raise CredentialsNotValid(
                         "You need to be logged in to ZenML Pro in order to "

{zenml_nightly-0.72.0.dev20250115.dist-info → zenml_nightly-0.72.0.dev20250117.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: zenml-nightly
-Version: 0.72.0.dev20250115
+Version: 0.72.0.dev20250117
 Summary: ZenML: Write production-ready ML code.
 License: Apache-2.0
 Keywords: machine learning,production,pipeline,mlops,devops
@@ -73,6 +73,7 @@ Requires-Dist: hvac (>=0.11.2) ; extra == "secrets-hashicorp"
 Requires-Dist: hypothesis (>=6.43.1,<7.0.0) ; extra == "dev"
 Requires-Dist: importlib_metadata (<=7.0.0) ; python_version < "3.10"
 Requires-Dist: ipinfo (>=4.4.3) ; extra == "server"
+Requires-Dist: itsdangerous (>=2.2.0,<2.3.0) ; extra == "server"
 Requires-Dist: jinja2-time (>=0.2.0,<0.3.0) ; extra == "templates"
 Requires-Dist: kfp (>=2.6.0) ; extra == "vertex"
 Requires-Dist: kubernetes (>=18.20.0) ; extra == "connectors-kubernetes" or extra == "connectors-aws" or extra == "connectors-gcp" or extra == "connectors-azure"
@@ -112,6 +113,7 @@ Requires-Dist: setuptools
 Requires-Dist: sqlalchemy (>=2.0.0,<3.0.0)
 Requires-Dist: sqlalchemy_utils
 Requires-Dist: sqlmodel (==0.0.18)
+Requires-Dist: tldextract (>=5.1.0,<5.2.0) ; extra == "server"
 Requires-Dist: tox (>=3.24.3,<4.0.0) ; extra == "dev"
 Requires-Dist: types-Markdown (>=3.3.6,<4.0.0) ; extra == "dev"
 Requires-Dist: types-Pillow (>=9.2.1,<10.0.0) ; extra == "dev"

{zenml_nightly-0.72.0.dev20250115.dist-info → zenml_nightly-0.72.0.dev20250117.dist-info}/RECORD

@@ -1,5 +1,5 @@
 zenml/README.md,sha256=827dekbOWAs1BpW7VF1a4d7EbwPbjwccX-2zdXBENZo,1777
-zenml/VERSION,sha256=N1u7Rr619sZkD-klFk2ilvZ-wFvnMfoRJVDsLT2WiDM,19
+zenml/VERSION,sha256=uOurOZmI34Olte7XOk1ZCm78eSeXtyGeK_3vJxGpxKY,19
 zenml/__init__.py,sha256=SkMObQA41ajqdZqGErN00S1Vf3KAxpLvbZ-OBy5uYoo,2130
 zenml/actions/__init__.py,sha256=mrt6wPo73iKRxK754_NqsGyJ3buW7RnVeIGXr1xEw8Y,681
 zenml/actions/base_action.py,sha256=UcaHev6BTuLDwuswnyaPjdA8AgUqB5xPZ-lRtuvf2FU,25553
@@ -38,13 +38,13 @@ zenml/cli/downgrade.py,sha256=eTpXts8y4s3wEUwOlvZGWsTngoMV8Stuzj0K-SAQUGU,1887
 zenml/cli/feature.py,sha256=Q8tNvWBlRze3FUyn0_VpOdl316ZW87476j7ezJb16GA,4387
 zenml/cli/formatter.py,sha256=OoAzgxpOEXrzT9FSFG0Bs5VRtVFEYck0gDYo02GCjzI,6440
 zenml/cli/integration.py,sha256=IsG1WR9H7tEEQl2bxm-i2P5hWU63sGCKAMbraJtRN9s,16208
-zenml/cli/login.py,sha256=xZdDiTbFiHvzvd5o08VRL01Oim5-0NSeKlxpBayAvss,35232
+zenml/cli/login.py,sha256=ze-SY-ffaZCFN_oY5c67GRfsiLBuofWRGxdyQLODfLk,37939
 zenml/cli/model.py,sha256=hXRXkMUOOf0eTo07WnQHNeeDDQLiH0m76E-xypFLYF0,22993
 zenml/cli/model_registry.py,sha256=zzxWXXFhKu2B1Wp0u7prKVnN1ftM-JdGdQwlD-5G-QM,20786
 zenml/cli/pipeline.py,sha256=ZK_J_GE1OGZkLCBsdTiHnSh7BBp29QiigtxvQNifxAc,19228
 zenml/cli/secret.py,sha256=dn0Pbmlwt6XoLmdJp1JbBhv6nDRVJcUnPYHmUCTVNOk,20120
 zenml/cli/served_model.py,sha256=3w1UcAbg6Geu37fr7ej1_81GBCt3fF7j3Ge799YE4Mc,14974
-zenml/cli/server.py,sha256=JbvRyc1Ha2LPGNVtZhJzZeTK0w9Z54SHBbjos1UWu54,25438
+zenml/cli/server.py,sha256=XkLRSkhDZfJy2JJn0CFHm2iNUSbR9dU_zUj1TnAJm04,26048
 zenml/cli/service_accounts.py,sha256=kl2YSUCsfy3V5ezjbnEjy7i29QgHw-UhLr26rdvz938,17789
 zenml/cli/service_connectors.py,sha256=QMPEhRtPbL7ZbsjtmrLXfN7dSL538W4Qe-OtkFtqyEI,74319
 zenml/cli/stack.py,sha256=4e65i5XijPd7uLApauAnX1lGd_ZJsLpr1x0oz1vx7l4,67390
@@ -77,7 +77,7 @@ zenml/config/retry_config.py,sha256=4UH1xqw0G6fSEbXSNKfmiFEkwadxQef9BGMe3JBm6NI,
 zenml/config/schedule.py,sha256=QdezLjIONnEAVPrWFekbetNvkclx_5eJyhza0Prl5Y0,4944
 zenml/config/secret_reference_mixin.py,sha256=6DOgOH_w1_coNs1NasMjY8SOv0MqzWQZxwrWsfncsgs,5862
 zenml/config/secrets_store_config.py,sha256=y05zqyQhr_DGrs3IfBGa_FRoZ043hSYRT5wzrx-zHTU,2818
-zenml/config/server_config.py,sha256=DIQLjI2EeTn4DN9asECThJzDQUe0rg1iZpY6X8y0K9Y,25919
+zenml/config/server_config.py,sha256=poUo_sOAPAItNyOzOb7RjpocrHfEd9ITymWiSV-oyro,30564
 zenml/config/settings_resolver.py,sha256=PR9BRm_x1dy7nVKa9UqpeFdck8IEATSW6aWT8FKd-DI,4278
 zenml/config/source.py,sha256=RzUw8lin8QztUjz-AdoCzVM5Om_cSSPuroaPx-qAO4w,8226
 zenml/config/step_configurations.py,sha256=RtlMYDpQBwcOjXNvokSEBQnWJAwPoPMpBFep-SrMMFk,10408
@@ -85,7 +85,7 @@ zenml/config/step_run_info.py,sha256=KiVRSTtKmZ1GbvseDTap2imr7XwMHD3jSFVpyLNEK1I
 zenml/config/store_config.py,sha256=Cla5p5dTB6nNlo8_OZDs9hod5hspi64vxwtZj882XgU,3559
 zenml/config/strict_base_model.py,sha256=iHnO9qOmLUP_eiy9IjRr3JjIs1l1I_CsRQ76EyAneYU,860
 zenml/console.py,sha256=hj_KerPQKwnyKACj0ehSqUQX0mGVCJBKE1QvCt6ik3A,1160
-zenml/constants.py,sha256=FF3he2jxNuv2pnJ_2nDfYTsRql6JW6wYemWwXWsZCLk,15846
+zenml/constants.py,sha256=92ve54aAeSzoUdYEEAWhUpV5E_b54gnsVZo0PbsTlAk,15493
 zenml/container_registries/__init__.py,sha256=ZSPbBIOnzhg88kQSpYgKe_POLuru14m629665-kAVAA,2200
 zenml/container_registries/azure_container_registry.py,sha256=t1sfDa94Vzbyqtb1iPFNutJ2EXV5_p9CUNITasoiQ70,2667
 zenml/container_registries/base_container_registry.py,sha256=6c2e32wuqxYHJXm5OV2LY1MtX9yopB7WZtes9fmTAz0,7625
@@ -575,20 +575,21 @@ zenml/logger.py,sha256=me8hascHygTlfqR0w3MpvuSLGfjTLdCftK0aWf79uww,6647
 zenml/logging/__init__.py,sha256=lnqbOa31wAHwPP5f8vZazOrUwnP2QviLiIVwxoAefD8,975
 zenml/logging/step_logging.py,sha256=zN0qgj9iNt4yGf6DPmq_C9Z5zd98uYGYSFXtYw2FiQQ,17649
 zenml/login/__init__.py,sha256=Evi7hq8tpUn57IM3iX3hYP0r8oIeEWUhS471TAOyVGs,644
-zenml/login/credentials.py,sha256=CUdZ5Dk7RfkVzwoKzUGHhWDHsHsTTl3Yx-nD7i16GsQ,11455
-zenml/login/credentials_store.py,sha256=sHYfg4zUT4H7hwJ9hDVRuc_975zd3BVbZKEvSo_6cqA,22455
+zenml/login/credentials.py,sha256=PLm_IZS9fszMP46iplxiae0BZjQGldLVySLkChPWhuQ,12655
+zenml/login/credentials_store.py,sha256=mGifmB1FtFuFUEr0RXaZRubvxMhanC3Zv-Lgn-vygXg,23603
 zenml/login/pro/__init__.py,sha256=F92z11oddNdkGADOdJd08VidGpw1fqjojZFoLoM2sxQ,639
-zenml/login/pro/client.py,sha256=7Oizprc2bH2VzNdou-SMDziQAGhOvTIgEmilOtvj-OQ,16060
-zenml/login/pro/constants.py,sha256=2ozhvgYOdRnyPtLKrNorGS0EVtVSX-xY00NGRXyek68,1259
+zenml/login/pro/client.py,sha256=tgXkPEpNbBRi3oIoQXk0PGaDDQQ2azBpLPE699EhakY,15873
+zenml/login/pro/constants.py,sha256=hGWYNPZBRPTxyoKXT1b4RCXjogDZcXSuHwfbOjNLm5A,1016
 zenml/login/pro/models.py,sha256=UyIozykWVyq7SV1yk5e81YBcT7rAfkPmm226dVeSLNY,907
 zenml/login/pro/organization/__init__.py,sha256=zK8RdBK_vgL_pEJyUlbpjf7m61EQ12-C-Bm7gN4F2D8,650
 zenml/login/pro/organization/client.py,sha256=xcYfj2inTj_lTDIMwAFgMyZNo0QV5T55IAYlXUwyhKk,2148
 zenml/login/pro/organization/models.py,sha256=w3eNH3PCOEvpuy4F58OTjubtUbHCrcSIX_r_le2Xru8,983
 zenml/login/pro/tenant/__init__.py,sha256=PNSZ2ZNFBOSiMJTDoG7NK2p0p3QJvvCbw-TpKvmFlh0,644
 zenml/login/pro/tenant/client.py,sha256=g5ijYqTBoK8GvisjB3f2g6gLLaN8snyMFPblo0KAuhk,2734
-zenml/login/pro/tenant/models.py,sha256=OcYi2DjiIOzu4SPMD0HkOUHMOstB8zEjZKnDmqTSo_c,5313
-zenml/login/pro/utils.py,sha256=PnQ90eiqijY-VKqd974HzCtAosrHhzOxL1NpqZbpE5M,5355
-zenml/login/web_login.py,sha256=NkFKnPj8qDZAb2FfDW8ZvkWrGiurf7YiYEEs5m844aM,9037
+zenml/login/pro/tenant/models.py,sha256=KRZ7YPagBTilwwVukQoGLVXxQlo_1qavyhwWDikNpAc,5395
+zenml/login/pro/utils.py,sha256=-UdmcpCizkdp5ccrjT19NMvNu3FQBoZmJ-bH8vq_XU8,5148
+zenml/login/server_info.py,sha256=-_sK2L-curHdzUv1JDOwF6GoEeAXT5vFZN0J-5Ug4wU,1663
+zenml/login/web_login.py,sha256=Kq_fA9UQEravB2DtAkMmNvDttk8xppnxV617tCYUl6U,9186
 zenml/materializers/__init__.py,sha256=maME5CxKcgOkIFwG_iARt1-tuW8u8ZhTzfw50uyv_BA,1667
 zenml/materializers/base_materializer.py,sha256=M4hwkw7PB0LskCE92r-S35011l7DlFemit-EuUCW3Nc,14002
 zenml/materializers/built_in_materializer.py,sha256=AFrCW_LErey2ws6vfLTbBP5gx3IFi05YPEmT4BA-CzY,15668
@@ -659,13 +660,13 @@ zenml/models/v2/core/trigger_execution.py,sha256=1E0WNBy2qOGmDBtKCMeARan3SvW8wZq
 zenml/models/v2/core/user.py,sha256=TtMEt3AYGJUkNg7DUmY8LMkfHqxTD4zBxuJK5meJAVo,13947
 zenml/models/v2/core/workspace.py,sha256=2hA2t8E_d862pUfV-5fHdaTgxEEB5TG0kXHB2j1AvbQ,3530
 zenml/models/v2/misc/__init__.py,sha256=knhroJ2h0uHBCGzAiBBGJEiuhEA3cwI6XYBRIyXdbkQ,613
-zenml/models/v2/misc/auth_models.py,sha256=_YFUfQvE16DkVUH-Jgk0HsLM3vMxKI-FOwmJmbkKLtQ,3689
+zenml/models/v2/misc/auth_models.py,sha256=j31CQt9qfH20tyCjY0depqbKP0YZEewcgGKrixqBUbg,3688
 zenml/models/v2/misc/build_item.py,sha256=66Fywatv8bDY86pf0k530fsHTk_L4IkHL6uJL7-6dAo,1938
 zenml/models/v2/misc/external_user.py,sha256=_prXznyVA3Dc5pCK7nVE72K3tjxslG_e3JxIXLIikBo,944
 zenml/models/v2/misc/info_models.py,sha256=3X6HFIlFcz5uuT7qgU18HqZBPkgrGZQ7QcM2yBT0qeU,2645
 zenml/models/v2/misc/loaded_visualization.py,sha256=u6lapDNZDtU9eS-_EMzl00urj0yPNhiqhZcvjIz7DSg,946
 zenml/models/v2/misc/run_metadata.py,sha256=rN4ngnbm9I-1j4OiMOmxfVZPDllUMc74nOsd6IiUXcw,1371
-zenml/models/v2/misc/server_models.py,sha256=cRm_ItouirAPgCfX2YxrQrRSL2ABDmpqKtFEXKIQH8w,4474
+zenml/models/v2/misc/server_models.py,sha256=XJS7YIV8Bj84dcZq02Q0_l_O7Yb3vzbj4FmDaM61Oxw,5999
 zenml/models/v2/misc/service_connector_type.py,sha256=jGJLvIsBB87ZIEeZCLLueUf28H1QJFqXOJYtiRVvZ4M,28748
 zenml/models/v2/misc/stack_deployment.py,sha256=PgQevLjvHTpsecY_rzrXesr7jiGms7rH_QbFtI2ZIaA,3768
 zenml/models/v2/misc/user_auth.py,sha256=1-yafNA9qK4wL8ToROjaklTVI7Mj9va0t80_4wm7w3U,6988
@@ -791,9 +792,10 @@ zenml/utils/uuid_utils.py,sha256=aOGQ2SdREexcVQICPU2jUAgjvAJxTmh4ESdM52PEhck,204
 zenml/utils/visualization_utils.py,sha256=bRhgho_c8oKSBGtT4ptWj0PsXsKTKs64L3qZty3sBTM,4702
 zenml/utils/yaml_utils.py,sha256=DsbvKcJ_HYXDnNT2uSF1oKsPgP9xGpZ6G-qTFg6nQn4,5759
 zenml/zen_server/__init__.py,sha256=WyltI9TzFW2mEHZVOs6alLWMCQrrZaFALtrQXs83STA,1355
-zenml/zen_server/auth.py,sha256=dk48aS-egwDvBqpXHgEqertSESyvjZbOISe_0706DFc,35459
+zenml/zen_server/auth.py,sha256=oP5dN9pyBDbwPlQyjGKZptmmzMAfsefLMz9_Av7qLAQ,39131
 zenml/zen_server/cache.py,sha256=Tc4TSugmsU1bhThxlYfE8rv0KmltIX1CcVHgzrJ0Eus,6633
-zenml/zen_server/cloud_utils.py,sha256=Pb0tpLyVy1XkXZgZHqA9kUC5V39hikEW2YqReBtArS4,8458
+zenml/zen_server/cloud_utils.py,sha256=2uW0546xD_xiT5CyBhWvVC21a0kplcIaLQeEnBTYVKA,8467
+zenml/zen_server/csrf.py,sha256=Jsbi_IKriWCOuquSYTOEWqSXiGORJATIhR5Wrkm4XzQ,2684
 zenml/zen_server/dashboard/assets/404-Dfq64Boz.js,sha256=k1ezORVmKYeRipQFA-bn86K0T8m50WHcF6EiOwGqMxE,1033
 zenml/zen_server/dashboard/assets/@radix-DeK6qiuw.js,sha256=ksXSriMjxhb_3DmaEDYuN89nxeMJtseEkkkYso4iwE0,300701
 zenml/zen_server/dashboard/assets/@react-router-B3Z5rLr2.js,sha256=d_CQ3tJg4gfvGl6tXr87Bp8BWBIgl95vvUynXxoO5AQ,64845
@@ -990,8 +992,8 @@ zenml/zen_server/deploy/exceptions.py,sha256=tX0GNnLB_GMkeN7zGNlJRwtlrpZ5Slvyj_u
 zenml/zen_server/deploy/helm/.helmignore,sha256=u5h-ao70WpklXR1jKBJILV8PMlXqhBUgtNkDQt9f4rM,349
 zenml/zen_server/deploy/helm/Chart.yaml,sha256=sHnGnEriE1NihXNZq9h_Pj1EuKeLY97ehziZvI9nWfY,333
 zenml/zen_server/deploy/helm/README.md,sha256=376mM7AisqNyVY5FXWFps9e5FFrH_7n3dHV1AYDdqHs,1777
-zenml/zen_server/deploy/helm/templates/NOTES.txt,sha256=1APbGHYNf5cKDyDXkTYE3ybVBSPrYiSJh4dKCUOtH2c,1977
-zenml/zen_server/deploy/helm/templates/_environment.tpl,sha256=yZd8zNaWqaYjJOf19P_dfPBxyeAubGxJYt86U4LLcTA,18578
+zenml/zen_server/deploy/helm/templates/NOTES.txt,sha256=HnSMHerw1ck2BPxuO4JEttRsMZs18chDja8_WAf_Cbo,2611
+zenml/zen_server/deploy/helm/templates/_environment.tpl,sha256=KrStNrof9ciB-d6dkQO37vs9QN-k9jrQ-UX1TxARKhk,19374
 zenml/zen_server/deploy/helm/templates/_helpers.tpl,sha256=SuPztyigDEA00ftUfQtDCNK25RWZB4x58BXthNLJKJA,2042
 zenml/zen_server/deploy/helm/templates/cert-secret.yaml,sha256=VYA-0FfdF4hACYAsww8NjFL9JTuDUJTUbO8ud_Dveqo,1565
 zenml/zen_server/deploy/helm/templates/hpa.yaml,sha256=wORGNZ7-vL89PHPmyx5bR5TpDXECbyiqu0py1uOWsds,987
@@ -999,20 +1001,20 @@ zenml/zen_server/deploy/helm/templates/server-db-job.yaml,sha256=r4-mnxwuPpXtpg-
 zenml/zen_server/deploy/helm/templates/server-db-pvc.yaml,sha256=ghpBSOtjxL6aoVzYphTytAojCD0T5lP3c3dwIB-1SLw,820
 zenml/zen_server/deploy/helm/templates/server-deployment.yaml,sha256=cbVqp3dah5rsQFol1gqLU8asTj4-m3-Oz8utLChPDmw,4692
 zenml/zen_server/deploy/helm/templates/server-ingress.yaml,sha256=cgbxnKLPwzJyFqTW9GfT7nKRxG2LYPTsGBVnfrM7r3E,2274
-zenml/zen_server/deploy/helm/templates/server-secret.yaml,sha256=VBObN90vWC8z5iQq6uwLS2YnMlU_30MfRJ-A32WAzuQ,1873
+zenml/zen_server/deploy/helm/templates/server-secret.yaml,sha256=BxtCzQ4Dma5N1YomR8EfX8gqAshHXl7wfXd8oxViGW4,2391
 zenml/zen_server/deploy/helm/templates/server-service.yaml,sha256=PEYcze_4pKINKW_PbTv0vAvcXRctXuZ9jN0axqLfPFI,367
 zenml/zen_server/deploy/helm/templates/serviceaccount.yaml,sha256=4qX970F0O_SR1HSAMugyYAvwSN_63CqQofU4jV0YJO4,826
 zenml/zen_server/deploy/helm/templates/tests/test-connection.yaml,sha256=DPuf5tBRFLlxGLi8XgIbTG9nxaGn-8Z_oIkYJKjFBNs,379
-zenml/zen_server/deploy/helm/values.yaml,sha256=ci4vk2n3UbGoogcj-PgTcQRHF4nnGg9qMC-ZIGY8wWY,40445
+zenml/zen_server/deploy/helm/values.yaml,sha256=13_lzJP703GtMKs8gdQmlIJM7AgpcnTL4FWKz6SIJOs,42864
 zenml/zen_server/exceptions.py,sha256=HunRj5SZolEYBZjrGCQsJ17bO6RJwkHLU5V8HLsfucE,9859
 zenml/zen_server/feature_gate/__init__.py,sha256=yabe4fBY6NSusn-QlKQDLOvXVLERNpcAQgigsyWQIbQ,612
 zenml/zen_server/feature_gate/endpoint_utils.py,sha256=upQzEOlrD5c6cg0TfxYVM4BKDHle3LIV_c7NZHoaKvg,2220
-zenml/zen_server/feature_gate/feature_gate_interface.py,sha256=AdYAsHiNbGaloxSHpm7DLSsa8CDEGu8ikyxp9U3a0wU,1639
+zenml/zen_server/feature_gate/feature_gate_interface.py,sha256=XCgsqUN5yYCipodHCIDmCHRhyYMyt483Pp5mdpqzwHA,1647
 zenml/zen_server/feature_gate/zenml_cloud_feature_gate.py,sha256=X6sQGR70SHHFxQeLVis9v6SkBT7DiWYSbi07cbu5mxY,4406
-zenml/zen_server/jwt.py,sha256=cr-rGusdxuBss2kpCCFoUTdF0X4FmsMnmGde7EkON9c,6796
+zenml/zen_server/jwt.py,sha256=Y2DpiDSWF8pmjqTdlrJtBb3F1Bg03MD_W1_j9SjHba0,7377
 zenml/zen_server/rate_limit.py,sha256=rOg5H_6rOGQ_qiSCtMKPREmL1LL3bZyn4czKILtImJg,6057
 zenml/zen_server/rbac/__init__.py,sha256=nACbn_G7nZt_AWM3zeFL0FCmELvQnvaOFMwvTG3-6ZE,637
-zenml/zen_server/rbac/endpoint_utils.py,sha256=2cVmC1kawaCSsTT7YOupWTE9KDh6ikRE9JKgh1B6pRY,8021
+zenml/zen_server/rbac/endpoint_utils.py,sha256=PzHl3ShdpG5jIR2lyfFRdrweMagT1-TXhDAZkMg21Qs,8076
 zenml/zen_server/rbac/models.py,sha256=EpFDbsNeV5gV_9RHeNIAlvZR-lCRGU0zDmzf2g6ouW4,2434
 zenml/zen_server/rbac/rbac_interface.py,sha256=pNdfNtis5YhQgpWYkli7xNwfzNT_uXQlBaYKlSFi5HA,2881
 zenml/zen_server/rbac/rbac_sql_zen_store.py,sha256=noTb2egXJQwIH_MXMHwuhyVb2oFI0qZNw3B-uriHeQg,5670
@@ -1022,17 +1024,17 @@ zenml/zen_server/routers/__init__.py,sha256=ViyAhWL-ogHxE9wBvB_iMcur5H1NRVrzXkpo
 zenml/zen_server/routers/actions_endpoints.py,sha256=TgFFU9fMu43iqTu-ybnxJ5QokzV-eivRAd6pW2TrFe0,9549
 zenml/zen_server/routers/artifact_endpoint.py,sha256=XhbOat2pddDluiT0a_QH2RfNcqlwtf3yf2Fh92a6ZDw,5175
 zenml/zen_server/routers/artifact_version_endpoints.py,sha256=TZpBPhNI6njJLKI-QX8wbNIn9-fFFGXKWqaVdiXFZVw,8528
-zenml/zen_server/routers/auth_endpoints.py,sha256=tGYZ1Eg-U2lvtcZ-lnu6pVU0QW4e54h-cqIge_NeOmw,22353
+zenml/zen_server/routers/auth_endpoints.py,sha256=pT8sY4guywt0WUm8cjgO2aKG6suHyfbtmOl18GOib8c,23275
 zenml/zen_server/routers/code_repositories_endpoints.py,sha256=WFCRPsv3Qrm8QtCr5zxfSdgCS5WI1DPNCF4Y6vLXGow,4747
 zenml/zen_server/routers/devices_endpoints.py,sha256=luqj7owGuQ-2LyAC6Pka7GjEVpYpfi_QDdFKXhWlXek,10712
 zenml/zen_server/routers/event_source_endpoints.py,sha256=dXupWrySV3LtxsqMVoYpUJ-OK7q6o6ehfuYW_RU1JlA,10379
 zenml/zen_server/routers/flavors_endpoints.py,sha256=6NDfPmPyhtCqUhGc-moYFRugJNp7gjxbSN479iJ4gTI,5462
 zenml/zen_server/routers/logs_endpoints.py,sha256=WMw0ZOEsBeib6msMz4ye66Eozv9BdDvF8GIwEZetfH8,1912
 zenml/zen_server/routers/model_versions_endpoints.py,sha256=B4mHk275kiSiCrFWs3it6OT2_nTHigStaJ8q5I1VTcA,13113
-zenml/zen_server/routers/models_endpoints.py,sha256=_spMFdxSpeNff0CUnA-7ybqEcr_BSWTAfsyIEah_M_Q,6484
+zenml/zen_server/routers/models_endpoints.py,sha256=RTRCmzLuZK0c41SsE_ZikxUZLqvk5f7nykNkJvTwVjs,6474
 zenml/zen_server/routers/pipeline_builds_endpoints.py,sha256=SkFyhiknq0CqPEqfcrFw1fhAYy7bMe2KHe37bDJhsGM,3596
 zenml/zen_server/routers/pipeline_deployments_endpoints.py,sha256=SGnDSpgR9kfSPWTA0O-Gj-svlvOOdiNDpPxH8h7N0qA,4617
-zenml/zen_server/routers/pipelines_endpoints.py,sha256=Iy-qRAyrTHWnYR2cEzyMxWEqN5tR_EGOGIUGMTbsQcE,5643
+zenml/zen_server/routers/pipelines_endpoints.py,sha256=pVqEafoKOEmDEKrVgmvQqOz9KIhE68yn52oaOnPQzTY,5652
 zenml/zen_server/routers/plugin_endpoints.py,sha256=y3gzu3ESsBlLBOR4f8ET4WoOhgKmqTaVV53mRqG-BMU,3284
 zenml/zen_server/routers/run_templates_endpoints.py,sha256=2l5WCMo-kaGi0pxsAjvIz8NNTe1BhFHkL1pYrdbl3qE,6290
 zenml/zen_server/routers/runs_endpoints.py,sha256=G0BXeAiYsbND5ddJcUNV8qu1lpNwRZy47x7g2b61Dj8,9322
@@ -1043,23 +1045,23 @@ zenml/zen_server/routers/service_accounts_endpoints.py,sha256=CJOMngUPGnVILQr3fs
 zenml/zen_server/routers/service_connectors_endpoints.py,sha256=D8Y5-OSo9OpO0P6ugom1_pqxfxayiZ9kZfWIuOP79CU,14696
 zenml/zen_server/routers/service_endpoints.py,sha256=PyAxQLewVnpotfQI_OmyUTl7ohTCrWrteA9nfvxDRyU,5055
 zenml/zen_server/routers/stack_components_endpoints.py,sha256=Iv5sLWMq-25y58R5CepUNh2sFld6qiF3fbT4sS8OyTk,6110
-zenml/zen_server/routers/stack_deployment_endpoints.py,sha256=GzlNnGWbTfmZe5zEGZmE66Vovg7LSOSiDPRm3Txn7Mw,5187
+zenml/zen_server/routers/stack_deployment_endpoints.py,sha256=e7J3ZAlrg_8BKZ7tjYaIazXbFtgU-qB-KoeQ1tm4rYQ,5193
 zenml/zen_server/routers/stacks_endpoints.py,sha256=imL7s26xFOf4EY4zwSz8y8-Ggl6-Xfu-tJY75TNodeo,4540
 zenml/zen_server/routers/steps_endpoints.py,sha256=tNPd5A8gM6uN0cpmdSmEypF6vngv6jffENfCVXkOkNM,7937
 zenml/zen_server/routers/tags_endpoints.py,sha256=oK-A-EqAsrIXXgl7A870I2PT8_fct1dZVQDQ-g8GHys,4941
 zenml/zen_server/routers/triggers_endpoints.py,sha256=1fHKDRDr6WsjnGCvkV7SOcX9cggw1rvHzce27uAzM9A,10190
 zenml/zen_server/routers/users_endpoints.py,sha256=sXCu6fdSl5LtPl1ZCGesz96h2Ea3SA48MwpVLpJ05h0,25019
 zenml/zen_server/routers/webhook_endpoints.py,sha256=KOJsuykv_TMjL3oEItpC4OWWP75p-OEvVjRSUBd5Mro,3983
-zenml/zen_server/routers/workspaces_endpoints.py,sha256=sC8XXMeIT9AnayI1EgJRueG4J14wjW1Juiio8kjQnnc,46260
+zenml/zen_server/routers/workspaces_endpoints.py,sha256=9XmiwwF4ilSPdarE50kX06Qb_sLqH5DDfChmz_GS0Dc,46269
 zenml/zen_server/secure_headers.py,sha256=glh6QujnjyeoH1_FK-tAS-105G-qKS_34AqSzqJ6TRc,4182
 zenml/zen_server/template_execution/__init__.py,sha256=79knXLKfegsvVSVSWecpqrepq6iAavTUA4hKuiDk-WE,613
 zenml/zen_server/template_execution/runner_entrypoint_configuration.py,sha256=Y8aYJhqqs8Kv8I1q-dM1WemS5VBIfyoaaYH_YkzC7iY,1541
 zenml/zen_server/template_execution/utils.py,sha256=9GLuMV6pjEk40xfz9wDZsLo8ViiA9dllgDZx5NAOju8,16186
 zenml/zen_server/template_execution/workload_manager_interface.py,sha256=CL9c7z8ajuZE01DaHmdCDCZmsroDcTarvN-nE8jv6qQ,2590
-zenml/zen_server/utils.py,sha256=JtT41WjqEcR6XcG8NjIXYKMUapB5pj131w3AZzJ2iiE,17175
-zenml/zen_server/zen_server_api.py,sha256=NzwRCqZTTdzhCrFy2u1Rt5fbHEr_Wojw6z7Xto_JdGE,17466
+zenml/zen_server/utils.py,sha256=WqHi-J6AtZKGbkwWl_lP2jTLUk_5yIm2DNUNPBr9kdY,19383
+zenml/zen_server/zen_server_api.py,sha256=Zg4APLkMb9EftDsyeL5y1iw-tpQ9idPBkI1mkLpfuTA,17786
 zenml/zen_stores/__init__.py,sha256=6LTgH6XwDeDxKqVJ1JTfGhmS8II1NLopPloINGmdyI0,691
-zenml/zen_stores/base_zen_store.py,sha256=_e0NBL-UtJMbxUQ0VVG405LaIEPlzYgvSb7ft7CwD1E,15855
+zenml/zen_stores/base_zen_store.py,sha256=X58H-8Sn0cJU_1JuUVOIwaiDd3kARxc-2ysLZ8KCQNA,16702
 zenml/zen_stores/migrations/README.md,sha256=x04jsb6EOP6PBEGMQlDELiqKEham2O-iztAs9AylMFc,4898
 zenml/zen_stores/migrations/__init__.py,sha256=N9CHfdz0AZ6KniQ450VCIV3H0CuWtx83AloYy82woho,657
 zenml/zen_stores/migrations/alembic.py,sha256=JDqx7Md6DxnHtP3xrZG1I0cNv6NyTR0oO3tPRUPaS2I,7455
@@ -1239,7 +1241,7 @@ zenml/zen_stores/migrations/versions/ec6307720f92_simplify_model_version_links.p
 zenml/zen_stores/migrations/versions/f3b3964e3a0f_add_oauth_devices.py,sha256=2CR4R-7Vx6j_AXxo-e5Guy6OX-ZnS47HSKSGfqlO-y0,3065
 zenml/zen_stores/migrations/versions/f49904a80aa7_increase_length_of_artifact_table_sources.py,sha256=kLgfDUnQdAb5_SyFx3VKXDLC0YbuBKf9iXRDNeBin7Q,1618
 zenml/zen_stores/migrations/versions/fbd7f18ced1e_increase_step_run_field_lengths.py,sha256=kn-ng5EHe_mmLfffIFbz7T59z-to3oMx8III_4wOsz4,1956
-zenml/zen_stores/rest_zen_store.py,sha256=m49wNDBRUGcY0vndqdnabgj_KVPmkk8kGeT6zO6gO1U,159900
+zenml/zen_stores/rest_zen_store.py,sha256=Ry1VvFplj9_I_e9_nmWgR7uG7TMPYdhslkY2YEViJH4,160181
 zenml/zen_stores/schemas/__init__.py,sha256=vMb0Pyf94aMWZDWpPKuRaxNoMsTp9DZNKSQScm9Vd5I,4339
 zenml/zen_stores/schemas/action_schemas.py,sha256=vNnDF4zRy0eWgNwtcU7yY0JXyqe4I3Nns2LHRHWwiDs,6293
 zenml/zen_stores/schemas/api_key_schemas.py,sha256=pCvoTSXSHz-im6aRt-opSBnmIhw3wDRIsi-NJP5WtCQ,7243
@@ -1285,8 +1287,8 @@ zenml/zen_stores/secrets_stores/sql_secrets_store.py,sha256=Bq1djrUP9saoD7vECjS7
 zenml/zen_stores/sql_zen_store.py,sha256=uE5Rk-Fh_vsAaO0UjDY4hKZp7745R_RbkDAW9C-b5gY,416468
 zenml/zen_stores/template_utils.py,sha256=EKYBgmDLTS_PSMWaIO5yvHPLiQvMqHcsAe6NUCrv-i4,9068
 zenml/zen_stores/zen_store_interface.py,sha256=vf2gKBWfUUPtcGZC35oQB6pPNVzWVyQC8nWxVLjfrxM,92692
-zenml_nightly-0.72.0.dev20250115.dist-info/LICENSE,sha256=wbnfEnXnafPbqwANHkV6LUsPKOtdpsd-SNw37rogLtc,11359
-zenml_nightly-0.72.0.dev20250115.dist-info/METADATA,sha256=Z7UF-6cKnZHz1I-SzZPr5OsrJx95WIQadkfzBWOfWQI,21227
-zenml_nightly-0.72.0.dev20250115.dist-info/WHEEL,sha256=IYZQI976HJqqOpQU6PHkJ8fb3tMNBFjg-Cn-pwAbaFM,88
-zenml_nightly-0.72.0.dev20250115.dist-info/entry_points.txt,sha256=QK3ETQE0YswAM2mWypNMOv8TLtr7EjnqAFq1br_jEFE,43
-zenml_nightly-0.72.0.dev20250115.dist-info/RECORD,,
+zenml_nightly-0.72.0.dev20250117.dist-info/LICENSE,sha256=wbnfEnXnafPbqwANHkV6LUsPKOtdpsd-SNw37rogLtc,11359
+zenml_nightly-0.72.0.dev20250117.dist-info/METADATA,sha256=-AlPSJaDfuD7Gyzk_ppjGaDTZKjFQ5qyeL2Ms16fdpE,21355
+zenml_nightly-0.72.0.dev20250117.dist-info/WHEEL,sha256=IYZQI976HJqqOpQU6PHkJ8fb3tMNBFjg-Cn-pwAbaFM,88
+zenml_nightly-0.72.0.dev20250117.dist-info/entry_points.txt,sha256=QK3ETQE0YswAM2mWypNMOv8TLtr7EjnqAFq1br_jEFE,43
+zenml_nightly-0.72.0.dev20250117.dist-info/RECORD,,