hatchet-sdk 1.12.0__tar.gz → 1.12.1__tar.gz

{hatchet_sdk-1.12.0 → hatchet_sdk-1.12.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: hatchet-sdk
-Version: 1.12.0
+Version: 1.12.1
 Summary: 
 License: MIT
 Author: Alexander Belanger

{hatchet_sdk-1.12.0 → hatchet_sdk-1.12.1}/hatchet_sdk/clients/events.py

@@ -10,8 +10,12 @@ from hatchet_sdk.clients.rest.api.event_api import EventApi
 from hatchet_sdk.clients.rest.api.workflow_runs_api import WorkflowRunsApi
 from hatchet_sdk.clients.rest.api_client import ApiClient
 from hatchet_sdk.clients.rest.models.v1_event_list import V1EventList
+from hatchet_sdk.clients.rest.models.v1_task_status import V1TaskStatus
 from hatchet_sdk.clients.rest.tenacity_utils import tenacity_retry
-from hatchet_sdk.clients.v1.api_client import BaseRestClient
+from hatchet_sdk.clients.v1.api_client import (
+    BaseRestClient,
+    maybe_additional_metadata_to_kv,
+)
 from hatchet_sdk.config import ClientConfig
 from hatchet_sdk.connection import new_conn
 from hatchet_sdk.contracts.events_pb2 import (
@@ -211,9 +215,24 @@ class EventClient(BaseRestClient):
         keys: list[str] | None = None,
         since: datetime.datetime | None = None,
         until: datetime.datetime | None = None,
+        workflow_ids: list[str] | None = None,
+        workflow_run_statuses: list[V1TaskStatus] | None = None,
+        event_ids: list[str] | None = None,
+        additional_metadata: JSONSerializableMapping | None = None,
+        scopes: list[str] | None = None,
     ) -> V1EventList:
         return await asyncio.to_thread(
-            self.list, offset=offset, limit=limit, keys=keys, since=since, until=until
+            self.list,
+            offset=offset,
+            limit=limit,
+            keys=keys,
+            since=since,
+            until=until,
+            workflow_ids=workflow_ids,
+            workflow_run_statuses=workflow_run_statuses,
+            event_ids=event_ids,
+            additional_metadata=additional_metadata,
+            scopes=scopes,
         )
 
     def list(
@@ -223,6 +242,11 @@ class EventClient(BaseRestClient):
         keys: list[str] | None = None,
         since: datetime.datetime | None = None,
         until: datetime.datetime | None = None,
+        workflow_ids: list[str] | None = None,
+        workflow_run_statuses: list[V1TaskStatus] | None = None,
+        event_ids: list[str] | None = None,
+        additional_metadata: JSONSerializableMapping | None = None,
+        scopes: list[str] | None = None,
     ) -> V1EventList:
         with self.client() as client:
             return self._ea(client).v1_event_list(
@@ -232,4 +256,11 @@ class EventClient(BaseRestClient):
                 keys=keys,
                 since=since,
                 until=until,
+                workflow_ids=workflow_ids,
+                workflow_run_statuses=workflow_run_statuses,
+                event_ids=event_ids,
+                additional_metadata=maybe_additional_metadata_to_kv(
+                    additional_metadata
+                ),
+                scopes=scopes,
             )

{hatchet_sdk-1.12.0 → hatchet_sdk-1.12.1}/hatchet_sdk/clients/rest/__init__.py

@@ -237,6 +237,7 @@ from hatchet_sdk.clients.rest.models.v1_create_filter_request import (
 from hatchet_sdk.clients.rest.models.v1_dag_children import V1DagChildren
 from hatchet_sdk.clients.rest.models.v1_event import V1Event
 from hatchet_sdk.clients.rest.models.v1_event_list import V1EventList
+from hatchet_sdk.clients.rest.models.v1_event_triggered_run import V1EventTriggeredRun
 from hatchet_sdk.clients.rest.models.v1_event_workflow_run_summary import (
     V1EventWorkflowRunSummary,
 )

{hatchet_sdk-1.12.0 → hatchet_sdk-1.12.1}/hatchet_sdk/clients/rest/api/event_api.py

@@ -39,6 +39,7 @@ from hatchet_sdk.clients.rest.models.event_update_cancel200_response import (
 from hatchet_sdk.clients.rest.models.events import Events
 from hatchet_sdk.clients.rest.models.replay_event_request import ReplayEventRequest
 from hatchet_sdk.clients.rest.models.v1_event_list import V1EventList
+from hatchet_sdk.clients.rest.models.v1_task_status import V1TaskStatus
 from hatchet_sdk.clients.rest.models.workflow_run_status import WorkflowRunStatus
 from hatchet_sdk.clients.rest.rest import RESTResponseType
 
@@ -2549,6 +2550,269 @@ class EventApi:
             _request_auth=_request_auth,
         )
 
+    @validate_call
+    def v1_event_key_list(
+        self,
+        tenant: Annotated[
+            str,
+            Field(
+                min_length=36, strict=True, max_length=36, description="The tenant id"
+            ),
+        ],
+        _request_timeout: Union[
+            None,
+            Annotated[StrictFloat, Field(gt=0)],
+            Tuple[
+                Annotated[StrictFloat, Field(gt=0)], Annotated[StrictFloat, Field(gt=0)]
+            ],
+        ] = None,
+        _request_auth: Optional[Dict[StrictStr, Any]] = None,
+        _content_type: Optional[StrictStr] = None,
+        _headers: Optional[Dict[StrictStr, Any]] = None,
+        _host_index: Annotated[StrictInt, Field(ge=0, le=0)] = 0,
+    ) -> EventKeyList:
+        """List event keys
+
+        Lists all event keys for a tenant.
+
+        :param tenant: The tenant id (required)
+        :type tenant: str
+        :param _request_timeout: timeout setting for this request. If one
+                                 number provided, it will be total request
+                                 timeout. It can also be a pair (tuple) of
+                                 (connection, read) timeouts.
+        :type _request_timeout: int, tuple(int, int), optional
+        :param _request_auth: set to override the auth_settings for an a single
+                              request; this effectively ignores the
+                              authentication in the spec for a single request.
+        :type _request_auth: dict, optional
+        :param _content_type: force content-type for the request.
+        :type _content_type: str, Optional
+        :param _headers: set to override the headers for a single
+                         request; this effectively ignores the headers
+                         in the spec for a single request.
+        :type _headers: dict, optional
+        :param _host_index: set to override the host_index for a single
+                            request; this effectively ignores the host_index
+                            in the spec for a single request.
+        :type _host_index: int, optional
+        :return: Returns the result object.
+        """  # noqa: E501
+
+        _param = self._v1_event_key_list_serialize(
+            tenant=tenant,
+            _request_auth=_request_auth,
+            _content_type=_content_type,
+            _headers=_headers,
+            _host_index=_host_index,
+        )
+
+        _response_types_map: Dict[str, Optional[str]] = {
+            "200": "EventKeyList",
+            "400": "APIErrors",
+            "403": "APIErrors",
+        }
+        response_data = self.api_client.call_api(
+            *_param, _request_timeout=_request_timeout
+        )
+        response_data.read()
+        return self.api_client.response_deserialize(
+            response_data=response_data,
+            response_types_map=_response_types_map,
+        ).data
+
+    @validate_call
+    def v1_event_key_list_with_http_info(
+        self,
+        tenant: Annotated[
+            str,
+            Field(
+                min_length=36, strict=True, max_length=36, description="The tenant id"
+            ),
+        ],
+        _request_timeout: Union[
+            None,
+            Annotated[StrictFloat, Field(gt=0)],
+            Tuple[
+                Annotated[StrictFloat, Field(gt=0)], Annotated[StrictFloat, Field(gt=0)]
+            ],
+        ] = None,
+        _request_auth: Optional[Dict[StrictStr, Any]] = None,
+        _content_type: Optional[StrictStr] = None,
+        _headers: Optional[Dict[StrictStr, Any]] = None,
+        _host_index: Annotated[StrictInt, Field(ge=0, le=0)] = 0,
+    ) -> ApiResponse[EventKeyList]:
+        """List event keys
+
+        Lists all event keys for a tenant.
+
+        :param tenant: The tenant id (required)
+        :type tenant: str
+        :param _request_timeout: timeout setting for this request. If one
+                                 number provided, it will be total request
+                                 timeout. It can also be a pair (tuple) of
+                                 (connection, read) timeouts.
+        :type _request_timeout: int, tuple(int, int), optional
+        :param _request_auth: set to override the auth_settings for an a single
+                              request; this effectively ignores the
+                              authentication in the spec for a single request.
+        :type _request_auth: dict, optional
+        :param _content_type: force content-type for the request.
+        :type _content_type: str, Optional
+        :param _headers: set to override the headers for a single
+                         request; this effectively ignores the headers
+                         in the spec for a single request.
+        :type _headers: dict, optional
+        :param _host_index: set to override the host_index for a single
+                            request; this effectively ignores the host_index
+                            in the spec for a single request.
+        :type _host_index: int, optional
+        :return: Returns the result object.
+        """  # noqa: E501
+
+        _param = self._v1_event_key_list_serialize(
+            tenant=tenant,
+            _request_auth=_request_auth,
+            _content_type=_content_type,
+            _headers=_headers,
+            _host_index=_host_index,
+        )
+
+        _response_types_map: Dict[str, Optional[str]] = {
+            "200": "EventKeyList",
+            "400": "APIErrors",
+            "403": "APIErrors",
+        }
+        response_data = self.api_client.call_api(
+            *_param, _request_timeout=_request_timeout
+        )
+        response_data.read()
+        return self.api_client.response_deserialize(
+            response_data=response_data,
+            response_types_map=_response_types_map,
+        )
+
+    @validate_call
+    def v1_event_key_list_without_preload_content(
+        self,
+        tenant: Annotated[
+            str,
+            Field(
+                min_length=36, strict=True, max_length=36, description="The tenant id"
+            ),
+        ],
+        _request_timeout: Union[
+            None,
+            Annotated[StrictFloat, Field(gt=0)],
+            Tuple[
+                Annotated[StrictFloat, Field(gt=0)], Annotated[StrictFloat, Field(gt=0)]
+            ],
+        ] = None,
+        _request_auth: Optional[Dict[StrictStr, Any]] = None,
+        _content_type: Optional[StrictStr] = None,
+        _headers: Optional[Dict[StrictStr, Any]] = None,
+        _host_index: Annotated[StrictInt, Field(ge=0, le=0)] = 0,
+    ) -> RESTResponseType:
+        """List event keys
+
+        Lists all event keys for a tenant.
+
+        :param tenant: The tenant id (required)
+        :type tenant: str
+        :param _request_timeout: timeout setting for this request. If one
+                                 number provided, it will be total request
+                                 timeout. It can also be a pair (tuple) of
+                                 (connection, read) timeouts.
+        :type _request_timeout: int, tuple(int, int), optional
+        :param _request_auth: set to override the auth_settings for an a single
+                              request; this effectively ignores the
+                              authentication in the spec for a single request.
+        :type _request_auth: dict, optional
+        :param _content_type: force content-type for the request.
+        :type _content_type: str, Optional
+        :param _headers: set to override the headers for a single
+                         request; this effectively ignores the headers
+                         in the spec for a single request.
+        :type _headers: dict, optional
+        :param _host_index: set to override the host_index for a single
+                            request; this effectively ignores the host_index
+                            in the spec for a single request.
+        :type _host_index: int, optional
+        :return: Returns the result object.
+        """  # noqa: E501
+
+        _param = self._v1_event_key_list_serialize(
+            tenant=tenant,
+            _request_auth=_request_auth,
+            _content_type=_content_type,
+            _headers=_headers,
+            _host_index=_host_index,
+        )
+
+        _response_types_map: Dict[str, Optional[str]] = {
+            "200": "EventKeyList",
+            "400": "APIErrors",
+            "403": "APIErrors",
+        }
+        response_data = self.api_client.call_api(
+            *_param, _request_timeout=_request_timeout
+        )
+        return response_data.response
+
+    def _v1_event_key_list_serialize(
+        self,
+        tenant,
+        _request_auth,
+        _content_type,
+        _headers,
+        _host_index,
+    ) -> RequestSerialized:
+
+        _host = None
+
+        _collection_formats: Dict[str, str] = {}
+
+        _path_params: Dict[str, str] = {}
+        _query_params: List[Tuple[str, str]] = []
+        _header_params: Dict[str, Optional[str]] = _headers or {}
+        _form_params: List[Tuple[str, str]] = []
+        _files: Dict[
+            str, Union[str, bytes, List[str], List[bytes], List[Tuple[str, bytes]]]
+        ] = {}
+        _body_params: Optional[bytes] = None
+
+        # process the path parameters
+        if tenant is not None:
+            _path_params["tenant"] = tenant
+        # process the query parameters
+        # process the header parameters
+        # process the form parameters
+        # process the body parameter
+
+        # set the HTTP header `Accept`
+        if "Accept" not in _header_params:
+            _header_params["Accept"] = self.api_client.select_header_accept(
+                ["application/json"]
+            )
+
+        # authentication setting
+        _auth_settings: List[str] = ["cookieAuth", "bearerAuth"]
+
+        return self.api_client.param_serialize(
+            method="GET",
+            resource_path="/api/v1/stable/tenants/{tenant}/events/keys",
+            path_params=_path_params,
+            query_params=_query_params,
+            header_params=_header_params,
+            body=_body_params,
+            post_params=_form_params,
+            files=_files,
+            auth_settings=_auth_settings,
+            collection_formats=_collection_formats,
+            _host=_host,
+            _request_auth=_request_auth,
+        )
+
     @validate_call
     def v1_event_list(
         self,
@@ -2575,6 +2839,34 @@ class EventApi:
             Optional[datetime],
             Field(description="Consider events that occurred before this time"),
         ] = None,
+        workflow_ids: Annotated[
+            Optional[
+                List[Annotated[str, Field(min_length=36, strict=True, max_length=36)]]
+            ],
+            Field(
+                description="Filter to events that are associated with a specific workflow run"
+            ),
+        ] = None,
+        workflow_run_statuses: Annotated[
+            Optional[List[V1TaskStatus]],
+            Field(
+                description="Filter to events that are associated with workflow runs matching a certain status"
+            ),
+        ] = None,
+        event_ids: Annotated[
+            Optional[
+                List[Annotated[str, Field(min_length=36, strict=True, max_length=36)]]
+            ],
+            Field(description="Filter to specific events by their ids"),
+        ] = None,
+        additional_metadata: Annotated[
+            Optional[List[Annotated[str, Field(min_length=1, strict=True)]]],
+            Field(description="Filter by additional metadata on the events"),
+        ] = None,
+        scopes: Annotated[
+            Optional[List[Annotated[str, Field(min_length=1, strict=True)]]],
+            Field(description="The scopes to filter by"),
+        ] = None,
         _request_timeout: Union[
             None,
             Annotated[StrictFloat, Field(gt=0)],
@@ -2603,6 +2895,16 @@ class EventApi:
         :type since: datetime
         :param until: Consider events that occurred before this time
         :type until: datetime
+        :param workflow_ids: Filter to events that are associated with a specific workflow run
+        :type workflow_ids: List[str]
+        :param workflow_run_statuses: Filter to events that are associated with workflow runs matching a certain status
+        :type workflow_run_statuses: List[V1TaskStatus]
+        :param event_ids: Filter to specific events by their ids
+        :type event_ids: List[str]
+        :param additional_metadata: Filter by additional metadata on the events
+        :type additional_metadata: List[str]
+        :param scopes: The scopes to filter by
+        :type scopes: List[str]
         :param _request_timeout: timeout setting for this request. If one
                                  number provided, it will be total request
                                  timeout. It can also be a pair (tuple) of
@@ -2632,6 +2934,11 @@ class EventApi:
             keys=keys,
             since=since,
             until=until,
+            workflow_ids=workflow_ids,
+            workflow_run_statuses=workflow_run_statuses,
+            event_ids=event_ids,
+            additional_metadata=additional_metadata,
+            scopes=scopes,
             _request_auth=_request_auth,
             _content_type=_content_type,
             _headers=_headers,
@@ -2678,6 +2985,34 @@ class EventApi:
             Optional[datetime],
             Field(description="Consider events that occurred before this time"),
         ] = None,
+        workflow_ids: Annotated[
+            Optional[
+                List[Annotated[str, Field(min_length=36, strict=True, max_length=36)]]
+            ],
+            Field(
+                description="Filter to events that are associated with a specific workflow run"
+            ),
+        ] = None,
+        workflow_run_statuses: Annotated[
+            Optional[List[V1TaskStatus]],
+            Field(
+                description="Filter to events that are associated with workflow runs matching a certain status"
+            ),
+        ] = None,
+        event_ids: Annotated[
+            Optional[
+                List[Annotated[str, Field(min_length=36, strict=True, max_length=36)]]
+            ],
+            Field(description="Filter to specific events by their ids"),
+        ] = None,
+        additional_metadata: Annotated[
+            Optional[List[Annotated[str, Field(min_length=1, strict=True)]]],
+            Field(description="Filter by additional metadata on the events"),
+        ] = None,
+        scopes: Annotated[
+            Optional[List[Annotated[str, Field(min_length=1, strict=True)]]],
+            Field(description="The scopes to filter by"),
+        ] = None,
         _request_timeout: Union[
             None,
             Annotated[StrictFloat, Field(gt=0)],
@@ -2706,6 +3041,16 @@ class EventApi:
         :type since: datetime
         :param until: Consider events that occurred before this time
         :type until: datetime
+        :param workflow_ids: Filter to events that are associated with a specific workflow run
+        :type workflow_ids: List[str]
+        :param workflow_run_statuses: Filter to events that are associated with workflow runs matching a certain status
+        :type workflow_run_statuses: List[V1TaskStatus]
+        :param event_ids: Filter to specific events by their ids
+        :type event_ids: List[str]
+        :param additional_metadata: Filter by additional metadata on the events
+        :type additional_metadata: List[str]
+        :param scopes: The scopes to filter by
+        :type scopes: List[str]
         :param _request_timeout: timeout setting for this request. If one
                                  number provided, it will be total request
                                  timeout. It can also be a pair (tuple) of
@@ -2735,6 +3080,11 @@ class EventApi:
             keys=keys,
             since=since,
             until=until,
+            workflow_ids=workflow_ids,
+            workflow_run_statuses=workflow_run_statuses,
+            event_ids=event_ids,
+            additional_metadata=additional_metadata,
+            scopes=scopes,
             _request_auth=_request_auth,
             _content_type=_content_type,
             _headers=_headers,
@@ -2781,6 +3131,34 @@ class EventApi:
             Optional[datetime],
             Field(description="Consider events that occurred before this time"),
         ] = None,
+        workflow_ids: Annotated[
+            Optional[
+                List[Annotated[str, Field(min_length=36, strict=True, max_length=36)]]
+            ],
+            Field(
+                description="Filter to events that are associated with a specific workflow run"
+            ),
+        ] = None,
+        workflow_run_statuses: Annotated[
+            Optional[List[V1TaskStatus]],
+            Field(
+                description="Filter to events that are associated with workflow runs matching a certain status"
+            ),
+        ] = None,
+        event_ids: Annotated[
+            Optional[
+                List[Annotated[str, Field(min_length=36, strict=True, max_length=36)]]
+            ],
+            Field(description="Filter to specific events by their ids"),
+        ] = None,
+        additional_metadata: Annotated[
+            Optional[List[Annotated[str, Field(min_length=1, strict=True)]]],
+            Field(description="Filter by additional metadata on the events"),
+        ] = None,
+        scopes: Annotated[
+            Optional[List[Annotated[str, Field(min_length=1, strict=True)]]],
+            Field(description="The scopes to filter by"),
+        ] = None,
         _request_timeout: Union[
             None,
             Annotated[StrictFloat, Field(gt=0)],
@@ -2809,6 +3187,16 @@ class EventApi:
         :type since: datetime
         :param until: Consider events that occurred before this time
         :type until: datetime
+        :param workflow_ids: Filter to events that are associated with a specific workflow run
+        :type workflow_ids: List[str]
+        :param workflow_run_statuses: Filter to events that are associated with workflow runs matching a certain status
+        :type workflow_run_statuses: List[V1TaskStatus]
+        :param event_ids: Filter to specific events by their ids
+        :type event_ids: List[str]
+        :param additional_metadata: Filter by additional metadata on the events
+        :type additional_metadata: List[str]
+        :param scopes: The scopes to filter by
+        :type scopes: List[str]
         :param _request_timeout: timeout setting for this request. If one
                                  number provided, it will be total request
                                  timeout. It can also be a pair (tuple) of
@@ -2838,6 +3226,11 @@ class EventApi:
             keys=keys,
             since=since,
             until=until,
+            workflow_ids=workflow_ids,
+            workflow_run_statuses=workflow_run_statuses,
+            event_ids=event_ids,
+            additional_metadata=additional_metadata,
+            scopes=scopes,
             _request_auth=_request_auth,
             _content_type=_content_type,
             _headers=_headers,
@@ -2862,6 +3255,11 @@ class EventApi:
         keys,
         since,
         until,
+        workflow_ids,
+        workflow_run_statuses,
+        event_ids,
+        additional_metadata,
+        scopes,
         _request_auth,
         _content_type,
         _headers,
@@ -2872,6 +3270,11 @@ class EventApi:
 
         _collection_formats: Dict[str, str] = {
             "keys": "multi",
+            "workflowIds": "multi",
+            "workflowRunStatuses": "multi",
+            "eventIds": "multi",
+            "additionalMetadata": "multi",
+            "scopes": "multi",
         }
 
         _path_params: Dict[str, str] = {}
@@ -2921,6 +3324,26 @@ class EventApi:
             else:
                 _query_params.append(("until", until))
 
+        if workflow_ids is not None:
+
+            _query_params.append(("workflowIds", workflow_ids))
+
+        if workflow_run_statuses is not None:
+
+            _query_params.append(("workflowRunStatuses", workflow_run_statuses))
+
+        if event_ids is not None:
+
+            _query_params.append(("eventIds", event_ids))
+
+        if additional_metadata is not None:
+
+            _query_params.append(("additionalMetadata", additional_metadata))
+
+        if scopes is not None:
+
+            _query_params.append(("scopes", scopes))
+
         # process the header parameters
         # process the form parameters
         # process the body parameter

{hatchet_sdk-1.12.0 → hatchet_sdk-1.12.1}/hatchet_sdk/clients/rest/models/__init__.py

@@ -201,6 +201,7 @@ from hatchet_sdk.clients.rest.models.v1_create_filter_request import (
 from hatchet_sdk.clients.rest.models.v1_dag_children import V1DagChildren
 from hatchet_sdk.clients.rest.models.v1_event import V1Event
 from hatchet_sdk.clients.rest.models.v1_event_list import V1EventList
+from hatchet_sdk.clients.rest.models.v1_event_triggered_run import V1EventTriggeredRun
 from hatchet_sdk.clients.rest.models.v1_event_workflow_run_summary import (
     V1EventWorkflowRunSummary,
 )

{hatchet_sdk-1.12.0 → hatchet_sdk-1.12.1}/hatchet_sdk/clients/rest/models/create_tenant_request.py

@@ -23,6 +23,7 @@ from pydantic import BaseModel, ConfigDict, Field, StrictStr
 from typing_extensions import Self
 
 from hatchet_sdk.clients.rest.models.tenant_ui_version import TenantUIVersion
+from hatchet_sdk.clients.rest.models.tenant_version import TenantVersion
 
 
 class CreateTenantRequest(BaseModel):
@@ -37,7 +38,12 @@ class CreateTenantRequest(BaseModel):
         description="The UI version of the tenant. Defaults to V0.",
         alias="uiVersion",
     )
-    __properties: ClassVar[List[str]] = ["name", "slug", "uiVersion"]
+    engine_version: Optional[TenantVersion] = Field(
+        default=None,
+        description="The engine version of the tenant. Defaults to V0.",
+        alias="engineVersion",
+    )
+    __properties: ClassVar[List[str]] = ["name", "slug", "uiVersion", "engineVersion"]
 
     model_config = ConfigDict(
         populate_by_name=True,
@@ -92,6 +98,7 @@ class CreateTenantRequest(BaseModel):
                 "name": obj.get("name"),
                 "slug": obj.get("slug"),
                 "uiVersion": obj.get("uiVersion"),
+                "engineVersion": obj.get("engineVersion"),
             }
         )
         return _obj