Skip to content

feat(adk): Support ToolContext for auth_token_getters in toolbox-adk#579

Open
anubhav756 wants to merge 8 commits intomainfrom
anubhav-adk-auth
Open

feat(adk): Support ToolContext for auth_token_getters in toolbox-adk#579
anubhav756 wants to merge 8 commits intomainfrom
anubhav-adk-auth

Conversation

@anubhav756
Copy link
Contributor

@anubhav756 anubhav756 commented Mar 6, 2026

This PR centralizes our parameter validation logic, adds ToolContext support for ADK auth token getters, and fixes flaky CI/CD test pipelines.

Background

  • The unused token/parameter validation logic was duplicated across toolbox_core and toolbox_adk. This made error messages slightly inconsistent and the code harder to maintain.
  • In toolbox-adk, auth_token_getters couldn't access the invocation context. We needed a way to pass the current ToolContext into these fetchers so they can retrieve tokens dynamically based on the current agent state.
  • *Our automated tests were randomly failing. The CI workflows were occasionally pulling older versions of the packages from the internet instead of properly testing the local code changes from the PR.

Changes

  • Pulled the duplicate validation logic into a single, shared helper function (validate_unused_requirements) in the toolbox-core package to ensure error messages are clear and consistent everywhere.
  • Upgraded toolbox-adk to support 1-argument functions for auth token getters, allowing ToolContext to be passed directly in.
  • Separated the dependency installation steps in our .cloudbuild.yaml workflows so local package builds are correctly prioritized during integration tests.

@anubhav756 anubhav756 self-assigned this Mar 6, 2026
@anubhav756 anubhav756 requested a review from a team as a code owner March 6, 2026 13:47
@anubhav756 anubhav756 changed the title feat(adk): Support ToolContext for aut_token_getters in toolbox-adk feat(adk): Support ToolContext for auth_token_getters in toolbox-adk Mar 6, 2026
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant