From 8703582160a82af34ec33f9a67a433bba894ef49 Mon Sep 17 00:00:00 2001 From: Christian Tzolov Date: Sat, 2 Mar 2024 22:17:36 +0100 Subject: [PATCH] Fix broken tests after updateding the models from DALL-E-2 to DALL-E-3 --- .../springframework/ai/openai/image/OpenAiImageClientIT.java | 2 +- .../ai/autoconfigure/openai/OpenAiAutoConfigurationIT.java | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/models/spring-ai-openai/src/test/java/org/springframework/ai/openai/image/OpenAiImageClientIT.java b/models/spring-ai-openai/src/test/java/org/springframework/ai/openai/image/OpenAiImageClientIT.java index bfdcbff8a..7cf49e8fc 100644 --- a/models/spring-ai-openai/src/test/java/org/springframework/ai/openai/image/OpenAiImageClientIT.java +++ b/models/spring-ai-openai/src/test/java/org/springframework/ai/openai/image/OpenAiImageClientIT.java @@ -58,7 +58,7 @@ public class OpenAiImageClientIT extends AbstractIT { OpenAiImageGenerationMetadata openAiImageGenerationMetadata = (OpenAiImageGenerationMetadata) imageGenerationMetadata; assertThat(openAiImageGenerationMetadata).isNotNull(); - assertThat(openAiImageGenerationMetadata.getRevisedPrompt()).isNull(); + assertThat(openAiImageGenerationMetadata.getRevisedPrompt()).isNotBlank(); } } diff --git a/spring-ai-spring-boot-autoconfigure/src/test/java/org/springframework/ai/autoconfigure/openai/OpenAiAutoConfigurationIT.java b/spring-ai-spring-boot-autoconfigure/src/test/java/org/springframework/ai/autoconfigure/openai/OpenAiAutoConfigurationIT.java index 2374b0c1b..1ec2d1859 100644 --- a/spring-ai-spring-boot-autoconfigure/src/test/java/org/springframework/ai/autoconfigure/openai/OpenAiAutoConfigurationIT.java +++ b/spring-ai-spring-boot-autoconfigure/src/test/java/org/springframework/ai/autoconfigure/openai/OpenAiAutoConfigurationIT.java @@ -92,7 +92,7 @@ public class OpenAiAutoConfigurationIT { @Test void generateImage() { - contextRunner.withPropertyValues("spring.ai.openai.image.options.size=256x256").run(context -> { + contextRunner.withPropertyValues("spring.ai.openai.image.options.size=1024x1024").run(context -> { OpenAiImageClient client = context.getBean(OpenAiImageClient.class); ImageResponse imageResponse = client.call(new ImagePrompt("forest")); assertThat(imageResponse.getResults()).hasSize(1);