Clean up tests with Flash 2.0 and 2.5

Signed-off-by: ddobrin <[email protected]>

Author: ddobrin

models/spring-ai-google-genai/src/test/java/org/springframework/ai/vertexai/gemini/VertexAiGeminiChatModelIT.java

@@ -368,7 +368,7 @@ void jsonTextToolCallingTest() {
 			.genAiClient(genAiClient())
 			.toolCallingManager(toolCallingManager)
 			.defaultOptions(VertexAiGeminiChatOptions.builder()
-				.model(VertexAiGeminiChatModel.ChatModel.GEMINI_2_5_FLASH)
+				.model(VertexAiGeminiChatModel.ChatModel.GEMINI_2_0_FLASH)
 				.temperature(0.1)
 				.build())
 			.build();
@@ -384,7 +384,7 @@ void jsonTextToolCallingTest() {
 			.content();
 
 		assertThat(response).isNotEmpty();
-		assertThat(response).contains("2025-05-08T10:10:10+02:00[Europe/Berlin]");
+		assertThat(response).contains("2025-05-08T10:10:10+02:00");
 	}
 
 	/**

models/spring-ai-google-genai/src/test/java/org/springframework/ai/vertexai/gemini/tool/VertexAiGeminiChatModelToolCallingIT.java

@@ -170,7 +170,7 @@ public void functionCallTestInferredOpenApiSchemaStream() {
 	}
 
 	@Test
-	public void functionCallUsageTestInferredOpenApiSchemaStream() {
+	public void functionCallUsageTestInferredOpenApiSchemaStreamFlash20() {
 
 		UserMessage userMessage = new UserMessage(
 				"What's the weather like in San Francisco, Paris and in Tokyo? Return the temperature in Celsius.");
@@ -204,6 +204,41 @@ public void functionCallUsageTestInferredOpenApiSchemaStream() {
 
 	}
 
+	@Test
+	public void functionCallUsageTestInferredOpenApiSchemaStreamFlash25() {
+
+		UserMessage userMessage = new UserMessage(
+				"What's the weather like in San Francisco, Paris and in Tokyo? Return the temperature in Celsius.");
+
+		List<Message> messages = new ArrayList<>(List.of(userMessage));
+
+		var promptOptions = VertexAiGeminiChatOptions.builder()
+				.model(VertexAiGeminiChatModel.ChatModel.GEMINI_2_5_FLASH)
+				.toolCallbacks(List.of(
+						FunctionToolCallback.builder("get_current_weather", new MockWeatherService())
+								.description("Get the current weather in a given location.")
+								.inputType(MockWeatherService.Request.class)
+								.build(),
+						FunctionToolCallback.builder("get_payment_status", new PaymentStatus())
+								.description(
+										"Retrieves the payment status for transaction. For example what is the payment status for transaction 700?")
+								.inputType(PaymentInfoRequest.class)
+								.build()))
+				.build();
+
+		Flux<ChatResponse> response = this.chatModel.stream(new Prompt(messages, promptOptions));
+
+		ChatResponse chatResponse = response.blockLast();
+
+		logger.info("Response: {}", chatResponse);
+
+		assertThat(chatResponse).isNotNull();
+		assertThat(chatResponse.getMetadata()).isNotNull();
+		assertThat(chatResponse.getMetadata().getUsage()).isNotNull();
+		assertThat(chatResponse.getMetadata().getUsage().getTotalTokens()).isGreaterThan(150).isLessThan(600);
+
+	}
+
 	public record PaymentInfoRequest(String id) {
 
 	}