From 8362ca4588e2ae2eac5ab2fe967ac6878d87504c Mon Sep 17 00:00:00 2001 From: Sara Robinson Date: Wed, 10 Jun 2026 13:52:41 -0700 Subject: [PATCH] fix: GenAI client - Make vertex_rag_store a required arg for client.rag.retrieve_contexts PiperOrigin-RevId: 930058524 --- agentplatform/_genai/rag.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/agentplatform/_genai/rag.py b/agentplatform/_genai/rag.py index f00db97a5e..90cf95a112 100644 --- a/agentplatform/_genai/rag.py +++ b/agentplatform/_genai/rag.py @@ -2227,7 +2227,7 @@ def _update_config( def retrieve_contexts( self, *, - vertex_rag_store: Optional[types.VertexRagStoreOrDict] = None, + vertex_rag_store: types.VertexRagStoreOrDict, query: types.RagQueryOrDict, config: Optional[types.RetrieveContextsConfigOrDict] = None, ) -> types.RetrieveContextsResponse: @@ -3258,7 +3258,7 @@ async def _update_config( async def retrieve_contexts( self, *, - vertex_rag_store: Optional[types.VertexRagStoreOrDict] = None, + vertex_rag_store: types.VertexRagStoreOrDict, query: types.RagQueryOrDict, config: Optional[types.RetrieveContextsConfigOrDict] = None, ) -> types.RetrieveContextsResponse: