hatchet-sdk 1.18.1__py3-none-any.whl → 1.20.0__py3-none-any.whl

hatchet_sdk/v0/clients/events.py

@@ -1,182 +0,0 @@
-import asyncio
-import datetime
-import json
-from typing import Any, Dict, List, Optional, TypedDict
-from uuid import uuid4
-
-import grpc
-from google.protobuf import timestamp_pb2
-
-from hatchet_sdk.contracts.events_pb2 import (
-    BulkPushEventRequest,
-    Event,
-    PushEventRequest,
-    PutLogRequest,
-    PutStreamEventRequest,
-)
-from hatchet_sdk.contracts.events_pb2_grpc import EventsServiceStub
-from hatchet_sdk.v0.clients.rest.tenacity_utils import tenacity_retry
-
-from ..loader import ClientConfig
-from ..metadata import get_metadata
-
-
-def new_event(conn, config: ClientConfig):
-    return EventClient(
-        client=EventsServiceStub(conn),
-        config=config,
-    )
-
-
-def proto_timestamp_now():
-    t = datetime.datetime.now().timestamp()
-    seconds = int(t)
-    nanos = int(t % 1 * 1e9)
-
-    return timestamp_pb2.Timestamp(seconds=seconds, nanos=nanos)
-
-
-class PushEventOptions(TypedDict, total=False):
-    additional_metadata: Dict[str, str] | None = None
-    namespace: str | None = None
-
-
-class BulkPushEventOptions(TypedDict, total=False):
-    namespace: str | None = None
-
-
-class BulkPushEventWithMetadata(TypedDict, total=False):
-    key: str
-    payload: Any
-    additional_metadata: Optional[Dict[str, Any]]  # Optional metadata
-
-
-class EventClient:
-    def __init__(self, client: EventsServiceStub, config: ClientConfig):
-        self.client = client
-        self.token = config.token
-        self.namespace = config.namespace
-
-    async def async_push(
-        self, event_key, payload, options: Optional[PushEventOptions] = None
-    ) -> Event:
-        return await asyncio.to_thread(
-            self.push, event_key=event_key, payload=payload, options=options
-        )
-
-    async def async_bulk_push(
-        self,
-        events: List[BulkPushEventWithMetadata],
-        options: Optional[BulkPushEventOptions] = None,
-    ) -> List[Event]:
-        return await asyncio.to_thread(self.bulk_push, events=events, options=options)
-
-    ## IMPORTANT: Keep this method's signature in sync with the wrapper in the OTel instrumentor
-    @tenacity_retry
-    def push(self, event_key, payload, options: PushEventOptions = None) -> Event:
-        namespace = self.namespace
-
-        if (
-            options is not None
-            and "namespace" in options
-            and options["namespace"] is not None
-        ):
-            namespace = options.pop("namespace")
-
-        namespaced_event_key = namespace + event_key
-
-        try:
-            meta = dict() if options is None else options["additional_metadata"]
-            meta_bytes = None if meta is None else json.dumps(meta).encode("utf-8")
-        except Exception as e:
-            raise ValueError(f"Error encoding meta: {e}")
-
-        try:
-            payload_bytes = json.dumps(payload).encode("utf-8")
-        except json.UnicodeEncodeError as e:
-            raise ValueError(f"Error encoding payload: {e}")
-
-        request = PushEventRequest(
-            key=namespaced_event_key,
-            payload=payload_bytes,
-            eventTimestamp=proto_timestamp_now(),
-            additionalMetadata=meta_bytes,
-        )
-
-        return self.client.Push(request, metadata=get_metadata(self.token))
-
-    ## IMPORTANT: Keep this method's signature in sync with the wrapper in the OTel instrumentor
-    @tenacity_retry
-    def bulk_push(
-        self,
-        events: List[BulkPushEventWithMetadata],
-        options: BulkPushEventOptions = None,
-    ) -> List[Event]:
-        namespace = self.namespace
-
-        if (
-            options is not None
-            and "namespace" in options
-            and options["namespace"] is not None
-        ):
-            namespace = options.pop("namespace")
-
-        bulk_events = []
-        for event in events:
-            event_key = namespace + event["key"]
-            payload = event["payload"]
-
-            try:
-                meta = event.get("additional_metadata", {})
-                meta_bytes = json.dumps(meta).encode("utf-8") if meta else None
-            except Exception as e:
-                raise ValueError(f"Error encoding meta: {e}")
-
-            try:
-                payload_bytes = json.dumps(payload).encode("utf-8")
-            except json.UnicodeEncodeError as e:
-                raise ValueError(f"Error encoding payload: {e}")
-
-            request = PushEventRequest(
-                key=event_key,
-                payload=payload_bytes,
-                eventTimestamp=proto_timestamp_now(),
-                additionalMetadata=meta_bytes,
-            )
-            bulk_events.append(request)
-
-        bulk_request = BulkPushEventRequest(events=bulk_events)
-
-        response = self.client.BulkPush(bulk_request, metadata=get_metadata(self.token))
-
-        return response.events
-
-    def log(self, message: str, step_run_id: str):
-        try:
-            request = PutLogRequest(
-                stepRunId=step_run_id,
-                createdAt=proto_timestamp_now(),
-                message=message,
-            )
-
-            self.client.PutLog(request, metadata=get_metadata(self.token))
-        except Exception as e:
-            raise ValueError(f"Error logging: {e}")
-
-    def stream(self, data: str | bytes, step_run_id: str):
-        try:
-            if isinstance(data, str):
-                data_bytes = data.encode("utf-8")
-            elif isinstance(data, bytes):
-                data_bytes = data
-            else:
-                raise ValueError("Invalid data type. Expected str, bytes, or file.")
-
-            request = PutStreamEventRequest(
-                stepRunId=step_run_id,
-                createdAt=proto_timestamp_now(),
-                message=data_bytes,
-            )
-            self.client.PutStreamEvent(request, metadata=get_metadata(self.token))
-        except Exception as e:
-            raise ValueError(f"Error putting stream event: {e}")

