fix: narrow A2A warning filter to exact remote_a2a_agent module

Address Copilot review: scope the module regex from the broad
`^google\.adk\.` prefix to the exact `^google\.adk\.agents\.remote_a2a_agent$`
module where the warning originates. This prevents accidentally suppressing
experimental warnings from other google.adk submodules.

Also adds a test verifying that warnings from other google.adk modules are
not affected by the filter.

Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
Signed-off-by: Optimus (AI Agent) <agent@fulcria.com>

Author: optimus-fulcria

python/packages/kagent-adk/src/kagent/adk/types.py

@@ -11,8 +11,13 @@
 
 # isort: off
 # Show the A2A experimental warning once per process instead of on every call (#1379).
-# Scoped to UserWarning from google.adk modules to avoid suppressing unrelated warnings.
-warnings.filterwarnings("once", message=r"\[EXPERIMENTAL\].*A2A", category=UserWarning, module=r"^google\.adk\.")
+# Scoped to UserWarning from the specific remote_a2a_agent module to avoid suppressing unrelated warnings.
+warnings.filterwarnings(
+    "once",
+    message=r"\[EXPERIMENTAL\].*A2A",
+    category=UserWarning,
+    module=r"^google\.adk\.agents\.remote_a2a_agent$",
+)
 from google.adk.agents.remote_a2a_agent import AGENT_CARD_WELL_KNOWN_PATH, DEFAULT_TIMEOUT, RemoteA2aAgent  # noqa: E402
 # isort: on
 from google.adk.models.anthropic_llm import Claude as ClaudeLLM

python/packages/kagent-adk/tests/unittests/test_a2a_warning_filter.py

@@ -8,17 +8,18 @@ def test_a2a_warning_shown_once():
     with warnings.catch_warnings(record=True) as caught:
         warnings.simplefilter("always")
 
-        # Re-install the filter as types.py does at import time (scoped to google.adk)
+        # Re-install the filter as types.py does at import time
+        # (scoped to the specific remote_a2a_agent module)
         warnings.filterwarnings(
             "once",
             message=r"\[EXPERIMENTAL\].*A2A",
             category=UserWarning,
-            module=r"^google\.adk\.",
+            module=r"^google\.adk\.agents\.remote_a2a_agent$",
         )
 
-        # Simulate a warning from a google.adk module by executing in a
+        # Simulate a warning from the exact upstream module by executing in a
         # namespace whose __name__ matches the module filter.
-        adk_globals = {"__name__": "google.adk.a2a.remote", "__file__": __file__}
+        adk_globals = {"__name__": "google.adk.agents.remote_a2a_agent", "__file__": __file__}
         code = compile(
             'import warnings; warnings.warn("[EXPERIMENTAL] A2A support is experimental", UserWarning, stacklevel=1)',
             "<test>",
@@ -29,11 +30,11 @@ def test_a2a_warning_shown_once():
         assert len(caught) == 1, "First A2A warning should be recorded"
 
         exec(code, adk_globals)  # noqa: S102
-        assert len(caught) == 1, "Duplicate A2A warning from google.adk should be suppressed"
+        assert len(caught) == 1, "Duplicate A2A warning from remote_a2a_agent should be suppressed"
 
 
 def test_filter_does_not_suppress_non_adk_warnings():
-    """The filter should not suppress warnings from modules outside google.adk."""
+    """The filter should not suppress warnings from modules outside remote_a2a_agent."""
     with warnings.catch_warnings(record=True) as caught:
         warnings.simplefilter("always")
 
@@ -42,13 +43,42 @@ def test_filter_does_not_suppress_non_adk_warnings():
             "once",
             message=r"\[EXPERIMENTAL\].*A2A",
             category=UserWarning,
-            module=r"^google\.adk\.",
+            module=r"^google\.adk\.agents\.remote_a2a_agent$",
         )
 
-        # A warning from the current test module (not google.adk.*) should not be suppressed
+        # A warning from the current test module (not remote_a2a_agent) should not be suppressed
         warnings.warn("[EXPERIMENTAL] A2A support is experimental", UserWarning, stacklevel=1)
         assert len(caught) == 1
 
         warnings.warn("[EXPERIMENTAL] A2A support is experimental", UserWarning, stacklevel=1)
-        # Both should be recorded because this module is not google.adk.*
-        assert len(caught) == 2, "Warnings from non-google.adk modules should not be suppressed"
+        # Both should be recorded because this module is not google.adk.agents.remote_a2a_agent
+        assert len(caught) == 2, "Warnings from non-remote_a2a_agent modules should not be suppressed"
+
+
+def test_filter_does_not_suppress_other_adk_modules():
+    """The filter should not suppress warnings from other google.adk submodules."""
+    with warnings.catch_warnings(record=True) as caught:
+        warnings.simplefilter("always")
+
+        # Install the scoped filter (exact module match only)
+        warnings.filterwarnings(
+            "once",
+            message=r"\[EXPERIMENTAL\].*A2A",
+            category=UserWarning,
+            module=r"^google\.adk\.agents\.remote_a2a_agent$",
+        )
+
+        # A warning from a different google.adk submodule should NOT be suppressed
+        other_adk_globals = {"__name__": "google.adk.a2a.executor", "__file__": __file__}
+        code = compile(
+            'import warnings; warnings.warn("[EXPERIMENTAL] A2A support is experimental", UserWarning, stacklevel=1)',
+            "<test>",
+            "exec",
+        )
+
+        exec(code, other_adk_globals)  # noqa: S102
+        assert len(caught) == 1
+
+        exec(code, other_adk_globals)  # noqa: S102
+        # Both should be recorded because the module doesn't match remote_a2a_agent exactly
+        assert len(caught) == 2, "Warnings from other google.adk modules should not be suppressed"