diff --git a/models/spring-ai-vertex-ai-gemini/src/main/java/org/springframework/ai/vertexai/gemini/VertexAiGeminiChatModel.java b/models/spring-ai-vertex-ai-gemini/src/main/java/org/springframework/ai/vertexai/gemini/VertexAiGeminiChatModel.java index 30c2b7c79..7607e76ee 100644 --- a/models/spring-ai-vertex-ai-gemini/src/main/java/org/springframework/ai/vertexai/gemini/VertexAiGeminiChatModel.java +++ b/models/spring-ai-vertex-ai-gemini/src/main/java/org/springframework/ai/vertexai/gemini/VertexAiGeminiChatModel.java @@ -714,11 +714,8 @@ public class VertexAiGeminiChatModel implements ChatModel, DisposableBean { } if (prompt.getOptions() instanceof VertexAiGeminiChatOptions options && options.getGoogleSearchRetrieval()) { - // final var googleSearchRetrieval = GoogleSearchRetrieval.newBuilder().getDefaultInstanceForType(); var googleSearch = GoogleSearch.newBuilder().getDefaultInstanceForType(); - final var googleSearchRetrievalTool = Tool.newBuilder() - .setGoogleSearch(googleSearch) - .build(); + final var googleSearchRetrievalTool = Tool.newBuilder().setGoogleSearch(googleSearch).build(); tools.add(googleSearchRetrievalTool); } diff --git a/models/spring-ai-vertex-ai-gemini/src/test/java/org/springframework/ai/vertexai/gemini/VertexAiGeminiChatModelIT.java b/models/spring-ai-vertex-ai-gemini/src/test/java/org/springframework/ai/vertexai/gemini/VertexAiGeminiChatModelIT.java index c7adc68b1..3dec24540 100644 --- a/models/spring-ai-vertex-ai-gemini/src/test/java/org/springframework/ai/vertexai/gemini/VertexAiGeminiChatModelIT.java +++ b/models/spring-ai-vertex-ai-gemini/src/test/java/org/springframework/ai/vertexai/gemini/VertexAiGeminiChatModelIT.java @@ -103,9 +103,9 @@ class VertexAiGeminiChatModelIT { @Test void googleSearchToolFlash() { Prompt prompt = createPrompt(VertexAiGeminiChatOptions.builder() - .model(ChatModel.GEMINI_2_0_FLASH) - .googleSearchRetrieval(true) - .build()); + .model(ChatModel.GEMINI_2_0_FLASH) + .googleSearchRetrieval(true) + .build()); ChatResponse response = this.chatModel.call(prompt); assertThat(response.getResult().getOutput().getText()).containsAnyOf("Blackbeard", "Bartholomew", "Bob"); }