diff --git a/spring-ai-model/src/main/java/org/springframework/ai/image/ImagePrompt.java b/spring-ai-model/src/main/java/org/springframework/ai/image/ImagePrompt.java index a212c2cf4..6cc6efd8d 100644 --- a/spring-ai-model/src/main/java/org/springframework/ai/image/ImagePrompt.java +++ b/spring-ai-model/src/main/java/org/springframework/ai/image/ImagePrompt.java @@ -29,12 +29,7 @@ public class ImagePrompt implements ModelRequest> { private ImageOptions imageModelOptions; public ImagePrompt(List messages) { - this.messages = messages; - } - - public ImagePrompt(List messages, ImageOptions imageModelOptions) { - this.messages = messages; - this.imageModelOptions = imageModelOptions; + this(messages, ImageOptionsBuilder.builder().build()); } public ImagePrompt(ImageMessage imageMessage, ImageOptions imageOptions) { @@ -49,6 +44,11 @@ public class ImagePrompt implements ModelRequest> { this(new ImageMessage(instructions), ImageOptionsBuilder.builder().build()); } + public ImagePrompt(List messages, ImageOptions imageModelOptions) { + this.messages = messages; + this.imageModelOptions = imageModelOptions != null ? imageModelOptions : ImageOptionsBuilder.builder().build(); + } + @Override public List getInstructions() { return this.messages;