From c07e85295d413fc2993afac2d3b7aae702b48962 Mon Sep 17 00:00:00 2001 From: Gareth Evans Date: Mon, 3 Mar 2025 16:42:53 +0000 Subject: [PATCH] chore: add missing toolCallbacks(vargs) to vertex chat options Signed-off-by: Gareth Evans --- .../ai/vertexai/gemini/VertexAiGeminiChatOptions.java | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/models/spring-ai-vertex-ai-gemini/src/main/java/org/springframework/ai/vertexai/gemini/VertexAiGeminiChatOptions.java b/models/spring-ai-vertex-ai-gemini/src/main/java/org/springframework/ai/vertexai/gemini/VertexAiGeminiChatOptions.java index 246530b1725..ae8ca2aeadc 100644 --- a/models/spring-ai-vertex-ai-gemini/src/main/java/org/springframework/ai/vertexai/gemini/VertexAiGeminiChatOptions.java +++ b/models/spring-ai-vertex-ai-gemini/src/main/java/org/springframework/ai/vertexai/gemini/VertexAiGeminiChatOptions.java @@ -16,6 +16,7 @@ package org.springframework.ai.vertexai.gemini; +import java.util.Arrays; import java.util.ArrayList; import java.util.HashMap; import java.util.HashSet; @@ -456,6 +457,12 @@ public Builder toolCallbacks(List toolCallbacks) { return this; } + public Builder toolCallbacks(FunctionCallback... toolCallbacks) { + Assert.notNull(toolCallbacks, "toolCallbacks cannot be null"); + this.options.toolCallbacks.addAll(Arrays.asList(toolCallbacks)); + return this; + } + @Deprecated public Builder functions(Set functionNames) { return this.toolNames(functionNames);