From 6eeae1fbc108bf51f8d901565c2056e56d762d43 Mon Sep 17 00:00:00 2001 From: jacoblee93 Date: Fri, 13 Dec 2024 21:02:07 -0800 Subject: [PATCH] Use 1.5 pro --- .../docs/integrations/chat/google_vertex_ai.ipynb | 8 +++++--- 1 file changed, 5 insertions(+), 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 index c7b1dc9742af..056e0d22aa80 100644 --- 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",