From 16c531c36c143babdda59eff32b24ac05b291bc6 Mon Sep 17 00:00:00 2001 From: Christian Tzolov Date: Tue, 18 Jun 2024 18:39:57 +0200 Subject: [PATCH] Azure ITs adjustment: prove streaming does chunks the response content --- .../openai/function/AzureOpenAiChatModelFunctionCallIT.java | 4 +++- .../autoconfigure/azure/AzureOpenAiAutoConfigurationIT.java | 2 +- 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/models/spring-ai-azure-openai/src/test/java/org/springframework/ai/azure/openai/function/AzureOpenAiChatModelFunctionCallIT.java b/models/spring-ai-azure-openai/src/test/java/org/springframework/ai/azure/openai/function/AzureOpenAiChatModelFunctionCallIT.java index 219ba6114..f4349e172 100644 --- a/models/spring-ai-azure-openai/src/test/java/org/springframework/ai/azure/openai/function/AzureOpenAiChatModelFunctionCallIT.java +++ b/models/spring-ai-azure-openai/src/test/java/org/springframework/ai/azure/openai/function/AzureOpenAiChatModelFunctionCallIT.java @@ -113,7 +113,9 @@ class AzureOpenAiChatModelFunctionCallIT { .map(AssistantMessage::getContent) .collect(Collectors.joining()); logger.info("Response: {}", content); - assertThat(counter.get()).isGreaterThan(2); + + assertThat(counter.get()).isGreaterThan(30).as("The response should be chunked in more than 30 messages"); + assertThat(content).containsAnyOf("30.0", "30"); assertThat(content).containsAnyOf("10.0", "10"); assertThat(content).containsAnyOf("15.0", "15"); diff --git a/spring-ai-spring-boot-autoconfigure/src/test/java/org/springframework/ai/autoconfigure/azure/AzureOpenAiAutoConfigurationIT.java b/spring-ai-spring-boot-autoconfigure/src/test/java/org/springframework/ai/autoconfigure/azure/AzureOpenAiAutoConfigurationIT.java index 222eddfff..2bc258997 100644 --- a/spring-ai-spring-boot-autoconfigure/src/test/java/org/springframework/ai/autoconfigure/azure/AzureOpenAiAutoConfigurationIT.java +++ b/spring-ai-spring-boot-autoconfigure/src/test/java/org/springframework/ai/autoconfigure/azure/AzureOpenAiAutoConfigurationIT.java @@ -92,7 +92,7 @@ public class AzureOpenAiAutoConfigurationIT { Flux response = chatModel.stream(new Prompt(List.of(userMessage, systemMessage))); List responses = response.collectList().block(); - assertThat(responses.size()).isGreaterThan(1); + assertThat(responses.size()).isGreaterThan(10); String stitchedResponseContent = responses.stream() .map(ChatResponse::getResults)