chore: Unify CallbackContext and ToolContext into the Context class

Co-authored-by: Xuan Yang <xygoogle@google.com>
PiperOrigin-RevId: 869400758
This commit is contained in:
Xuan Yang
2026-02-12 15:01:51 -08:00
committed by Copybara-Service
parent 4f71b45425
commit 647d3b16a3
4 changed files with 23 additions and 320 deletions
@@ -172,7 +172,7 @@ async def test_openid_connect_with_auth_response(
oauth2=OAuth2Auth(auth_response_uri='test_auth_response_uri'),
)
mock_auth_handler.get_auth_response.return_value = returned_credential
mock_auth_handler_path = 'google.adk.tools.tool_context.AuthHandler'
mock_auth_handler_path = 'google.adk.auth.auth_handler.AuthHandler'
monkeypatch.setattr(
mock_auth_handler_path, lambda *args, **kwargs: mock_auth_handler
)