When trying to run the embeddings code cell of "notebooks/1_Setup_OpenDataQnA.ipynb"
it throws an error that the Gemini 1.5 LLM model is not available anymore.
Embeddings Code cell:
from env_setup import get_embeddings
'# Generate embeddings for tables and columns
table_schema_embeddings, col_schema_embeddings = get_embeddings()
Full Error:
NotFound: 404 Publisher Model projects/nl2sql-proto/locations/us-central1/publishers/google/models/gemini-1.5-pro-001 was not found or your project does not have access to it. Please ensure you are using a valid model version. For more information, see: https://cloud.google.com/vertex-ai/generative-ai/docs/learn/model-versions
After investigating Google's documentation, it says the Gemini 1.5 pro model has been retired since May 24th 2025 (I am posting this issue on the 4th of June 2025).
https://cloud.google.com/vertex-ai/generative-ai/docs/learn/model-versions#retired-models
I assumed that the config section would fix the issue, to switch for example to:
description_model = 'gemini-2.0-flash-lite-001'
But this did not change the error message. Searching through the project files, there are 43 hardcoded(?) occurrences of "gemini-1.5" (in 13 files).