Skip to content

Commit 4dbc4c8

Browse files
zucchivantzolov
authored andcommitted
Add a default model to the initial autoconfiguration chat properties
1 parent fefc6f4 commit 4dbc4c8

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

spring-ai-spring-boot-autoconfigure/src/main/java/org/springframework/ai/autoconfigure/vertexai/gemini/VertexAiGeminiChatProperties.java

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -15,6 +15,7 @@
1515
*/
1616
package org.springframework.ai.autoconfigure.vertexai.gemini;
1717

18+
import org.springframework.ai.vertexai.gemini.VertexAiGeminiChatClient;
1819
import org.springframework.ai.vertexai.gemini.VertexAiGeminiChatOptions;
1920
import org.springframework.boot.context.properties.ConfigurationProperties;
2021

@@ -29,12 +30,15 @@ public class VertexAiGeminiChatProperties {
2930

3031
public static final String CONFIG_PREFIX = "spring.ai.vertex.ai.gemini.chat";
3132

33+
public static final String DEFAULT_MODEL = VertexAiGeminiChatClient.ChatModel.GEMINI_PRO_VISION.getValue();
34+
3235
/**
3336
* Vertex AI Gemini API generative options.
3437
*/
3538
private VertexAiGeminiChatOptions options = VertexAiGeminiChatOptions.builder()
3639
.withTemperature(0.7f)
3740
.withCandidateCount(1)
41+
.withModel(DEFAULT_MODEL)
3842
.build();
3943

4044
public VertexAiGeminiChatOptions getOptions() {

0 commit comments

Comments
 (0)