From 7c60d3efa5676e05d20a64716f9b5dc985c190e4 Mon Sep 17 00:00:00 2001 From: Soby Chacko Date: Wed, 9 Apr 2025 13:31:17 -0400 Subject: [PATCH] Model update in VertexAiGeminiChatModelIT from GEMINI_2_5_PRO to GEMINI_2_0_FLASH Update gemini-2.5-pro-exp-03-25 to gemini-2.5-pro-exp-03-28 Signed-off-by: Soby Chacko --- .../ai/vertexai/gemini/VertexAiGeminiChatModel.java | 2 +- .../ai/vertexai/gemini/VertexAiGeminiChatModelIT.java | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/models/spring-ai-vertex-ai-gemini/src/main/java/org/springframework/ai/vertexai/gemini/VertexAiGeminiChatModel.java b/models/spring-ai-vertex-ai-gemini/src/main/java/org/springframework/ai/vertexai/gemini/VertexAiGeminiChatModel.java index 94dd1a0d859..a356f88128c 100644 --- a/models/spring-ai-vertex-ai-gemini/src/main/java/org/springframework/ai/vertexai/gemini/VertexAiGeminiChatModel.java +++ b/models/spring-ai-vertex-ai-gemini/src/main/java/org/springframework/ai/vertexai/gemini/VertexAiGeminiChatModel.java @@ -869,7 +869,7 @@ public enum ChatModel implements ChatModelDescription { GEMINI_2_0_FLASH_LIGHT("gemini-2.0-flash-lite"), - GEMINI_2_5_PRO("gemini-2.5-pro-exp-03-25"); + GEMINI_2_5_PRO("gemini-2.5-pro-exp-03-28"); public final String value; diff --git a/models/spring-ai-vertex-ai-gemini/src/test/java/org/springframework/ai/vertexai/gemini/VertexAiGeminiChatModelIT.java b/models/spring-ai-vertex-ai-gemini/src/test/java/org/springframework/ai/vertexai/gemini/VertexAiGeminiChatModelIT.java index e2c42ac4846..1b6e3e8a00e 100644 --- a/models/spring-ai-vertex-ai-gemini/src/test/java/org/springframework/ai/vertexai/gemini/VertexAiGeminiChatModelIT.java +++ b/models/spring-ai-vertex-ai-gemini/src/test/java/org/springframework/ai/vertexai/gemini/VertexAiGeminiChatModelIT.java @@ -304,7 +304,7 @@ public VertexAiGeminiChatModel vertexAiEmbedding(VertexAI vertexAi) { return VertexAiGeminiChatModel.builder() .vertexAI(vertexAi) .defaultOptions(VertexAiGeminiChatOptions.builder() - .model(VertexAiGeminiChatModel.ChatModel.GEMINI_2_5_PRO) + .model(VertexAiGeminiChatModel.ChatModel.GEMINI_2_0_FLASH) .build()) .build(); }