From 16c3fb6c7e7704868b9ff6193f55212a10e5b1a3 Mon Sep 17 00:00:00 2001
From: Darshan Mehta <darshanmehta@google.com>
Date: Tue, 13 Oct 2026 14:32:26 -0700
Subject: [PATCH] docs: Update gemini model in docstrings for RAG Engine

PiperOrigin-RevId: 998896283
---
 agentplatform/rag/rag_data.py      | 2 +-
 agentplatform/rag/rag_retrieval.py | 4 ++--
 agentplatform/rag/rag_store.py     | 2 +-
 vertexai/rag/rag_data.py           | 2 +-
 vertexai/rag/rag_retrieval.py      | 2 +-
 vertexai/rag/rag_store.py          | 2 +-
 6 files changed, 7 insertions(+), 7 deletions(-)

diff --git a/agentplatform/rag/rag_data.py b/agentplatform/rag/rag_data.py
index 39e6661a5d..cdd849dedc 100644
--- a/agentplatform/rag/rag_data.py
+++ b/agentplatform/rag/rag_data.py
@@ -1021,7 +1021,7 @@ def add_inline_citations_and_references(
       )
 
     rag_model = GenerativeModel(
-        model_name="gemini-2.5-pro-preview-05-06", tools=[rag_retrieval_tool]
+        model_name="gemini-3.8-flash", tools=[rag_retrieval_tool]
     )
     response = rag_model.generate_content("Why is the sky blue?")
 
diff --git a/agentplatform/rag/rag_retrieval.py b/agentplatform/rag/rag_retrieval.py
index 70ead25de4..679a2111d0 100644
--- a/agentplatform/rag/rag_retrieval.py
+++ b/agentplatform/rag/rag_retrieval.py
@@ -46,8 +46,8 @@ def retrieval_query(
             vector_distance_threshold=0.5
         ),
         ranking=agentplatform.rag.Ranking(
-            llm_ranker=rag.LlmRanker(
-                model_name="gemini-2.5-flash"
+            llm_ranker=agentplatform.rag.LlmRanker(
+                model_name="gemini-3.8-flash"
             )
         )
     )
diff --git a/agentplatform/rag/rag_store.py b/agentplatform/rag/rag_store.py
index b8f0b93f49..705b9e5559 100644
--- a/agentplatform/rag/rag_store.py
+++ b/agentplatform/rag/rag_store.py
@@ -61,7 +61,7 @@ def __init__(
             ),
             ranking=vertex.rag.Ranking(
                 llm_ranker=agentplatform.rag.LlmRanker(
-                    model_name="gemini-1.5-flash-002"
+                    model_name="gemini-3.8-flash"
                 )
             )
         )
diff --git a/vertexai/rag/rag_data.py b/vertexai/rag/rag_data.py
index fd4ca0d7e6..ea5b7e007a 100644
--- a/vertexai/rag/rag_data.py
+++ b/vertexai/rag/rag_data.py
@@ -1045,7 +1045,7 @@ def add_inline_citations_and_references(
       )
 
     rag_model = GenerativeModel(
-        model_name="gemini-2.5-pro-preview-05-06", tools=[rag_retrieval_tool]
+        model_name="gemini-3.8-flash", tools=[rag_retrieval_tool]
     )
     response = rag_model.generate_content("Why is the sky blue?")
 
diff --git a/vertexai/rag/rag_retrieval.py b/vertexai/rag/rag_retrieval.py
index 0640d413ba..2c7ff53fcb 100644
--- a/vertexai/rag/rag_retrieval.py
+++ b/vertexai/rag/rag_retrieval.py
@@ -49,7 +49,7 @@ def retrieval_query(
         ),
         ranking=vertex.rag.Ranking(
             llm_ranker=vertexai.rag.LlmRanker(
-                model_name="gemini-1.5-flash-002"
+                model_name="gemini-3.8-flash"
             )
         )
     )
diff --git a/vertexai/rag/rag_store.py b/vertexai/rag/rag_store.py
index 497b4969eb..b10eb81cc5 100644
--- a/vertexai/rag/rag_store.py
+++ b/vertexai/rag/rag_store.py
@@ -64,7 +64,7 @@ def __init__(
             ),
             ranking=vertex.rag.Ranking(
                 llm_ranker=vertexai.rag.LlmRanker(
-                    model_name="gemini-1.5-flash-002"
+                    model_name="gemini-3.8-flash"
                 )
             )
         )