Add Feature Store, Vector Search, Pinecone Vector DB option for corpu…

…s creation to rag notebook

PiperOrigin-RevId: 675703407

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 [email protected], 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",