From 4dbc4c8a8f31e7928e95c186fe2c17a829d71322 Mon Sep 17 00:00:00 2001 From: Ivan Date: Sun, 31 Mar 2024 20:22:27 +0000 Subject: [PATCH] Add a default model to the initial autoconfiguration chat properties --- .../vertexai/gemini/VertexAiGeminiChatProperties.java | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/spring-ai-spring-boot-autoconfigure/src/main/java/org/springframework/ai/autoconfigure/vertexai/gemini/VertexAiGeminiChatProperties.java b/spring-ai-spring-boot-autoconfigure/src/main/java/org/springframework/ai/autoconfigure/vertexai/gemini/VertexAiGeminiChatProperties.java index a47f4cf32..866f4f352 100644 --- a/spring-ai-spring-boot-autoconfigure/src/main/java/org/springframework/ai/autoconfigure/vertexai/gemini/VertexAiGeminiChatProperties.java +++ b/spring-ai-spring-boot-autoconfigure/src/main/java/org/springframework/ai/autoconfigure/vertexai/gemini/VertexAiGeminiChatProperties.java @@ -15,6 +15,7 @@ */ package org.springframework.ai.autoconfigure.vertexai.gemini; +import org.springframework.ai.vertexai.gemini.VertexAiGeminiChatClient; import org.springframework.ai.vertexai.gemini.VertexAiGeminiChatOptions; import org.springframework.boot.context.properties.ConfigurationProperties; @@ -29,12 +30,15 @@ public class VertexAiGeminiChatProperties { public static final String CONFIG_PREFIX = "spring.ai.vertex.ai.gemini.chat"; + public static final String DEFAULT_MODEL = VertexAiGeminiChatClient.ChatModel.GEMINI_PRO_VISION.getValue(); + /** * Vertex AI Gemini API generative options. */ private VertexAiGeminiChatOptions options = VertexAiGeminiChatOptions.builder() .withTemperature(0.7f) .withCandidateCount(1) + .withModel(DEFAULT_MODEL) .build(); public VertexAiGeminiChatOptions getOptions() {