hatchet-sdk 1.0.0a1__py3-none-any.whl → 1.0.1__py3-none-any.whl

hatchet_sdk/__init__.py

@@ -98,6 +98,7 @@ from hatchet_sdk.clients.rest.models.user_tenant_memberships_list import (
     UserTenantMembershipsList,
 )
 from hatchet_sdk.clients.rest.models.user_tenant_public import UserTenantPublic
+from hatchet_sdk.clients.rest.models.v1_task_status import V1TaskStatus
 from hatchet_sdk.clients.rest.models.worker_list import WorkerList
 from hatchet_sdk.clients.rest.models.workflow import Workflow
 from hatchet_sdk.clients.rest.models.workflow_deployment_config import (
@@ -135,6 +136,7 @@ from hatchet_sdk.contracts.workflows_pb2 import (
     RateLimitDuration,
     WorkerLabelComparator,
 )
+from hatchet_sdk.features.runs import BulkCancelReplayOpts, RunFilter
 from hatchet_sdk.hatchet import Hatchet
 from hatchet_sdk.runnables.task import Task
 from hatchet_sdk.runnables.types import (
@@ -264,4 +266,7 @@ __all__ = [
     "DurableContext",
     "RegisterDurableEventRequest",
     "TaskDefaults",
+    "BulkCancelReplayOpts",
+    "RunFilter",
+    "V1TaskStatus",
 ]

hatchet_sdk/client.py

@@ -5,11 +5,18 @@ import grpc
 from hatchet_sdk.clients.admin import AdminClient
 from hatchet_sdk.clients.dispatcher.dispatcher import DispatcherClient
 from hatchet_sdk.clients.events import EventClient, new_event
-from hatchet_sdk.clients.rest_client import RestApi
 from hatchet_sdk.clients.run_event_listener import RunEventListenerClient
 from hatchet_sdk.clients.workflow_listener import PooledWorkflowRunListener
 from hatchet_sdk.config import ClientConfig
 from hatchet_sdk.connection import new_conn
+from hatchet_sdk.features.cron import CronClient
+from hatchet_sdk.features.logs import LogsClient
+from hatchet_sdk.features.metrics import MetricsClient
+from hatchet_sdk.features.rate_limits import RateLimitsClient
+from hatchet_sdk.features.runs import RunsClient
+from hatchet_sdk.features.scheduled import ScheduledClient
+from hatchet_sdk.features.workers import WorkersClient
+from hatchet_sdk.features.workflows import WorkflowsClient
 
 
 class Client:
@@ -20,7 +27,6 @@ class Client:
         admin_client: AdminClient | None = None,
         dispatcher_client: DispatcherClient | None = None,
         workflow_listener: PooledWorkflowRunListener | None | None = None,
-        rest_client: RestApi | None = None,
         debug: bool = False,
     ):
         try:
@@ -35,10 +41,16 @@ class Client:
         self.admin = admin_client or AdminClient(config)
         self.dispatcher = dispatcher_client or DispatcherClient(config)
         self.event = event_client or new_event(conn, config)
-        self.rest = rest_client or RestApi(
-            config.server_url, config.token, config.tenant_id
-        )
         self.listener = RunEventListenerClient(config)
         self.workflow_listener = workflow_listener
         self.logInterceptor = config.logger
         self.debug = debug
+
+        self.cron = CronClient(self.config)
+        self.logs = LogsClient(self.config)
+        self.metrics = MetricsClient(self.config)
+        self.rate_limits = RateLimitsClient(self.config)
+        self.runs = RunsClient(self.config)
+        self.scheduled = ScheduledClient(self.config)
+        self.workers = WorkersClient(self.config)
+        self.workflows = WorkflowsClient(self.config)

hatchet_sdk/clients/admin.py

@@ -163,12 +163,6 @@ class AdminClient:
         workflow: workflow_protos.CreateWorkflowVersionRequest,
         overrides: workflow_protos.CreateWorkflowVersionRequest | None = None,
     ) -> workflow_protos.CreateWorkflowVersionResponse:
-        ## IMPORTANT: The `pooled_workflow_listener` must be created 1) lazily, and not at `init` time, and 2) on the
-        ## main thread. If 1) is not followed, you'll get an error about something being attached to the wrong event
-        ## loop. If 2) is not followed, you'll get an error about the event loop not being set up.
-        if not self.pooled_workflow_listener:
-            self.pooled_workflow_listener = PooledWorkflowRunListener(self.config)
-
         return await asyncio.to_thread(self.put_workflow, name, workflow, overrides)
 
     @tenacity_retry
