diff --git a/src/strands_tools/handoff_to_user.py b/src/strands_tools/handoff_to_user.py index 9fa3b6ad..8a1a89d2 100644 --- a/src/strands_tools/handoff_to_user.py +++ b/src/strands_tools/handoff_to_user.py @@ -179,7 +179,9 @@ def handoff_to_user(tool: ToolUse, **kwargs: Any) -> ToolResult: else: # Wait for user input and continue try: - user_response = get_user_input(f"Agent requested user input: {message}\nYour response: ").strip() + user_response = get_user_input( + f"Agent requested user input: {message}\nYour response: " + ).strip() console.print() diff --git a/tests/test_handoff_to_user.py b/tests/test_handoff_to_user.py index 7dc627fd..d8c1ab0c 100644 --- a/tests/test_handoff_to_user.py +++ b/tests/test_handoff_to_user.py @@ -63,7 +63,9 @@ def test_handoff_with_breakout_false_direct(mock_get_user_input, mock_request_st result = handoff_to_user.handoff_to_user(tool=tool_use, request_state=mock_request_state) # Verify get_user_input was called - mock_get_user_input.assert_called_once_with("Agent requested user input: Please confirm the action.\nYour response: ") + mock_get_user_input.assert_called_once_with( + "Agent requested user input: Please confirm the action.\nYour response: " + ) # Verify the result has the expected structure assert result["toolUseId"] == "test-tool-use-id"