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

zenml/zen_server/routers/service_accounts_endpoints.py

@@ -49,7 +49,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_for_model
 from zenml.zen_server.utils import (
-    handle_exceptions,
+    async_fastapi_endpoint_wrapper,
     make_dependable,
     zen_store,
 )
@@ -73,7 +73,7 @@ router = APIRouter(
         422: error_response,
     },
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def create_service_account(
     service_account: ServiceAccountRequest,
     _: AuthContext = Security(authorize),
@@ -96,7 +96,7 @@ def create_service_account(
     "/{service_account_name_or_id}",
     responses={401: error_response, 404: error_response, 422: error_response},
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def get_service_account(
     service_account_name_or_id: Union[str, UUID],
     _: AuthContext = Security(authorize),
@@ -123,7 +123,7 @@ def get_service_account(
     "",
     responses={401: error_response, 404: error_response, 422: error_response},
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def list_service_accounts(
     filter_model: ServiceAccountFilter = Depends(
         make_dependable(ServiceAccountFilter)
@@ -158,7 +158,7 @@ def list_service_accounts(
         422: error_response,
     },
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def update_service_account(
     service_account_name_or_id: Union[str, UUID],
     service_account_update: ServiceAccountUpdate,
@@ -185,7 +185,7 @@ def update_service_account(
     "/{service_account_name_or_id}",
     responses={401: error_response, 404: error_response, 422: error_response},
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def delete_service_account(
     service_account_name_or_id: Union[str, UUID],
     _: AuthContext = Security(authorize),
@@ -211,7 +211,7 @@ def delete_service_account(
     "/{service_account_id}" + API_KEYS,
     responses={401: error_response, 409: error_response, 422: error_response},
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def create_api_key(
     service_account_id: UUID,
     api_key: APIKeyRequest,
@@ -249,7 +249,7 @@ def create_api_key(
     "/{service_account_id}" + API_KEYS + "/{api_key_name_or_id}",
     responses={401: error_response, 404: error_response, 422: error_response},
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def get_api_key(
     service_account_id: UUID,
     api_key_name_or_id: Union[str, UUID],
@@ -282,7 +282,7 @@ def get_api_key(
     "/{service_account_id}" + API_KEYS,
     responses={401: error_response, 404: error_response, 422: error_response},
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def list_api_keys(
     service_account_id: UUID,
     filter_model: APIKeyFilter = Depends(make_dependable(APIKeyFilter)),
@@ -316,7 +316,7 @@ def list_api_keys(
     "/{service_account_id}" + API_KEYS + "/{api_key_name_or_id}",
     responses={401: error_response, 409: error_response, 422: error_response},
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def update_api_key(
     service_account_id: UUID,
     api_key_name_or_id: Union[str, UUID],
@@ -350,7 +350,7 @@ def update_api_key(
     + API_KEY_ROTATE,
     responses={401: error_response, 409: error_response, 422: error_response},
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def rotate_api_key(
     service_account_id: UUID,
     api_key_name_or_id: Union[str, UUID],
@@ -381,7 +381,7 @@ def rotate_api_key(
     "/{service_account_id}" + API_KEYS + "/{api_key_name_or_id}",
     responses={401: error_response, 404: error_response, 422: error_response},
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def delete_api_key(
     service_account_id: UUID,
     api_key_name_or_id: Union[str, UUID],

zenml/zen_server/routers/service_connectors_endpoints.py

@@ -61,7 +61,7 @@ from zenml.zen_server.rbac.utils 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,
     zen_store,
 )
@@ -91,7 +91,7 @@ types_router = APIRouter(
     deprecated=True,
     tags=["service_connectors"],
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def create_service_connector(
     connector: ServiceConnectorRequest,
     project_name_or_id: Optional[Union[str, UUID]] = None,
@@ -124,7 +124,7 @@ def create_service_connector(
     deprecated=True,
     tags=["service_connectors"],
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def list_service_connectors(
     connector_filter_model: ServiceConnectorFilter = Depends(
         make_dependable(ServiceConnectorFilter)
@@ -198,7 +198,7 @@ def list_service_connectors(
     deprecated=True,
     tags=["service_connectors"],
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def list_service_connector_resources(
     filter_model: ServiceConnectorFilter = Depends(
         make_dependable(ServiceConnectorFilter)
@@ -234,7 +234,7 @@ def list_service_connector_resources(
     "/{connector_id}",
     responses={401: error_response, 404: error_response, 422: error_response},
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def get_service_connector(
     connector_id: UUID,
     expand_secrets: bool = True,
@@ -276,7 +276,7 @@ def get_service_connector(
     "/{connector_id}",
     responses={401: error_response, 404: error_response, 422: error_response},
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def update_service_connector(
     connector_id: UUID,
     connector_update: ServiceConnectorUpdate,
@@ -303,7 +303,7 @@ def update_service_connector(
     "/{connector_id}",
     responses={401: error_response, 404: error_response, 422: error_response},
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def delete_service_connector(
     connector_id: UUID,
     _: AuthContext = Security(authorize),
@@ -324,7 +324,7 @@ def delete_service_connector(
     SERVICE_CONNECTOR_VERIFY,
     responses={401: error_response, 409: error_response, 422: error_response},
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def validate_and_verify_service_connector_config(
     connector: ServiceConnectorRequest,
     list_resources: bool = True,
@@ -357,7 +357,7 @@ def validate_and_verify_service_connector_config(
     "/{connector_id}" + SERVICE_CONNECTOR_VERIFY,
     responses={401: error_response, 404: error_response, 422: error_response},
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def validate_and_verify_service_connector(
     connector_id: UUID,
     resource_type: Optional[str] = None,
@@ -398,7 +398,7 @@ def validate_and_verify_service_connector(
     "/{connector_id}" + SERVICE_CONNECTOR_CLIENT,
     responses={401: error_response, 404: error_response, 422: error_response},
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def get_service_connector_client(
     connector_id: UUID,
     resource_type: Optional[str] = None,
@@ -435,7 +435,7 @@ def get_service_connector_client(
     "",
     responses={401: error_response, 404: error_response, 422: error_response},
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def list_service_connector_types(
     connector_type: Optional[str] = None,
     resource_type: Optional[str] = None,
@@ -465,7 +465,7 @@ def list_service_connector_types(
     "/{connector_type}",
     responses={401: error_response, 404: error_response, 422: error_response},
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def get_service_connector_type(
     connector_type: str,
     _: AuthContext = Security(authorize),
@@ -485,7 +485,7 @@ def get_service_connector_type(
     SERVICE_CONNECTOR_FULL_STACK,
     responses={401: error_response, 409: error_response, 422: error_response},
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def get_resources_based_on_service_connector_info(
     connector_info: Optional[ServiceConnectorInfo] = None,
     connector_uuid: Optional[UUID] = None,

zenml/zen_server/routers/service_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=["services"],
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def create_service(
     service: ServiceRequest,
     project_name_or_id: Optional[Union[str, UUID]] = None,
@@ -91,7 +91,7 @@ def create_service(
     "",
     responses={401: error_response, 404: error_response, 422: error_response},
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def list_services(
     filter_model: ServiceFilter = Depends(make_dependable(ServiceFilter)),
     hydrate: bool = False,
@@ -120,7 +120,7 @@ def list_services(
     "/{service_id}",
     responses={401: error_response, 404: error_response, 422: error_response},
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def get_service(
     service_id: UUID,
     hydrate: bool = True,
@@ -147,7 +147,7 @@ def get_service(
     "/{service_id}",
     responses={401: error_response, 404: error_response, 422: error_response},
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def update_service(
     service_id: UUID,
     update: ServiceUpdate,
@@ -174,7 +174,7 @@ def update_service(
     "/{service_id}",
     responses={401: error_response, 404: error_response, 422: error_response},
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def delete_service(
     service_id: UUID,
     _: AuthContext = Security(authorize),

zenml/zen_server/routers/stack_components_endpoints.py

@@ -40,7 +40,7 @@ from zenml.zen_server.rbac.models import Action, ResourceType
 from zenml.zen_server.rbac.utils import verify_permission_for_model
 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,
 )
@@ -70,7 +70,7 @@ types_router = APIRouter(
     deprecated=True,
     tags=["stack_components"],
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def create_stack_component(
     component: ComponentRequest,
     project_name_or_id: Optional[Union[str, UUID]] = None,
@@ -120,7 +120,7 @@ def create_stack_component(
     deprecated=True,
     tags=["stack_components"],
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def list_stack_components(
     component_filter_model: ComponentFilter = Depends(
         make_dependable(ComponentFilter)
@@ -153,7 +153,7 @@ def list_stack_components(
     "/{component_id}",
     responses={401: error_response, 404: error_response, 422: error_response},
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def get_stack_component(
     component_id: UUID,
     hydrate: bool = True,
@@ -180,7 +180,7 @@ def get_stack_component(
     "/{component_id}",
     responses={401: error_response, 404: error_response, 422: error_response},
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def update_stack_component(
     component_id: UUID,
     component_update: ComponentUpdate,
@@ -226,7 +226,7 @@ def update_stack_component(
     "/{component_id}",
     responses={401: error_response, 404: error_response, 422: error_response},
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def deregister_stack_component(
     component_id: UUID,
     _: AuthContext = Security(authorize),
@@ -247,7 +247,7 @@ def deregister_stack_component(
     "",
     responses={401: error_response, 404: error_response, 422: error_response},
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def get_stack_component_types(
     _: AuthContext = Security(authorize),
 ) -> List[str]:

zenml/zen_server/routers/stack_deployment_endpoints.py

@@ -39,7 +39,7 @@ from zenml.zen_server.exceptions import error_response
 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,
     server_config,
 )
 
@@ -53,7 +53,7 @@ router = APIRouter(
 @router.get(
     INFO,
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def get_stack_deployment_info(
     provider: StackDeploymentProvider,
     _: AuthContext = Security(authorize),
@@ -73,7 +73,7 @@ def get_stack_deployment_info(
 @router.get(
     CONFIG,
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def get_stack_deployment_config(
     request: Request,
     provider: StackDeploymentProvider,
@@ -139,7 +139,7 @@ def get_stack_deployment_config(
 @router.get(
     STACK,
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def get_deployed_stack(
     provider: StackDeploymentProvider,
     stack_name: str,

zenml/zen_server/routers/stacks_endpoints.py

@@ -42,7 +42,7 @@ from zenml.zen_server.rbac.utils 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,
     zen_store,
 )
@@ -66,7 +66,7 @@ router = APIRouter(
     deprecated=True,
     tags=["stacks"],
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def create_stack(
     stack: StackRequest,
     project_name_or_id: Optional[Union[str, UUID]] = None,
@@ -136,7 +136,7 @@ def create_stack(
     deprecated=True,
     tags=["stacks"],
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def list_stacks(
     project_name_or_id: Optional[Union[str, UUID]] = None,
     stack_filter_model: StackFilter = Depends(make_dependable(StackFilter)),
@@ -167,7 +167,7 @@ def list_stacks(
     "/{stack_id}",
     responses={401: error_response, 404: error_response, 422: error_response},
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def get_stack(
     stack_id: UUID,
     hydrate: bool = True,
@@ -192,7 +192,7 @@ def get_stack(
     "/{stack_id}",
     responses={401: error_response, 404: error_response, 422: error_response},
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def update_stack(
     stack_id: UUID,
     stack_update: StackUpdate,
@@ -230,7 +230,7 @@ def update_stack(
     "/{stack_id}",
     responses={401: error_response, 404: error_response, 422: error_response},
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def delete_stack(
     stack_id: UUID,
     _: AuthContext = Security(authorize),

zenml/zen_server/routers/steps_endpoints.py

@@ -48,7 +48,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,
     set_filter_project_scope,
     zen_store,
@@ -65,7 +65,7 @@ router = APIRouter(
     "",
     responses={401: error_response, 404: error_response, 422: error_response},
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def list_run_steps(
     step_run_filter_model: StepRunFilter = Depends(
         make_dependable(StepRunFilter)
@@ -109,7 +109,7 @@ def list_run_steps(
     "",
     responses={401: error_response, 409: error_response, 422: error_response},
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def create_run_step(
     step: StepRunRequest,
     _: AuthContext = Security(authorize),
@@ -136,7 +136,7 @@ def create_run_step(
     "/{step_id}",
     responses={401: error_response, 404: error_response, 422: error_response},
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def get_step(
     step_id: UUID,
     hydrate: bool = True,
@@ -169,7 +169,7 @@ def get_step(
     "/{step_id}",
     responses={401: error_response, 404: error_response, 422: error_response},
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def update_step(
     step_id: UUID,
     step_model: StepRunUpdate,
@@ -198,7 +198,7 @@ def update_step(
     "/{step_id}" + STEP_CONFIGURATION,
     responses={401: error_response, 404: error_response, 422: error_response},
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def get_step_configuration(
     step_id: UUID,
     _: AuthContext = Security(authorize),
@@ -222,7 +222,7 @@ def get_step_configuration(
     "/{step_id}" + STATUS,
     responses={401: error_response, 404: error_response, 422: error_response},
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def get_step_status(
     step_id: UUID,
     _: AuthContext = Security(authorize),
@@ -246,7 +246,7 @@ def get_step_status(
     "/{step_id}" + LOGS,
     responses={401: error_response, 404: error_response, 422: error_response},
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def get_step_logs(
     step_id: UUID,
     offset: int = 0,

zenml/zen_server/routers/tag_resource_endpoints.py

@@ -27,7 +27,7 @@ from zenml.models import TagResourceRequest, TagResourceResponse
 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,
     zen_store,
 )
 
@@ -42,7 +42,7 @@ router = APIRouter(
     "",
     responses={401: error_response, 404: error_response, 422: error_response},
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def create_tag_resource(
     tag_resource: TagResourceRequest,
     _: AuthContext = Security(authorize),
@@ -62,7 +62,7 @@ def create_tag_resource(
     BATCH,
     responses={401: error_response, 409: error_response, 422: error_response},
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def batch_create_tag_resource(
     tag_resources: List[TagResourceRequest],
     _: AuthContext = Security(authorize),
@@ -85,7 +85,7 @@ def batch_create_tag_resource(
     "",
     responses={401: error_response, 404: error_response, 422: error_response},
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def delete_tag_resource(
     tag_resource: TagResourceRequest,
     _: AuthContext = Security(authorize),
@@ -102,7 +102,7 @@ def delete_tag_resource(
     BATCH,
     responses={401: error_response, 404: error_response, 422: error_response},
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def batch_delete_tag_resource(
     tag_resources: List[TagResourceRequest],
     _: AuthContext = Security(authorize),

zenml/zen_server/routers/tags_endpoints.py

@@ -38,7 +38,7 @@ from zenml.zen_server.rbac.endpoint_utils import (
     verify_permissions_and_update_entity,
 )
 from zenml.zen_server.utils import (
-    handle_exceptions,
+    async_fastapi_endpoint_wrapper,
     make_dependable,
     zen_store,
 )
@@ -58,7 +58,7 @@ router = APIRouter(
     "",
     responses={401: error_response, 409: error_response, 422: error_response},
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def create_tag(
     tag: TagRequest,
     _: AuthContext = Security(authorize),
@@ -78,7 +78,7 @@ def create_tag(
     "",
     responses={401: error_response, 404: error_response, 422: error_response},
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def list_tags(
     tag_filter_model: TagFilter = Depends(make_dependable(TagFilter)),
     hydrate: bool = False,
@@ -105,7 +105,7 @@ def list_tags(
     "/{tag_name_or_id}",
     responses={401: error_response, 404: error_response, 422: error_response},
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def get_tag(
     tag_name_or_id: Union[str, UUID],
     hydrate: bool = True,
@@ -132,7 +132,7 @@ def get_tag(
     "/{tag_id}",
     responses={401: error_response, 404: error_response, 422: error_response},
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def update_tag(
     tag_id: UUID,
     tag_update_model: TagUpdate,
@@ -159,7 +159,7 @@ def update_tag(
     "/{tag_name_or_id}",
     responses={401: error_response, 404: error_response, 422: error_response},
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def delete_tag(
     tag_name_or_id: Union[str, UUID],
     _: AuthContext = Security(authorize),

zenml/zen_server/routers/triggers_endpoints.py

@@ -43,7 +43,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,
@@ -60,7 +60,7 @@ router = APIRouter(
     "",
     responses={401: error_response, 404: error_response, 422: error_response},
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def list_triggers(
     trigger_filter_model: TriggerFilter = Depends(
         make_dependable(TriggerFilter)
@@ -91,7 +91,7 @@ def list_triggers(
     "/{trigger_id}",
     responses={401: error_response, 404: error_response, 422: error_response},
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def get_trigger(
     trigger_id: UUID,
     hydrate: bool = True,
@@ -116,7 +116,7 @@ def get_trigger(
     "",
     responses={401: error_response, 409: error_response, 422: error_response},
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def create_trigger(
     trigger: TriggerRequest,
     _: AuthContext = Security(authorize),
@@ -167,7 +167,7 @@ def create_trigger(
     "/{trigger_id}",
     responses={401: error_response, 404: error_response, 422: error_response},
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def update_trigger(
     trigger_id: UUID,
     trigger_update: TriggerUpdate,
@@ -230,7 +230,7 @@ def update_trigger(
     "/{trigger_id}",
     responses={401: error_response, 404: error_response, 422: error_response},
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def delete_trigger(
     trigger_id: UUID,
     _: AuthContext = Security(authorize),
@@ -258,7 +258,7 @@ executions_router = APIRouter(
     "",
     responses={401: error_response, 404: error_response, 422: error_response},
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def list_trigger_executions(
     trigger_execution_filter_model: TriggerExecutionFilter = Depends(
         make_dependable(TriggerExecutionFilter)
@@ -289,7 +289,7 @@ def list_trigger_executions(
     "/{trigger_execution_id}",
     responses={401: error_response, 404: error_response, 422: error_response},
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def get_trigger_execution(
     trigger_execution_id: UUID,
     hydrate: bool = True,
@@ -316,7 +316,7 @@ def get_trigger_execution(
     "/{trigger_execution_id}",
     responses={401: error_response, 404: error_response, 422: error_response},
 )
-@handle_exceptions
+@async_fastapi_endpoint_wrapper
 def delete_trigger_execution(
     trigger_execution_id: UUID,
     _: AuthContext = Security(authorize),