Spring/spring-ai
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Fix broken tests after updateding the models from DALL-E-2 to DALL-E-3

Browse Source
This commit is contained in:
Christian Tzolov
2024-03-02 22:17:36 +01:00
parent e819b7c313
commit 8703582160
2 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
models/spring-ai-openai/src/test/java/org/springframework/ai/openai/image/OpenAiImageClientIT.java
View File

@@ -58,7 +58,7 @@ public class OpenAiImageClientIT extends AbstractIT {
OpenAiImageGenerationMetadata openAiImageGenerationMetadata = (OpenAiImageGenerationMetadata) imageGenerationMetadata;
assertThat(openAiImageGenerationMetadata).isNotNull();
assertThat(openAiImageGenerationMetadata.getRevisedPrompt()).isNull();
assertThat(openAiImageGenerationMetadata.getRevisedPrompt()).isNotBlank();
}
}

2
spring-ai-spring-boot-autoconfigure/src/test/java/org/springframework/ai/autoconfigure/openai/OpenAiAutoConfigurationIT.java
View File

@@ -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);