This is an automated email from the ASF dual-hosted git repository.
Abacn pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/beam.git
The following commit(s) were added to refs/heads/master by this push:
new b452ad63fb0 Update Gemini text classification to gemini-2.5-flash
(#38839)
b452ad63fb0 is described below
commit b452ad63fb0d832b161b96a2388b77f2b5095731
Author: Abdelrahman Ibrahim <[email protected]>
AuthorDate: Mon Jun 8 21:08:23 2026 +0200
Update Gemini text classification to gemini-2.5-flash (#38839)
---
.github/trigger_files/beam_PostCommit_Python.json | 2 +-
sdks/python/apache_beam/examples/inference/gemini_image_generation.py | 4 ++--
.../apache_beam/examples/inference/gemini_text_classification.py | 4 ++--
3 files changed, 5 insertions(+), 5 deletions(-)
diff --git a/.github/trigger_files/beam_PostCommit_Python.json
b/.github/trigger_files/beam_PostCommit_Python.json
index 4f79e635cc0..b12889937d5 100644
--- a/.github/trigger_files/beam_PostCommit_Python.json
+++ b/.github/trigger_files/beam_PostCommit_Python.json
@@ -1,5 +1,5 @@
{
"comment": "Modify this file in a trivial way to cause this test suite to
run.",
"pr": "38069",
- "modification": 43
+ "modification": 44
}
diff --git
a/sdks/python/apache_beam/examples/inference/gemini_image_generation.py
b/sdks/python/apache_beam/examples/inference/gemini_image_generation.py
index 29b2d562e63..069b3fca3a9 100644
--- a/sdks/python/apache_beam/examples/inference/gemini_image_generation.py
+++ b/sdks/python/apache_beam/examples/inference/gemini_image_generation.py
@@ -17,8 +17,8 @@
""" A sample pipeline using the RunInference API to classify text using an LLM.
This pipeline creates a set of prompts and sends it to a Gemini service then
-returns the predictions from the classifier model. This example uses the
-gemini-2.0-flash-001 model.
+returns generated images from the model. This example uses the
+gemini-2.5-flash-image model.
"""
import argparse
diff --git
a/sdks/python/apache_beam/examples/inference/gemini_text_classification.py
b/sdks/python/apache_beam/examples/inference/gemini_text_classification.py
index 0072dfc50b2..55b603bc2ab 100644
--- a/sdks/python/apache_beam/examples/inference/gemini_text_classification.py
+++ b/sdks/python/apache_beam/examples/inference/gemini_text_classification.py
@@ -18,7 +18,7 @@
""" A sample pipeline using the RunInference API to classify text using an LLM.
This pipeline creates a set of prompts and sends it to a Gemini service then
returns the predictions from the classifier model. This example uses the
-gemini-2.0-flash-001 model.
+gemini-2.5-flash model.
"""
import argparse
@@ -88,7 +88,7 @@ def run(
pipeline_options = PipelineOptions(pipeline_args)
pipeline_options.view_as(SetupOptions).save_main_session = save_main_session
model_handler = GeminiModelHandler(
- model_name='gemini-2.0-flash-001',
+ model_name='gemini-2.5-flash',
request_fn=generate_from_string,
api_key=known_args.api_key,
project=known_args.project,
