feat: support gemini-3.x models in loader and update default model to gemini-3.5-flash#17557
feat: support gemini-3.x models in loader and update default model to gemini-3.5-flash#17557shuoweil wants to merge 15 commits into
Conversation
![gemini-code-assist[bot]](https://avatars.githubusercontent.com/in/956858?s=60&v=4){kind=small}
There was a problem hiding this comment.
Code Review
This pull request adds support for the new Gemini models gemini-3.1-flash-lite and gemini-3.5-flash across BigQuery DataFrames ML, updates the default model for GeminiTextGenerator to gemini-2.5-pro, and updates deprecation warnings, notebooks, and tests accordingly. Feedback was provided to update the GeminiTextGenerator docstring to reflect the new default model, as it still refers to the old default.
|
|
||
| if model_name is None: | ||
| model_name = "gemini-2.0-flash-001" | ||
| model_name = "gemini-2.5-pro" |
There was a problem hiding this comment.
The default model has been updated to "gemini-2.5-pro" here, but the docstring in GeminiTextGenerator (lines 451-452) still states that "gemini-2.0-flash-001" will be used by default. Please update the docstring to match this new default.
There was a problem hiding this comment.
All docstrings have been updated.
shuoweil
changed the title
1 participant
** STILL IN CLEAN UP STAGE"
GeminiTextGeneratortogemini-3.5-flash(instead ofgemini-2.5-pro).gemini-3.1-flash-liteandgemini-3.5-flashendpoints in BQML model loader to preventKeyErrorduring model loading.gemini-3.5-flashand fixed notebook formatting syntax to pass pre-commit checks.Fixes #<519302924> 🦕