durable-workflow 0.4.95__tar.gz → 0.4.97__tar.gz

{durable_workflow-0.4.95/src/durable_workflow.egg-info → durable_workflow-0.4.97}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: durable-workflow
-Version: 0.4.95
+Version: 0.4.97
 Summary: Python SDK for the Durable Workflow server (language-neutral HTTP protocol)
 Author: Durable Workflow Contributors
 License-Expression: MIT

{durable_workflow-0.4.95 → durable_workflow-0.4.97}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "durable-workflow"
-version = "0.4.95"
+version = "0.4.97"
 description = "Python SDK for the Durable Workflow server (language-neutral HTTP protocol)"
 readme = "README.md"
 requires-python = ">=3.10"

{durable_workflow-0.4.95 → durable_workflow-0.4.97}/src/durable_workflow/client.py

@@ -164,6 +164,8 @@ class WorkflowExecution:
     input: Any = None
     output: Any = None
     payload_codec: str | None = None
+    memo: dict[str, Any] | None = None
+    search_attributes: dict[str, Any] | None = None
 
 
 @dataclass
@@ -2386,6 +2388,10 @@ class Client:
             input=input_val,
             output=output_val,
             payload_codec=data.get("payload_codec"),
+            memo=data.get("memo") if isinstance(data.get("memo"), dict) else None,
+            search_attributes=(
+                data.get("search_attributes") if isinstance(data.get("search_attributes"), dict) else None
+            ),
         )
 
     async def list_workflows(
@@ -2425,6 +2431,13 @@ class Client:
                 run_id=item.get("run_id"),
                 workflow_type=item.get("workflow_type", ""),
                 status=item.get("status"),
+                namespace=item.get("namespace"),
+                task_queue=item.get("task_queue"),
+                payload_codec=item.get("payload_codec"),
+                memo=item.get("memo") if isinstance(item.get("memo"), dict) else None,
+                search_attributes=(
+                    item.get("search_attributes") if isinstance(item.get("search_attributes"), dict) else None
+                ),
             )
             for item in items
         ]

{durable_workflow-0.4.95 → durable_workflow-0.4.97}/src/durable_workflow/workflow.py

