style: appease linter

Signed-off-by: Samantha Coyle <sam@diagrid.io>

Author: sicoyle

ext/dapr-ext-workflow/dapr/ext/workflow/aio/mcp.py

@@ -104,14 +104,10 @@ async def connect(self, mcpserver_name: str) -> None:
             except Exception as exc:  # noqa: BLE001 — classified by helper
                 if not _is_transient_schedule_error(exc):
                     raise
-                sleep_for = min(
-                    _SCHEDULE_RETRY_INTERVAL_SECONDS, deadline - time.monotonic()
-                )
+                sleep_for = min(_SCHEDULE_RETRY_INTERVAL_SECONDS, deadline - time.monotonic())
                 if sleep_for <= 0:
                     raise
-                logger.debug(
-                    'schedule_new_workflow returned transient error %s; retrying', exc
-                )
+                logger.debug('schedule_new_workflow returned transient error %s; retrying', exc)
                 await asyncio.sleep(sleep_for)
 
         remaining = deadline - time.monotonic()

ext/dapr-ext-workflow/dapr/ext/workflow/mcp.py

@@ -39,7 +39,6 @@
 from typing import Any, Dict, List, Optional, Set
 
 import grpc
-
 from dapr.ext.workflow.dapr_workflow_client import DaprWorkflowClient
 from dapr.ext.workflow.workflow_state import WorkflowStatus
 
@@ -53,10 +52,12 @@
 _MCP_METHOD_LIST_TOOLS = '.ListTools'
 _MCP_METHOD_CALL_TOOL = '.CallTool'
 
-_TRANSIENT_GRPC_CODES = frozenset({
-    grpc.StatusCode.CANCELLED,
-    grpc.StatusCode.UNAVAILABLE,
-})
+_TRANSIENT_GRPC_CODES = frozenset(
+    {
+        grpc.StatusCode.CANCELLED,
+        grpc.StatusCode.UNAVAILABLE,
+    }
+)
 _SCHEDULE_RETRY_INTERVAL_SECONDS = 0.5
 
 
@@ -247,14 +248,10 @@ def connect(self, mcpserver_name: str) -> None:
             except Exception as exc:  # noqa: BLE001 — classified by helper
                 if not _is_transient_schedule_error(exc):
                     raise
-                sleep_for = min(
-                    _SCHEDULE_RETRY_INTERVAL_SECONDS, deadline - time.monotonic()
-                )
+                sleep_for = min(_SCHEDULE_RETRY_INTERVAL_SECONDS, deadline - time.monotonic())
                 if sleep_for <= 0:
                     raise
-                logger.debug(
-                    'schedule_new_workflow returned transient error %s; retrying', exc
-                )
+                logger.debug('schedule_new_workflow returned transient error %s; retrying', exc)
                 time.sleep(sleep_for)
 
         remaining = deadline - time.monotonic()

ext/dapr-ext-workflow/tests/test_mcp_client.py

@@ -19,7 +19,6 @@
 from unittest.mock import AsyncMock, MagicMock, patch
 
 import grpc
-
 from dapr.ext.workflow._durabletask import client
 from dapr.ext.workflow.aio.mcp import DaprMCPClient as AioDaprMCPClient
 from dapr.ext.workflow.mcp import MCP_WORKFLOW_PREFIX, DaprMCPClient, MCPToolDef
@@ -440,9 +439,7 @@ def test_retries_on_unavailable(self):
     def test_non_transient_propagates_immediately(self):
         """A non-CANCELLED/UNAVAILABLE error must not be retried."""
         mock_wf = MagicMock()
-        mock_wf.schedule_new_workflow.side_effect = _StubRpcError(
-            grpc.StatusCode.PERMISSION_DENIED
-        )
+        mock_wf.schedule_new_workflow.side_effect = _StubRpcError(grpc.StatusCode.PERMISSION_DENIED)
 
         mcp_client = DaprMCPClient(timeout_in_seconds=30, wf_client=mock_wf)
         with patch('dapr.ext.workflow.mcp.time.sleep') as sleep_mock:
