hatchet-sdk 1.0.2__tar.gz → 1.0.3__tar.gz

{hatchet_sdk-1.0.2 → hatchet_sdk-1.0.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: hatchet-sdk
-Version: 1.0.2
+Version: 1.0.3
 Summary: 
 Author: Alexander Belanger
 Author-email: alexander@hatchet.run

{hatchet_sdk-1.0.2 → hatchet_sdk-1.0.3}/hatchet_sdk/clients/admin.py

@@ -36,6 +36,7 @@ class ScheduleTriggerWorkflowOptions(BaseModel):
     child_index: int | None = None
     child_key: str | None = None
     namespace: str | None = None
+    additional_metadata: JSONSerializableMapping = Field(default_factory=dict)
 
 
 class TriggerWorkflowOptions(ScheduleTriggerWorkflowOptions):
@@ -153,7 +154,11 @@ class AdminClient:
             name=name,
             schedules=[self._parse_schedule(schedule) for schedule in schedules],
             input=json.dumps(input),
-            **options.model_dump(),
+            parent_id=options.parent_id,
+            parent_step_run_id=options.parent_step_run_id,
+            child_index=options.child_index,
+            child_key=options.child_key,
+            additional_metadata=json.dumps(options.additional_metadata),
         )
 
     @tenacity_retry

{hatchet_sdk-1.0.2 → hatchet_sdk-1.0.3}/hatchet_sdk/clients/dispatcher/action_listener.py

@@ -298,7 +298,9 @@ class ActionListener:
                             )
                         )
                     except (ValueError, json.JSONDecodeError) as e:
-                        raise ValueError(f"Error decoding payload: {e}")
+                        logger.error(f"Error decoding payload: {e}")
+
+                        action_payload = ActionPayload()
 
                     action = Action(
                         tenant_id=assigned_action.tenantId,

{hatchet_sdk-1.0.2 → hatchet_sdk-1.0.3}/hatchet_sdk/clients/run_event_listener.py

@@ -1,5 +1,4 @@
 import asyncio
-import json
 from enum import Enum
 from typing import Any, AsyncGenerator, Callable, Generator, cast
 
@@ -128,18 +127,10 @@ class RunEventListener:
                             raise Exception(
                                 f"Unknown event type: {workflow_event.eventType}"
                             )
-                        payload = None
 
-                        try:
-                            if workflow_event.eventPayload:
-                                payload = json.loads(workflow_event.eventPayload)
-                        except Exception:
-                            payload = workflow_event.eventPayload
-                            pass
-
-                        assert isinstance(payload, str)
-
-                        yield StepRunEvent(type=eventType, payload=payload)
+                        yield StepRunEvent(
+                            type=eventType, payload=workflow_event.eventPayload
+                        )
                     elif workflow_event.resourceType == RESOURCE_TYPE_WORKFLOW_RUN:
                         if workflow_event.eventType in step_run_event_type_mapping:
                             workflowRunEventType = step_run_event_type_mapping[
@@ -150,17 +141,10 @@ class RunEventListener:
                                 f"Unknown event type: {workflow_event.eventType}"
                             )
 
-                        payload = None
-
-                        try:
-                            if workflow_event.eventPayload:
-                                payload = json.loads(workflow_event.eventPayload)
-                        except Exception:
-                            pass
-
-                        assert isinstance(payload, str)
-
-                        yield StepRunEvent(type=workflowRunEventType, payload=payload)
+                        yield StepRunEvent(
+                            type=workflowRunEventType,
+                            payload=workflow_event.eventPayload,
+                        )
 
                     if workflow_event.hangup:
                         listener = None
@@ -236,9 +220,6 @@ class RunEventListenerClient:
         return self.stream(workflow_run_id)
 
     def stream(self, workflow_run_id: str) -> RunEventListener:
-        if not isinstance(workflow_run_id, str) and hasattr(workflow_run_id, "__str__"):
-            workflow_run_id = str(workflow_run_id)
-
         if not self.client:
             aio_conn = new_conn(self.config, True)
             self.client = DispatcherStub(aio_conn)  # type: ignore[no-untyped-call]

{hatchet_sdk-1.0.2 → hatchet_sdk-1.0.3}/hatchet_sdk/context/context.py

@@ -113,6 +113,14 @@ class Context:
 
         return parent_step_data
 
+    def aio_task_output(self, task: "Task[TWorkflowInput, R]") -> "R":
+        if task.is_async_function:
+            return self.task_output(task)
+
+        raise ValueError(
+            f"Task '{task.name}' is not an async function. Use `task_output` instead."
+        )
+
     @property
     def was_triggered_by_event(self) -> bool:
         return self.data.triggered_by == "event"
@@ -157,6 +165,7 @@ class Context:
 
         def handle_result(future: Future[tuple[bool, Exception | None]]) -> None:
             success, exception = future.result()
+
             if not success and exception:
                 if raise_on_error:
                     raise exception

{hatchet_sdk-1.0.2 → hatchet_sdk-1.0.3}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "hatchet-sdk"
-version = "1.0.2"
+version = "1.0.3"
 description = ""
 authors = ["Alexander Belanger <alexander@hatchet.run>"]
 readme = "README.md"
@@ -155,3 +155,4 @@ v2_simple = "examples.v2.simple.worker:main"
 otel = "examples.opentelemetry_instrumentation.worker:main"
 waits = "examples.waits.worker:main"
 durable = "examples.durable.worker:main"
+streaming = "examples.streaming.worker:main"