fix: linting

Author: dbschmigelski

src/strands/multiagent/swarm.py

@@ -563,7 +563,9 @@ async def _execute_swarm(self, invocation_state: dict[str, Any]) -> None:
             f"{elapsed_time:.2f}",
         )
 
-    async def _execute_node(self, node: SwarmNode, task: str | list[ContentBlock], invocation_state: dict[str, Any]) -> AgentResult:
+    async def _execute_node(
+        self, node: SwarmNode, task: str | list[ContentBlock], invocation_state: dict[str, Any]
+    ) -> AgentResult:
         """Execute swarm node."""
         start_time = time.time()
         node_name = node.node_id

tests/strands/agent/test_agent.py

@@ -18,7 +18,6 @@
 from strands.handlers.callback_handler import PrintingCallbackHandler, null_callback_handler
 from strands.models.bedrock import DEFAULT_BEDROCK_MODEL_ID, BedrockModel
 from strands.session.repository_session_manager import RepositorySessionManager
-from strands.telemetry.tracer import serialize
 from strands.types._events import EventLoopStopEvent, ModelStreamEvent
 from strands.types.content import Messages
 from strands.types.exceptions import ContextWindowOverflowException, EventLoopException

tests/strands/multiagent/test_graph.py

@@ -1088,30 +1088,31 @@ async def test_state_reset_only_with_cycles_enabled():
         # With reset_on_revisit enabled, reset should be called
         mock_reset.assert_called_once()
 
+
 @pytest.mark.asyncio
 async def test_graph_kwargs_passing_agent(mock_strands_tracer, mock_use_span):
     """Test that kwargs are passed through to underlying Agent nodes."""
     # Create a mock agent that captures kwargs
     kwargs_agent = create_mock_agent("kwargs_agent", "Response with kwargs")
-    
+
     async def capture_kwargs(*args, **kwargs):
         # Store kwargs for verification
         capture_kwargs.captured_kwargs = kwargs
         return kwargs_agent.return_value
-    
+
     kwargs_agent.invoke_async = MagicMock(side_effect=capture_kwargs)
-    
+
     # Create graph
     builder = GraphBuilder()
     builder.add_node(kwargs_agent, "kwargs_node")
     graph = builder.build()
-    
+
     # Execute with custom kwargs
     test_kwargs = {"custom_param": "test_value", "another_param": 42}
     result = await graph.invoke_async("Test kwargs passing", **test_kwargs)
-    
+
     # Verify kwargs were passed to agent
-    assert hasattr(capture_kwargs, 'captured_kwargs')
+    assert hasattr(capture_kwargs, "captured_kwargs")
     assert capture_kwargs.captured_kwargs == test_kwargs
     assert result.status == Status.COMPLETED
 
