Fix e2e tests by switching to native Gemini integration#1375
Merged
josemontespg merged 4 commits intogoogle:mainfrom May 8, 2026
Merged
Fix e2e tests by switching to native Gemini integration#1375josemontespg merged 4 commits intogoogle:mainfrom
josemontespg merged 4 commits intogoogle:mainfrom
Conversation
Contributor
There was a problem hiding this comment.
Code Review
This pull request migrates the restaurant finder agent from LiteLlm to the native Gemini model integration. The review feedback identifies a typo in the default model name (gemini-2.5-flash) and suggests a more robust approach for parsing model IDs from environment variables to accommodate various provider prefixes.
nan-yu
approved these changes
May 8, 2026
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.This suggestion is invalid because no changes were made to the code.Suggestions cannot be applied while the pull request is closed.Suggestions cannot be applied while viewing a subset of changes.Only one suggestion per line can be applied in a batch.Add this suggestion to a batch that can be applied as a single commit.Applying suggestions on deleted lines is not supported.You must change the existing code in this line in order to create a valid suggestion.Outdated suggestions cannot be applied.This suggestion has been applied or marked resolved.Suggestions cannot be applied from pending reviews.Suggestions cannot be applied on multi-line comments.Suggestions cannot be applied while the pull request is queued to merge.Suggestion cannot be applied right now. Please check back later.
Description
This PR fixes the failed e2e tests for
restaurant_finderin CI.Error
The tests failed with the following error from LiteLLM:
Fix
As suggested by the warning in the logs, I switched from using
LiteLlmto nativeGeminiintegration inRestaurantAgent(samples/agent/adk/restaurant_finder/agent.py).This avoids the issue with LiteLLM trying to use an unavailable API version.
Verification
I verified that the tests pass locally after:
betachannel (version3.44.0-0.3.pre) to match CI.gemini-3-flash-previewas the model (by settingMODEL_NAMEenvironment variable), asgemini-2.5-flashcontinued to return 404 errors even with the native SDK.All 4 tests passed locally.