Use 1.5 pro

langchain-ai · Dec 14, 2024 · 6eeae1f · 6eeae1f
1 parent 1967663
commit 6eeae1f
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/docs/core_docs/docs/integrations/chat/google_vertex_ai.ipynb b/docs/core_docs/docs/integrations/chat/google_vertex_ai.ipynb
@@ -198,7 +198,9 @@
       "source": [
         "## Tool Calling with Google Search Retrieval\n",
         "\n",
-        "It is possible to call the model with a Google search tool which you can use to [ground](https://cloud.google.com/vertex-ai/generative-ai/docs/model-reference/grounding) content generation with real-world information and reduce hallucinations.  \n",
+        "It is possible to call the model with a Google search tool which you can use to [ground](https://cloud.google.com/vertex-ai/generative-ai/docs/model-reference/grounding) content generation with real-world information and reduce hallucinations.\n",
+        "\n",
+        "Grounding is currently not supported by `gemini-2.0-flash-exp`.\n",
         "\n",
         "You can choose to either ground using Google Search or by using a custom data store. Here are examples of both:  "
       ]
@@ -241,7 +243,7 @@
         "};\n",
         "\n",
         "const searchRetrievalModel = new ChatVertexAI({\n",
-        "  model: \"gemini-2.0-flash-exp\",\n",
+        "  model: \"gemini-1.5-pro\",\n",
         "  temperature: 0,\n",
         "  maxRetries: 0,\n",
         "}).bindTools([searchRetrievalTool]);\n",
@@ -301,7 +303,7 @@
         "};\n",
         "\n",
         "const searchRetrievalModelWithDataset = new ChatVertexAI({\n",
-        "  model: \"gemini-2.0-flash-exp\",\n",
+        "  model: \"gemini-1.5-pro\",\n",
         "  temperature: 0,\n",
         "  maxRetries: 0,\n",
         "}).bindTools([searchRetrievalToolWithDataset]);\n",