applied autoformat.sh

Author: timof1308

src/google/adk/tools/mcp_tool/mcp_toolset.py

@@ -60,7 +60,6 @@
 from .mcp_tool import MCPTool
 from .types import HeaderProvider
 
-
 logger = logging.getLogger("google_adk." + __name__)
 
 
@@ -80,9 +79,9 @@ def _validate_header_value(
   if not isinstance(value, (str, int, float, bool)):
     msg = (
         f'Value for state key "{state_key}" is of type'
-        f' {type(value).__name__}, which may not serialize correctly into a'
-        ' header. Consider pre-serializing complex values or using'
-        ' state_header_format.'
+        f" {type(value).__name__}, which may not serialize correctly into a"
+        " header. Consider pre-serializing complex values or using"
+        " state_header_format."
     )
     if strict:
       raise ValueError(msg)

tests/unittests/tools/mcp_tool/test_jwt_token_propagation.py

@@ -143,19 +143,22 @@ def test_empty_string_value_returns_empty(self):
   def test_strict_mode_with_primitive_types(self):
     """Test that strict mode works properly with primitive types."""
     mock_context = Mock(spec=ReadonlyContext)
-    
+
     # Test with string
     mock_context.state = {"token": "my-token"}
     provider = create_session_state_header_provider(
         state_key="token", strict=True
     )
     headers = provider(mock_context)
     assert headers == {"Authorization": "Bearer my-token"}
-    
+
     # Test with int
     mock_context.state = {"count": 42}
     provider = create_session_state_header_provider(
-        state_key="count", header_name="X-Count", header_format="{value}", strict=True
+        state_key="count",
+        header_name="X-Count",
+        header_format="{value}",
+        strict=True,
     )
     headers = provider(mock_context)
     assert headers == {"X-Count": "42"}
@@ -171,13 +174,12 @@ def test_strict_mode_raises_on_non_primitive_types(self):
 
     with pytest.raises(ValueError) as exc_info:
       provider(mock_context)
-    
+
     assert "complex_data" in str(exc_info.value)
     assert "dict" in str(exc_info.value)
     assert "may not serialize correctly" in str(exc_info.value)
 
 
-
 class TestMcpToolsetConfigStateHeaderMapping:
   """Test suite for state_header_mapping configuration."""
 
@@ -341,4 +343,3 @@ def test_from_config_with_strict_mode(self):
 
     assert "data" in str(exc_info.value)
     assert "dict" in str(exc_info.value)
-