fix: Address PR comments

Author: zastrowm

tests/strands/agent/test_agent.py

@@ -2091,7 +2091,10 @@ def shell(command: str):
     )
 
     agent = Agent(tools=[shell], model=model)
-    agent("Test")
+    result = agent("Test")
+
+    # Ensure the stop_reason is
+    assert result.stop_reason == "end_turn"
 
     # Assert that there exists a message with a toolResponse
     assert agent.messages[-2] == {
@@ -2106,3 +2109,6 @@ def shell(command: str):
         ],
         "role": "user",
     }
+
+    # And that it continued to the LLM call
+    assert agent.messages[-1] == {"content": [{"text": "I invoked a tool!"}], "role": "assistant"}

tests/strands/event_loop/test_event_loop.py

@@ -400,36 +400,6 @@ async def test_event_loop_cycle_tool_result_error(
         await alist(stream)
 
 
-@pytest.mark.asyncio
-async def test_event_loop_with_tool_result_valid(
-    agent,
-    model,
-    tool_stream,
-    mock_tracer,
-    agenerator,
-    alist,
-):
-    # Setup
-
-    # Set up model to return tool use and then text response
-    model.stream.side_effect = [
-        agenerator(tool_stream),
-        agenerator(
-            [
-                {"contentBlockDelta": {"delta": {"text": "test text"}}},
-                {"contentBlockStop": {}},
-            ]
-        ),
-    ]
-
-    # Call event_loop_cycle which should execute a tool
-    stream = strands.event_loop.event_loop.event_loop_cycle(
-        agent=agent,
-        invocation_state={},
-    )
-    await alist(stream)
-
-
 @pytest.mark.asyncio
 async def test_event_loop_cycle_tool_result_no_tool_handler(
     agent,