Batch embedding sample does not work after update to gemini-embedding-001 #13393
Labels
priority: p2
Moderately-important priority. Fix may not be included in next release.
samples
Issues that are directly related to samples.
triage me
I really want to be triaged.
type: bug
Error or flaw in code with unintended results or allowing sub-optimal usage patterns.
TL;DR: The recently updated batch text embedding sample throws an error due to unsupported model
gemini-embedding-001
.Introduced in: #13388
In which file did you encounter the issue?
generative_ai/embeddings/batch_example.py
Did you change the file? If so, how?
Only to set the project ID and output bucket.
Describe the issue
google-cloud-aiplatform
v1.95 into a Python 3.13 virtual environment.python3 batch_example.py
and received the error message400 Do not support publisher model gemini-embedding-001
Full error traceback
To establish whether this is an issue with the model or something in my setup, I tried changing the model name to a previous model,
text-embedding-005
withAnd the batch prediction job completed successfully as expected. Therefore, I conclude it is something to do with the model
gemini-embedding-001
.The text was updated successfully, but these errors were encountered: