Skip to content

feat: support gemini-3.x models in loader and update default model to gemini-3.5-flash#17557

Merged
shuoweil merged 20 commits into
mainfrom
shuowei-deprecate-gemini2p5
Jun 25, 2026
Merged

feat: support gemini-3.x models in loader and update default model to gemini-3.5-flash#17557
shuoweil merged 20 commits into
mainfrom
shuowei-deprecate-gemini2p5

Conversation

@shuoweil

@shuoweil shuoweil commented Jun 23, 2026

Copy link
Copy Markdown
Contributor
  • Updated the default model in BQML GeminiTextGenerator to gemini-3.5-flash (instead of gemini-2.5-pro).
  • Registered gemini-3.1-flash-lite and gemini-3.5-flash endpoints in BQML model loader to prevent KeyError during model loading.
  • Updated all related docstrings, deprecation notes, warnings, and tests to align with the new defaults.
  • Updated Gemini model references across all sample notebooks to gemini-3.5-flash and fixed notebook formatting syntax to pass pre-commit checks.

Fixes #<519302924> 🦕

@shuoweil shuoweil self-assigned this Jun 23, 2026
@shuoweil shuoweil requested review from a team as code owners June 23, 2026 22:59
@shuoweil shuoweil requested review from GarrettWu, sycai and tswast and removed request for a team and tswast June 23, 2026 22:59
@shuoweil shuoweil marked this pull request as draft June 23, 2026 23:00

@gemini-code-assist gemini-code-assist Bot left a comment

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

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.

Comment thread packages/bigframes/bigframes/ml/llm.py Outdated

if model_name is None:
model_name = "gemini-2.0-flash-001"
model_name = "gemini-2.5-pro"

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

medium

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.

Copy link
Copy Markdown
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

All docstrings have been updated.

@shuoweil shuoweil removed request for GarrettWu and sycai June 23, 2026 23:02
@shuoweil shuoweil changed the title feat: support gemini-3.x models in loader and update tests/notebooks feat: support gemini-3.x models in loader and update default model to gemini-3.5-flash Jun 23, 2026
@shuoweil shuoweil marked this pull request as ready for review June 24, 2026 20:27

@sycai sycai left a comment

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Thanks! Especially given that this is a labor-intensive work.

@shuoweil shuoweil merged commit 3619b29 into main Jun 25, 2026
30 checks passed
@shuoweil shuoweil deleted the shuowei-deprecate-gemini2p5 branch June 25, 2026 17:18
shuoweil pushed a commit that referenced this pull request Jun 25, 2026
🤖 I have created a release *beep* *boop*
---


##
[2.44.0](bigframes-v2.43.0...bigframes-v2.44.0)
(2026-06-25)


### Features

* add date functions to `bigframes.bigquery` module
([#17514](#17514))
([e5d2e35](e5d2e35))
* **bigframes:** add AI TVFs to the pandas bq accessor
([#17402](#17402))
([ee74e31](ee74e31))
* Experimental transpilation of unannotated python callables
([#17419](#17419))
([ea9aad9](ea9aad9))
* support gemini-3.x models in loader and update default model to
gemini-3.5-flash
([#17557](#17557))
([3619b29](3619b29))
* support interactive execution of deferred DataFrames in TableWidget
([#17486](#17486))
([421eebd](421eebd))


### Bug Fixes

* avoid invalid CAST(NULL AS NULL) in SQLGlot compiler
([#17487](#17487))
([3b79caa](3b79caa))
* **bigframes:** world-readable temp zip in create_cloud_function
([#17522](#17522))
([e726878](e726878))
* bump @angular/common, @angular/forms, @angular/platform-browser and
@angular/router in
/packages/bigframes/bigframes/display/table_widget_angular
([#17525](#17525))
([2f893b1](2f893b1))
* bump langsmith from 0.8.0 to 0.8.18 in /packages/bigframes
([#17518](#17518))
([f23063f](f23063f))
* bump msgpack from 1.1.1 to 1.2.1 in /packages/bigframes
([#17520](#17520))
([36b5b7e](36b5b7e))
* bump undici and @angular/build in
/packages/bigframes/bigframes/display/table_widget_angular
([#17519](#17519))
([6fc45e3](6fc45e3))
* handle empty endpoints during cloud function reuse
([#17501](#17501))
([4f5593a](4f5593a))


### Documentation

* ensure that PlotAccessor is included in the API reference
([#17513](#17513))
([6febabf](6febabf))

---
This PR was generated with [Release
Please](https://github.com/googleapis/release-please). See
[documentation](https://github.com/googleapis/release-please#release-please).

Co-authored-by: release-please[bot] <55107282+release-please[bot]@users.noreply.github.com>
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.

4 participants