Skip to content

Commit 2c96d24

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 2c96d24

File tree

2 files changed

+7
-5
lines changed

2 files changed

+7
-5
lines changed

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

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -25,6 +25,7 @@
2525
import org.junit.jupiter.api.BeforeEach;
2626
import org.junit.jupiter.api.Test;
2727
import org.junit.jupiter.api.condition.EnabledIfEnvironmentVariable;
28+
import org.springframework.ai.google.genai.GoogleGenAiChatModel.ChatModel;
2829
import reactor.core.publisher.Flux;
2930

3031
import org.springframework.ai.chat.metadata.ChatResponseMetadata;
@@ -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(
@@ -175,7 +176,7 @@ public GoogleGenAiChatModel vertexAiEmbedding(Client genAiClient, TestObservatio
175176
.genAiClient(genAiClient)
176177
.observationRegistry(observationRegistry)
177178
.defaultOptions(GoogleGenAiChatOptions.builder()
178-
.model(GoogleGenAiChatModel.ChatModel.GEMINI_3_PRO_PREVIEW)
179+
.model(ChatModel.GEMINI_2_5_PRO)
179180
.build())
180181
.build();
181182
}

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)