@@ -178,12 +172,6 @@ class AdminClient:
         limit: int,
         duration: RateLimitDuration = RateLimitDuration.SECOND,
     ) -> None:
-        ## IMPORTANT: The `pooled_workflow_listener` must be created 1) lazily, and not at `init` time, and 2) on the
-        ## main thread. If 1) is not followed, you'll get an error about something being attached to the wrong event
-        ## loop. If 2) is not followed, you'll get an error about the event loop not being set up.
-        if not self.pooled_workflow_listener:
-            self.pooled_workflow_listener = PooledWorkflowRunListener(self.config)
-
         return await asyncio.to_thread(self.put_rate_limit, key, limit, duration)
 
     @tenacity_retry
@@ -194,12 +182,6 @@ class AdminClient:
         input: JSONSerializableMapping = {},
         options: ScheduleTriggerWorkflowOptions = ScheduleTriggerWorkflowOptions(),
     ) -> v0_workflow_protos.WorkflowVersion:
-        ## IMPORTANT: The `pooled_workflow_listener` must be created 1) lazily, and not at `init` time, and 2) on the
-        ## main thread. If 1) is not followed, you'll get an error about something being attached to the wrong event
-        ## loop. If 2) is not followed, you'll get an error about the event loop not being set up.
-        if not self.pooled_workflow_listener:
-            self.pooled_workflow_listener = PooledWorkflowRunListener(self.config)
-
         return await asyncio.to_thread(
             self.schedule_workflow, name, schedules, input, options
         )

hatchet_sdk/clients/rest/models/workflow_runs_metrics.py

@@ -22,13 +22,17 @@ from typing import Any, ClassVar, Dict, List, Optional, Set
 from pydantic import BaseModel, ConfigDict
 from typing_extensions import Self
 
+from hatchet_sdk.clients.rest.models.workflow_runs_metrics_counts import (
+    WorkflowRunsMetricsCounts,
+)
+
 
 class WorkflowRunsMetrics(BaseModel):
     """
     WorkflowRunsMetrics
     """  # noqa: E501
 
-    counts: Optional[Dict[str, Any]] = None
+    counts: Optional[WorkflowRunsMetricsCounts] = None
     __properties: ClassVar[List[str]] = ["counts"]
 
     model_config = ConfigDict(

hatchet_sdk/clients/v1/api_client.py

@@ -0,0 +1,81 @@
+import asyncio
+from concurrent.futures import ThreadPoolExecutor
+from typing import AsyncContextManager, Callable, Coroutine, ParamSpec, TypeVar
+
+from hatchet_sdk.clients.rest.api_client import ApiClient
+from hatchet_sdk.clients.rest.configuration import Configuration
+from hatchet_sdk.config import ClientConfig
+from hatchet_sdk.utils.typing import JSONSerializableMapping
+
+## Type variables to use with coroutines.
+## See https://stackoverflow.com/questions/73240620/the-right-way-to-type-hint-a-coroutine-function
+## Return type
+R = TypeVar("R")
+
+## Yield type
+Y = TypeVar("Y")
+
+## Send type
+S = TypeVar("S")
+
+P = ParamSpec("P")
+
+
+def maybe_additional_metadata_to_kv(
+    additional_metadata: dict[str, str] | JSONSerializableMapping | None
+) -> list[str] | None:
+    if not additional_metadata:
+        return None
+
+    return [f"{k}:{v}" for k, v in additional_metadata.items()]
+
+
+class BaseRestClient:
+    def __init__(self, config: ClientConfig) -> None:
+        self.tenant_id = config.tenant_id
+
+        self.client_config = config
+        self.api_config = Configuration(
+            host=config.server_url,
+            access_token=config.token,
+        )
+
+        self.api_config.datetime_format = "%Y-%m-%dT%H:%M:%S.%fZ"
+
+    def client(self) -> AsyncContextManager[ApiClient]:
+        return ApiClient(self.api_config)
+
+    def _run_async_function_do_not_use_directly(
+        self,
+        async_func: Callable[P, Coroutine[Y, S, R]],
+        *args: P.args,
+        **kwargs: P.kwargs,
+    ) -> R:
+        loop = asyncio.new_event_loop()
+        asyncio.set_event_loop(loop)
+        try:
+            return loop.run_until_complete(async_func(*args, **kwargs))
+        finally:
+            loop.close()
+
+    def _run_async_from_sync(
+        self,
+        async_func: Callable[P, Coroutine[Y, S, R]],
+        *args: P.args,
+        **kwargs: P.kwargs,
+    ) -> R:
+        try:
+            loop = asyncio.get_event_loop()
+        except RuntimeError:
+            loop = None
+
+        if loop and loop.is_running():
+            return loop.run_until_complete(async_func(*args, **kwargs))
+        else:
+            with ThreadPoolExecutor() as executor:
+                future = executor.submit(
+                    lambda: self._run_async_function_do_not_use_directly(
+                        async_func, *args, **kwargs
+                    )
+                )
+                return future.result()

hatchet_sdk/context/context.py

@@ -17,9 +17,6 @@ from hatchet_sdk.clients.durable_event_listener import (
     RegisterDurableEventRequest,
 )
 from hatchet_sdk.clients.events import EventClient
-from hatchet_sdk.clients.rest_client import RestApi
-from hatchet_sdk.clients.run_event_listener import RunEventListenerClient
-from hatchet_sdk.clients.workflow_listener import PooledWorkflowRunListener
 from hatchet_sdk.context.worker_context import WorkerContext
 from hatchet_sdk.logger import logger
 from hatchet_sdk.utils.timedelta_to_expression import Duration, timedelta_to_expr
@@ -53,12 +50,8 @@ class Context:
         dispatcher_client: DispatcherClient,
         admin_client: AdminClient,
         event_client: EventClient,
-        rest_client: RestApi,
-        workflow_listener: PooledWorkflowRunListener | None,
         durable_event_listener: DurableEventListener | None,
-        workflow_run_event_listener: RunEventListenerClient,
         worker: WorkerContext,
-        namespace: str = "",
         validator_registry: dict[str, WorkflowValidator] = {},
     ):
         self.worker = worker
