Spring/spring-ai
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Add a default model to the initial autoconfiguration chat properties

Browse Source
This commit is contained in:
Ivan
2024-03-31 20:22:27 +00:00
committed by Christian Tzolov
parent fefc6f4173
commit 4dbc4c8a8f
1 changed files with 4 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -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() {