Skip to content

feat(bigframes): implement ai.similarity#16771

Open
sycai wants to merge 2 commits intomainfrom
sycai_ai_similarity
Open

feat(bigframes): implement ai.similarity#16771
sycai wants to merge 2 commits intomainfrom
sycai_ai_similarity

Conversation

@sycai
Copy link
Copy Markdown
Contributor

@sycai sycai commented Apr 22, 2026

Fixes b/497837587

@sycai sycai requested review from a team as code owners April 22, 2026 23:55
@sycai sycai requested review from mpovoa and removed request for a team April 22, 2026 23:55
@sycai sycai marked this pull request as draft April 22, 2026 23:55
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 similarity function to the BigQuery AI module, enabling cosine similarity calculations between strings or series using Vertex AI models. The implementation includes the necessary operation definitions, compiler registrations for Ibis and SQLGlot, and comprehensive system and unit tests. Review feedback identified a consistent typo in the model name 'embedding-gemma-300m' within the documentation and test cases.

Comment thread packages/bigframes/bigframes/bigquery/_operations/ai.py
Comment thread packages/bigframes/tests/system/small/bigquery/test_ai.py
@sycai sycai marked this pull request as ready for review April 23, 2026 00:10
@sycai sycai requested review from TrevorBergeron and chelsea-lin and removed request for mpovoa April 23, 2026 00:10
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.

1 participant