feat(genai): update to use gemini 2.5 flash

msampathkumar · msampathkumar · commit 79c5ce28f581 · 2025-05-20T17:41:43.000+02:00
diff --git a/genai/batch_prediction/batchpredict_with_bq.py b/genai/batch_prediction/batchpredict_with_bq.py
@@ -26,7 +26,7 @@ def generate_content(output_uri: str) -> str:
     # output_uri = f"bq://your-project.your_dataset.your_table"
 
     job = client.batches.create(
-        model="gemini-2.0-flash-001",
+        model="gemini-2.5-flash-preview-05-20",
         src="bq://storage-samples.generative_ai.batch_requests_for_multimodal_input",
         config=CreateBatchJobConfig(dest=output_uri),
     )
diff --git a/genai/batch_prediction/batchpredict_with_gcs.py b/genai/batch_prediction/batchpredict_with_gcs.py
@@ -26,7 +26,7 @@ def generate_content(output_uri: str) -> str:
 
     # See the documentation: https://googleapis.github.io/python-genai/genai.html#genai.batches.Batches.create
     job = client.batches.create(
-        model="gemini-2.0-flash-001",
+        model="gemini-2.5-flash-preview-05-20",
         # Source link: https://storage.cloud.google.com/cloud-samples-data/batch/prompt_for_batch_gemini_predict.jsonl
         src="gs://cloud-samples-data/batch/prompt_for_batch_gemini_predict.jsonl",
         config=CreateBatchJobConfig(dest=output_uri),
diff --git a/genai/bounding_box/boundingbox_with_txt_img.py b/genai/bounding_box/boundingbox_with_txt_img.py
@@ -94,7 +94,7 @@ def plot_bounding_boxes(image_uri: str, bounding_boxes: list[BoundingBox]) -> No
     image_uri = "https://storage.googleapis.com/generativeai-downloads/images/socks.jpg"
 
     response = client.models.generate_content(
-        model="gemini-2.0-flash-001",
+        model="gemini-2.5-flash-preview-05-20",
         contents=[
             Part.from_uri(
                 file_uri=image_uri,
diff --git a/genai/content_cache/contentcache_create_with_txt_gcs_pdf.py b/genai/content_cache/contentcache_create_with_txt_gcs_pdf.py
@@ -42,7 +42,7 @@ def create_content_cache() -> str:
     ]
 
     content_cache = client.caches.create(
-        model="gemini-2.0-flash-001",
+        model="gemini-2.5-flash-preview-05-20",
         config=CreateCachedContentConfig(
             contents=contents,
             system_instruction=system_instruction,
diff --git a/genai/content_cache/contentcache_use_with_txt.py b/genai/content_cache/contentcache_use_with_txt.py
@@ -23,7 +23,7 @@ def generate_content(cache_name: str) -> str:
     # Use content cache to generate text response
     # E.g cache_name = 'projects/111111111111/locations/us-central1/cachedContents/1111111111111111111'
     response = client.models.generate_content(
-        model="gemini-2.0-flash-001",
+        model="gemini-2.5-flash-preview-05-20",
         contents="Summarize the pdfs",
         config=GenerateContentConfig(
             cached_content=cache_name,
diff --git a/genai/controlled_generation/ctrlgen_with_class_schema.py b/genai/controlled_generation/ctrlgen_with_class_schema.py
@@ -26,7 +26,7 @@ class Recipe(BaseModel):
 
     client = genai.Client(http_options=HttpOptions(api_version="v1"))
     response = client.models.generate_content(
-        model="gemini-2.0-flash-001",
+        model="gemini-2.5-flash-preview-05-20",
         contents="List a few popular cookie recipes.",
         config=GenerateContentConfig(
             response_mime_type="application/json",
diff --git a/genai/controlled_generation/ctrlgen_with_enum_class_schema.py b/genai/controlled_generation/ctrlgen_with_enum_class_schema.py
@@ -29,7 +29,7 @@ class InstrumentClass(enum.Enum):
 
     client = genai.Client(http_options=HttpOptions(api_version="v1"))
     response = client.models.generate_content(
-        model="gemini-2.0-flash-001",
+        model="gemini-2.5-flash-preview-05-20",
         contents="What type of instrument is a guitar?",
         config={
             "response_mime_type": "text/x.enum",
diff --git a/genai/controlled_generation/ctrlgen_with_enum_schema.py b/genai/controlled_generation/ctrlgen_with_enum_schema.py
@@ -20,7 +20,7 @@ def generate_content() -> str:
 
     client = genai.Client(http_options=HttpOptions(api_version="v1"))
     response = client.models.generate_content(
-        model="gemini-2.0-flash-001",
+        model="gemini-2.5-flash-preview-05-20",
         contents="What type of instrument is an oboe?",
         config=GenerateContentConfig(
             response_mime_type="text/x.enum",
diff --git a/genai/controlled_generation/ctrlgen_with_nested_class_schema.py b/genai/controlled_generation/ctrlgen_with_nested_class_schema.py
@@ -36,7 +36,7 @@ class Recipe(BaseModel):
 
     client = genai.Client(http_options=HttpOptions(api_version="v1"))
     response = client.models.generate_content(
-        model="gemini-2.0-flash-001",
+        model="gemini-2.5-flash-preview-05-20",
         contents="List about 10 home-baked cookies and give them grades based on tastiness.",
         config=GenerateContentConfig(
             response_mime_type="application/json",
diff --git a/genai/controlled_generation/ctrlgen_with_nullable_schema.py b/genai/controlled_generation/ctrlgen_with_nullable_schema.py
@@ -51,7 +51,7 @@ def generate_content() -> str:
 
     client = genai.Client(http_options=HttpOptions(api_version="v1"))
     response = client.models.generate_content(
-        model="gemini-2.0-flash-001",
+        model="gemini-2.5-flash-preview-05-20",
         contents=prompt,
         config=GenerateContentConfig(
             response_mime_type="application/json",
diff --git a/genai/controlled_generation/ctrlgen_with_resp_schema.py b/genai/controlled_generation/ctrlgen_with_resp_schema.py
@@ -36,7 +36,7 @@ def generate_content() -> str:
 
     client = genai.Client(http_options=HttpOptions(api_version="v1"))
     response = client.models.generate_content(
-        model="gemini-2.0-flash-001",
+        model="gemini-2.5-flash-preview-05-20",
         contents=prompt,
         config={
             "response_mime_type": "application/json",
diff --git a/genai/count_tokens/counttoken_compute_with_txt.py b/genai/count_tokens/counttoken_compute_with_txt.py
@@ -20,7 +20,7 @@ def compute_tokens_example() -> int:
 
     client = genai.Client(http_options=HttpOptions(api_version="v1"))
     response = client.models.compute_tokens(
-        model="gemini-2.0-flash-001",
+        model="gemini-2.5-flash-preview-05-20",
         contents="What's the longest word in the English language?",
     )
 
diff --git a/genai/count_tokens/counttoken_resp_with_txt.py b/genai/count_tokens/counttoken_resp_with_txt.py
@@ -24,7 +24,7 @@ def count_tokens_example() -> int:
 
     # Send text to Gemini
     response = client.models.generate_content(
-        model="gemini-2.0-flash-001", contents=prompt
+        model="gemini-2.5-flash-preview-05-20", contents=prompt
     )
 
     # Prompt and response tokens count
diff --git a/genai/count_tokens/counttoken_with_txt.py b/genai/count_tokens/counttoken_with_txt.py
@@ -20,7 +20,7 @@ def count_tokens() -> int:
 
     client = genai.Client(http_options=HttpOptions(api_version="v1"))
     response = client.models.count_tokens(
-        model="gemini-2.0-flash-001",
+        model="gemini-2.5-flash-preview-05-20",
         contents="What's the highest mountain in Africa?",
     )
     print(response)
diff --git a/genai/count_tokens/counttoken_with_txt_vid.py b/genai/count_tokens/counttoken_with_txt_vid.py
@@ -29,7 +29,7 @@ def count_tokens() -> int:
     ]
 
     response = client.models.count_tokens(
-        model="gemini-2.0-flash-001",
+        model="gemini-2.5-flash-preview-05-20",
         contents=contents,
     )
     print(response)
diff --git a/genai/express_mode/api_key_example.py b/genai/express_mode/api_key_example.py
@@ -23,7 +23,7 @@ def generate_content() -> str:
     client = genai.Client(vertexai=True, api_key=API_KEY)
 
     response = client.models.generate_content(
-        model="gemini-2.0-flash-001",
+        model="gemini-2.5-flash-preview-05-20",
         contents="Explain bubble sort to me.",
     )
 
diff --git a/genai/provisioned_throughput/provisionedthroughput_with_txt.py b/genai/provisioned_throughput/provisionedthroughput_with_txt.py
@@ -31,7 +31,7 @@ def generate_content() -> str:
         )
     )
     response = client.models.generate_content(
-        model="gemini-2.0-flash-001",
+        model="gemini-2.5-flash-preview-05-20",
         contents="How does AI work?",
     )
     print(response.text)
diff --git a/genai/safety/safety_with_txt.py b/genai/safety/safety_with_txt.py
@@ -54,7 +54,7 @@ def generate_content() -> GenerateContentResponse:
     ]
 
     response = client.models.generate_content(
-        model="gemini-2.0-flash-001",
+        model="gemini-2.5-flash-preview-05-20",
         contents=prompt,
         config=GenerateContentConfig(
             system_instruction=system_instruction,
diff --git a/genai/template_folder/templatefolder_with_txt.py b/genai/template_folder/templatefolder_with_txt.py
@@ -1,4 +1,4 @@
-# Copyright 2024 Google LLC
+# Copyright 2025 Google LLC
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
diff --git a/genai/text_generation/textgen_async_with_txt.py b/genai/text_generation/textgen_async_with_txt.py
@@ -1,4 +1,4 @@
-# Copyright 2024 Google LLC
+# Copyright 2025 Google LLC
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
@@ -21,7 +21,7 @@ async def generate_content() -> str:
     from google.genai.types import GenerateContentConfig, HttpOptions
 
     client = genai.Client(http_options=HttpOptions(api_version="v1"))
-    model_id = "gemini-2.0-flash-001"
+    model_id = "gemini-2.5-flash-preview-05-20"
 
     response = await client.aio.models.generate_content(
         model=model_id,
diff --git a/genai/text_generation/textgen_chat_stream_with_txt.py b/genai/text_generation/textgen_chat_stream_with_txt.py
@@ -13,25 +13,23 @@
 # limitations under the License.
 
 
-def generate_content() -> str:
+def generate_content() -> bool:
     # [START googlegenaisdk_textgen_chat_stream_with_txt]
     from google import genai
-    from google.genai.types import HttpOptions
+    from google.genai.types import HttpOptions, GenerateContentConfig
 
     client = genai.Client(http_options=HttpOptions(api_version="v1"))
-    chat_session = client.chats.create(model="gemini-2.0-flash-001")
-    response_text = ""
+    chat_session = client.chats.create(model="gemini-2.5-flash-preview-05-20")
 
     for chunk in chat_session.send_message_stream("Why is the sky blue?"):
         print(chunk.text, end="")
-        response_text += chunk.text
     # Example response:
     # The
     #  sky appears blue due to a phenomenon called **Rayleigh scattering**. Here's
     #  a breakdown of why:
     # ...
     # [END googlegenaisdk_textgen_chat_stream_with_txt]
-    return response_text
+    return True
 
 
 if __name__ == "__main__":
diff --git a/genai/text_generation/textgen_chat_with_txt.py b/genai/text_generation/textgen_chat_with_txt.py
@@ -20,7 +20,7 @@ def generate_content() -> str:
 
     client = genai.Client(http_options=HttpOptions(api_version="v1"))
     chat_session = client.chats.create(
-        model="gemini-2.0-flash-001",
+        model="gemini-2.5-flash-preview-05-20",
         history=[
             UserContent(parts=[Part(text="Hello")]),
             ModelContent(
diff --git a/genai/text_generation/textgen_config_with_txt.py b/genai/text_generation/textgen_config_with_txt.py
@@ -20,9 +20,10 @@ def generate_content() -> str:
 
     client = genai.Client(http_options=HttpOptions(api_version="v1"))
     response = client.models.generate_content(
-        model="gemini-2.0-flash-001",
+        model="gemini-2.5-flash-preview-05-20",
         contents="Why is the sky blue?",
-        # See the documentation: https://googleapis.github.io/python-genai/genai.html#genai.types.GenerateContentConfig
+        # See the SDK documentation at
+        # https://googleapis.github.io/python-genai/genai.html#genai.types.GenerateContentConfig
         config=GenerateContentConfig(
             temperature=0,
             candidate_count=1,
diff --git a/genai/text_generation/textgen_sys_instr_with_txt.py b/genai/text_generation/textgen_sys_instr_with_txt.py
@@ -20,7 +20,7 @@ def generate_content() -> str:
 
     client = genai.Client(http_options=HttpOptions(api_version="v1"))
     response = client.models.generate_content(
-        model="gemini-2.0-flash-001",
+        model="gemini-2.5-flash-preview-05-20",
         contents="Why is the sky blue?",
         config=GenerateContentConfig(
             system_instruction=[
diff --git a/genai/text_generation/textgen_transcript_with_gcs_audio.py b/genai/text_generation/textgen_transcript_with_gcs_audio.py
@@ -24,7 +24,7 @@ def generate_content() -> str:
     Use speaker A, speaker B, etc. to identify speakers.
     """
     response = client.models.generate_content(
-        model="gemini-2.0-flash-001",
+        model="gemini-2.5-flash-preview-05-20",
         contents=[
             prompt,
             Part.from_uri(
diff --git a/genai/text_generation/textgen_with_gcs_audio.py b/genai/text_generation/textgen_with_gcs_audio.py
@@ -23,7 +23,7 @@ def generate_content() -> str:
     Provide a concise summary of the main points in the audio file.
     """
     response = client.models.generate_content(
-        model="gemini-2.0-flash-001",
+        model="gemini-2.5-flash-preview-05-20",
         contents=[
             prompt,
             Part.from_uri(
diff --git a/genai/text_generation/textgen_with_local_video.py b/genai/text_generation/textgen_with_local_video.py
@@ -1,4 +1,4 @@
-# Copyright 2024 Google LLC
+# Copyright 2025 Google LLC
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
@@ -19,7 +19,7 @@ def generate_content() -> str:
     from google.genai.types import HttpOptions, Part
 
     client = genai.Client(http_options=HttpOptions(api_version="v1"))
-    model_id = "gemini-2.0-flash-001"
+    model_id = "gemini-2.5-flash-preview-05-20"
 
     # Read local video file content
     with open("test_data/describe_video_content.mp4", "rb") as fp:
@@ -29,6 +29,7 @@ def generate_content() -> str:
     response = client.models.generate_content(
         model=model_id,
         contents=[
+            Part.from_text(text="hello-world"),
             Part.from_bytes(data=video_content, mime_type="video/mp4"),
             "Write a short and engaging blog post based on this video.",
         ],
diff --git a/genai/text_generation/textgen_with_multi_img.py b/genai/text_generation/textgen_with_multi_img.py
@@ -28,7 +28,7 @@ def generate_content() -> str:
         local_file_img_bytes = f.read()
 
     response = client.models.generate_content(
-        model="gemini-2.0-flash-001",
+        model="gemini-2.5-flash-preview-05-20",
         contents=[
             "Generate a list of all the objects contained in both images.",
             Part.from_uri(file_uri=gcs_file_img_path, mime_type="image/jpeg"),
diff --git a/genai/text_generation/textgen_with_multi_local_img.py b/genai/text_generation/textgen_with_multi_local_img.py
@@ -28,7 +28,7 @@ def generate_content(image_path_1: str, image_path_2: str) -> str:
         image_2_bytes = f.read()
 
     response = client.models.generate_content(
-        model="gemini-2.0-flash-001",
+        model="gemini-2.5-flash-preview-05-20",
         contents=[
             "Generate a list of all the objects contained in both images.",
             Part.from_bytes(data=image_1_bytes, mime_type="image/jpeg"),
diff --git a/genai/text_generation/textgen_with_mute_video.py b/genai/text_generation/textgen_with_mute_video.py
@@ -20,7 +20,7 @@ def generate_content() -> str:
 
     client = genai.Client(http_options=HttpOptions(api_version="v1"))
     response = client.models.generate_content(
-        model="gemini-2.0-flash-001",
+        model="gemini-2.5-flash-preview-05-20",
         contents=[
             Part.from_uri(
                 file_uri="gs://cloud-samples-data/generative-ai/video/ad_copy_from_video.mp4",
diff --git a/genai/text_generation/textgen_with_pdf.py b/genai/text_generation/textgen_with_pdf.py
@@ -1,4 +1,4 @@
-# Copyright 2024 Google LLC
+# Copyright 2025 Google LLC
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
@@ -21,7 +21,7 @@ def generate_content() -> str:
     from google.genai.types import HttpOptions, Part
 
     client = genai.Client(http_options=HttpOptions(api_version="v1"))
-    model_id = "gemini-2.0-flash-001"
+    model_id = "gemini-2.5-flash-preview-05-20"
 
     prompt = """
     You are a highly skilled document summarization specialist.
diff --git a/genai/text_generation/textgen_with_txt.py b/genai/text_generation/textgen_with_txt.py
@@ -20,7 +20,7 @@ def generate_content() -> str:
 
     client = genai.Client(http_options=HttpOptions(api_version="v1"))
     response = client.models.generate_content(
-        model="gemini-2.0-flash-001",
+        model="gemini-2.5-flash-preview-05-20",
         contents="How does AI work?",
     )
     print(response.text)
diff --git a/genai/text_generation/textgen_with_txt_img.py b/genai/text_generation/textgen_with_txt_img.py
@@ -20,7 +20,7 @@ def generate_content() -> str:
 
     client = genai.Client(http_options=HttpOptions(api_version="v1"))
     response = client.models.generate_content(
-        model="gemini-2.0-flash-001",
+        model="gemini-2.5-flash-preview-05-20",
         contents=[
             "What is shown in this image?",
             Part.from_uri(
diff --git a/genai/text_generation/textgen_with_txt_stream.py b/genai/text_generation/textgen_with_txt_stream.py
@@ -13,26 +13,25 @@
 # limitations under the License.
 
 
-def generate_content() -> str:
+def generate_content() -> bool:
     # [START googlegenaisdk_textgen_with_txt_stream]
     from google import genai
     from google.genai.types import HttpOptions
 
     client = genai.Client(http_options=HttpOptions(api_version="v1"))
-    response_text = ""
+
     for chunk in client.models.generate_content_stream(
-        model="gemini-2.0-flash-001",
+        model="gemini-2.5-flash-preview-05-20",
         contents="Why is the sky blue?",
     ):
         print(chunk.text, end="")
-        response_text += chunk.text
     # Example response:
     # The
     #  sky appears blue due to a phenomenon called **Rayleigh scattering**. Here's
     #  a breakdown of why:
     # ...
     # [END googlegenaisdk_textgen_with_txt_stream]
-    return response_text
+    return True
 
 
 if __name__ == "__main__":
diff --git a/genai/text_generation/textgen_with_video.py b/genai/text_generation/textgen_with_video.py
@@ -25,7 +25,7 @@ def generate_content() -> str:
     Create a chapter breakdown with timestamps for key sections or topics discussed.
     """
     response = client.models.generate_content(
-        model="gemini-2.0-flash-001",
+        model="gemini-2.5-flash-preview-05-20",
         contents=[
             Part.from_uri(
                 file_uri="gs://cloud-samples-data/generative-ai/video/pixel8.mp4",
diff --git a/genai/text_generation/textgen_with_youtube_video.py b/genai/text_generation/textgen_with_youtube_video.py
@@ -1,4 +1,4 @@
-# Copyright 2024 Google LLC
+# Copyright 2025 Google LLC
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
@@ -21,7 +21,7 @@ def generate_content() -> str:
     from google.genai.types import HttpOptions, Part
 
     client = genai.Client(http_options=HttpOptions(api_version="v1"))
-    model_id = "gemini-2.0-flash-001"
+    model_id = "gemini-2.5-flash-preview-05-20"
 
     response = client.models.generate_content(
         model=model_id,
diff --git a/genai/text_generation/thinking_textgen_with_txt.py b/genai/text_generation/thinking_textgen_with_txt.py
diff --git a/genai/thinking/thinking_budget_with_txt.py b/genai/thinking/thinking_budget_with_txt.py
diff --git a/genai/thinking/thinking_with_txt.py b/genai/thinking/thinking_with_txt.py
diff --git a/genai/tools/tools_code_exec_with_txt.py b/genai/tools/tools_code_exec_with_txt.py
diff --git a/genai/tools/tools_code_exec_with_txt_local_img.py b/genai/tools/tools_code_exec_with_txt_local_img.py
diff --git a/genai/tools/tools_func_def_with_txt.py b/genai/tools/tools_func_def_with_txt.py
diff --git a/genai/tools/tools_func_desc_with_txt.py b/genai/tools/tools_func_desc_with_txt.py
diff --git a/genai/tools/tools_google_search_with_txt.py b/genai/tools/tools_google_search_with_txt.py
diff --git a/genai/tools/tools_vais_with_txt.py b/genai/tools/tools_vais_with_txt.py

Original file line number	Diff line number	Diff line change
`@@ -26,7 +26,7 @@ def generate_content(output_uri: str) -> str:`
`26`	`26`	`# output_uri = f"bq://your-project.your_dataset.your_table"`
`27`	`27`
`28`	`28`	`job = client.batches.create(`
`29`		`- model="gemini-2.0-flash-001",`
	`29`	`+ model="gemini-2.5-flash-preview-05-20",`
`30`	`30`	`src="bq://storage-samples.generative_ai.batch_requests_for_multimodal_input",`
`31`	`31`	`config=CreateBatchJobConfig(dest=output_uri),`
`32`	`32`	`)`
Original file line number	Diff line number	Diff line change
`@@ -42,7 +42,7 @@ def create_content_cache() -> str:`
`42`	`42`	`]`
`43`	`43`
`44`	`44`	`content_cache = client.caches.create(`
`45`		`- model="gemini-2.0-flash-001",`
	`45`	`+ model="gemini-2.5-flash-preview-05-20",`
`46`	`46`	`config=CreateCachedContentConfig(`
`47`	`47`	`contents=contents,`
`48`	`48`	`system_instruction=system_instruction,`