Skip to content

chore(aiplatform): Update CreateBatchPredictionGeminiBigqueryJobSample to use gemini 2.5#10229

Open
amcolin wants to merge 1 commit intoGoogleCloudPlatform:mainfrom
amcolin:create-batch-prediction-gemini-bigquery
Open

chore(aiplatform): Update CreateBatchPredictionGeminiBigqueryJobSample to use gemini 2.5#10229
amcolin wants to merge 1 commit intoGoogleCloudPlatform:mainfrom
amcolin:create-batch-prediction-gemini-bigquery

Conversation

@amcolin
Copy link

@amcolin amcolin commented Feb 11, 2026

Description

The sample uses an outdated or inconsistent Gemini model version (gemini-2.0-flash-001). It needs to be updated to a supported version like gemini-2.5-flash

Internal: b/481401081

Note: Before submitting a pull request, please open an issue for discussion if you are not associated with Google.

Checklist

  • I have followed Sample Format Guide
  • pom.xml parent set to latest shared-configuration
  • Appropriate changes to README are included in PR
  • These samples need a new API enabled in testing projects to pass (let us know which ones)
  • These samples need a new/updated env vars in testing projects set to pass (let us know which ones)
  • Tests pass: mvn clean verify required
  • Lint passes: mvn -P lint checkstyle:check required
  • Static Analysis: mvn -P lint clean compile pmd:cpd-check spotbugs:check advisory only
  • This sample adds a new sample directory, and I updated the CODEOWNERS file with the codeowners for this sample
  • This sample adds a new Product API, and I updated the Blunderbuss issue/PR auto-assigner with the codeowners for this sample
  • Please merge this PR for me once it is approved

@product-auto-label product-auto-label bot added api: aiplatform Issues related to the AI Platform API. samples Issues that are directly related to samples. labels Feb 11, 2026
@gemini-code-assist
Copy link
Contributor

Summary of Changes

Hello @amcolin, I'm Gemini Code Assist1! I'm currently reviewing this pull request and will post my feedback shortly. In the meantime, here's a summary to help you and other reviewers quickly get up to speed!

This pull request updates a Java sample for Google Cloud AI Platform to ensure it uses a current and supported version of the Gemini model. The change specifically targets the CreateBatchPredictionGeminiBigqueryJobSample, replacing an outdated model reference with gemini-2.5-flash to maintain sample functionality and consistency with the latest model offerings.

Highlights

  • Gemini Model Update: The CreateBatchPredictionGeminiBigqueryJobSample has been updated to utilize the gemini-2.5-flash model, replacing the previously used gemini-2.0-flash-001.
Changelog
  • aiplatform/src/main/java/aiplatform/batchpredict/CreateBatchPredictionGeminiBigqueryJobSample.java
    • Updated the Gemini model version from gemini-2.0-flash-001 to gemini-2.5-flash.
Activity
  • No activity found for this pull request yet.
Using Gemini Code Assist

The full guide for Gemini Code Assist can be found on our documentation page, here are some quick tips.

Invoking Gemini

You can request assistance from Gemini at any point by creating a comment using either /gemini <command> or @gemini-code-assist <command>. Below is a summary of the supported commands on the current page.

Feature Command Description
Code Review /gemini review Performs a code review for the current pull request in its current state.
Pull Request Summary /gemini summary Provides a summary of the current pull request in its current state.
Comment @gemini-code-assist Responds in comments when explicitly tagged, both in pull request comments and review comments.
Help /gemini help Displays a list of available commands.

Customization

To customize Gemini Code Assist for GitHub experience, repository maintainers can create a configuration file and/or provide a custom code review style guide (such as PEP-8 for Python) by creating and adding files to a .gemini/ folder in the base of the repository. Detailed instructions can be found here.

Limitations & Feedback

Gemini Code Assist may make mistakes. Please leave feedback on any instances where its feedback is incorrect or counter productive. You can react with 👍 and 👎 on @gemini-code-assist comments. If you're interested in giving your feedback about your experience with Gemini Code Assist for Github and other Google products, sign up here.

You can also get AI-powered code generation, chat, as well as code reviews directly in the IDE at no cost with the Gemini Code Assist IDE Extension.

Footnotes

  1. Review the Privacy Notices, Generative AI Prohibited Use Policy, Terms of Service, and learn how to configure Gemini Code Assist in GitHub here. Gemini can make mistakes, so double check it and use code with caution.

Copy link
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 correctly updates the Gemini model version to gemini-2.5-flash in CreateBatchPredictionGeminiBigqueryJobSample.java. However, a related sample file, aiplatform/src/main/java/aiplatform/batchpredict/CreateBatchPredictionGeminiJobSample.java, still references the outdated gemini-2.0-flash-001 model. To ensure consistency and that all samples use the supported model version as stated in the PR description, I recommend updating the model in that file as well. After updating both files, consider extracting the model version string into a shared constant to improve maintainability and avoid duplication.

@amcolin amcolin marked this pull request as ready for review February 11, 2026 17:25
@amcolin amcolin requested review from a team and yoshi-approver as code owners February 11, 2026 17:25
@amcolin amcolin added the kokoro:force-run Add this label to force Kokoro to re-run the tests. label Feb 11, 2026
@kokoro-team kokoro-team removed the kokoro:force-run Add this label to force Kokoro to re-run the tests. label Feb 11, 2026
@alarconesparza alarconesparza added the kokoro:force-run Add this label to force Kokoro to re-run the tests. label Feb 11, 2026
@kokoro-team kokoro-team removed the kokoro:force-run Add this label to force Kokoro to re-run the tests. label Feb 11, 2026
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

api: aiplatform Issues related to the AI Platform API. samples Issues that are directly related to samples.

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants