zenml-nightly 0.82.1.dev20250521__py3-none-any.whl → 0.82.1.dev20250524__py3-none-any.whl

zenml/zen_stores/sql_zen_store.py

@@ -14,6 +14,7 @@
 """SQL Zen Store implementation."""
 
 import base64
+import inspect
 import json
 import logging
 import math
@@ -21,6 +22,7 @@ import os
 import random
 import re
 import sys
+import threading
 import time
 from collections import defaultdict
 from datetime import datetime
@@ -58,7 +60,7 @@ from pydantic import (
     field_validator,
     model_validator,
 )
-from sqlalchemy import func
+from sqlalchemy import QueuePool, func
 from sqlalchemy.engine import URL, Engine, make_url
 from sqlalchemy.exc import (
     ArgumentError,
@@ -66,6 +68,7 @@ from sqlalchemy.exc import (
 )
 from sqlalchemy.orm import Mapped, noload
 from sqlalchemy.util import immutabledict
+from sqlmodel import Session as SqlModelSession
 
 # Important to note: The select function of SQLModel works slightly differently
 # from the select function of sqlalchemy. If you input only one entity on the
@@ -74,7 +77,6 @@ from sqlalchemy.util import immutabledict
 # the tuple. While this is convenient in most cases, in unique cases like using
 # the "add_columns" functionality, one might encounter unexpected results.
 from sqlmodel import (
-    Session,
     SQLModel,
     and_,
     col,
@@ -412,6 +414,81 @@ def exponential_backoff_with_jitter(
     return random.uniform(0, exponential_backoff)
 
 
+class Session(SqlModelSession):
+    """Session subclass that automatically tracks duration and calling context."""
+
+    def __enter__(self) -> "Session":
+        """Enter the context manager.
+
+        Returns:
+            The SqlModel session.
+        """
+        if logger.isEnabledFor(logging.DEBUG):
+            # Get the request ID from the current thread object
+            self.request_id = threading.current_thread().name
+
+            # Get SQLAlchemy connection pool info
+            assert isinstance(self.bind, Engine)
+            assert isinstance(self.bind.pool, QueuePool)
+            checked_out_connections = self.bind.pool.checkedout()
+            available_connections = self.bind.pool.checkedin()
+            overflow = self.bind.pool.overflow()
+
+            # Look up the stack to find the SQLZenStore method
+            for frame in inspect.stack():
+                if "self" in frame.frame.f_locals:
+                    instance = frame.frame.f_locals["self"]
+                    if isinstance(instance, SqlZenStore):
+                        self.caller_method = (
+                            f"{instance.__class__.__name__}.{frame.function}"
+                        )
+                        break
+            else:
+                self.caller_method = "unknown"
+
+            logger.debug(
+                f"[{self.request_id}] SQL STATS - "
+                f"'{self.caller_method}' started [ conn(active): "
+                f"{checked_out_connections} conn(idle): "
+                f"{available_connections} conn(overflow): {overflow} ]"
+            )
+
+            self.start_time = time.time()
+
+        return super().__enter__()
+
+    def __exit__(
+        self,
+        exc_type: Optional[Any],
+        exc_val: Optional[Any],
+        exc_tb: Optional[Any],
+    ) -> None:
+        """Exit the context manager.
+
+        Args:
+            exc_type: The exception type.
+            exc_val: The exception value.
+            exc_tb: The exception traceback.
+        """
+        if logger.isEnabledFor(logging.DEBUG):
+            duration = (time.time() - self.start_time) * 1000
+
+            # Get SQLAlchemy connection pool info
+            assert isinstance(self.bind, Engine)
+            assert isinstance(self.bind.pool, QueuePool)
+            checked_out_connections = self.bind.pool.checkedout()
+            available_connections = self.bind.pool.checkedin()
+            overflow = self.bind.pool.overflow()
+            logger.debug(
+                f"[{self.request_id}] SQL STATS - "
+                f"'{self.caller_method}' completed in "
+                f"{duration:.2f}ms [ conn(active): "
+                f"{checked_out_connections} conn(idle): "
+                f"{available_connections} conn(overflow): {overflow} ]"
+            )
+        super().__exit__(exc_type, exc_val, exc_tb)
+
+
 class SQLDatabaseDriver(StrEnum):
     """SQL database drivers supported by the SQL ZenML store."""
 

{zenml_nightly-0.82.1.dev20250521.dist-info → zenml_nightly-0.82.1.dev20250524.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: zenml-nightly
-Version: 0.82.1.dev20250521
+Version: 0.82.1.dev20250524
 Summary: ZenML: Write production-ready ML code.
 License: Apache-2.0
 Keywords: machine learning,production,pipeline,mlops,devops

{zenml_nightly-0.82.1.dev20250521.dist-info → zenml_nightly-0.82.1.dev20250524.dist-info}/RECORD

@@ -1,5 +1,5 @@
 zenml/README.md,sha256=827dekbOWAs1BpW7VF1a4d7EbwPbjwccX-2zdXBENZo,1777
-zenml/VERSION,sha256=FX-9_pQVSQnGY2Fhoo6Eck1dVTnyq1TCpchQLE2aZAI,19
+zenml/VERSION,sha256=MexsT1YdyW8E4qM4hiGrG8oqIdUf48PHwk4-TGyiebg,19
 zenml/__init__.py,sha256=CKEyepFK-7akXYiMrNVh92Nb01Cjs23w4_YyI6sgdc8,2242
 zenml/actions/__init__.py,sha256=mrt6wPo73iKRxK754_NqsGyJ3buW7RnVeIGXr1xEw8Y,681
 zenml/actions/base_action.py,sha256=UcaHev6BTuLDwuswnyaPjdA8AgUqB5xPZ-lRtuvf2FU,25553
@@ -55,7 +55,7 @@ zenml/cli/text_utils.py,sha256=bY1GIjoULt1cW2FyrPlMoAXNS2R7cSOjDFEZQqrpVQ8,3553
 zenml/cli/user_management.py,sha256=sNnhaUxH-cHecbZBR1L0mEU0TnLNZHzI6ZBCUSQa7OY,13078
 zenml/cli/utils.py,sha256=vMAb9f6GDfNVGmZWOz9UOyPRpKI3KfnYpRl_w9YUBNE,86501
 zenml/cli/version.py,sha256=nm1iSU_1V6-MUwpMKeXcwFhLYGUMLswvQL67cEuCpxA,3635
-zenml/client.py,sha256=bQqMJVZcg5TJxwQLsB2zmKaTQ6PdYdUALIqytqO-j1k,293057
+zenml/client.py,sha256=aCj2HEE6M724js6t4QJQq1S9R6C-_hxQe4UMUN0gjbw,293202
 zenml/client_lazy_loader.py,sha256=MOBgS1ITYqGvPUnWQ6edn9s8Hr_72YfWbwEIfHKUr9g,7104
 zenml/code_repositories/__init__.py,sha256=W5bDfzAG8OXIKZSV1L-VHuzMcSCYa9qzTdPb3jqfyYw,920
 zenml/code_repositories/base_code_repository.py,sha256=Id6VjbUu8N3ZpNvBGhOgbahtoMiCAtYXed3G7YQ_iAc,5225
@@ -64,10 +64,10 @@ zenml/code_repositories/git/local_git_repository_context.py,sha256=hfX7zVDQ27Le0
 zenml/code_repositories/local_repository_context.py,sha256=1VyiYkJBDVg0iGusgRQDToGRPJuu9lx7jTBDpplukDg,2816
 zenml/config/__init__.py,sha256=DZEic7euSbwI9Yb3FMRQhTgfhqz-C6OdAiYmOb0-opI,1519
 zenml/config/base_settings.py,sha256=itoLqc1cOwEYhgSGdZmSKSaBevQkvYH7NQh7PUamazc,1700
-zenml/config/build_configuration.py,sha256=me2jXT91Ndbrq6OFFTz2zbKimUWp6APQiE_goYONLgM,6417
+zenml/config/build_configuration.py,sha256=jGGNwP0Cb7a80JXArNxDgxzxl9ytSZRtv-WW7_psLbM,6870
 zenml/config/compiler.py,sha256=bK3LCDkrFc9SapJYH-vuQZ_o8scHNs-FdC5DblIUU4U,23024
 zenml/config/constants.py,sha256=QvSgMwXWxtspcJ45CrFDP1ZY3w6gS3bIhXLOtIDAbZA,713
-zenml/config/docker_settings.py,sha256=B3pYlnyQDiwSEHO6y965Jx6BMrVqKARjp7v23uBziLc,17068
+zenml/config/docker_settings.py,sha256=WhoocDGn9cegEd4mKb_oo7pZ87sXZ0NvqkOVnWK8CY4,17985
 zenml/config/global_config.py,sha256=ZD3WodfcWMBJZOl1FNn3ztzwilGDDv9EKdHClLqSO8s,29562
 zenml/config/pipeline_configurations.py,sha256=7trCbElpqGGgawii2FrdLW8fKaAWCR8jACkNqdG_vcQ,3983
 zenml/config/pipeline_run_configuration.py,sha256=Y9C5mVUH-w4gc1w1PCQFpjEmfBBpSMvb8riA_sL78hY,2311
@@ -77,7 +77,7 @@ zenml/config/retry_config.py,sha256=4UH1xqw0G6fSEbXSNKfmiFEkwadxQef9BGMe3JBm6NI,
 zenml/config/schedule.py,sha256=-83j99U9OyiG7E322XWA7QvuLSwQzF21whwpeiF0b30,5348
 zenml/config/secret_reference_mixin.py,sha256=YvY68MTd1gE23IVprf0BLkNn62hoxcvb5nqGgc8jMkU,5871
 zenml/config/secrets_store_config.py,sha256=y05zqyQhr_DGrs3IfBGa_FRoZ043hSYRT5wzrx-zHTU,2818
-zenml/config/server_config.py,sha256=or-LKYFl-9yE3MEq_yPa5aCCu3d5_f0tL9MqPtb_3-c,31852
+zenml/config/server_config.py,sha256=DYYQ10HddkvlCWobBpwpg8ggO6imMZCnfFxzqlkQg_U,32336
 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=vjGtCoHneUJeaMkTiuzl7TTPBO94MefxNlFIAL1l5rU,13038
@@ -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=t_ULrtJF2eW7TgyYBRobl1fscwwIZXATYky8ER97ev4,860
 zenml/console.py,sha256=hj_KerPQKwnyKACj0ehSqUQX0mGVCJBKE1QvCt6ik3A,1160
-zenml/constants.py,sha256=nLjJUhBqFHWHlGKbZZw-_tn7x99y6q4vmk5LOe7GhJI,16597
+zenml/constants.py,sha256=O8b67hvsObx42r2REHA_X4XqDwqm_Hcsq7hv6q96HDs,16650
 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=-9RIkD6oXNPaU59R3PB_PtyCqsFoLPLSn5xYZmEmzbc,8915
@@ -100,8 +100,8 @@ zenml/entrypoints/base_entrypoint_configuration.py,sha256=t7sU2MEEPVomX9dZCpPhk1
 zenml/entrypoints/entrypoint.py,sha256=XNgXBCMKoidmP0_AYgMpqo-neG8Y8jG0rj43ofTDZ9E,2033
 zenml/entrypoints/pipeline_entrypoint_configuration.py,sha256=To-vTP29qAE36ndJDF1fRw9wL2Nk2bsBuO-ayAwvSmo,1646
 zenml/entrypoints/step_entrypoint_configuration.py,sha256=fJuTvJnGuhKc60CH1VMQL5EHomGXkYZulv6pVgd9M6w,7316
-zenml/enums.py,sha256=m18yol20jS9JNhuf7UPBYUsgNRCtbyCaWPNpeEAJeXY,11199
-zenml/environment.py,sha256=UtriBwD0ylEK1cmSjEkDTlUUXBifBEukAn6gEdcnv4M,11827
+zenml/enums.py,sha256=xXp6VkaY03NlGKEBs9Bf7oVfUrtESECmhkQYAaKRS5g,11239
+zenml/environment.py,sha256=_iS9o4rTxRCjLocSw_LEe9pZ9jfyMhXBY47LIUF3VCI,12255
 zenml/event_hub/__init__.py,sha256=-fD9mPOslf4J-_XFBPp5gYlPz7-6ZaAKHa5jxf_nyAo,757
 zenml/event_hub/base_event_hub.py,sha256=PqKrnvOye0UUS3s09zGgjI5dtj0IwzEBDbavA_PgfZQ,6579
 zenml/event_hub/event_hub.py,sha256=e1eCRB1qAabOFIx2OJCAoNqN1QQIW0CSTukiWuGMOi8,6502
@@ -253,7 +253,7 @@ zenml/integrations/feast/feature_stores/__init__.py,sha256=Wi3NBBBPJg6CjgtxmBjoU
 zenml/integrations/feast/feature_stores/feast_feature_store.py,sha256=jV6WznuKT3y1aikI3OEwoI8r_l8bEu5waX0LKePPuU8,5880
 zenml/integrations/feast/flavors/__init__.py,sha256=gbCZ4tKgLZSI4-gzOCR2xihiPNmpe-lMUxwvMrhYL-w,858
 zenml/integrations/feast/flavors/feast_feature_store_flavor.py,sha256=E0k2iwgNti8lOVr9W8n2nTVQQf2wSeJWeaQD1vwAIbU,3060
-zenml/integrations/gcp/__init__.py,sha256=4bl_mutvdwXw-ZVdWdTHi3spTRH1JRCtBNPFxq5c74k,2957
+zenml/integrations/gcp/__init__.py,sha256=9o7TXcQlNebImis821VXcOVFxKau86mbQ7H8d8e6mIw,2984
 zenml/integrations/gcp/artifact_stores/__init__.py,sha256=zYQkZBI4-COGX-E0NS7G-hLT88wbQBnYY6Oh1gruSRs,798
 zenml/integrations/gcp/artifact_stores/gcp_artifact_store.py,sha256=XfSIJ4HtsZvaUrUtzXvUp7QHr3WbgVDNsz7_q1h-DCo,10988
 zenml/integrations/gcp/constants.py,sha256=ZBQS_ZEjerUrJq-hH3UusgZAvB45FLgxNv11TSt3qhw,1334
@@ -271,7 +271,7 @@ zenml/integrations/gcp/image_builders/gcp_image_builder.py,sha256=5T6BXsHxLhvp1B
 zenml/integrations/gcp/orchestrators/__init__.py,sha256=6xLFJKZKQk73fHPF-XdpbQO87zjQNGTsNHjJjLfG_Kg,805
 zenml/integrations/gcp/orchestrators/vertex_orchestrator.py,sha256=qoMCr36buZUz0y4CyTFQde3RDslkaGLAG0FjXc0XEPU,42100
 zenml/integrations/gcp/service_connectors/__init__.py,sha256=fdydawaor8KAtMYvRZieiTuA1i5QATxXXgI-yV1lsn8,788
-zenml/integrations/gcp/service_connectors/gcp_service_connector.py,sha256=JGrFTKkQV4ZDn_yTEqX-AtBMraFasFgzLVws2mvhS64,94915
+zenml/integrations/gcp/service_connectors/gcp_service_connector.py,sha256=9u-vEHbmSyN5IGwYI8v39TcFZg5ObgkxlbwSPz-e5zE,95018
 zenml/integrations/gcp/step_operators/__init__.py,sha256=iPkob2LtPIQ-OHszhbNz_ojhoovL6SprmTx37It4EJ8,808
 zenml/integrations/gcp/step_operators/vertex_step_operator.py,sha256=X8CCniyAo7NHiy3Mv_YSKQ4Hw3UYMXob6B3uWKsCJ-0,13567
 zenml/integrations/gcp/vertex_custom_job_parameters.py,sha256=B5RLkw7KDOi4ZfWHFnC6TGLphXMzToMjROxszCEAS9c,3676
@@ -317,7 +317,7 @@ zenml/integrations/hyperai/__init__.py,sha256=6ed5rgRgiRr2Ksi3DDFUaNQwPkCfE4BRSn
 zenml/integrations/hyperai/flavors/__init__.py,sha256=PUGBPmQ7y3H7QU2zAj7Ri0rrUBbOWnM_L59AIVWUYwU,800
 zenml/integrations/hyperai/flavors/hyperai_orchestrator_flavor.py,sha256=erjbYJ5E3qkifyGd_ArtbY1JdfyOEz4vgl9QrZK0R9Y,5578
 zenml/integrations/hyperai/orchestrators/__init__.py,sha256=kSYpMZPEWwNu2vxoOC6PeyQ9RLzsPAgTHxL35K36MiE,784
-zenml/integrations/hyperai/orchestrators/hyperai_orchestrator.py,sha256=2bKFlJE6HUU3oItkj3-siVxTKUpextN25t8aX7du4ZU,20286
+zenml/integrations/hyperai/orchestrators/hyperai_orchestrator.py,sha256=6Z4GAXfkC2Ia4czQrgCy85xuocC9TYXTLmaNP8xxv_s,20287
 zenml/integrations/hyperai/service_connectors/__init__.py,sha256=oHuCNC09z5C7Wlb3vV1d4zJjftttHg364eoEBVRDOdo,803
 zenml/integrations/hyperai/service_connectors/hyperai_service_connector.py,sha256=7Ql5cVoSY3SE4j7uzeVi5TWuoKFDvsHFTn72k9_wPUY,13400
 zenml/integrations/integration.py,sha256=ljUK2GHy-LNqsdloOvf8mIfUQsqoi0yBJYocMyS-DX8,6879
@@ -337,7 +337,7 @@ zenml/integrations/kubernetes/flavors/__init__.py,sha256=a5gU45qCj3FkLwl_uVjlIkL
 zenml/integrations/kubernetes/flavors/kubernetes_orchestrator_flavor.py,sha256=JH_Kxfh1eoiSlz-OpixJ6itEsDiGPvbkA5_PffMTa54,10215
 zenml/integrations/kubernetes/flavors/kubernetes_step_operator_flavor.py,sha256=xFO7cSusji-mgbRrt4mU29gdyC9iEjEHKtomdFLp9mM,6265
 zenml/integrations/kubernetes/orchestrators/__init__.py,sha256=TJID3OTieZBox36WpQpzD0jdVRA_aZVcs_bNtfXS8ik,811
-zenml/integrations/kubernetes/orchestrators/kube_utils.py,sha256=8BVOpg40dGeRSkbzG_66xOksry5xpMVVmcXVDqb1bSs,18747
+zenml/integrations/kubernetes/orchestrators/kube_utils.py,sha256=N66GH5ac22Xm_A3nr162kbFBhMeypSFaQjOQRHlGXIQ,18942
 zenml/integrations/kubernetes/orchestrators/kubernetes_orchestrator.py,sha256=qubV13fSGE_vo8wSzIfcp66McclLw-zDtmB4yCCaS08,25974
 zenml/integrations/kubernetes/orchestrators/kubernetes_orchestrator_entrypoint.py,sha256=Nd1YZsP6FZ4LTpoeFIM3YlFUugHX7F9axb7U9ZgGNdQ,12692
 zenml/integrations/kubernetes/orchestrators/kubernetes_orchestrator_entrypoint_configuration.py,sha256=KjHfQK9VQEQkkkM2i9w51AzqolgIU01M5dgb2YGamvY,2754
@@ -596,7 +596,7 @@ zenml/login/server_info.py,sha256=-_sK2L-curHdzUv1JDOwF6GoEeAXT5vFZN0J-5Ug4wU,16
 zenml/login/web_login.py,sha256=Kq_fA9UQEravB2DtAkMmNvDttk8xppnxV617tCYUl6U,9186
 zenml/materializers/__init__.py,sha256=C3lZaTmIFxwIPwCKF8oLQUkLaX2o7Dbj9hvYVFrSzt8,1758
 zenml/materializers/base_materializer.py,sha256=M4hwkw7PB0LskCE92r-S35011l7DlFemit-EuUCW3Nc,14002
-zenml/materializers/built_in_materializer.py,sha256=KjW3p8LWORZTjHqoGGC-Hlf-bdNSir3zsbRD2C_oQBQ,16951
+zenml/materializers/built_in_materializer.py,sha256=HgcCHg3GpuQ4t-jecYcdg0kldUfLeUJvIpm8-wcS11I,17189
 zenml/materializers/cloudpickle_materializer.py,sha256=x8a6jEMTky6N2YVHiwrnGWSfVJUpiy-4kQsD2Aqj_E0,4837
 zenml/materializers/materializer_registry.py,sha256=ic-aWhJ2Ex9F_rml2dDVAxhRfW3nd71QMxzfTPP6BIM,4002
 zenml/materializers/numpy_materializer.py,sha256=OLcHF9Z0tAqQ_U8TraA0vGmZjHoT7eT_XevncIutt0M,1715
@@ -805,7 +805,7 @@ zenml/utils/yaml_utils.py,sha256=RvEr-N84wwG1Aq8rfdPAP7lev_SEaSQlqcL9fB_vHk8,584
 zenml/zen_server/__init__.py,sha256=WyltI9TzFW2mEHZVOs6alLWMCQrrZaFALtrQXs83STA,1355
 zenml/zen_server/auth.py,sha256=sJ_UAG58Qa0sUx8P1QkWmwp5zfAafFnm8WBHAOnPi4w,40070
 zenml/zen_server/cache.py,sha256=Tc4TSugmsU1bhThxlYfE8rv0KmltIX1CcVHgzrJ0Eus,6633
-zenml/zen_server/cloud_utils.py,sha256=qKMtHjnkVfAk_yxF2K9iC0TkMCjp8s0gdz_cEsbB4DE,10908
+zenml/zen_server/cloud_utils.py,sha256=vKsjWadgPctZDtfeG9qJC-5gKOir_tZsglYMBkC8QqU,11874
 zenml/zen_server/csrf.py,sha256=Jsbi_IKriWCOuquSYTOEWqSXiGORJATIhR5Wrkm4XzQ,2684
 zenml/zen_server/dashboard/assets/404-_AtuLtaX.js,sha256=Y2jAo7K2feru-k7QcMxjnfUNkDrO7xv341YyKKpjZiU,1033
 zenml/zen_server/dashboard/assets/@radix-C7hRs6Kx.js,sha256=cNA9UX8LGrKUQubGrls3E3Wq1fCrlK51W14yh22FE_U,296700
@@ -1027,47 +1027,47 @@ zenml/zen_server/rbac/rbac_sql_zen_store.py,sha256=ZBjAbCUlhMsp8GZiIkULl2ztJ8K2W
 zenml/zen_server/rbac/utils.py,sha256=09r1dSZ04hcag21rg6Ug2fMCdYfNjSBnx_W8C336GzE,24254
 zenml/zen_server/rbac/zenml_cloud_rbac.py,sha256=L1Tio4IcNNxNWnVpAqBWYgdVfrqR20ddvdhWAo1i3f4,6055
 zenml/zen_server/routers/__init__.py,sha256=ViyAhWL-ogHxE9wBvB_iMcur5H1NRVrzXkpogVY7FBA,641
-zenml/zen_server/routers/actions_endpoints.py,sha256=MFausi27i5QBB5s4vbjQQTVqFiRlk5jF5lRM6vsRM2o,9422
-zenml/zen_server/routers/artifact_endpoint.py,sha256=huFufMN-hHzBUKqYmn1WeWVaJic1ypvqSpTWFxklUDQ,4976
-zenml/zen_server/routers/artifact_version_endpoints.py,sha256=zda_iINad3VnKcyGgCW2ZveblEzZa5YjwrCAsuEN13s,11254
-zenml/zen_server/routers/auth_endpoints.py,sha256=LKYtYbuuJljL6VdPNTeunfb7puS8iWwM0QTGVYJdBmY,22315
-zenml/zen_server/routers/code_repositories_endpoints.py,sha256=_D1896lrEgYD516y69iiFrPn5b62MxlrTcmDt1AyqBA,6476
-zenml/zen_server/routers/devices_endpoints.py,sha256=GluKziLikUn_IoflcBwV5wB8ui71STjDsg2Mjgn43MU,10551
-zenml/zen_server/routers/event_source_endpoints.py,sha256=x7XL70ybrrWbkq_pM7xv67zHLMi6SN0GpXqKSxO_ORQ,10164
-zenml/zen_server/routers/flavors_endpoints.py,sha256=Nq24PcvDyRLRAsdy0u3WGdXqLiHK6iE_uN29VLw-_U0,5273
-zenml/zen_server/routers/logs_endpoints.py,sha256=qq7vLfJzmjIZ9C4DSw_IPjqHi5LdPbE0xHN8tDPa2uA,1879
-zenml/zen_server/routers/model_versions_endpoints.py,sha256=VhmBUuMFef8bXvDIRncPD5hLa851lmIvmASI_lw2F5o,15483
-zenml/zen_server/routers/models_endpoints.py,sha256=Bh_f7wvw1ZNwmZ51LRiiz3ftatkp_J5t622bJinmOaY,5680
-zenml/zen_server/routers/pipeline_builds_endpoints.py,sha256=PQRSrDcyor6laZyOl74BODcdg0JSdFr-hbNvEJkJIG0,5333
-zenml/zen_server/routers/pipeline_deployments_endpoints.py,sha256=f45O7dQqZnFWA7C62N5VGAtBsqpt38cRYsGgTaBS3Dw,9534
-zenml/zen_server/routers/pipelines_endpoints.py,sha256=R6OM-0Tc6ZNPzQmLs5MWTJIE4FaksHt6xuDqCBtmr1U,7782
-zenml/zen_server/routers/plugin_endpoints.py,sha256=vgIdA0kdYsf8PeF_pqDaqEYnIi2mlYALAv4uB9Nn7S0,3239
-zenml/zen_server/routers/projects_endpoints.py,sha256=iIgCg_0_iR5OddgULvsP4xxMOOmEGmtCuSKYZVRE9f4,8464
-zenml/zen_server/routers/run_metadata_endpoints.py,sha256=UTWQvweZTGuZnLrqhbIRYQ-z_TfOPY3Kn9p4YwU0tqw,3662
-zenml/zen_server/routers/run_templates_endpoints.py,sha256=AdBvwbXefo9nKU2erkNZNFjZ3b8LitPuWR22rgWF3JM,8469
-zenml/zen_server/routers/runs_endpoints.py,sha256=YoTqWm-Mx97q8ZMYXgIY8gr2ICJpRlk6DxxHKN1utFU,12631
-zenml/zen_server/routers/schedule_endpoints.py,sha256=UXt6TIEX0PYhLoBgU-n5kIvWckX1BY_P5uUA_lK9J9A,5782
-zenml/zen_server/routers/secrets_endpoints.py,sha256=diYTC-jl6Hxd_BHVaNsPf2ZcWUwC99naWHsM1xRjR-g,9198
-zenml/zen_server/routers/server_endpoints.py,sha256=4KHL11w0pPEjs1ZE1SYNb80eojhou3Q5AabnpaWmYS0,8006
-zenml/zen_server/routers/service_accounts_endpoints.py,sha256=egShMu776yf0jmoMLmdEnC29HQ7glHV8-MwWflcOcak,11930
-zenml/zen_server/routers/service_connectors_endpoints.py,sha256=e3iarrmjalEhL1ErXc9ZemqksPsEJ01o6xlZPgL2AKQ,17401
-zenml/zen_server/routers/service_endpoints.py,sha256=VmcmTXXkW2K6-_WC-Te8679Dr5uY0nsbIQUapUvrUfc,5488
-zenml/zen_server/routers/stack_components_endpoints.py,sha256=8g8CA88uxOnWDMN4tZgZpSzkyThfDEhJiyJB16bjb9A,8143
-zenml/zen_server/routers/stack_deployment_endpoints.py,sha256=dzSGkX8bnQAhdS5cfjN6dyklCporIo6cun0B4evEnII,5329
-zenml/zen_server/routers/stacks_endpoints.py,sha256=iF9SANF2JCdl4YCrL0PpIG8LzoGO8C5-A1UnOZyBXwc,7449
-zenml/zen_server/routers/steps_endpoints.py,sha256=By-9_VKrJeSo6UUqrggDhGKtrnhxiElORK_T-Ngn-OM,8135
-zenml/zen_server/routers/tag_resource_endpoints.py,sha256=AdmbujrwKK6arnSx2VHpJWbZXnlviGlay7LCov4x9Fc,3203
-zenml/zen_server/routers/tags_endpoints.py,sha256=lRjnMaJOdTnooMHoQYIViKfgu32cMq7rI3Lyeaxq4QI,4467
-zenml/zen_server/routers/triggers_endpoints.py,sha256=5fb0EITa3bs--AtKQRQZ5QswHu4pb37OrG5y0OlnGXE,9983
-zenml/zen_server/routers/users_endpoints.py,sha256=2SfqHutGp0OqwYVytx3WeIwF51nZ3f874uCiCc_m_7g,24531
-zenml/zen_server/routers/webhook_endpoints.py,sha256=KOJsuykv_TMjL3oEItpC4OWWP75p-OEvVjRSUBd5Mro,3983
+zenml/zen_server/routers/actions_endpoints.py,sha256=-zzeTawwTrkWOnp8ltrGS_tUnWnyiXMFNaBRXI3zddk,9500
+zenml/zen_server/routers/artifact_endpoint.py,sha256=d-5yPQO4N9O2W97j0f_5fghNThnNe0izT9WpZLaJyuA,5054
+zenml/zen_server/routers/artifact_version_endpoints.py,sha256=1M1DpSFEDBDUVAd69eXf6Y-2vCiq8_MN5gi5nu0f3Iw,11397
+zenml/zen_server/routers/auth_endpoints.py,sha256=QInOg3IPn2ErzccmOLw8xJQd-jtyVR2VFZ9D-virM0U,22386
+zenml/zen_server/routers/code_repositories_endpoints.py,sha256=BEjnbUbH2CM7q-VfrKES-d3ngSipnvfTuvAsd88yRCo,6554
+zenml/zen_server/routers/devices_endpoints.py,sha256=ehOVJZwXYtZZh6VeXcUlhOgPq0yoZk2-mumKEeOn6Do,10629
+zenml/zen_server/routers/event_source_endpoints.py,sha256=-H5li0QTviGA2vFD6yW1N580KCuB9JFvosiXIdB0N_4,10242
+zenml/zen_server/routers/flavors_endpoints.py,sha256=S3RqxH68ytKlB8nLJooFpjb2RGNJa2xfPSz9zuYkS3Q,5364
+zenml/zen_server/routers/logs_endpoints.py,sha256=55nWKudfsE86xGKUf-XcG35EWldb8TSMEyEM1IRcr5Q,1905
+zenml/zen_server/routers/model_versions_endpoints.py,sha256=4gKbJC_vSYY6JZfJfmqDB0zwUM40vMGmLhrzznbTpko,15652
+zenml/zen_server/routers/models_endpoints.py,sha256=fF7q3znX5KaQiMtFK32yq4CeMASf7lt_ygl2QeNDzVo,5758
+zenml/zen_server/routers/pipeline_builds_endpoints.py,sha256=oYMIiGRGmWgF467YcvupGJvmkUHiygsYjl1lxk0D7RE,5398
+zenml/zen_server/routers/pipeline_deployments_endpoints.py,sha256=tY48uFvzlU2tJRG-QRHAe7HB6iYTB5881SMdaPQvVDI,9612
+zenml/zen_server/routers/pipelines_endpoints.py,sha256=Nnh5jPLF3hV-muRPQMt_8j9dBmjOloT6a4o29PBTQj0,7873
+zenml/zen_server/routers/plugin_endpoints.py,sha256=vyKKQIiLSzC5WkZoQdzS7sU-qYFSGn-fwCt917RxX_M,3278
+zenml/zen_server/routers/projects_endpoints.py,sha256=uL7oK8_0-H2nhjvPYHfYPvJOU4id7-n81MwjiSzHToE,8555
+zenml/zen_server/routers/run_metadata_endpoints.py,sha256=hUatO7c6LUBHpMDZYgDqiUbVUSBpt6-ZxaSC9eAaF9I,3688
+zenml/zen_server/routers/run_templates_endpoints.py,sha256=7MZjreJikAcwGfUmSbw4dunZISSKlf1iy33oYsXYXRI,8560
+zenml/zen_server/routers/runs_endpoints.py,sha256=CTLrVvYdcnbVgv04WpiJGervAI7nGiFn8MCpTdqDvJk,12774
+zenml/zen_server/routers/schedule_endpoints.py,sha256=m8TVhncapdtQqPm5EEm7CAeGr_i_0Tqa61EA8p1bIl4,5860
+zenml/zen_server/routers/secrets_endpoints.py,sha256=VgepInQHZXt-d8IAinftHs1n9il6hTD1qca6w5Pnnsk,9302
+zenml/zen_server/routers/server_endpoints.py,sha256=Aci-7TB5VZylmasEYc2tVpG7Npgtp6Dj56nrmy7mMmc,8133
+zenml/zen_server/routers/service_accounts_endpoints.py,sha256=q-vFzzYYorqaOPUnESQU06h2I-BaZvIyPxCAhQ4gLPw,12086
+zenml/zen_server/routers/service_connectors_endpoints.py,sha256=CkwhnNI49_64lqG7aWHtyE41iqKlY-lumiOrWU7aDU0,17570
+zenml/zen_server/routers/service_endpoints.py,sha256=37CY-22h4hTscEkBByKJqdHcOWsP2lTGAiY2WgBPB0w,5566
+zenml/zen_server/routers/stack_components_endpoints.py,sha256=MBQ3iz29WN1n9T_BLpd2BtWxWoLIBP6QT7BfaAkGEMY,8580
+zenml/zen_server/routers/stack_deployment_endpoints.py,sha256=r7N4UaDAwj9ACYfdyUofRHpNPRrzWl-vmGoSErN9oYI,5381
+zenml/zen_server/routers/stacks_endpoints.py,sha256=ci-WZ774Q2T3-XN6cZARJ95wmRUW2-W1mh7gKlbNE54,7527
+zenml/zen_server/routers/steps_endpoints.py,sha256=KZpVsEsycLMmiEGnwT2EkmQvaTIIyL4QOUJBXMu7NSo,8239
+zenml/zen_server/routers/tag_resource_endpoints.py,sha256=54HBt31whSj7wjzhw9Xzb6aJLBDVPd5iyg5fJ2Ej5qU,3268
+zenml/zen_server/routers/tags_endpoints.py,sha256=H0aXms8iAkcxo9lsBwpDrVhJnY5n2cqBd3Iz1OI8r_k,4545
+zenml/zen_server/routers/triggers_endpoints.py,sha256=Z9ry01MMpCZ2wcjcvJjnEu1eDhkcGACKuBL_F9m8smI,10100
+zenml/zen_server/routers/users_endpoints.py,sha256=-0wjKK0LYOUY7gmmbTRU1AHIej99dRvy3TI0bsGk9hQ,24687
+zenml/zen_server/routers/webhook_endpoints.py,sha256=4Ca6k_qyE5lZpflEqV0P4mcuXnhlW9N-YemZIwchRJg,4009
 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=CiN7d8jCUTV5oKuoNF9Z_eDkitZrhL6V_Tz4CjU7em8,19410
 zenml/zen_server/template_execution/workload_manager_interface.py,sha256=CL9c7z8ajuZE01DaHmdCDCZmsroDcTarvN-nE8jv6qQ,2590
-zenml/zen_server/utils.py,sha256=yOwPI0DxzDIxF7Ppnty_DuGMWYJ_ubHUNWfbTQ1_q9Q,18461
-zenml/zen_server/zen_server_api.py,sha256=1_YHOUmBulXuHWCxDLYtDCMxGze159UmU3f05643PSg,18182
+zenml/zen_server/utils.py,sha256=roJuND2KMVPue7fztVOKn4Tvoj1bvQ54TFpXgIa78Y0,20474
+zenml/zen_server/zen_server_api.py,sha256=Bhdn2AwGINEpKJiTL3QpC0EK8pvgGNK1IZNJr6E6ijE,23229
 zenml/zen_stores/__init__.py,sha256=6LTgH6XwDeDxKqVJ1JTfGhmS8II1NLopPloINGmdyI0,691
 zenml/zen_stores/base_zen_store.py,sha256=AplsW2NR-G9_CU54XvNTQJo4W0KJ5TJV22cjKW4n2BY,16124
 zenml/zen_stores/migrations/README.md,sha256=x04jsb6EOP6PBEGMQlDELiqKEham2O-iztAs9AylMFc,4898
@@ -1273,7 +1273,7 @@ zenml/zen_stores/migrations/versions/f49904a80aa7_increase_length_of_artifact_ta
 zenml/zen_stores/migrations/versions/f76a368a25a5_add_stack_description.py,sha256=u8fRomaasFeGhxvM2zU-Ab-AEpVsWm5zRcixxKFXdRw,904
 zenml/zen_stores/migrations/versions/fbd7f18ced1e_increase_step_run_field_lengths.py,sha256=kn-ng5EHe_mmLfffIFbz7T59z-to3oMx8III_4wOsz4,1956
 zenml/zen_stores/migrations/versions/ff538a321a92_migrate_onboarding_state.py,sha256=gsUFLJQ32_o9U35JCVqkqJVVk-zfq3yel25hXhzVFm4,3829
-zenml/zen_stores/rest_zen_store.py,sha256=d5JHoGeO_FSg55MmVNIFYmwN1JdAnJ2QbE6R590fKZU,158599
+zenml/zen_stores/rest_zen_store.py,sha256=ljz1iIc8szDh1dlX1129Q817gcDBuzFzQnw6hkSMYCg,159628
 zenml/zen_stores/schemas/__init__.py,sha256=4EXqExiVyxdnGxhQ_Hz79mOdRuMD0LsGlw0PaP2Ef6o,4333
 zenml/zen_stores/schemas/action_schemas.py,sha256=sv2J2TP12MeyGPQR2JsOPIivbPQ5OImg64exYS7CZBM,6496
 zenml/zen_stores/schemas/api_key_schemas.py,sha256=0pK7b9HlJuQL3DuKT4eGjFb87tyd4x-E2VyxJLpRv3o,7459
@@ -1316,11 +1316,11 @@ zenml/zen_stores/secrets_stores/hashicorp_secrets_store.py,sha256=5err1a-TrV3SR5
 zenml/zen_stores/secrets_stores/secrets_store_interface.py,sha256=Q2Jbnt2Pp7NGlR-u1YBfRZV2g8su2Fd0ArBMdksAE-Q,2819
 zenml/zen_stores/secrets_stores/service_connector_secrets_store.py,sha256=S87ne23D08PAwtfRVlVnBn8R0ilTpEh6r8blauNV5WQ,6941
 zenml/zen_stores/secrets_stores/sql_secrets_store.py,sha256=LPFW757WCJLP1S8vrvjsrl2Tf1yo281xUTjSBsos4qk,8788
-zenml/zen_stores/sql_zen_store.py,sha256=oUB9u3tMVmRrSxslC1oEN5kHQa8i7g4bNEpJKgsYmqY,441911
+zenml/zen_stores/sql_zen_store.py,sha256=jsx_1_MJZE4dnNXxj8bAMX8pT5JfoJliADSBF9n6kJc,444809
 zenml/zen_stores/template_utils.py,sha256=GbJ7LgGVYHSCKPEA8RNTxPoVTWqpC77F_lGzjJ4O1Fw,9220
 zenml/zen_stores/zen_store_interface.py,sha256=fF_uL_FplnvGvM5o3jOQ8i1zHXhuhKLL2n4nvIKSR7E,92090
-zenml_nightly-0.82.1.dev20250521.dist-info/LICENSE,sha256=wbnfEnXnafPbqwANHkV6LUsPKOtdpsd-SNw37rogLtc,11359
-zenml_nightly-0.82.1.dev20250521.dist-info/METADATA,sha256=0YoGPYbCHpQHx_aj7HeAvDvkV3GGQIWqJOn5UFaitPs,24317
-zenml_nightly-0.82.1.dev20250521.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-zenml_nightly-0.82.1.dev20250521.dist-info/entry_points.txt,sha256=QK3ETQE0YswAM2mWypNMOv8TLtr7EjnqAFq1br_jEFE,43
-zenml_nightly-0.82.1.dev20250521.dist-info/RECORD,,
+zenml_nightly-0.82.1.dev20250524.dist-info/LICENSE,sha256=wbnfEnXnafPbqwANHkV6LUsPKOtdpsd-SNw37rogLtc,11359
+zenml_nightly-0.82.1.dev20250524.dist-info/METADATA,sha256=OFFcRSYChQvW4vCBOO5FJDfx8UkZbhmq9DKwtmBN2s0,24317
+zenml_nightly-0.82.1.dev20250524.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+zenml_nightly-0.82.1.dev20250524.dist-info/entry_points.txt,sha256=QK3ETQE0YswAM2mWypNMOv8TLtr7EjnqAFq1br_jEFE,43
+zenml_nightly-0.82.1.dev20250524.dist-info/RECORD,,