Skip to content

fix(models): the judge's Vertex client uses ADC even when agent API keys are set - #81

Open
geojaz wants to merge 2 commits into
kubernetes-sigs:mainfrom
geojaz:ehole/judge-vertex-adc
Open

fix(models): the judge's Vertex client uses ADC even when agent API keys are set#81
geojaz wants to merge 2 commits into
kubernetes-sigs:mainfrom
geojaz:ehole/judge-vertex-adc

Conversation

@geojaz

@geojaz geojaz commented Aug 7, 2026

Copy link
Copy Markdown
Contributor

The judge's Vertex client was picking up an agent's API key from the environment instead of using application default credentials.

This matters because the agent under evaluation and the judge scoring it are not the same principal and should not share credentials. When an agent API key was present in the environment, the judge would authenticate as that agent, which both crosses an isolation boundary and fails in any setup where the agent key has no access to the judge's model.

The judge now resolves credentials through ADC regardless of what agent keys are in the environment.

…e in the environment

The judge failed 401 UNAUTHENTICATED against aiplatform.googleapis.com
("API keys are not supported by this API") even with valid ADC on the
host. google-genai's Vertex mode silently prefers API-key/express auth
over Application Default Credentials whenever GOOGLE_API_KEY or
GEMINI_API_KEY is present in the environment, and those are exactly the
variables exported for the sandboxed coding agent this judge scores.
Resolving credentials explicitly via google.auth.default() and handing
them to genai.Client(..., credentials=...) closes that path: the client
can no longer fall back to an env API key it was never told to look for.

Signed-off-by: Eric Hole <ehole@onixnet.com>
@kubernetes-prow

Copy link
Copy Markdown

[APPROVALNOTIFIER] This PR is NOT APPROVED

This pull-request has been approved by: geojaz
Once this PR has been reviewed and has the lgtm label, please assign janetkuo for approval. For more information see the Code Review Process.

The full list of commands accepted by this bot can be found here.

Details Needs approval from an approver in each of these files:

Approvers can indicate their approval by writing /approve in a comment
Approvers can cancel approval by writing /approve cancel in a comment

@kubernetes-prow
kubernetes-prow Bot requested a review from janetkuo August 7, 2026 13:10
@coderabbitai

coderabbitai Bot commented Aug 7, 2026

Copy link
Copy Markdown

Warning

Review limit reached

@geojaz, you've reached your PR review limit, so we couldn't start this review.

Next review available in: 59 minutes

You've used all free OSS reviews for now. Wait for the free limit to reset to keep reviewing this public repository.

How can I continue?

After more reviews become available, a review can be triggered using the @coderabbitai review command as a PR comment. Alternatively, push new commits to this PR.

To avoid repeated limits, reduce automatic review volume by pausing incremental auto-reviews earlier, using label-based review opt-in, excluding WIP or generated PR titles, or requesting reviews manually when the PR is ready. If your team needs uninterrupted high-volume reviews, an organization admin can enable usage-based reviews.

How do review limits work?

CodeRabbit enforces per-developer PR review limits for each organization. Most developers receive the normal plan review availability.

For paid Pro and Pro+ PR reviews, CodeRabbit uses adaptive limits for sustained high-volume activity. When a developer's recent PR review activity reaches the 95th percentile or higher among CodeRabbit users, additional reviews become available more gradually as earlier reviews age out of the rolling window.

Please refer docs for additional details.

Review details
⚙️ Run configuration

Configuration used: Path: .coderabbit.yaml

Review profile: CHILL

Plan: Pro Plus

Run ID: cf20c0f5-f55e-4cf7-96a7-878f190c183e

📥 Commits

Reviewing files that changed from the base of the PR and between 4670d76 and 490d9d9.

📒 Files selected for processing (2)
  • devops_bench/models/gemini.py
  • tests/unit/models/test_models_gemini.py

Thanks for using CodeRabbit! It's free for OSS, and your support helps us grow. If you like it, consider giving us a shout-out.

❤️ Share

Comment @coderabbitai help to get the list of available commands.

@kubernetes-prow kubernetes-prow Bot added cncf-cla: yes Indicates the PR's author has signed the CNCF CLA. needs-ok-to-test Indicates a PR that requires an org member to verify it is safe to test. labels Aug 7, 2026
@kubernetes-prow

Copy link
Copy Markdown

Hi @geojaz. Thanks for your PR.

I'm waiting for a kubernetes-sigs member to verify that this patch is reasonable to test. If it is, they should reply with /ok-to-test on its own line. Until that is done, I will not automatically test new commits in this PR, but the usual testing commands by org members will still work.

Regular contributors should join the org to skip this step.

Once the patch is verified, the new status will be reflected by the ok-to-test label.

I understand the commands that are listed here.

Details

Instructions for interacting with me using PR comments are available here. If you have questions or suggestions related to my behavior, please file an issue against the kubernetes-sigs/prow repository.

@kubernetes-prow kubernetes-prow Bot added the size/M Denotes a PR that changes 30-99 lines, ignoring generated files. label Aug 7, 2026
@janetkuo janetkuo added ok-to-test Indicates a non-member PR verified by an org member that is safe to test. and removed needs-ok-to-test Indicates a PR that requires an org member to verify it is safe to test. labels Aug 11, 2026
test_get_model_resolves_gemini_aliases reaches the new Vertex ADC path
via the google-vertex/google_vertex provider aliases and was doing a
real credentials lookup in CI. Mock google_auth.default there like the
other Vertex-path tests. Also reorders the try-block imports in
gemini.py so isort/ruff stops flagging the plain import.
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

cncf-cla: yes Indicates the PR's author has signed the CNCF CLA. ok-to-test Indicates a non-member PR verified by an org member that is safe to test. size/M Denotes a PR that changes 30-99 lines, ignoring generated files.

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants