diff --git a/notebooks/community/model_garden/model_garden_rag.ipynb b/notebooks/community/model_garden/model_garden_rag.ipynb index 8946de264..e03085855 100644 --- a/notebooks/community/model_garden/model_garden_rag.ipynb +++ b/notebooks/community/model_garden/model_garden_rag.ipynb @@ -31,7 +31,7 @@ "source": [ "## Model Garden RAG API\n", "\n", - "Last updated: 9/11/2024\n", + "Last updated: 9/17/2024\n", "\n", "## Onboarding\n", "If you have any questions, please reach out to *Vertex RAG API * team vertex-rag-eng@google.com, for the onboarding process.\n", @@ -181,6 +181,31 @@ " api_key=API_KEY,\n", ")\n", "\n", + "# Configure a Vertex Feature Store Database Instance for the corpus\n", + "RESOURCE_NAME = \"feature-view-resource-name\" # @param {type:\"string\"}\n", + "\n", + "vector_db = rag.VertexFeatureStore(\n", + " resource_name=RESOURCE_NAME,\n", + ")\n", + "\n", + "# Configure a Pinecone Vector Database Instance for the corpus\n", + "INDEX_NAME = \"pinecone-index-name\" # @param {type:\"string\"}\n", + "API_KEY = \"your-secret-manager-resource-name\" # @param {type:\"string\"}\n", + "\n", + "vector_db = rag.Pinecone(\n", + " index_name=INDEX_NAME,\n", + " api_key=API_KEY,\n", + ")\n", + "\n", + "# Configure a Vertex Vector Search Database Instance for the corpus\n", + "INDEX_ENDPOINT = \"vector-search-index-endpoint\" # @param {type:\"string\"}\n", + "INDEX = \"vector-search-index-resource-name\" # @param {type:\"string\"}\n", + "\n", + "vector_db = rag.VertexVectorSearch(\n", + " index_endpoint=INDEX_ENDPOINT,\n", + " index=INDEX,\n", + ")\n", + "\n", "\n", "# Name your corpus\n", "DISPLAY_NAME = \"your-corpus-name\" # @param {type:\"string\"}\n",