@@ -455,16 +452,17 @@ def test_non_transient_propagates_immediately(self):
     def test_deadline_exhausted_raises_last_error(self):
         """When the timeout budget runs out mid-retry, propagate the last error."""
         mock_wf = MagicMock()
-        mock_wf.schedule_new_workflow.side_effect = _StubRpcError(
-            grpc.StatusCode.CANCELLED
-        )
+        mock_wf.schedule_new_workflow.side_effect = _StubRpcError(grpc.StatusCode.CANCELLED)
 
         mcp_client = DaprMCPClient(timeout_in_seconds=1, wf_client=mock_wf)
         # Patch monotonic to advance past the deadline immediately so we don't
         # actually sleep for a second in tests.
-        with patch('dapr.ext.workflow.mcp.time.sleep'), patch(
-            'dapr.ext.workflow.mcp.time.monotonic',
-            side_effect=[0.0, 2.0],
+        with (
+            patch('dapr.ext.workflow.mcp.time.sleep'),
+            patch(
+                'dapr.ext.workflow.mcp.time.monotonic',
+                side_effect=[0.0, 2.0],
+            ),
         ):
             with self.assertRaises(grpc.RpcError):
                 mcp_client.connect('weather')
@@ -482,9 +480,12 @@ def test_budget_exhausted_after_schedule_succeeds(self):
         mcp_client = DaprMCPClient(timeout_in_seconds=1, wf_client=mock_wf)
         # monotonic: 0.0 → deadline = 1.0; 0.4 → sleep_for = 0.5 (still in budget);
         # 2.0 → post-loop remaining = -1.0 → raise.
-        with patch('dapr.ext.workflow.mcp.time.sleep'), patch(
-            'dapr.ext.workflow.mcp.time.monotonic',
-            side_effect=[0.0, 0.4, 2.0],
+        with (
+            patch('dapr.ext.workflow.mcp.time.sleep'),
+            patch(
+                'dapr.ext.workflow.mcp.time.monotonic',
+                side_effect=[0.0, 0.4, 2.0],
+            ),
         ):
             with self.assertRaises(RuntimeError) as ctx:
                 mcp_client.connect('weather')
@@ -514,14 +515,15 @@ async def test_retries_then_succeeds_on_cancelled(self):
 
     async def test_deadline_exhausted_raises(self):
         mock_wf = AsyncMock()
-        mock_wf.schedule_new_workflow.side_effect = _StubRpcError(
-            grpc.StatusCode.CANCELLED
-        )
+        mock_wf.schedule_new_workflow.side_effect = _StubRpcError(grpc.StatusCode.CANCELLED)
 
         mcp_client = AioDaprMCPClient(timeout_in_seconds=1, wf_client=mock_wf)
-        with patch('dapr.ext.workflow.aio.mcp.asyncio.sleep', new=AsyncMock()), patch(
-            'dapr.ext.workflow.aio.mcp.time.monotonic',
-            side_effect=[0.0, 2.0],
+        with (
+            patch('dapr.ext.workflow.aio.mcp.asyncio.sleep', new=AsyncMock()),
+            patch(
+                'dapr.ext.workflow.aio.mcp.time.monotonic',
+                side_effect=[0.0, 2.0],
+            ),
         ):
             with self.assertRaises(grpc.RpcError):
                 await mcp_client.connect('weather')
@@ -535,9 +537,12 @@ async def test_budget_exhausted_after_schedule_succeeds(self):
         ]
 
         mcp_client = AioDaprMCPClient(timeout_in_seconds=1, wf_client=mock_wf)
-        with patch('dapr.ext.workflow.aio.mcp.asyncio.sleep', new=AsyncMock()), patch(
-            'dapr.ext.workflow.aio.mcp.time.monotonic',
-            side_effect=[0.0, 0.4, 2.0],
+        with (
+            patch('dapr.ext.workflow.aio.mcp.asyncio.sleep', new=AsyncMock()),
+            patch(
+                'dapr.ext.workflow.aio.mcp.time.monotonic',
+                side_effect=[0.0, 0.4, 2.0],
+            ),
         ):
             with self.assertRaises(RuntimeError) as ctx:
                 await mcp_client.connect('weather')