11package com .volcengine .ark .runtime .model .images .generation ;
22
33import com .fasterxml .jackson .annotation .JsonProperty ;
4- import com .volcengine .ark .runtime .model .content .generation .CreateContentGenerationTaskRequest ;
5-
6- import java .util .List ;
74
85public class GenerateImagesRequest {
96
@@ -13,6 +10,9 @@ public class GenerateImagesRequest {
1310 @ JsonProperty ("prompt" )
1411 private String prompt ;
1512
13+ @ JsonProperty ("image" )
14+ private String image ;
15+
1616 @ JsonProperty ("response_format" )
1717 private String responseFormat ;
1818
@@ -28,12 +28,16 @@ public class GenerateImagesRequest {
2828 @ JsonProperty ("watermark" )
2929 private Boolean watermark ;
3030
31+ @ JsonProperty ("optimize_prompt" )
32+ private Boolean optimizePrompt ;
33+
3134 public GenerateImagesRequest () {
3235 }
3336
34- public GenerateImagesRequest (String model , String prompt , String responseFormat , Integer seed , Double guidanceScale , String size ) {
37+ public GenerateImagesRequest (String model , String prompt , String image , String responseFormat , Integer seed , Double guidanceScale , String size ) {
3538 this .model = model ;
3639 this .prompt = prompt ;
40+ this .image = image ;
3741 this .responseFormat = responseFormat ;
3842 this .seed = seed ;
3943 this .guidanceScale = guidanceScale ;
@@ -57,6 +61,14 @@ public void setPrompt(String prompt) {
5761 this .prompt = prompt ;
5862 }
5963
64+ public String getImage () {
65+ return this .image ;
66+ }
67+
68+ public void setImage (String image ) {
69+ this .image = image ;
70+ }
71+
6072 public String getResponseFormat () {
6173 return this .responseFormat ;
6274 }
@@ -97,16 +109,26 @@ public void setWatermark(Boolean watermark) {
97109 this .watermark = watermark ;
98110 }
99111
112+ public Boolean getOptimizePrompt () {
113+ return this .optimizePrompt ;
114+ }
115+
116+ public void setOptimizePrompt (Boolean optimizePrompt ) {
117+ this .optimizePrompt = optimizePrompt ;
118+ }
119+
100120 @ Override
101121 public String toString () {
102122 return "GenerateImagesRequest{" +
103123 "model='" + model + '\'' +
104124 ", prompt=" + prompt +
125+ ", image=" + image +
105126 ", responseFormat=" + responseFormat +
106127 ", seed=" + seed +
107128 ", guidanceScale=" + guidanceScale +
108129 ", size=" + size +
109130 ", watermark=" + watermark +
131+ ", optimizePrompt=" + optimizePrompt +
110132 '}' ;
111133 }
112134
@@ -117,11 +139,13 @@ public static GenerateImagesRequest.Builder builder() {
117139 public static class Builder {
118140 private String model ;
119141 private String prompt ;
142+ private String image ;
120143 private String responseFormat ;
121144 private Integer seed ;
122145 private Double guidanceScale ;
123146 private String size ;
124147 private Boolean watermark ;
148+ private Boolean optimizePrompt ;
125149
126150 private Builder () {
127151 }
@@ -136,6 +160,11 @@ public GenerateImagesRequest.Builder prompt(String prompt) {
136160 return this ;
137161 }
138162
163+ public GenerateImagesRequest .Builder image (String image ) {
164+ this .image = image ;
165+ return this ;
166+ }
167+
139168 public GenerateImagesRequest .Builder responseFormat (String responseFormat ) {
140169 this .responseFormat = responseFormat ;
141170 return this ;
@@ -161,15 +190,22 @@ public GenerateImagesRequest.Builder watermark(Boolean watermark) {
161190 return this ;
162191 }
163192
193+ public GenerateImagesRequest .Builder optimizePrompt (Boolean optimizePrompt ) {
194+ this .optimizePrompt = optimizePrompt ;
195+ return this ;
196+ }
197+
164198 public GenerateImagesRequest build () {
165199 GenerateImagesRequest generateImagesRequest = new GenerateImagesRequest ();
166200 generateImagesRequest .setModel (model );
167201 generateImagesRequest .setPrompt (prompt );
202+ generateImagesRequest .setImage (image );
168203 generateImagesRequest .setResponseFormat (responseFormat );
169204 generateImagesRequest .setSeed (seed );
170205 generateImagesRequest .setGuidanceScale (guidanceScale );
171206 generateImagesRequest .setSize (size );
172207 generateImagesRequest .setWatermark (watermark );
208+ generateImagesRequest .setOptimizePrompt (optimizePrompt );
173209 return generateImagesRequest ;
174210 }
175211 }
0 commit comments