From 39b7fe58263dd5237fa04f6fb118d4ac3fe84c50 Mon Sep 17 00:00:00 2001 From: Darshan Mehta Date: Tue, 17 Mar 2026 10:32:48 -0700 Subject: [PATCH] feat: RAG - Remove VectorDb defaulting in SDK PiperOrigin-RevId: 885094404 --- vertexai/preview/rag/utils/_gapic_utils.py | 4 +--- vertexai/rag/utils/_gapic_utils.py | 4 ++-- 2 files changed, 3 insertions(+), 5 deletions(-) diff --git a/vertexai/preview/rag/utils/_gapic_utils.py b/vertexai/preview/rag/utils/_gapic_utils.py index 4150111c60..6eae67fdfa 100644 --- a/vertexai/preview/rag/utils/_gapic_utils.py +++ b/vertexai/preview/rag/utils/_gapic_utils.py @@ -869,9 +869,7 @@ def set_vector_db( ) -> None: """Sets the vector db configuration for the rag corpus.""" if vector_db is None: - rag_corpus.rag_vector_db_config = GapicRagVectorDbConfig( - rag_managed_db=GapicRagVectorDbConfig.RagManagedDb(), - ) + rag_corpus.rag_vector_db_config = GapicRagVectorDbConfig() elif isinstance(vector_db, RagManagedDb): rag_corpus.rag_vector_db_config = GapicRagVectorDbConfig( rag_managed_db=_convert_rag_managed_db_to_gapic(vector_db) diff --git a/vertexai/rag/utils/_gapic_utils.py b/vertexai/rag/utils/_gapic_utils.py index 3ee39a7a0f..dfcd46989c 100644 --- a/vertexai/rag/utils/_gapic_utils.py +++ b/vertexai/rag/utils/_gapic_utils.py @@ -648,7 +648,7 @@ def set_backend_config( if backend_config.vector_db is not None: vector_config = backend_config.vector_db - if vector_config is None or isinstance(vector_config, RagManagedDb): + if isinstance(vector_config, RagManagedDb): rag_corpus.vector_db_config.rag_managed_db.CopyFrom( GapicRagVectorDbConfig.RagManagedDb() ) @@ -668,7 +668,7 @@ def set_backend_config( rag_corpus.vector_db_config.api_auth.api_key_config.api_key_secret_version = ( api_key ) - else: + elif vector_config is not None: raise TypeError( "backend_config must be a VertexFeatureStore," "RagManagedDb, or Pinecone."