Skip to content

feat(bigframes): add ai.embed and ai.similarity to bigquery accessor - #17927

Merged
sycai merged 3 commits into
mainfrom
sycai_ai_embed_and_similarity
Jul 29, 2026
Merged

feat(bigframes): add ai.embed and ai.similarity to bigquery accessor#17927
sycai merged 3 commits into
mainfrom
sycai_ai_embed_and_similarity

Conversation

@sycai

@sycaisycai commented Jul 28, 2026

Copy link
Copy Markdown
Contributor

I chose to define functions under the mixin instead of the code generator because of several function gaps.

We may onboard the ai functions to the code generator some day in the future. Tracking: b/540011825

Internal issue: b/517233441

@sycai
sycai requested review from a team as code ownersJuly 28, 2026 18:50
@sycai
sycai requested review from GarrettWu and tswast and removed request for a teamJuly 28, 2026 18:50
gemini-code-assist[bot]
gemini-code-assistBot reviewed Jul 28, 2026
View reviewed changes

@gemini-code-assistgemini-code-assistBot 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 renames AITVFMixin to AIMixin and adds embed and similarity methods to the AIMixin class in series_mixins.py to support BigQuery AI embeddings and similarity calculations, along with corresponding unit tests. The review feedback highlights a shadowing issue in the new methods where the session parameter shadows the imported session module, which would cause failures when session is None. To resolve this, it is recommended to import bigframes at the module level and fall back to bigframes.session when the parameter is not provided.

Comment threadpackages/bigframes/bigframes/extensions/core/series_mixins.py Outdated
Comment threadpackages/bigframes/bigframes/extensions/core/series_mixins.py
Comment threadpackages/bigframes/bigframes/extensions/core/series_mixins.py
@sycaisycai added the kokoro:force-run Add this label to force Kokoro to re-run the tests. label Jul 29, 2026
@yoshi-kokoroyoshi-kokoro removed the kokoro:force-run Add this label to force Kokoro to re-run the tests. label Jul 29, 2026
@sycai
sycai merged commit 2d1372b into mainJul 29, 2026
38 of 39 checks passed
@sycai
sycai deleted the sycai_ai_embed_and_similarity branch July 29, 2026 19:41
@release-pleaserelease-pleaseBot mentioned this pull request Jul 29, 2026
Merged
tswast pushed a commit that referenced this pull request Aug 3, 2026
@release-please
chore(main): release bigframes 2.47.0 (#17800)
654a96f
🤖 I have created a release *beep* *boop*
---
##
[2.47.0](bigframes-v2.46.0...bigframes-v2.47.0)
(2026-08-03)
### Features
* **bigframes:** add ai.embed and ai.similarity to bigquery accessor
([#17927](#17927))
([2d1372b](2d1372b))
### Bug Fixes
* **bigframes:** fix mypy errors in _magics.py related to get_ipython
([#17979](#17979))
([24d955e](24d955e))
* bump brace-expansion from 5.0.6 to 5.0.7 in
/packages/bigframes/bigframes/display/table_widget_angular
([#17794](#17794))
([2df1bb5](2df1bb5))
* bump fast-uri from 3.1.1 to 3.1.4 in
/packages/bigframes/bigframes/display/table_widget_angular
([#17828](#17828))
([8ce495a](8ce495a))
* bump hono from 4.12.16 to 4.12.31 in
/packages/bigframes/bigframes/display/table_widget_angular
([#17829](#17829))
([6d6fa39](6d6fa39))
* bump immutable from 5.1.7 to 5.1.9 in
/packages/bigframes/bigframes/display/table_widget_angular
([#17830](#17830))
([6abb1da](6abb1da))
* bump postcss from 8.5.14 to 8.5.23 in
/packages/bigframes/bigframes/display/table_widget_angular
([#17908](#17908))
([e68eb8f](e68eb8f))
* bump tar from 7.5.16 to 7.5.20 in
/packages/bigframes/bigframes/display/table_widget_angular
([#17793](#17793))
([3502d41](3502d41))
* require Protobuf 6.33.5+
([#17743](#17743))
([d267342](d267342))
---
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>
copybara-serviceBot pushed a commit to google/bigframes that referenced this pull request Aug 12, 2026
@release-please@copybara-github
chore(main): release bigframes 2.47.0 (#17800)
7cf63e5
🤖 I have created a release *beep* *boop*
---
##
[2.47.0](googleapis/google-cloud-python@bigframes-v2.46.0...bigframes-v2.47.0)
(2026-08-03)
### Features
* **bigframes:** add ai.embed and ai.similarity to bigquery accessor
([#17927](googleapis/google-cloud-python#17927))
([2d1372b](googleapis/google-cloud-python@2d1372b))
### Bug Fixes
* **bigframes:** fix mypy errors in _magics.py related to get_ipython
([#17979](googleapis/google-cloud-python#17979))
([24d955e](googleapis/google-cloud-python@24d955e))
* bump brace-expansion from 5.0.6 to 5.0.7 in
/packages/bigframes/bigframes/display/table_widget_angular
([#17794](googleapis/google-cloud-python#17794))
([2df1bb5](googleapis/google-cloud-python@2df1bb5))
* bump fast-uri from 3.1.1 to 3.1.4 in
/packages/bigframes/bigframes/display/table_widget_angular
([#17828](googleapis/google-cloud-python#17828))
([8ce495a](googleapis/google-cloud-python@8ce495a))
* bump hono from 4.12.16 to 4.12.31 in
/packages/bigframes/bigframes/display/table_widget_angular
([#17829](googleapis/google-cloud-python#17829))
([6d6fa39](googleapis/google-cloud-python@6d6fa39))
* bump immutable from 5.1.7 to 5.1.9 in
/packages/bigframes/bigframes/display/table_widget_angular
([#17830](googleapis/google-cloud-python#17830))
([6abb1da](googleapis/google-cloud-python@6abb1da))
* bump postcss from 8.5.14 to 8.5.23 in
/packages/bigframes/bigframes/display/table_widget_angular
([#17908](googleapis/google-cloud-python#17908))
([e68eb8f](googleapis/google-cloud-python@e68eb8f))
* bump tar from 7.5.16 to 7.5.20 in
/packages/bigframes/bigframes/display/table_widget_angular
([#17793](googleapis/google-cloud-python#17793))
([3502d41](googleapis/google-cloud-python@3502d41))
* require Protobuf 6.33.5+
([#17743](googleapis/google-cloud-python#17743))
([d267342](googleapis/google-cloud-python@d267342))
---
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>
GitOrigin-RevId: 654a96f867c8b749702027d6c3d39f3864c48081
Change-Id: I942ec3bf4556b4c325f4b0acd3a0759ce4139dc2
Sign up for freeto join this conversation on GitHub. Already have an account? Sign in to comment

Reviewers

@GarrettWuGarrettWuGarrettWu approved these changes
@tswasttswastAwaiting requested review from tswast
+1 more reviewer
@gemini-code-assistgemini-code-assist[bot]gemini-code-assist[bot] left review comments
Reviewers whose approvals may not affect merge requirements

Assignees

@tswasttswast

Labels

None yet

Projects

None yet

Milestone

No milestone

Development

Successfully merging this pull request may close these issues.

4 participants

@sycai@GarrettWu@tswast@yoshi-kokoro