Make ImageOptions non-null when constructing ImagePrompt

- Pass non-null ImageOptions when instantiating ImagePrompt

Signed-off-by: Ilayaperumal Gopinathan <ilayaperumal.gopinathan@broadcom.com>
This commit is contained in:
Ilayaperumal Gopinathan
2025-04-28 18:51:10 +01:00
parent 3b752bbb54
commit 0c0787b849

View File

@@ -29,12 +29,7 @@ public class ImagePrompt implements ModelRequest<List<ImageMessage>> {
private ImageOptions imageModelOptions;
public ImagePrompt(List<ImageMessage> messages) {
this.messages = messages;
}
public ImagePrompt(List<ImageMessage> 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<List<ImageMessage>> {
this(new ImageMessage(instructions), ImageOptionsBuilder.builder().build());
}
public ImagePrompt(List<ImageMessage> messages, ImageOptions imageModelOptions) {
this.messages = messages;
this.imageModelOptions = imageModelOptions != null ? imageModelOptions : ImageOptionsBuilder.builder().build();
}
@Override
public List<ImageMessage> getInstructions() {
return this.messages;