diff --git a/src/durable_workflow/worker.py b/src/durable_workflow/worker.py index 577d89e..1eacfb5 100644 --- a/src/durable_workflow/worker.py +++ b/src/durable_workflow/worker.py @@ -137,6 +137,7 @@ TYPED_SEARCH_ATTRIBUTES_CAPABILITY, DURABLE_SELECTION_CAPABILITY, LOCAL_ACTIVITIES_CAPABILITY, + "workflow_entry_arguments_validation", ) _WORKFLOW_WORK_TASK_KINDS = ("workflow", "update_validation") diff --git a/tests/test_worker.py b/tests/test_worker.py index eda7573..daf6bd9 100644 --- a/tests/test_worker.py +++ b/tests/test_worker.py @@ -430,6 +430,7 @@ async def test_register(self, mock_client: AsyncMock) -> None: "typed_search_attributes", DURABLE_SELECTION_CAPABILITY, "local_activities", + "workflow_entry_arguments_validation", "query_tasks", MESSAGE_STREAMS_CAPABILITY, "worker_sessions", @@ -577,6 +578,7 @@ def run(self, ctx): # type: ignore[no-untyped-def] TYPED_SEARCH_ATTRIBUTES_CAPABILITY, DURABLE_SELECTION_CAPABILITY, "local_activities", + "workflow_entry_arguments_validation", "query_tasks", WORKFLOW_UPDATES_CAPABILITY, MESSAGE_STREAMS_CAPABILITY, @@ -664,6 +666,7 @@ def run(self, ctx): # type: ignore[no-untyped-def] TYPED_SEARCH_ATTRIBUTES_CAPABILITY, DURABLE_SELECTION_CAPABILITY, "local_activities", + "workflow_entry_arguments_validation", "query_tasks", UPDATE_VALIDATION_TASKS_CAPABILITY, WORKFLOW_UPDATES_CAPABILITY, @@ -774,6 +777,7 @@ async def test_register_keeps_baseline_capabilities_when_server_does_not_support TYPED_SEARCH_ATTRIBUTES_CAPABILITY, DURABLE_SELECTION_CAPABILITY, "local_activities", + "workflow_entry_arguments_validation", MESSAGE_STREAMS_CAPABILITY, "worker_sessions", ]