feat: add max tokens reached test

Author: dbschmigelski

src/strands/agent/conversation_manager/recover_tool_use_on_max_tokens_reached.py

@@ -32,6 +32,7 @@ async def recover_tool_use_on_max_tokens_reached(agent: "Agent", exception: MaxT
     logger.info("handling MaxTokensReachedException - inspecting incomplete message for invalid tool uses")
 
     incomplete_message: Message = exception.incomplete_message
+    logger.warning(f"incomplete message {incomplete_message}")
 
     if not incomplete_message["content"]:
         # Cannot correct invalid content block if content is empty
@@ -63,10 +64,6 @@ async def recover_tool_use_on_max_tokens_reached(agent: "Agent", exception: MaxT
             # ToolUse was invalid for an unknown reason. Cannot correct, return without modifying
             raise exception
 
-    if not has_corrected_content:
-        # No ToolUse were modified, meaning this method could not have resolved the root cause
-        raise exception
-
     valid_message: Message = {"content": valid_content, "role": incomplete_message["role"]}
     agent.messages.append(valid_message)
     agent.hooks.invoke_callbacks(MessageAddedEvent(agent=agent, message=valid_message))

tests/strands/agent/conversation_manager/test_recover_tool_use_on_max_tokens_reached.py

@@ -128,23 +128,16 @@ async def test_recover_tool_use_on_max_tokens_reached_with_valid_tool_use():
     mock_invoke_callbacks.assert_not_called()
 
 
-@pytest.mark.parametrize(
-    "content,description",
-    [
-        ([], "empty content"),
-        ([{"text": "Just some text with no tools to edit."}], "text-only content"),
-    ],
-)
 @pytest.mark.asyncio
-async def test_recover_tool_use_on_max_tokens_reached_with_empty_content(content, description):
+async def test_recover_tool_use_on_max_tokens_reached_with_empty_content():
     """Test that an exception that is raised without recoverability, re-raises exception."""
     agent = Agent()
     # Mock the hooks.invoke_callbacks method
     mock_invoke_callbacks = Mock()
     agent.hooks.invoke_callbacks = mock_invoke_callbacks
     initial_message_count = len(agent.messages)
 
-    incomplete_message: Message = {"role": "assistant", "content": content}
+    incomplete_message: Message = {"role": "assistant", "content": []}
 
     exception = MaxTokensReachedException(message="Token limit reached", incomplete_message=incomplete_message)