Skip to content

feat(bigframes): implement ai.embed#16759

Open
sycai wants to merge 5 commits intomainfrom
sycai_ai_embed
Open

feat(bigframes): implement ai.embed#16759
sycai wants to merge 5 commits intomainfrom
sycai_ai_embed

Conversation

@sycai
Copy link
Copy Markdown
Contributor

@sycai sycai commented Apr 21, 2026

Fixes b/497836685 🦕

@sycai sycai requested review from a team as code owners April 21, 2026 23:32
@sycai sycai requested review from TrevorBergeron and removed request for a team April 21, 2026 23:32
@sycai sycai marked this pull request as draft April 21, 2026 23:32
Copy link
Copy Markdown
Contributor

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

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 introduces the embed function to support creating embeddings from text or image data in BigQuery. The implementation includes the core operation logic, integration with Ibis and SQLGlot compilers, and comprehensive unit tests. Feedback was provided to correct the type hint for connection_id, optimize the parameter validation logic using a generator expression, and fix a docstring copy-paste error in the vendored Ibis operations.

Comment thread packages/bigframes/bigframes/bigquery/_operations/ai.py Outdated
Comment thread packages/bigframes/bigframes/bigquery/_operations/ai.py Outdated
Comment thread packages/bigframes/third_party/bigframes_vendored/ibis/expr/operations/ai_ops.py Outdated
@sycai sycai force-pushed the sycai_ai_embed branch 2 times, most recently from 75fda1f to d225c0c Compare April 21, 2026 23:52
@sycai
Copy link
Copy Markdown
Contributor Author

sycai commented Apr 22, 2026

Pre-release tests failed at divmod cases, which is unrelated to this change.

@sycai sycai marked this pull request as ready for review April 22, 2026 19:49
@sycai sycai added the kokoro:force-run Add this label to force Kokoro to re-run the tests. label Apr 22, 2026
@sycai sycai requested a review from chelsea-lin April 22, 2026 19:51
@yoshi-kokoro yoshi-kokoro removed the kokoro:force-run Add this label to force Kokoro to re-run the tests. label Apr 22, 2026
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.

2 participants