fix: fix unit tests

Author: Unshure

src/strands/agent/agent.py

@@ -14,7 +14,7 @@
 import os
 import random
 from concurrent.futures import ThreadPoolExecutor
-from typing import Any, AsyncIterator, Callable, Generator, Mapping, Optional, Type, TypeVar, Union, cast
+from typing import Any, AsyncIterator, Callable, Generator, List, Mapping, Optional, Type, TypeVar, Union, cast
 
 from opentelemetry import trace
 from pydantic import BaseModel
@@ -512,10 +512,10 @@ def _execute_event_loop_cycle(
 
     def _record_tool_execution(
         self,
-        tool: dict[str, Any],
-        tool_result: dict[str, Any],
+        tool: ToolUse,
+        tool_result: ToolResult,
         user_message_override: Optional[str],
-        messages: list[dict[str, Any]],
+        messages: Messages,
     ) -> None:
         """Record a tool execution in the message history.
 

tests/strands/models/test_bedrock.py

@@ -121,9 +121,11 @@ def test__init__default_model_id(bedrock_client):
 
 def test__init__with_default_region(session_cls, mock_client_method):
     """Test that BedrockModel uses the provided region."""
-    BedrockModel()
-
-    session_cls.return_value.client.assert_called_with(region_name=DEFAULT_BEDROCK_REGION, config=ANY, service_name=ANY)
+    with unittest.mock.patch.object(os, "environ", {}):
+        BedrockModel()
+        session_cls.return_value.client.assert_called_with(
+            region_name=DEFAULT_BEDROCK_REGION, config=ANY, service_name=ANY
+        )
 
 
 def test__init__with_session_region(session_cls, mock_client_method):
@@ -152,7 +154,7 @@ def test__init__with_default_environment_variable_region(mock_client_method):
 
 def test__init__region_precedence(mock_client_method, session_cls):
     """Test that BedrockModel uses the correct ordering of precedence when determining region."""
-    with unittest.mock.patch.object(os, "environ", {"AWS_REGION": "us-environment-1"}):
+    with unittest.mock.patch.object(os, "environ", {"AWS_REGION": "us-environment-1"}) as mock_os_environ:
         session_cls.return_value.region_name = "us-session-1"
 
         # specifying a region always wins out
@@ -168,9 +170,10 @@ def test__init__region_precedence(mock_client_method, session_cls):
         BedrockModel()
         mock_client_method.assert_called_with(region_name="us-environment-1", config=ANY, service_name=ANY)
 
-    # Finally default
-    BedrockModel()
-    mock_client_method.assert_called_with(region_name=DEFAULT_BEDROCK_REGION, config=ANY, service_name=ANY)
+        mock_os_environ.pop("AWS_REGION")
+        session_cls.return_value.region_name = None  # No session region
+        BedrockModel()
+        mock_client_method.assert_called_with(region_name=DEFAULT_BEDROCK_REGION, config=ANY, service_name=ANY)
 
 
 def test__init__with_region_and_session_raises_value_error():