Fix AzureOpenAiChatModelFunctionCallIT#streamFunctionCallUsageTest

  - Fix the test to identify the right usage metadata from the stream content

Signed-off-by: Ilayaperumal Gopinathan <[email protected]>
Signed-off-by: 家娃 <[email protected]>

Author: ilayaperumalg

models/spring-ai-azure-openai/src/test/java/org/springframework/ai/azure/openai/function/AzureOpenAiChatModelFunctionCallIT.java

@@ -163,12 +163,18 @@ void streamFunctionCallUsageTest() {
 			.streamOptions(streamOptions)
 			.build();
 
-		Flux<ChatResponse> response = this.chatModel.stream(new Prompt(messages, promptOptions));
+		List<ChatResponse> responses = this.chatModel.stream(new Prompt(messages, promptOptions)).collectList().block();
+
+		assertThat(responses).isNotEmpty();
+
+		ChatResponse finalResponse = responses.get(responses.size() - 2);
+
+		logger.info("Final Response: {}", finalResponse);
 
-		ChatResponse chatResponse = response.last().block();
-		logger.info("Response: {}", chatResponse);
+		assertThat(finalResponse.getMetadata()).isNotNull();
+		assertThat(finalResponse.getMetadata().getUsage()).isNotNull();
 
-		assertThat(chatResponse.getMetadata().getUsage().getTotalTokens()).isGreaterThan(600).isLessThan(800);
+		assertThat(finalResponse.getMetadata().getUsage().getTotalTokens()).isGreaterThan(600).isLessThan(800);
 
 	}