revert to older model because gemini-embedding-001 doesn't support batch_predict at the moment. (#13396)

Feiyang1 · web-flow · commit 50c338654360 · 2025-05-30T10:34:40.000+10:00
diff --git a/generative_ai/embeddings/batch_example.py b/generative_ai/embeddings/batch_example.py
@@ -39,7 +39,7 @@ def embed_text_batch() -> BatchPredictionJob:
     output_uri = OUTPUT_URI
 
     textembedding_model = language_models.TextEmbeddingModel.from_pretrained(
-        "gemini-embedding-001"
+        "textembedding-gecko@003"
     )
 
     batch_prediction_job = textembedding_model.batch_predict(

Original file line number	Diff line number	Diff line change
`@@ -39,7 +39,7 @@ def embed_text_batch() -> BatchPredictionJob:`
`39`	`39`	`output_uri = OUTPUT_URI`
`40`	`40`
`41`	`41`	`textembedding_model = language_models.TextEmbeddingModel.from_pretrained(`
`42`		`- "gemini-embedding-001"`
	`42`	`+ "textembedding-gecko@003"`
`43`	`43`	`)`
`44`	`44`
`45`	`45`	`batch_prediction_job = textembedding_model.batch_predict(`