@@ -1121,28 +1122,28 @@ async def test_graph_kwargs_passing_multiagent(mock_strands_tracer, mock_use_spa
     """Test that kwargs are passed through to underlying MultiAgentBase nodes."""
     # Create a mock MultiAgentBase that captures kwargs
     kwargs_multiagent = create_mock_multi_agent("kwargs_multiagent", "MultiAgent response with kwargs")
-    
+
     # Store the original return value
     original_result = kwargs_multiagent.invoke_async.return_value
-    
+
     async def capture_kwargs(*args, **kwargs):
         # Store kwargs for verification
         capture_kwargs.captured_kwargs = kwargs
         return original_result
-    
+
     kwargs_multiagent.invoke_async = AsyncMock(side_effect=capture_kwargs)
-    
+
     # Create graph
     builder = GraphBuilder()
     builder.add_node(kwargs_multiagent, "multiagent_node")
     graph = builder.build()
-    
+
     # Execute with custom kwargs
     test_kwargs = {"custom_param": "test_value", "another_param": 42}
     result = await graph.invoke_async("Test kwargs passing to multiagent", **test_kwargs)
-    
+
     # Verify kwargs were passed to multiagent
-    assert hasattr(capture_kwargs, 'captured_kwargs')
+    assert hasattr(capture_kwargs, "captured_kwargs")
     assert capture_kwargs.captured_kwargs == test_kwargs
     assert result.status == Status.COMPLETED
 
@@ -1151,24 +1152,24 @@ def test_graph_kwargs_passing_sync(mock_strands_tracer, mock_use_span):
     """Test that kwargs are passed through to underlying nodes in sync execution."""
     # Create a mock agent that captures kwargs
     kwargs_agent = create_mock_agent("kwargs_agent", "Response with kwargs")
-    
+
     async def capture_kwargs(*args, **kwargs):
         # Store kwargs for verification
         capture_kwargs.captured_kwargs = kwargs
         return kwargs_agent.return_value
-    
+
     kwargs_agent.invoke_async = MagicMock(side_effect=capture_kwargs)
-    
+
     # Create graph
     builder = GraphBuilder()
     builder.add_node(kwargs_agent, "kwargs_node")
     graph = builder.build()
-    
+
     # Execute with custom kwargs
     test_kwargs = {"custom_param": "test_value", "another_param": 42}
     result = graph("Test kwargs passing sync", **test_kwargs)
-    
+
     # Verify kwargs were passed to agent
-    assert hasattr(capture_kwargs, 'captured_kwargs')
+    assert hasattr(capture_kwargs, "captured_kwargs")
     assert capture_kwargs.captured_kwargs == test_kwargs
-    assert result.status == Status.COMPLETED
+    assert result.status == Status.COMPLETED

tests/strands/multiagent/test_swarm.py

@@ -470,42 +470,29 @@ def test_swarm_validate_unsupported_features():
     with pytest.raises(ValueError, match="Session persistence is not supported for Swarm agents yet"):
         Swarm([agent_with_session])
 
-    # Test with callbacks (should fail)
-    class TestHookProvider(HookProvider):
-        def register_hooks(self, registry, **kwargs):
-            registry.add_callback(AgentInitializedEvent, lambda e: None)
-
-    agent_with_hooks = create_mock_agent("agent_with_hooks")
-    agent_with_hooks._session_manager = None
-    agent_with_hooks.hooks = HookRegistry()
-    agent_with_hooks.hooks.add_hook(TestHookProvider())
-
-    with pytest.raises(ValueError, match="Agent callbacks are not supported for Swarm agents yet"):
-        Swarm([agent_with_hooks])
-
 
 @pytest.mark.asyncio
 async def test_swarm_kwargs_passing(mock_strands_tracer, mock_use_span):
     """Test that kwargs are passed through to underlying agents."""
     # Create a mock agent that captures kwargs
     kwargs_agent = create_mock_agent("kwargs_agent", "Response with kwargs")
-    
+
     async def capture_kwargs(*args, **kwargs):
         # Store kwargs for verification
         capture_kwargs.captured_kwargs = kwargs
         return kwargs_agent.return_value
-    
+
     kwargs_agent.invoke_async = MagicMock(side_effect=capture_kwargs)
-    
+
     # Create swarm
     swarm = Swarm(nodes=[kwargs_agent])
-    
+
     # Execute with custom kwargs
     test_kwargs = {"custom_param": "test_value", "another_param": 42}
     result = await swarm.invoke_async("Test kwargs passing", **test_kwargs)
-    
+
     # Verify kwargs were passed to agent
-    assert hasattr(capture_kwargs, 'captured_kwargs')
+    assert hasattr(capture_kwargs, "captured_kwargs")
     assert capture_kwargs.captured_kwargs == test_kwargs
     assert result.status == Status.COMPLETED
 
@@ -514,22 +501,22 @@ def test_swarm_kwargs_passing_sync(mock_strands_tracer, mock_use_span):
     """Test that kwargs are passed through to underlying agents in sync execution."""
     # Create a mock agent that captures kwargs
     kwargs_agent = create_mock_agent("kwargs_agent", "Response with kwargs")
-    
+
     async def capture_kwargs(*args, **kwargs):
         # Store kwargs for verification
         capture_kwargs.captured_kwargs = kwargs
         return kwargs_agent.return_value
-    
+
     kwargs_agent.invoke_async = MagicMock(side_effect=capture_kwargs)
-    
+
     # Create swarm
     swarm = Swarm(nodes=[kwargs_agent])
-    
+
     # Execute with custom kwargs
     test_kwargs = {"custom_param": "test_value", "another_param": 42}
     result = swarm("Test kwargs passing sync", **test_kwargs)
-    
+
     # Verify kwargs were passed to agent
-    assert hasattr(capture_kwargs, 'captured_kwargs')
+    assert hasattr(capture_kwargs, "captured_kwargs")
     assert capture_kwargs.captured_kwargs == test_kwargs
     assert result.status == Status.COMPLETED