hatchet_sdk/v0/clients/rest/__init__.py

@@ -1,307 +0,0 @@
-# coding: utf-8
-
-# flake8: noqa
-
-"""
-    Hatchet API
-
-    The Hatchet API
-
-    The version of the OpenAPI document: 1.0.0
-    Generated by OpenAPI Generator (https://openapi-generator.tech)
-
-    Do not edit the class manually.
-"""  # noqa: E501
-
-
-__version__ = "1.0.0"
-
-# import apis into sdk package
-from hatchet_sdk.v0.clients.rest.api.api_token_api import APITokenApi
-from hatchet_sdk.v0.clients.rest.api.default_api import DefaultApi
-from hatchet_sdk.v0.clients.rest.api.event_api import EventApi
-from hatchet_sdk.v0.clients.rest.api.github_api import GithubApi
-from hatchet_sdk.v0.clients.rest.api.healthcheck_api import HealthcheckApi
-from hatchet_sdk.v0.clients.rest.api.log_api import LogApi
-from hatchet_sdk.v0.clients.rest.api.metadata_api import MetadataApi
-from hatchet_sdk.v0.clients.rest.api.rate_limits_api import RateLimitsApi
-from hatchet_sdk.v0.clients.rest.api.slack_api import SlackApi
-from hatchet_sdk.v0.clients.rest.api.sns_api import SNSApi
-from hatchet_sdk.v0.clients.rest.api.step_run_api import StepRunApi
-from hatchet_sdk.v0.clients.rest.api.tenant_api import TenantApi
-from hatchet_sdk.v0.clients.rest.api.user_api import UserApi
-from hatchet_sdk.v0.clients.rest.api.worker_api import WorkerApi
-from hatchet_sdk.v0.clients.rest.api.workflow_api import WorkflowApi
-from hatchet_sdk.v0.clients.rest.api.workflow_run_api import WorkflowRunApi
-from hatchet_sdk.v0.clients.rest.api_client import ApiClient
-
-# import ApiClient
-from hatchet_sdk.v0.clients.rest.api_response import ApiResponse
-from hatchet_sdk.v0.clients.rest.configuration import Configuration
-from hatchet_sdk.v0.clients.rest.exceptions import (
-    ApiAttributeError,
-    ApiException,
-    ApiKeyError,
-    ApiTypeError,
-    ApiValueError,
-    OpenApiException,
-)
-from hatchet_sdk.v0.clients.rest.models.accept_invite_request import AcceptInviteRequest
-
-# import models into sdk package
-from hatchet_sdk.v0.clients.rest.models.api_error import APIError
-from hatchet_sdk.v0.clients.rest.models.api_errors import APIErrors
-from hatchet_sdk.v0.clients.rest.models.api_meta import APIMeta
-from hatchet_sdk.v0.clients.rest.models.api_meta_auth import APIMetaAuth
-from hatchet_sdk.v0.clients.rest.models.api_meta_integration import APIMetaIntegration
-from hatchet_sdk.v0.clients.rest.models.api_meta_posthog import APIMetaPosthog
-from hatchet_sdk.v0.clients.rest.models.api_resource_meta import APIResourceMeta
-from hatchet_sdk.v0.clients.rest.models.api_token import APIToken
-from hatchet_sdk.v0.clients.rest.models.bulk_create_event_request import (
-    BulkCreateEventRequest,
-)
-from hatchet_sdk.v0.clients.rest.models.cancel_event_request import CancelEventRequest
-from hatchet_sdk.v0.clients.rest.models.concurrency_limit_strategy import (
-    ConcurrencyLimitStrategy,
-)
-from hatchet_sdk.v0.clients.rest.models.create_api_token_request import (
-    CreateAPITokenRequest,
-)
-from hatchet_sdk.v0.clients.rest.models.create_api_token_response import (
-    CreateAPITokenResponse,
-)
-from hatchet_sdk.v0.clients.rest.models.create_cron_workflow_trigger_request import (
-    CreateCronWorkflowTriggerRequest,
-)
-from hatchet_sdk.v0.clients.rest.models.create_event_request import CreateEventRequest
-from hatchet_sdk.v0.clients.rest.models.create_pull_request_from_step_run import (
-    CreatePullRequestFromStepRun,
-)
-from hatchet_sdk.v0.clients.rest.models.create_sns_integration_request import (
-    CreateSNSIntegrationRequest,
-)
-from hatchet_sdk.v0.clients.rest.models.create_tenant_alert_email_group_request import (
-    CreateTenantAlertEmailGroupRequest,
-)
-from hatchet_sdk.v0.clients.rest.models.create_tenant_invite_request import (
-    CreateTenantInviteRequest,
-)
-from hatchet_sdk.v0.clients.rest.models.create_tenant_request import CreateTenantRequest
-from hatchet_sdk.v0.clients.rest.models.cron_workflows import CronWorkflows
-from hatchet_sdk.v0.clients.rest.models.cron_workflows_list import CronWorkflowsList
-from hatchet_sdk.v0.clients.rest.models.cron_workflows_method import CronWorkflowsMethod
-from hatchet_sdk.v0.clients.rest.models.cron_workflows_order_by_field import (
-    CronWorkflowsOrderByField,
-)
-from hatchet_sdk.v0.clients.rest.models.event import Event
-from hatchet_sdk.v0.clients.rest.models.event_data import EventData
-from hatchet_sdk.v0.clients.rest.models.event_key_list import EventKeyList
-from hatchet_sdk.v0.clients.rest.models.event_list import EventList
-from hatchet_sdk.v0.clients.rest.models.event_order_by_direction import (
-    EventOrderByDirection,
-)
-from hatchet_sdk.v0.clients.rest.models.event_order_by_field import EventOrderByField
-from hatchet_sdk.v0.clients.rest.models.event_update_cancel200_response import (
-    EventUpdateCancel200Response,
-)
-from hatchet_sdk.v0.clients.rest.models.event_workflow_run_summary import (
-    EventWorkflowRunSummary,
-)
-from hatchet_sdk.v0.clients.rest.models.events import Events
-from hatchet_sdk.v0.clients.rest.models.get_step_run_diff_response import (
-    GetStepRunDiffResponse,
-)
-from hatchet_sdk.v0.clients.rest.models.info_get_version200_response import (
-    InfoGetVersion200Response,
-)
-from hatchet_sdk.v0.clients.rest.models.job import Job
-from hatchet_sdk.v0.clients.rest.models.job_run import JobRun
-from hatchet_sdk.v0.clients.rest.models.job_run_status import JobRunStatus
-from hatchet_sdk.v0.clients.rest.models.list_api_tokens_response import (
-    ListAPITokensResponse,
-)
-from hatchet_sdk.v0.clients.rest.models.list_pull_requests_response import (
-    ListPullRequestsResponse,
-)
-from hatchet_sdk.v0.clients.rest.models.list_slack_webhooks import ListSlackWebhooks
-from hatchet_sdk.v0.clients.rest.models.list_sns_integrations import ListSNSIntegrations
-from hatchet_sdk.v0.clients.rest.models.log_line import LogLine
-from hatchet_sdk.v0.clients.rest.models.log_line_level import LogLineLevel
-from hatchet_sdk.v0.clients.rest.models.log_line_list import LogLineList
-from hatchet_sdk.v0.clients.rest.models.log_line_order_by_direction import (
-    LogLineOrderByDirection,
-)
-from hatchet_sdk.v0.clients.rest.models.log_line_order_by_field import (
-    LogLineOrderByField,
-)
-from hatchet_sdk.v0.clients.rest.models.pagination_response import PaginationResponse
-from hatchet_sdk.v0.clients.rest.models.pull_request import PullRequest
-from hatchet_sdk.v0.clients.rest.models.pull_request_state import PullRequestState
-from hatchet_sdk.v0.clients.rest.models.queue_metrics import QueueMetrics
-from hatchet_sdk.v0.clients.rest.models.rate_limit import RateLimit
-from hatchet_sdk.v0.clients.rest.models.rate_limit_list import RateLimitList
-from hatchet_sdk.v0.clients.rest.models.rate_limit_order_by_direction import (
-    RateLimitOrderByDirection,
-)
-from hatchet_sdk.v0.clients.rest.models.rate_limit_order_by_field import (
-    RateLimitOrderByField,
-)
-from hatchet_sdk.v0.clients.rest.models.recent_step_runs import RecentStepRuns
-from hatchet_sdk.v0.clients.rest.models.reject_invite_request import RejectInviteRequest
-from hatchet_sdk.v0.clients.rest.models.replay_event_request import ReplayEventRequest
-from hatchet_sdk.v0.clients.rest.models.replay_workflow_runs_request import (
-    ReplayWorkflowRunsRequest,
-)
-from hatchet_sdk.v0.clients.rest.models.replay_workflow_runs_response import (
-    ReplayWorkflowRunsResponse,
-)
-from hatchet_sdk.v0.clients.rest.models.rerun_step_run_request import (
-    RerunStepRunRequest,
-)
-from hatchet_sdk.v0.clients.rest.models.schedule_workflow_run_request import (
-    ScheduleWorkflowRunRequest,
-)
-from hatchet_sdk.v0.clients.rest.models.scheduled_run_status import ScheduledRunStatus
-from hatchet_sdk.v0.clients.rest.models.scheduled_workflows import ScheduledWorkflows
-from hatchet_sdk.v0.clients.rest.models.scheduled_workflows_list import (
-    ScheduledWorkflowsList,
-)
-from hatchet_sdk.v0.clients.rest.models.scheduled_workflows_method import (
-    ScheduledWorkflowsMethod,
-)
-from hatchet_sdk.v0.clients.rest.models.scheduled_workflows_order_by_field import (
-    ScheduledWorkflowsOrderByField,
-)
-from hatchet_sdk.v0.clients.rest.models.semaphore_slots import SemaphoreSlots
-from hatchet_sdk.v0.clients.rest.models.slack_webhook import SlackWebhook
-from hatchet_sdk.v0.clients.rest.models.sns_integration import SNSIntegration
-from hatchet_sdk.v0.clients.rest.models.step import Step
-from hatchet_sdk.v0.clients.rest.models.step_run import StepRun
-from hatchet_sdk.v0.clients.rest.models.step_run_archive import StepRunArchive
-from hatchet_sdk.v0.clients.rest.models.step_run_archive_list import StepRunArchiveList
-from hatchet_sdk.v0.clients.rest.models.step_run_diff import StepRunDiff
-from hatchet_sdk.v0.clients.rest.models.step_run_event import StepRunEvent
-from hatchet_sdk.v0.clients.rest.models.step_run_event_list import StepRunEventList
-from hatchet_sdk.v0.clients.rest.models.step_run_event_reason import StepRunEventReason
-from hatchet_sdk.v0.clients.rest.models.step_run_event_severity import (
-    StepRunEventSeverity,
-)
-from hatchet_sdk.v0.clients.rest.models.step_run_status import StepRunStatus
-from hatchet_sdk.v0.clients.rest.models.tenant import Tenant
-from hatchet_sdk.v0.clients.rest.models.tenant_alert_email_group import (
-    TenantAlertEmailGroup,
-)
-from hatchet_sdk.v0.clients.rest.models.tenant_alert_email_group_list import (
-    TenantAlertEmailGroupList,
-)
-from hatchet_sdk.v0.clients.rest.models.tenant_alerting_settings import (
-    TenantAlertingSettings,
-)
-from hatchet_sdk.v0.clients.rest.models.tenant_invite import TenantInvite
-from hatchet_sdk.v0.clients.rest.models.tenant_invite_list import TenantInviteList
-from hatchet_sdk.v0.clients.rest.models.tenant_list import TenantList
-from hatchet_sdk.v0.clients.rest.models.tenant_member import TenantMember
-from hatchet_sdk.v0.clients.rest.models.tenant_member_list import TenantMemberList
-from hatchet_sdk.v0.clients.rest.models.tenant_member_role import TenantMemberRole
-from hatchet_sdk.v0.clients.rest.models.tenant_queue_metrics import TenantQueueMetrics
-from hatchet_sdk.v0.clients.rest.models.tenant_resource import TenantResource
-from hatchet_sdk.v0.clients.rest.models.tenant_resource_limit import TenantResourceLimit
-from hatchet_sdk.v0.clients.rest.models.tenant_resource_policy import (
-    TenantResourcePolicy,
-)
-from hatchet_sdk.v0.clients.rest.models.tenant_step_run_queue_metrics import (
-    TenantStepRunQueueMetrics,
-)
-from hatchet_sdk.v0.clients.rest.models.trigger_workflow_run_request import (
-    TriggerWorkflowRunRequest,
-)
-from hatchet_sdk.v0.clients.rest.models.update_tenant_alert_email_group_request import (
-    UpdateTenantAlertEmailGroupRequest,
-)
-from hatchet_sdk.v0.clients.rest.models.update_tenant_invite_request import (
-    UpdateTenantInviteRequest,
-)
-from hatchet_sdk.v0.clients.rest.models.update_tenant_request import UpdateTenantRequest
-from hatchet_sdk.v0.clients.rest.models.update_worker_request import UpdateWorkerRequest
-from hatchet_sdk.v0.clients.rest.models.user import User
-from hatchet_sdk.v0.clients.rest.models.user_change_password_request import (
-    UserChangePasswordRequest,
-)
-from hatchet_sdk.v0.clients.rest.models.user_login_request import UserLoginRequest
-from hatchet_sdk.v0.clients.rest.models.user_register_request import UserRegisterRequest
-from hatchet_sdk.v0.clients.rest.models.user_tenant_memberships_list import (
-    UserTenantMembershipsList,
-)
-from hatchet_sdk.v0.clients.rest.models.user_tenant_public import UserTenantPublic
-from hatchet_sdk.v0.clients.rest.models.webhook_worker import WebhookWorker
-from hatchet_sdk.v0.clients.rest.models.webhook_worker_create_request import (
-    WebhookWorkerCreateRequest,
-)
-from hatchet_sdk.v0.clients.rest.models.webhook_worker_create_response import (
-    WebhookWorkerCreateResponse,
-)
-from hatchet_sdk.v0.clients.rest.models.webhook_worker_created import (
-    WebhookWorkerCreated,
-)
-from hatchet_sdk.v0.clients.rest.models.webhook_worker_list_response import (
-    WebhookWorkerListResponse,
-)
-from hatchet_sdk.v0.clients.rest.models.webhook_worker_request import (
-    WebhookWorkerRequest,
-)
-from hatchet_sdk.v0.clients.rest.models.webhook_worker_request_list_response import (
-    WebhookWorkerRequestListResponse,
-)
-from hatchet_sdk.v0.clients.rest.models.webhook_worker_request_method import (
-    WebhookWorkerRequestMethod,
-)
-from hatchet_sdk.v0.clients.rest.models.worker import Worker
-from hatchet_sdk.v0.clients.rest.models.worker_label import WorkerLabel
-from hatchet_sdk.v0.clients.rest.models.worker_list import WorkerList
-from hatchet_sdk.v0.clients.rest.models.worker_runtime_info import WorkerRuntimeInfo
-from hatchet_sdk.v0.clients.rest.models.worker_runtime_sdks import WorkerRuntimeSDKs
-from hatchet_sdk.v0.clients.rest.models.worker_type import WorkerType
-from hatchet_sdk.v0.clients.rest.models.workflow import Workflow
-from hatchet_sdk.v0.clients.rest.models.workflow_concurrency import WorkflowConcurrency
-from hatchet_sdk.v0.clients.rest.models.workflow_kind import WorkflowKind
-from hatchet_sdk.v0.clients.rest.models.workflow_list import WorkflowList
-from hatchet_sdk.v0.clients.rest.models.workflow_metrics import WorkflowMetrics
-from hatchet_sdk.v0.clients.rest.models.workflow_run import WorkflowRun
-from hatchet_sdk.v0.clients.rest.models.workflow_run_list import WorkflowRunList
-from hatchet_sdk.v0.clients.rest.models.workflow_run_order_by_direction import (
-    WorkflowRunOrderByDirection,
-)
-from hatchet_sdk.v0.clients.rest.models.workflow_run_order_by_field import (
-    WorkflowRunOrderByField,
-)
-from hatchet_sdk.v0.clients.rest.models.workflow_run_shape import WorkflowRunShape
-from hatchet_sdk.v0.clients.rest.models.workflow_run_status import WorkflowRunStatus
-from hatchet_sdk.v0.clients.rest.models.workflow_run_triggered_by import (
-    WorkflowRunTriggeredBy,
-)
-from hatchet_sdk.v0.clients.rest.models.workflow_runs_cancel_request import (
-    WorkflowRunsCancelRequest,
-)
-from hatchet_sdk.v0.clients.rest.models.workflow_runs_metrics import WorkflowRunsMetrics
-from hatchet_sdk.v0.clients.rest.models.workflow_runs_metrics_counts import (
-    WorkflowRunsMetricsCounts,
-)
-from hatchet_sdk.v0.clients.rest.models.workflow_tag import WorkflowTag
-from hatchet_sdk.v0.clients.rest.models.workflow_trigger_cron_ref import (
-    WorkflowTriggerCronRef,
-)
-from hatchet_sdk.v0.clients.rest.models.workflow_trigger_event_ref import (
-    WorkflowTriggerEventRef,
-)
-from hatchet_sdk.v0.clients.rest.models.workflow_triggers import WorkflowTriggers
-from hatchet_sdk.v0.clients.rest.models.workflow_update_request import (
-    WorkflowUpdateRequest,
-)
-from hatchet_sdk.v0.clients.rest.models.workflow_version import WorkflowVersion
-from hatchet_sdk.v0.clients.rest.models.workflow_version_definition import (
-    WorkflowVersionDefinition,
-)
-from hatchet_sdk.v0.clients.rest.models.workflow_version_meta import WorkflowVersionMeta
-from hatchet_sdk.v0.clients.rest.models.workflow_workers_count import (
-    WorkflowWorkersCount,
-)

