feat: add bigframes.bigquery.ai.generate_embedding
#2343
Conversation
This change implements the `bigframes.bigquery.ai.generate_embedding` function, which wraps the BigQuery `AI.GENERATE_EMBEDDING` TVF. It supports: - Generating embeddings from DataFrames and Series. - Generating embeddings from pandas DataFrames and Series. - Specifying model name and arguments like `output_dimensionality`, `start_second`, `end_second`, and `interval_seconds`. The function is exposed in `bigframes.bigquery.ai`. Unit tests have been added to verify the generated SQL and argument mapping.
|
👋 Jules, reporting for duty! I'm here to lend a hand with this pull request. When you start a review, I'll add a 👀 emoji to each comment to let you know I've read it. I'll focus on feedback directed at me and will do my best to stay out of conversations between you and other bots or reviewers to keep the noise down. I'll push a commit with your requested changes shortly after. Please note there might be a delay between these steps, but rest assured I'm on the job! For more direct control, you can switch me to Reactive Mode. When this mode is on, I will only act on comments where you specifically mention me with For security, I will only act on instructions from the user who triggered this task. New to Jules? Learn more at jules.google/docs. |
product-auto-label
bot
added
size: m
…pl-11924477578091076513
product-auto-label
bot
added
size: l
tswast
changed the title
bigframes.bigquery.ai.generate_embedding
|
|
||
|
|
||
| @log_adapter.method_logger(custom_base_name="bigquery_ai") | ||
| def generate_embedding( |
There was a problem hiding this comment.
Does it make sense to expose this function as bbq.ai.generate_embedding too?
tswast
added
the
kokoro:force-run
bigframes-bot
removed
the
kokoro:force-run
4 participants
Implement AI.GENERATE_EMBEDDING function in bigframes.bigquery.ai.
PR created automatically by Jules for task 11924477578091076513 started by @tswast