fix: pass context to client inceptors

PiperOrigin-RevId: 840918430

Author: google-genai-bot

src/google/adk/agents/remote_a2a_agent.py

@@ -32,6 +32,7 @@
 from a2a.client.client import ClientConfig as A2AClientConfig
 from a2a.client.client_factory import ClientFactory as A2AClientFactory
 from a2a.client.errors import A2AClientHTTPError
+from a2a.client.middleware import ClientCallContext
 from a2a.types import AgentCard
 from a2a.types import Message as A2AMessage
 from a2a.types import Part as A2APart
@@ -533,6 +534,7 @@ async def _run_async_impl(
       async for a2a_response in self._a2a_client.send_message(
           request=a2a_request,
           request_metadata=request_metadata,
+          context=ClientCallContext(state=ctx.session.state),
       ):
         logger.debug(build_a2a_response_log(a2a_response))
 

tests/unittests/agents/test_remote_a2a_agent.py

@@ -23,6 +23,7 @@
 from a2a.client.client import ClientConfig
 from a2a.client.client import Consumer
 from a2a.client.client_factory import ClientFactory
+from a2a.client.middleware import ClientCallContext
 from a2a.types import AgentCapabilities
 from a2a.types import AgentCard
 from a2a.types import AgentSkill
@@ -1429,6 +1430,7 @@ def setup_method(self):
     self.mock_session = Mock(spec=Session)
     self.mock_session.id = "session-123"
     self.mock_session.events = []
+    self.mock_session.state = {}
 
     self.mock_context = Mock(spec=InvocationContext)
     self.mock_context.session = self.mock_session
@@ -1682,6 +1684,7 @@ async def test_run_async_impl_with_meta_provider(self):
                   mock_a2a_client.send_message.assert_called_once_with(
                       request=mock_message,
                       request_metadata=request_metadata,
+                      context=ClientCallContext(state=self.mock_session.state),
                   )
 
 
@@ -1703,6 +1706,7 @@ def setup_method(self):
     self.mock_session = Mock(spec=Session)
     self.mock_session.id = "session-123"
     self.mock_session.events = []
+    self.mock_session.state = {}
 
     self.mock_context = Mock(spec=InvocationContext)
     self.mock_context.session = self.mock_session
@@ -1989,6 +1993,7 @@ async def test_full_workflow_with_direct_agent_card(self):
     mock_session = Mock(spec=Session)
     mock_session.id = "session-123"
     mock_session.events = [mock_event]
+    mock_session.state = {}
 
     mock_context = Mock(spec=InvocationContext)
     mock_context.session = mock_session
@@ -2084,6 +2089,7 @@ async def test_full_workflow_with_direct_agent_card_and_factory(self):
     mock_session = Mock(spec=Session)
     mock_session.id = "session-123"
     mock_session.events = [mock_event]
+    mock_session.state = {}
 
     mock_context = Mock(spec=InvocationContext)
     mock_context.session = mock_session