Expose downstream task queue budget-group admission in Python SDK

Expose budget group admission fields in task queue admission responses.

Author: durable-workflow-ops

README.md

@@ -174,7 +174,8 @@ for queue in queues.task_queues:
 The workflow and activity admission objects expose both queue-level and
 namespace-level server budgets, including active lease caps and per-minute
 dispatch-rate limits, so automation can detect whether local worker slots,
-queue caps, or namespace caps are constraining throughput.
+queue caps, namespace caps, or downstream dispatch budget groups are
+constraining throughput.
 
 ## Replay captured histories
 

src/durable_workflow/client.py

@@ -127,6 +127,10 @@ class TaskQueueTaskAdmission:
     server_max_dispatches_per_minute_per_namespace: int | None = None
     server_namespace_dispatch_count_this_minute: int | None = None
     server_remaining_namespace_dispatch_capacity: int | None = None
+    server_dispatch_budget_group: str | None = None
+    server_max_dispatches_per_minute_per_budget_group: int | None = None
+    server_budget_group_dispatch_count_this_minute: int | None = None
+    server_remaining_budget_group_dispatch_capacity: int | None = None
     server_lock_required: bool | None = None
     server_lock_supported: bool | None = None
 
@@ -159,6 +163,16 @@ def from_dict(cls, data: dict[str, Any] | None) -> TaskQueueTaskAdmission | None
             ),
             server_namespace_dispatch_count_this_minute=data.get("server_namespace_dispatch_count_this_minute"),
             server_remaining_namespace_dispatch_capacity=data.get("server_remaining_namespace_dispatch_capacity"),
+            server_dispatch_budget_group=data.get("server_dispatch_budget_group"),
+            server_max_dispatches_per_minute_per_budget_group=data.get(
+                "server_max_dispatches_per_minute_per_budget_group"
+            ),
+            server_budget_group_dispatch_count_this_minute=data.get(
+                "server_budget_group_dispatch_count_this_minute"
+            ),
+            server_remaining_budget_group_dispatch_capacity=data.get(
+                "server_remaining_budget_group_dispatch_capacity"
+            ),
             server_lock_required=data.get("server_lock_required"),
             server_lock_supported=data.get("server_lock_supported"),
         )

tests/test_client.py

@@ -459,6 +459,10 @@ async def test_list_task_queues_parses_admission(self, client: Client) -> None:
                             "server_max_dispatches_per_minute_per_namespace": 240,
                             "server_namespace_dispatch_count_this_minute": 200,
                             "server_remaining_namespace_dispatch_capacity": 40,
+                            "server_dispatch_budget_group": "downstream-openai",
+                            "server_max_dispatches_per_minute_per_budget_group": 75,
+                            "server_budget_group_dispatch_count_this_minute": 75,
+                            "server_remaining_budget_group_dispatch_capacity": 0,
                             "server_lock_required": True,
                             "server_lock_supported": True,
                         },
@@ -497,6 +501,10 @@ async def test_list_task_queues_parses_admission(self, client: Client) -> None:
             assert queue.admission.workflow_tasks.server_max_dispatches_per_minute_per_namespace == 240
             assert queue.admission.workflow_tasks.server_namespace_dispatch_count_this_minute == 200
             assert queue.admission.workflow_tasks.server_remaining_namespace_dispatch_capacity == 40
+            assert queue.admission.workflow_tasks.server_dispatch_budget_group == "downstream-openai"
+            assert queue.admission.workflow_tasks.server_max_dispatches_per_minute_per_budget_group == 75
+            assert queue.admission.workflow_tasks.server_budget_group_dispatch_count_this_minute == 75
+            assert queue.admission.workflow_tasks.server_remaining_budget_group_dispatch_capacity == 0
             assert queue.admission.activity_tasks is not None
             assert queue.admission.activity_tasks.configured_slot_count == 5
             assert queue.admission.query_tasks is not None