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

zenml/VERSION

@@ -1 +1 @@
-0.82.1.dev20250521
+0.82.1.dev20250522

zenml/config/build_configuration.py

@@ -108,6 +108,13 @@ class BuildConfiguration(BaseModel):
             )
             if digest:
                 hash_.update(digest.encode())
+            elif self.settings.skip_build:
+                # If the user has specified to skip the build, the image being
+                # used for the run is whatever they set for `parent_image`.
+                # In this case, the bevavior depends on the orchestrators image
+                # pull policy, and whether there is any caching involved. This
+                # is out of our control here, so we don't log any warning.
+                pass
             else:
                 logger.warning(
                     "Unable to fetch parent image digest for image `%s`. "

zenml/zen_server/routers/actions_endpoints.py

@@ -44,7 +44,7 @@ from zenml.zen_server.rbac.utils import (
     verify_permission_for_model,
 )
 from zenml.zen_server.utils import (
-    handle_exceptions,
+    async_fastapi_endpoint_wrapper,
     make_dependable,
     plugin_flavor_registry,
     zen_store,
@@ -61,7 +61,7 @@ router = APIRouter(
     "",
     responses={401: error_response, 404: error_response, 422: error_response},
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def list_actions(
     action_filter_model: ActionFilter = Depends(make_dependable(ActionFilter)),
     hydrate: bool = False,
@@ -132,7 +132,7 @@ def list_actions(
     "/{action_id}",
     responses={401: error_response, 404: error_response, 422: error_response},
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def get_action(
     action_id: UUID,
     hydrate: bool = True,
@@ -179,7 +179,7 @@ def get_action(
     "",
     responses={401: error_response, 409: error_response, 422: error_response},
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def create_action(
     action: ActionRequest,
     _: AuthContext = Security(authorize),
@@ -225,7 +225,7 @@ def create_action(
     "/{action_id}",
     responses={401: error_response, 404: error_response, 422: error_response},
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def update_action(
     action_id: UUID,
     action_update: ActionUpdate,
@@ -280,7 +280,7 @@ def update_action(
     "/{action_id}",
     responses={401: error_response, 404: error_response, 422: error_response},
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def delete_action(
     action_id: UUID,
     force: bool = False,

zenml/zen_server/routers/artifact_endpoint.py

@@ -36,7 +36,7 @@ from zenml.zen_server.rbac.endpoint_utils import (
 )
 from zenml.zen_server.rbac.models import ResourceType
 from zenml.zen_server.utils import (
-    handle_exceptions,
+    async_fastapi_endpoint_wrapper,
     make_dependable,
     zen_store,
 )
@@ -52,7 +52,7 @@ artifact_router = APIRouter(
     "",
     responses={401: error_response, 404: error_response, 422: error_response},
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def list_artifacts(
     artifact_filter_model: ArtifactFilter = Depends(
         make_dependable(ArtifactFilter)
@@ -83,7 +83,7 @@ def list_artifacts(
     "",
     responses={401: error_response, 409: error_response, 422: error_response},
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def create_artifact(
     artifact: ArtifactRequest,
     _: AuthContext = Security(authorize),
@@ -106,7 +106,7 @@ def create_artifact(
     "/{artifact_id}",
     responses={401: error_response, 404: error_response, 422: error_response},
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def get_artifact(
     artifact_id: UUID,
     hydrate: bool = True,
@@ -133,7 +133,7 @@ def get_artifact(
     "/{artifact_id}",
     responses={401: error_response, 404: error_response, 422: error_response},
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def update_artifact(
     artifact_id: UUID,
     artifact_update: ArtifactUpdate,
@@ -160,7 +160,7 @@ def update_artifact(
     "/{artifact_id}",
     responses={401: error_response, 404: error_response, 422: error_response},
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def delete_artifact(
     artifact_id: UUID,
     _: AuthContext = Security(authorize),

zenml/zen_server/routers/artifact_version_endpoints.py

@@ -66,7 +66,7 @@ from zenml.zen_server.rbac.utils import (
     get_allowed_resource_ids,
 )
 from zenml.zen_server.utils import (
-    handle_exceptions,
+    async_fastapi_endpoint_wrapper,
     make_dependable,
     set_filter_project_scope,
     zen_store,
@@ -83,7 +83,7 @@ artifact_version_router = APIRouter(
     "",
     responses={401: error_response, 404: error_response, 422: error_response},
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def list_artifact_versions(
     artifact_version_filter_model: ArtifactVersionFilter = Depends(
         make_dependable(ArtifactVersionFilter)
@@ -127,7 +127,7 @@ def list_artifact_versions(
     "",
     responses={401: error_response, 409: error_response, 422: error_response},
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def create_artifact_version(
     artifact_version: ArtifactVersionRequest,
     _: AuthContext = Security(authorize),
@@ -150,7 +150,7 @@ def create_artifact_version(
     BATCH,
     responses={401: error_response, 409: error_response, 422: error_response},
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def batch_create_artifact_version(
     artifact_versions: List[ArtifactVersionRequest],
     _: AuthContext = Security(authorize),
@@ -173,7 +173,7 @@ def batch_create_artifact_version(
     "/{artifact_version_id}",
     responses={401: error_response, 404: error_response, 422: error_response},
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def get_artifact_version(
     artifact_version_id: UUID,
     hydrate: bool = True,
@@ -200,7 +200,7 @@ def get_artifact_version(
     "/{artifact_version_id}",
     responses={401: error_response, 404: error_response, 422: error_response},
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def update_artifact_version(
     artifact_version_id: UUID,
     artifact_version_update: ArtifactVersionUpdate,
@@ -227,7 +227,7 @@ def update_artifact_version(
     "/{artifact_version_id}",
     responses={401: error_response, 404: error_response, 422: error_response},
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def delete_artifact_version(
     artifact_version_id: UUID,
     _: AuthContext = Security(authorize),
@@ -248,7 +248,7 @@ def delete_artifact_version(
     "",
     responses={401: error_response, 404: error_response, 422: error_response},
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def prune_artifact_versions(
     project_name_or_id: Union[str, UUID],
     only_versions: bool = True,
@@ -275,7 +275,7 @@ def prune_artifact_versions(
     "/{artifact_version_id}" + VISUALIZE,
     responses={401: error_response, 404: error_response, 422: error_response},
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def get_artifact_visualization(
     artifact_version_id: UUID,
     index: int = 0,
@@ -303,7 +303,7 @@ def get_artifact_visualization(
     "/{artifact_version_id}" + DOWNLOAD_TOKEN,
     responses={401: error_response, 404: error_response, 422: error_response},
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def get_artifact_download_token(
     artifact_version_id: UUID,
     _: AuthContext = Security(authorize),
@@ -334,7 +334,7 @@ def get_artifact_download_token(
     "/{artifact_version_id}" + DATA,
     responses={401: error_response, 404: error_response, 422: error_response},
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def download_artifact_data(
     artifact_version_id: UUID, token: str
 ) -> FileResponse:

zenml/zen_server/routers/auth_endpoints.py

@@ -71,8 +71,8 @@ from zenml.zen_server.rbac.utils import (
     verify_permission_for_model,
 )
 from zenml.zen_server.utils import (
+    async_fastapi_endpoint_wrapper,
     get_ip_location,
-    handle_exceptions,
     server_config,
     zen_store,
 )
@@ -222,11 +222,11 @@ class OAuthLoginRequestForm:
     LOGIN,
     response_model=Union[OAuthTokenResponse, OAuthRedirectResponse],
 )
+@async_fastapi_endpoint_wrapper
 @rate_limit_requests(
     day_limit=server_config().login_rate_limit_day,
     minute_limit=server_config().login_rate_limit_minute,
 )
-@handle_exceptions
 def token(
     request: Request,
     response: Response,
@@ -341,6 +341,7 @@ def logout(
     DEVICE_AUTHORIZATION,
     response_model=OAuthDeviceAuthorizationResponse,
 )
+@async_fastapi_endpoint_wrapper
 def device_authorization(
     request: Request,
     client_id: UUID = Form(...),
@@ -471,7 +472,7 @@ def device_authorization(
     API_TOKEN,
     response_model=str,
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def api_token(
     token_type: APITokenType = APITokenType.GENERIC,
     expires_in: Optional[int] = None,

zenml/zen_server/routers/code_repositories_endpoints.py

@@ -38,7 +38,7 @@ from zenml.zen_server.rbac.endpoint_utils import (
 from zenml.zen_server.rbac.models import ResourceType
 from zenml.zen_server.routers.projects_endpoints import workspace_router
 from zenml.zen_server.utils import (
-    handle_exceptions,
+    async_fastapi_endpoint_wrapper,
     make_dependable,
     zen_store,
 )
@@ -62,7 +62,7 @@ router = APIRouter(
     deprecated=True,
     tags=["code_repositories"],
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def create_code_repository(
     code_repository: CodeRepositoryRequest,
     project_name_or_id: Optional[Union[str, UUID]] = None,
@@ -99,7 +99,7 @@ def create_code_repository(
     deprecated=True,
     tags=["code_repositories"],
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def list_code_repositories(
     filter_model: CodeRepositoryFilter = Depends(
         make_dependable(CodeRepositoryFilter)
@@ -135,7 +135,7 @@ def list_code_repositories(
     "/{code_repository_id}",
     responses={401: error_response, 404: error_response, 422: error_response},
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def get_code_repository(
     code_repository_id: UUID,
     hydrate: bool = True,
@@ -162,7 +162,7 @@ def get_code_repository(
     "/{code_repository_id}",
     responses={401: error_response, 404: error_response, 422: error_response},
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def update_code_repository(
     code_repository_id: UUID,
     update: CodeRepositoryUpdate,
@@ -189,7 +189,7 @@ def update_code_repository(
     "/{code_repository_id}",
     responses={401: error_response, 404: error_response, 422: error_response},
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def delete_code_repository(
     code_repository_id: UUID,
     _: AuthContext = Security(authorize),

zenml/zen_server/routers/devices_endpoints.py

@@ -39,7 +39,7 @@ from zenml.utils.time_utils import utc_now
 from zenml.zen_server.auth import AuthContext, authorize
 from zenml.zen_server.exceptions import error_response
 from zenml.zen_server.utils import (
-    handle_exceptions,
+    async_fastapi_endpoint_wrapper,
     make_dependable,
     server_config,
     zen_store,
@@ -56,7 +56,7 @@ router = APIRouter(
     "",
     responses={401: error_response, 404: error_response, 422: error_response},
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def list_authorized_devices(
     filter_model: OAuthDeviceFilter = Depends(
         make_dependable(OAuthDeviceFilter)
@@ -86,7 +86,7 @@ def list_authorized_devices(
     "/{device_id}",
     responses={401: error_response, 404: error_response, 422: error_response},
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def get_authorization_device(
     device_id: UUID,
     user_code: Optional[str] = None,
@@ -136,7 +136,7 @@ def get_authorization_device(
     "/{device_id}",
     responses={401: error_response, 404: error_response, 422: error_response},
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def update_authorized_device(
     device_id: UUID,
     update: OAuthDeviceUpdate,
@@ -172,7 +172,7 @@ def update_authorized_device(
     "/{device_id}" + DEVICE_VERIFY,
     responses={401: error_response, 404: error_response, 422: error_response},
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def verify_authorized_device(
     device_id: UUID,
     request: OAuthDeviceVerificationRequest,
@@ -276,7 +276,7 @@ def verify_authorized_device(
     "/{device_id}",
     responses={401: error_response, 404: error_response, 422: error_response},
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def delete_authorized_device(
     device_id: UUID,
     auth_context: AuthContext = Security(authorize),

zenml/zen_server/routers/event_source_endpoints.py

@@ -39,7 +39,7 @@ from zenml.zen_server.rbac.utils import (
     verify_permission_for_model,
 )
 from zenml.zen_server.utils import (
-    handle_exceptions,
+    async_fastapi_endpoint_wrapper,
     make_dependable,
     plugin_flavor_registry,
     zen_store,
@@ -59,7 +59,7 @@ event_source_router = APIRouter(
     "",
     responses={401: error_response, 404: error_response, 422: error_response},
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def list_event_sources(
     event_source_filter_model: EventSourceFilter = Depends(
         make_dependable(EventSourceFilter)
@@ -133,7 +133,7 @@ def list_event_sources(
     "/{event_source_id}",
     responses={401: error_response, 404: error_response, 422: error_response},
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def get_event_source(
     event_source_id: UUID,
     hydrate: bool = True,
@@ -185,7 +185,7 @@ def get_event_source(
     "",
     responses={401: error_response, 409: error_response, 422: error_response},
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def create_event_source(
     event_source: EventSourceRequest,
     _: AuthContext = Security(authorize),
@@ -227,7 +227,7 @@ def create_event_source(
     "/{event_source_id}",
     responses={401: error_response, 404: error_response, 422: error_response},
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def update_event_source(
     event_source_id: UUID,
     event_source_update: EventSourceUpdate,
@@ -279,7 +279,7 @@ def update_event_source(
     "/{event_source_id}",
     responses={401: error_response, 404: error_response, 422: error_response},
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def delete_event_source(
     event_source_id: UUID,
     force: bool = False,

zenml/zen_server/routers/flavors_endpoints.py

@@ -37,7 +37,7 @@ from zenml.zen_server.rbac.endpoint_utils import (
 from zenml.zen_server.rbac.models import Action, ResourceType
 from zenml.zen_server.rbac.utils import verify_permission
 from zenml.zen_server.utils import (
-    handle_exceptions,
+    async_fastapi_endpoint_wrapper,
     make_dependable,
     zen_store,
 )
@@ -53,7 +53,7 @@ router = APIRouter(
     "",
     responses={401: error_response, 404: error_response, 422: error_response},
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def list_flavors(
     flavor_filter_model: FlavorFilter = Depends(make_dependable(FlavorFilter)),
     hydrate: bool = False,
@@ -82,7 +82,7 @@ def list_flavors(
     "/{flavor_id}",
     responses={401: error_response, 404: error_response, 422: error_response},
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def get_flavor(
     flavor_id: UUID,
     hydrate: bool = True,
@@ -107,7 +107,7 @@ def get_flavor(
     "",
     responses={401: error_response, 409: error_response, 422: error_response},
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def create_flavor(
     flavor: FlavorRequest,
     _: AuthContext = Security(authorize),
@@ -130,7 +130,7 @@ def create_flavor(
     "/{flavor_id}",
     responses={401: error_response, 409: error_response, 422: error_response},
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def update_flavor(
     flavor_id: UUID,
     flavor_update: FlavorUpdate,
@@ -159,7 +159,7 @@ def update_flavor(
     "/{flavor_id}",
     responses={401: error_response, 404: error_response, 422: error_response},
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def delete_flavor(
     flavor_id: UUID,
     _: AuthContext = Security(authorize),
@@ -180,7 +180,7 @@ def delete_flavor(
     "/sync",
     responses={401: error_response, 404: error_response, 422: error_response},
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def sync_flavors(
     _: AuthContext = Security(authorize),
 ) -> None:

zenml/zen_server/routers/logs_endpoints.py

@@ -29,7 +29,7 @@ from zenml.zen_server.rbac.endpoint_utils import (
     verify_permissions_and_get_entity,
 )
 from zenml.zen_server.utils import (
-    handle_exceptions,
+    async_fastapi_endpoint_wrapper,
     zen_store,
 )
 
@@ -44,7 +44,7 @@ router = APIRouter(
     "/{logs_id}",
     responses={401: error_response, 404: error_response, 422: error_response},
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def get_logs(
     logs_id: UUID,
     hydrate: bool = True,

zenml/zen_server/routers/model_versions_endpoints.py

@@ -59,7 +59,7 @@ from zenml.zen_server.routers.models_endpoints import (
 )
 from zenml.zen_server.routers.projects_endpoints import workspace_router
 from zenml.zen_server.utils import (
-    handle_exceptions,
+    async_fastapi_endpoint_wrapper,
     make_dependable,
     set_filter_project_scope,
     zen_store,
@@ -89,7 +89,7 @@ router = APIRouter(
     deprecated=True,
     tags=["model_versions"],
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def create_model_version(
     model_version: ModelVersionRequest,
     model_id: Optional[UUID] = None,
@@ -127,7 +127,7 @@ def create_model_version(
     "",
     responses={401: error_response, 404: error_response, 422: error_response},
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def list_model_versions(
     model_version_filter_model: ModelVersionFilter = Depends(
         make_dependable(ModelVersionFilter)
@@ -176,7 +176,7 @@ def list_model_versions(
     "/{model_version_id}",
     responses={401: error_response, 404: error_response, 422: error_response},
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def get_model_version(
     model_version_id: UUID,
     hydrate: bool = True,
@@ -204,7 +204,7 @@ def get_model_version(
     "/{model_version_id}",
     responses={401: error_response, 404: error_response, 422: error_response},
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def update_model_version(
     model_version_id: UUID,
     model_version_update_model: ModelVersionUpdate,
@@ -238,7 +238,7 @@ def update_model_version(
     "/{model_version_id}",
     responses={401: error_response, 404: error_response, 422: error_response},
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def delete_model_version(
     model_version_id: UUID,
     _: AuthContext = Security(authorize),
@@ -270,7 +270,7 @@ model_version_artifacts_router = APIRouter(
     "",
     responses={401: error_response, 409: error_response, 422: error_response},
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def create_model_version_artifact_link(
     model_version_artifact_link: ModelVersionArtifactRequest,
     _: AuthContext = Security(authorize),
@@ -300,7 +300,7 @@ def create_model_version_artifact_link(
     "",
     responses={401: error_response, 404: error_response, 422: error_response},
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def list_model_version_artifact_links(
     model_version_artifact_link_filter_model: ModelVersionArtifactFilter = Depends(
         make_dependable(ModelVersionArtifactFilter)
@@ -331,7 +331,7 @@ def list_model_version_artifact_links(
     + "/{model_version_artifact_link_name_or_id}",
     responses={401: error_response, 404: error_response, 422: error_response},
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def delete_model_version_artifact_link(
     model_version_id: UUID,
     model_version_artifact_link_name_or_id: Union[str, UUID],
@@ -357,7 +357,7 @@ def delete_model_version_artifact_link(
     "/{model_version_id}" + ARTIFACTS,
     responses={401: error_response, 404: error_response, 422: error_response},
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def delete_all_model_version_artifact_links(
     model_version_id: UUID,
     only_links: bool = True,
@@ -392,7 +392,7 @@ model_version_pipeline_runs_router = APIRouter(
     "",
     responses={401: error_response, 409: error_response, 422: error_response},
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def create_model_version_pipeline_run_link(
     model_version_pipeline_run_link: ModelVersionPipelineRunRequest,
     _: AuthContext = Security(authorize),
@@ -423,7 +423,7 @@ def create_model_version_pipeline_run_link(
     "",
     responses={401: error_response, 404: error_response, 422: error_response},
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def list_model_version_pipeline_run_links(
     model_version_pipeline_run_link_filter_model: ModelVersionPipelineRunFilter = Depends(
         make_dependable(ModelVersionPipelineRunFilter)
@@ -454,7 +454,7 @@ def list_model_version_pipeline_run_links(
     + "/{model_version_pipeline_run_link_name_or_id}",
     responses={401: error_response, 404: error_response, 422: error_response},
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def delete_model_version_pipeline_run_link(
     model_version_id: UUID,
     model_version_pipeline_run_link_name_or_id: Union[str, UUID],

zenml/zen_server/routers/models_endpoints.py

@@ -43,7 +43,7 @@ from zenml.zen_server.rbac.endpoint_utils import (
 from zenml.zen_server.rbac.models import ResourceType
 from zenml.zen_server.routers.projects_endpoints import workspace_router
 from zenml.zen_server.utils import (
-    handle_exceptions,
+    async_fastapi_endpoint_wrapper,
     make_dependable,
     server_config,
     zen_store,
@@ -72,7 +72,7 @@ router = APIRouter(
     deprecated=True,
     tags=["models"],
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def create_model(
     model: ModelRequest,
     project_name_or_id: Optional[Union[str, UUID]] = None,
@@ -101,7 +101,7 @@ def create_model(
     "",
     responses={401: error_response, 404: error_response, 422: error_response},
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def list_models(
     model_filter_model: ModelFilter = Depends(make_dependable(ModelFilter)),
     hydrate: bool = False,
@@ -130,7 +130,7 @@ def list_models(
     "/{model_id}",
     responses={401: error_response, 404: error_response, 422: error_response},
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def get_model(
     model_id: UUID,
     hydrate: bool = True,
@@ -155,7 +155,7 @@ def get_model(
     "/{model_id}",
     responses={401: error_response, 404: error_response, 422: error_response},
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def update_model(
     model_id: UUID,
     model_update: ModelUpdate,
@@ -182,7 +182,7 @@ def update_model(
     "/{model_id}",
     responses={401: error_response, 404: error_response, 422: error_response},
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def delete_model(
     model_id: UUID,
     _: AuthContext = Security(authorize),