Skip to content

Commit 5914550

Browse files
committed
Use Gemini 2.5 Pro in tests instead of Gemini 3 Pro Preview
Signed-off-by: ddobrin <[email protected]>
1 parent 6827503 commit 5914550

File tree

2 files changed

+7
-7
lines changed

2 files changed

+7
-7
lines changed

models/spring-ai-google-genai/src/test/java/org/springframework/ai/google/genai/GoogleGenAiChatModelObservationApiKeyIT.java

Lines changed: 5 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -32,6 +32,7 @@
3232
import org.springframework.ai.chat.observation.ChatModelObservationDocumentation;
3333
import org.springframework.ai.chat.observation.DefaultChatModelObservationConvention;
3434
import org.springframework.ai.chat.prompt.Prompt;
35+
import org.springframework.ai.google.genai.GoogleGenAiChatModel.ChatModel;
3536
import org.springframework.ai.observation.conventions.AiOperationType;
3637
import org.springframework.ai.observation.conventions.AiProvider;
3738
import org.springframework.beans.factory.annotation.Autowired;
@@ -64,7 +65,7 @@ void beforeEach() {
6465
void observationForChatOperation() {
6566

6667
var options = GoogleGenAiChatOptions.builder()
67-
.model(GoogleGenAiChatModel.ChatModel.GEMINI_3_PRO_PREVIEW.getValue())
68+
.model(ChatModel.GEMINI_2_5_PRO.getValue())
6869
.temperature(0.7)
6970
.stopSequences(List.of("this-is-the-end"))
7071
.maxOutputTokens(2048)
@@ -86,7 +87,7 @@ void observationForChatOperation() {
8687
void observationForStreamingOperation() {
8788

8889
var options = GoogleGenAiChatOptions.builder()
89-
.model(GoogleGenAiChatModel.ChatModel.GEMINI_3_PRO_PREVIEW.getValue())
90+
.model(ChatModel.GEMINI_2_5_PRO.getValue())
9091
.temperature(0.7)
9192
.stopSequences(List.of("this-is-the-end"))
9293
.maxOutputTokens(2048)
@@ -126,7 +127,7 @@ private void validate(ChatResponseMetadata responseMetadata) {
126127
AiProvider.GOOGLE_GENAI_AI.value())
127128
.hasLowCardinalityKeyValue(
128129
ChatModelObservationDocumentation.LowCardinalityKeyNames.REQUEST_MODEL.asString(),
129-
GoogleGenAiChatModel.ChatModel.GEMINI_3_PRO_PREVIEW.getValue())
130+
ChatModel.GEMINI_2_5_PRO.getValue())
130131
.hasHighCardinalityKeyValue(
131132
ChatModelObservationDocumentation.HighCardinalityKeyNames.REQUEST_MAX_TOKENS.asString(), "2048")
132133
.hasHighCardinalityKeyValue(
@@ -174,9 +175,7 @@ public GoogleGenAiChatModel vertexAiEmbedding(Client genAiClient, TestObservatio
174175
return GoogleGenAiChatModel.builder()
175176
.genAiClient(genAiClient)
176177
.observationRegistry(observationRegistry)
177-
.defaultOptions(GoogleGenAiChatOptions.builder()
178-
.model(GoogleGenAiChatModel.ChatModel.GEMINI_3_PRO_PREVIEW)
179-
.build())
178+
.defaultOptions(GoogleGenAiChatOptions.builder().model(ChatModel.GEMINI_2_5_PRO).build())
180179
.build();
181180
}
182181

models/spring-ai-google-genai/src/test/java/org/springframework/ai/google/genai/GoogleGenAiRetryTests.java

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -26,6 +26,7 @@
2626
import org.mockito.Mock;
2727
import org.mockito.junit.jupiter.MockitoExtension;
2828

29+
import org.springframework.ai.google.genai.GoogleGenAiChatModel.ChatModel;
2930
import org.springframework.ai.retry.RetryUtils;
3031
import org.springframework.core.retry.RetryListener;
3132
import org.springframework.core.retry.RetryPolicy;
@@ -61,7 +62,7 @@ public void setUp() {
6162
GoogleGenAiChatOptions.builder()
6263
.temperature(0.7)
6364
.topP(1.0)
64-
.model(GoogleGenAiChatModel.ChatModel.GEMINI_3_PRO_PREVIEW.getValue())
65+
.model(ChatModel.GEMINI_2_5_PRO.getValue())
6566
.build(),
6667
this.retryTemplate);
6768

0 commit comments

Comments
 (0)