fix(genai): error

msampathkumar · msampathkumar · commit dd56ba2630c9 · 2025-05-20T20:07:12.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.5-flash-preview-05-20",
+        model="gemini-2.0-flash-001",
         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.5-flash-preview-05-20",
+        model="gemini-2.0-flash-001",
         # 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),

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.5-flash-preview-05-20",`
	`29`	`+ model="gemini-2.0-flash-001",`
`30`	`30`	`src="bq://storage-samples.generative_ai.batch_requests_for_multimodal_input",`
`31`	`31`	`config=CreateBatchJobConfig(dest=output_uri),`
`32`	`32`	`)`