@@ -68,16 +61,12 @@ class Context:
 
         self.action = action
 
-        self.step_run_id: str = action.step_run_id
+        self.step_run_id = action.step_run_id
         self.exit_flag = False
         self.dispatcher_client = dispatcher_client
         self.admin_client = admin_client
         self.event_client = event_client
-        self.rest_client = rest_client
-        self.workflow_listener = workflow_listener
         self.durable_event_listener = durable_event_listener
-        self.workflow_run_event_listener = workflow_run_event_listener
-        self.namespace = namespace
 
         # FIXME: this limits the number of concurrent log requests to 1, which means we can do about
         # 100 log lines per second but this depends on network.

hatchet_sdk/features/cron.py

@@ -1,8 +1,11 @@
-from typing import List, Union
-
 from pydantic import BaseModel, Field, field_validator
 
-from hatchet_sdk.client import Client
+from hatchet_sdk.clients.rest.api.workflow_api import WorkflowApi
+from hatchet_sdk.clients.rest.api.workflow_run_api import WorkflowRunApi
+from hatchet_sdk.clients.rest.api_client import ApiClient
+from hatchet_sdk.clients.rest.models.create_cron_workflow_trigger_request import (
+    CreateCronWorkflowTriggerRequest,
+)
 from hatchet_sdk.clients.rest.models.cron_workflows import CronWorkflows
 from hatchet_sdk.clients.rest.models.cron_workflows_list import CronWorkflowsList
 from hatchet_sdk.clients.rest.models.cron_workflows_order_by_field import (
@@ -11,6 +14,10 @@ from hatchet_sdk.clients.rest.models.cron_workflows_order_by_field import (
 from hatchet_sdk.clients.rest.models.workflow_run_order_by_direction import (
     WorkflowRunOrderByDirection,
 )
+from hatchet_sdk.clients.v1.api_client import (
+    BaseRestClient,
+    maybe_additional_metadata_to_kv,
+)
 from hatchet_sdk.utils.typing import JSONSerializableMapping
 
 
@@ -62,27 +69,14 @@ class CreateCronTriggerConfig(BaseModel):
         return v
 
 
-class CronClient:
-    """
-    Client for managing workflow cron triggers synchronously.
-
-    Attributes:
-        _client (Client): The underlying client used to interact with the REST API.
-        aio (CronClientAsync): Asynchronous counterpart of CronClient.
-    """
-
-    _client: Client
-
-    def __init__(self, _client: Client):
-        """
-        Initializes the CronClient with a given Client instance.
+class CronClient(BaseRestClient):
+    def _wra(self, client: ApiClient) -> WorkflowRunApi:
+        return WorkflowRunApi(client)
 
-        Args:
-            _client (Client): The client instance to be used for REST interactions.
-        """
-        self._client = _client
+    def _wa(self, client: ApiClient) -> WorkflowApi:
+        return WorkflowApi(client)
 
-    def create(
+    async def aio_create(
         self,
         workflow_name: str,
         cron_name: str,
@@ -91,7 +85,7 @@ class CronClient:
         additional_metadata: JSONSerializableMapping,
     ) -> CronWorkflows:
         """
-        Creates a new workflow cron trigger.
+        Asynchronously creates a new workflow cron trigger.
 
         Args:
             workflow_name (str): The name of the workflow to trigger.
@@ -107,38 +101,61 @@ class CronClient:
             expression=expression, input=input, additional_metadata=additional_metadata
         )
 
-        return self._client.rest.cron_create(
+        async with self.client() as client:
+            return await self._wra(client).cron_workflow_trigger_create(
+                tenant=self.client_config.tenant_id,
+                workflow=workflow_name,
+                create_cron_workflow_trigger_request=CreateCronWorkflowTriggerRequest(
+                    cronName=cron_name,
+                    cronExpression=validated_input.expression,
+                    input=dict(validated_input.input),
+                    additionalMetadata=dict(validated_input.additional_metadata),
+                ),
+            )
+
+    def create(
+        self,
+        workflow_name: str,
+        cron_name: str,
+        expression: str,
+        input: JSONSerializableMapping,
+        additional_metadata: JSONSerializableMapping,
+    ) -> CronWorkflows:
+        return self._run_async_from_sync(
+            self.aio_create,
             workflow_name,
             cron_name,
-            validated_input.expression,
-            validated_input.input,
-            validated_input.additional_metadata,
+            expression,
+            input,
+            additional_metadata,
         )
 
-    def delete(self, cron_trigger: Union[str, CronWorkflows]) -> None:
+    async def aio_delete(self, cron_id: str) -> None:
         """
-        Deletes a workflow cron trigger.
+        Asynchronously deletes a workflow cron trigger.
 
         Args:
-            cron_trigger (Union[str, CronWorkflows]): The cron trigger ID or CronWorkflows instance to delete.
+            cron_id (str): The cron trigger ID or CronWorkflows instance to delete.
         """
-        self._client.rest.cron_delete(
-            cron_trigger.metadata.id
-            if isinstance(cron_trigger, CronWorkflows)
-            else cron_trigger
-        )
+        async with self.client() as client:
+            await self._wa(client).workflow_cron_delete(
+                tenant=self.client_config.tenant_id, cron_workflow=str(cron_id)
+            )
 
-    def list(
+    def delete(self, cron_id: str) -> None:
+        return self._run_async_from_sync(self.aio_delete, cron_id)
+
+    async def aio_list(
         self,
         offset: int | None = None,
         limit: int | None = None,
         workflow_id: str | None = None,
-        additional_metadata: list[str] | None = None,
+        additional_metadata: JSONSerializableMapping | None = None,
         order_by_field: CronWorkflowsOrderByField | None = None,
         order_by_direction: WorkflowRunOrderByDirection | None = None,
     ) -> CronWorkflowsList:
         """
-        Retrieves a list of all workflow cron triggers matching the criteria.
+        Asynchronously retrieves a list of all workflow cron triggers matching the criteria.
 
         Args:
             offset (int | None): The offset to start the list from.
@@ -151,88 +168,30 @@ class CronClient:
         Returns:
             CronWorkflowsList: A list of cron workflows.
         """
-        return self._client.rest.cron_list(
-            offset=offset,
-            limit=limit,
-            workflow_id=workflow_id,
-            additional_metadata=additional_metadata,
-            order_by_field=order_by_field,
-            order_by_direction=order_by_direction,
-        )
-
-    def get(self, cron_trigger: Union[str, CronWorkflows]) -> CronWorkflows:
-        """
-        Retrieves a specific workflow cron trigger by ID.
-
-        Args:
-            cron_trigger (Union[str, CronWorkflows]): The cron trigger ID or CronWorkflows instance to retrieve.
-
-        Returns:
-            CronWorkflows: The requested cron workflow instance.
-        """
-        return self._client.rest.cron_get(
-            cron_trigger.metadata.id
-            if isinstance(cron_trigger, CronWorkflows)
-            else cron_trigger
-        )
-
-    async def aio_create(
-        self,
-        workflow_name: str,
-        cron_name: str,
-        expression: str,
-        input: JSONSerializableMapping,
-        additional_metadata: JSONSerializableMapping,
-    ) -> CronWorkflows:
-        """
-        Asynchronously creates a new workflow cron trigger.
-
-        Args:
-            workflow_name (str): The name of the workflow to trigger.
-            cron_name (str): The name of the cron trigger.
-            expression (str): The cron expression defining the schedule.
-            input (dict): The input data for the cron workflow.
-            additional_metadata (dict[str, str]): Additional metadata associated with the cron trigger (e.g. {"key1": "value1", "key2": "value2"}).
-
-        Returns:
-            CronWorkflows: The created cron workflow instance.
-        """
-        validated_input = CreateCronTriggerConfig(
-            expression=expression, input=input, additional_metadata=additional_metadata
-        )
-
-        return await self._client.rest.aio_create_cron(
-            workflow_name=workflow_name,
-            cron_name=cron_name,
-            expression=validated_input.expression,
-            input=validated_input.input,
-            additional_metadata=validated_input.additional_metadata,
-        )
-
-    async def aio_delete(self, cron_trigger: Union[str, CronWorkflows]) -> None:
-        """
-        Asynchronously deletes a workflow cron trigger.
-
-        Args:
-            cron_trigger (Union[str, CronWorkflows]): The cron trigger ID or CronWorkflows instance to delete.
-        """
-        await self._client.rest.aio_delete_cron(
-            cron_trigger.metadata.id
-            if isinstance(cron_trigger, CronWorkflows)
-            else cron_trigger
-        )
+        async with self.client() as client:
+            return await self._wa(client).cron_workflow_list(
+                tenant=self.client_config.tenant_id,
+                offset=offset,
+                limit=limit,
+                workflow_id=workflow_id,
+                additional_metadata=maybe_additional_metadata_to_kv(
+                    additional_metadata
+                ),
+                order_by_field=order_by_field,
+                order_by_direction=order_by_direction,
+            )
 
-    async def aio_list(
+    def list(
         self,
         offset: int | None = None,
         limit: int | None = None,
         workflow_id: str | None = None,
-        additional_metadata: List[str] | None = None,
+        additional_metadata: JSONSerializableMapping | None = None,
         order_by_field: CronWorkflowsOrderByField | None = None,
         order_by_direction: WorkflowRunOrderByDirection | None = None,
     ) -> CronWorkflowsList:
         """
-        Asynchronously retrieves a list of all workflow cron triggers matching the criteria.
+        Synchronously retrieves a list of all workflow cron triggers matching the criteria.
 
         Args:
             offset (int | None): The offset to start the list from.
@@ -245,7 +204,8 @@ class CronClient:
         Returns:
             CronWorkflowsList: A list of cron workflows.
         """
-        return await self._client.rest.aio_list_crons(
+        return self._run_async_from_sync(
+            self.aio_list,
             offset=offset,
             limit=limit,
             workflow_id=workflow_id,
@@ -254,19 +214,29 @@ class CronClient:
             order_by_direction=order_by_direction,
         )
 
-    async def aio_get(self, cron_trigger: Union[str, CronWorkflows]) -> CronWorkflows:
+    async def aio_get(self, cron_id: str) -> CronWorkflows:
         """
         Asynchronously retrieves a specific workflow cron trigger by ID.
 
         Args:
-            cron_trigger (Union[str, CronWorkflows]): The cron trigger ID or CronWorkflows instance to retrieve.
+            cron_id (str): The cron trigger ID or CronWorkflows instance to retrieve.
 
         Returns:
             CronWorkflows: The requested cron workflow instance.
         """
+        async with self.client() as client:
+            return await self._wa(client).workflow_cron_get(
+                tenant=self.client_config.tenant_id, cron_workflow=str(cron_id)
+            )
+
+    def get(self, cron_id: str) -> CronWorkflows:
+        """
+        Synchronously retrieves a specific workflow cron trigger by ID.
 
-        return await self._client.rest.aio_get_cron(
-            cron_trigger.metadata.id
-            if isinstance(cron_trigger, CronWorkflows)
-            else cron_trigger
-        )
+        Args:
+            cron_id (str): The cron trigger ID or CronWorkflows instance to retrieve.
+
+        Returns:
+            CronWorkflows: The requested cron workflow instance.
+        """
+        return self._run_async_from_sync(self.aio_get, cron_id)

hatchet_sdk/features/logs.py

@@ -0,0 +1,16 @@
+from hatchet_sdk.clients.rest.api.log_api import LogApi
+from hatchet_sdk.clients.rest.api_client import ApiClient
+from hatchet_sdk.clients.rest.models.v1_log_line_list import V1LogLineList
+from hatchet_sdk.clients.v1.api_client import BaseRestClient
+
+
+class LogsClient(BaseRestClient):
+    def _la(self, client: ApiClient) -> LogApi:
+        return LogApi(client)
+
+    async def aio_list(self, task_run_id: str) -> V1LogLineList:
+        async with self.client() as client:
+            return await self._la(client).v1_log_line_list(task=task_run_id)
+
+    def list(self, task_run_id: str) -> V1LogLineList:
+        return self._run_async_from_sync(self.aio_list, task_run_id)

hatchet_sdk/features/metrics.py

@@ -0,0 +1,75 @@
+from hatchet_sdk.clients.rest.api.tenant_api import TenantApi
+from hatchet_sdk.clients.rest.api.workflow_api import WorkflowApi
+from hatchet_sdk.clients.rest.api_client import ApiClient
+from hatchet_sdk.clients.rest.models.tenant_queue_metrics import TenantQueueMetrics
+from hatchet_sdk.clients.rest.models.tenant_step_run_queue_metrics import (
+    TenantStepRunQueueMetrics,
+)
+from hatchet_sdk.clients.rest.models.workflow_metrics import WorkflowMetrics
+from hatchet_sdk.clients.rest.models.workflow_run_status import WorkflowRunStatus
+from hatchet_sdk.clients.v1.api_client import (
+    BaseRestClient,
+    maybe_additional_metadata_to_kv,
+)
+from hatchet_sdk.utils.typing import JSONSerializableMapping
+
+
+class MetricsClient(BaseRestClient):
+    def _wa(self, client: ApiClient) -> WorkflowApi:
+        return WorkflowApi(client)
+
+    def _ta(self, client: ApiClient) -> TenantApi:
+        return TenantApi(client)
+
+    async def aio_get_workflow_metrics(
+        self,
+        workflow_id: str,
+        status: WorkflowRunStatus | None = None,
+        group_key: str | None = None,
+    ) -> WorkflowMetrics:
+        async with self.client() as client:
+            return await self._wa(client).workflow_get_metrics(
+                workflow=workflow_id, status=status, group_key=group_key
+            )
+
+    def get_workflow_metrics(
+        self,
+        workflow_id: str,
+        status: WorkflowRunStatus | None = None,
+        group_key: str | None = None,
+    ) -> WorkflowMetrics:
+        return self._run_async_from_sync(
+            self.aio_get_workflow_metrics, workflow_id, status, group_key
+        )
+
+    async def aio_get_queue_metrics(
+        self,
+        workflow_ids: list[str] | None = None,
+        additional_metadata: JSONSerializableMapping | None = None,
+    ) -> TenantQueueMetrics:
+        async with self.client() as client:
+            return await self._wa(client).tenant_get_queue_metrics(
+                tenant=self.client_config.tenant_id,
+                workflows=workflow_ids,
+                additional_metadata=maybe_additional_metadata_to_kv(
+                    additional_metadata
+                ),
+            )
+
+    def get_queue_metrics(
+        self,
+        workflow_ids: list[str] | None = None,
+        additional_metadata: JSONSerializableMapping | None = None,
+    ) -> TenantQueueMetrics:
+        return self._run_async_from_sync(
+            self.aio_get_queue_metrics, workflow_ids, additional_metadata
+        )
+
+    async def aio_get_task_metrics(self) -> TenantStepRunQueueMetrics:
+        async with self.client() as client:
+            return await self._ta(client).tenant_get_step_run_queue_metrics(
+                tenant=self.client_config.tenant_id
+            )
+
+    def get_task_metrics(self) -> TenantStepRunQueueMetrics:
+        return self._run_async_from_sync(self.aio_get_task_metrics)