hatchet_sdk/v0/clients/rest/api/__init__.py

@@ -1,19 +0,0 @@
-# flake8: noqa
-
-# import apis into api package
-from hatchet_sdk.v0.clients.rest.api.api_token_api import APITokenApi
-from hatchet_sdk.v0.clients.rest.api.default_api import DefaultApi
-from hatchet_sdk.v0.clients.rest.api.event_api import EventApi
-from hatchet_sdk.v0.clients.rest.api.github_api import GithubApi
-from hatchet_sdk.v0.clients.rest.api.healthcheck_api import HealthcheckApi
-from hatchet_sdk.v0.clients.rest.api.log_api import LogApi
-from hatchet_sdk.v0.clients.rest.api.metadata_api import MetadataApi
-from hatchet_sdk.v0.clients.rest.api.rate_limits_api import RateLimitsApi
-from hatchet_sdk.v0.clients.rest.api.slack_api import SlackApi
-from hatchet_sdk.v0.clients.rest.api.sns_api import SNSApi
-from hatchet_sdk.v0.clients.rest.api.step_run_api import StepRunApi
-from hatchet_sdk.v0.clients.rest.api.tenant_api import TenantApi
-from hatchet_sdk.v0.clients.rest.api.user_api import UserApi
-from hatchet_sdk.v0.clients.rest.api.worker_api import WorkerApi
-from hatchet_sdk.v0.clients.rest.api.workflow_api import WorkflowApi
-from hatchet_sdk.v0.clients.rest.api.workflow_run_api import WorkflowRunApi