@@ -2270,10 +2270,12 @@ def _child_workflow_failed_from_payload(
         message = _optional_str(exception.get("message"))
 
     exception_class = _optional_str(payload.get("exception_class"))
+    if exception_class is None and exception is not None:
+        exception_class = _optional_str(exception.get("class"))
     if exception_class is None:
         exception_class = _optional_str(payload.get("exception_type"))
     if exception_class is None and exception is not None:
-        exception_class = _optional_str(exception.get("class")) or _optional_str(exception.get("type"))
+        exception_class = _optional_str(exception.get("type"))
 
     child_workflow_run_id = _optional_str(payload.get("child_workflow_run_id"))
     child_workflow_type = _optional_str(payload.get("child_workflow_type"))
@@ -2823,7 +2825,12 @@ def _replay_state(
     result_cursor = 0
     pending_step_cursor = 0
     wait_yield_count = 0
-    gen = instance.run(ctx, *start_input)
+    try:
+        gen = instance.run(ctx, *start_input)
+    except NonDeterministicReplayError:
+        raise
+    except Exception as exc:
+        return _state([_fail_workflow_from_exception(exc)])
     if not hasattr(gen, "__next__"):
         if isinstance(gen, ContinueAsNew):
             _assert_no_unconsumed_history("continue as new")

{durable_workflow-0.4.95 → durable_workflow-0.4.97/src/durable_workflow.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: durable-workflow
-Version: 0.4.95
+Version: 0.4.97
 Summary: Python SDK for the Durable Workflow server (language-neutral HTTP protocol)
 Author: Durable Workflow Contributors
 License-Expression: MIT

{durable_workflow-0.4.95 → durable_workflow-0.4.97}/tests/test_client.py

@@ -464,6 +464,8 @@ class TestDescribeWorkflow:
         assert desc.status == response["status"]
         assert desc.payload_codec == response["payload_codec"]
         assert desc.input == response["input"]
+        assert desc.memo == response["memo"]
+        assert desc.search_attributes == response["search_attributes"]
 
     @pytest.mark.asyncio
     async def test_envelope_fields(self, client: Client) -> None:
@@ -1131,7 +1133,23 @@ class TestListWorkflows:
     async def test_list(self, client: Client) -> None:
         resp = _mock_response(200, {
             "workflows": [
-                {"workflow_id": "wf-1", "run_id": "r1", "workflow_type": "greeter", "status": "running"},
+                {
+                    "workflow_id": "wf-1",
+                    "run_id": "r1",
+                    "workflow_type": "greeter",
+                    "status": "running",
+                    "namespace": "ns1",
+                    "task_queue": "q1",
+                    "payload_codec": "avro",
+                    "memo": {"source": "php-writer"},
+                    "search_attributes": {
+                        "customer_id": "cust-php-1",
+                        "order_total_cents": 9200,
+                        "discount_ratio": 0.125,
+                        "is_vip": True,
+                        "tags": ["urgent", "renewal"],
+                    },
+                },
                 {"workflow_id": "wf-2", "run_id": "r2", "workflow_type": "greeter", "status": "completed"},
             ],
             "next_page_token": "abc",
@@ -1141,6 +1159,17 @@ class TestListWorkflows:
             assert len(result.executions) == 2
             assert result.next_page_token == "abc"
             assert result.executions[0].workflow_id == "wf-1"
+            assert result.executions[0].namespace == "ns1"
+            assert result.executions[0].task_queue == "q1"
+            assert result.executions[0].payload_codec == "avro"
+            assert result.executions[0].memo == {"source": "php-writer"}
+            assert result.executions[0].search_attributes == {
+                "customer_id": "cust-php-1",
+                "order_total_cents": 9200,
+                "discount_ratio": 0.125,
+                "is_vip": True,
+                "tags": ["urgent", "renewal"],
+            }
 
     @pytest.mark.asyncio
     async def test_list_request_matches_polyglot_fixture(self, client: Client) -> None:

{durable_workflow-0.4.95 → durable_workflow-0.4.97}/tests/test_replay.py

@@ -155,6 +155,18 @@ class FailingWorkflow:
         raise ValueError("something went wrong")
 
 
+@workflow.defn(name="construction-failing-workflow")
+class ConstructionFailingWorkflow:
+    def run(self, ctx: WorkflowContext):  # type: ignore[no-untyped-def]
+        raise RuntimeError("failed before first yield")
+
+
+@workflow.defn(name="construction-stop-iteration-workflow")
+class ConstructionStopIterationWorkflow:
+    def run(self, ctx: WorkflowContext):  # type: ignore[no-untyped-def]
+        raise StopIteration("stopped before first yield")
+
+
 @workflow.defn(name="typed-compensation-failure-workflow")
 class TypedCompensationFailureWorkflow:
     def run(self, ctx: WorkflowContext):  # type: ignore[no-untyped-def]
@@ -848,6 +860,24 @@ class TestFailingWorkflow:
         assert cmd.exception_type == "ValueError"
         assert cmd.exception_class == "builtins.ValueError"
 
+    def test_non_generator_exception_produces_fail_command(self) -> None:
+        outcome = replay(ConstructionFailingWorkflow, [], [])
+        assert len(outcome.commands) == 1
+        cmd = outcome.commands[0]
+        assert isinstance(cmd, FailWorkflow)
+        assert "failed before first yield" in cmd.message
+        assert cmd.exception_type == "RuntimeError"
+        assert cmd.exception_class == "builtins.RuntimeError"
+
+    def test_non_generator_stop_iteration_is_not_completion(self) -> None:
+        outcome = replay(ConstructionStopIterationWorkflow, [], [])
+        assert len(outcome.commands) == 1
+        cmd = outcome.commands[0]
+        assert isinstance(cmd, FailWorkflow)
+        assert "stopped before first yield" in cmd.message
+        assert cmd.exception_type == "StopIteration"
+        assert cmd.exception_class == "builtins.StopIteration"
+
     def test_fail_server_command_shape(self) -> None:
         history = [{"event_type": "ActivityCompleted", "payload": {"result": '"ok"'}}]
         outcome = replay(FailingWorkflow, history, [])
@@ -1363,6 +1393,29 @@ class TestChildWorkflow:
             "child_workflow_type": "sub-workflow",
         }
 
+    def test_child_failed_prefers_nested_exception_class_over_type(self) -> None:
+        history = [
+            {
+                "event_type": "ChildRunFailed",
+                "payload": {
+                    "message": "child failed",
+                    "exception_type": "NestedType",
+                    "exception": {
+                        "type": "NestedType",
+                        "class": "pkg.NestedClass",
+                        "message": "child failed",
+                    },
+                },
+            },
+        ]
+
+        outcome = replay(ChildWorkflowFailureDetailsWf, history, [])
+
+        assert len(outcome.commands) == 1
+        cmd = outcome.commands[0]
+        assert isinstance(cmd, CompleteWorkflow)
+        assert cmd.result["exception_class"] == "pkg.NestedClass"
+
     def test_child_cancelled_is_observed_by_parent_as_typed_cancellation(self) -> None:
         history = [
             {

{durable_workflow-0.4.95 → durable_workflow-0.4.97}/tests/test_sync.py

@@ -156,11 +156,23 @@ class TestSyncClientDescribe:
                 "run_id": "run-1",
                 "workflow_type": "greeter",
                 "status": "running",
+                "search_attributes": {
+                    "customer_id": "cust-php-sync",
+                    "order_total_cents": 6100,
+                    "is_vip": False,
+                    "tags": ["standard", "renewal"],
+                },
             },
         )
         with patch.object(client._async._http, "request", new_callable=AsyncMock, return_value=resp):
             desc = client.describe_workflow("wf-1")
             assert desc.status == "running"
+            assert desc.search_attributes == {
+                "customer_id": "cust-php-sync",
+                "order_total_cents": 6100,
+                "is_vip": False,
+                "tags": ["standard", "renewal"],
+            }
 
 
 class TestSyncClientSignal:
@@ -202,12 +214,21 @@ class TestSyncClientList:
         resp = _mock_response(
             200,
             {
-                "workflows": [{"workflow_id": "wf-1", "run_id": "r1", "workflow_type": "g", "status": "running"}],
+                "workflows": [
+                    {
+                        "workflow_id": "wf-1",
+                        "run_id": "r1",
+                        "workflow_type": "g",
+                        "status": "running",
+                        "search_attributes": {"customer_id": "cust-php-sync"},
+                    }
+                ],
             },
         )
         with patch.object(client._async._http, "request", new_callable=AsyncMock, return_value=resp):
             result = client.list_workflows(workflow_type="g")
             assert len(result.executions) == 1
+            assert result.executions[0].search_attributes == {"customer_id": "cust-php-sync"}
 
     def test_list_task_queues(self) -> None:
         client = Client("http://localhost:8080")