Skip to content

[Bug]: [Update Required] SDK Location Inconsistency and Preview Model ID in intro_rag_engine.ipynb #2778

@amiragamalyassin

Description

@amiragamalyassin

File Name

gemini/rag-engine/intro_rag_engine.ipynb

What happened?

Inconsistency in SDK initialization and outdated Model ID.
Current Behavior:
The notebook initializes Vertex AI with location="us-east1" but sets the genai.Client to location="global". This mismatch causes authentication and resource discovery errors in the latest RAG Engine SDK updates.
The MODEL_ID is still using the gemini-3-flash-preview tag.

Expected Behavior:
Locations should be synchronized ( all set to us-central1) to ensure the RAG corpus and client can communicate properly.
The MODEL_ID should be updated to the stable gemini-3.1-flash since the stable version has been released (April 2026).

Steps to Reproduce:
Run the first two cells of the notebook.
Observe the mismatch in location parameters between vertexai.init and genai.Client.

vertexai.init(project=PROJECT_ID, location="us-east1")
client = genai.Client(vertexai=True, project=PROJECT_ID, location="global")

As a Data Science Researcher (MSc Informatics), I have already implemented these fixes in my fork and verified the stability. I am happy to open a Pull Request if access is granted.

Relevant log output

Code of Conduct

  • I agree to follow this project's Code of Conduct

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions