Skip to content

Commit 1ca9440

Browse files
fabianlimxuhuaifabianlimxuhuai
committed
feat: ark generate images sdk
1 parent 2193538 commit 1ca9440

File tree

6 files changed

+457
-0
lines changed

6 files changed

+457
-0
lines changed

‎volcengine-java-sdk-ark-runtime/src/main/java/com/volcengine/ark/runtime/model/images/generation/GenerateImagesRequest.java‎

Lines changed: 176 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,176 @@
1+
package com.volcengine.ark.runtime.model.images.generation;
2+
3+
import com.fasterxml.jackson.annotation.JsonProperty;
4+
import com.volcengine.ark.runtime.model.content.generation.CreateContentGenerationTaskRequest;
5+
6+
import java.util.List;
7+
8+
public class GenerateImagesRequest {
9+
10+
@JsonProperty("model")
11+
private String model;
12+
13+
@JsonProperty("prompt")
14+
private String prompt;
15+
16+
@JsonProperty("response_format")
17+
private String responseFormat;
18+
19+
@JsonProperty("seed")
20+
private Integer seed;
21+
22+
@JsonProperty("guidance_scale")
23+
private Double guidanceScale;
24+
25+
@JsonProperty("size")
26+
private String size;
27+
28+
@JsonProperty("watermark")
29+
private Boolean watermark;
30+
31+
public GenerateImagesRequest() {
32+
}
33+
34+
public GenerateImagesRequest(String model, String prompt, String responseFormat, Integer seed, Double guidanceScale, String size) {
35+
this.model = model;
36+
this.prompt = prompt;
37+
this.responseFormat = responseFormat;
38+
this.seed = seed;
39+
this.guidanceScale = guidanceScale;
40+
this.size = size;
41+
this.watermark = false;
42+
}
43+
44+
public String getModel() {
45+
return this.model;
46+
}
47+
48+
public void setModel(String model) {
49+
this.model = model;
50+
}
51+
52+
public String getPrompt() {
53+
return this.prompt;
54+
}
55+
56+
public void setPrompt(String prompt) {
57+
this.prompt = prompt;
58+
}
59+
60+
public String getResponseFormat() {
61+
return this.responseFormat;
62+
}
63+
64+
public void setResponseFormat(String responseFormat) {
65+
this.responseFormat = responseFormat;
66+
}
67+
68+
public Integer getSeed() {
69+
return this.seed;
70+
}
71+
72+
public void setSeed(Integer seed) {
73+
this.seed = seed;
74+
}
75+
76+
public Double getGuidanceScale() {
77+
return this.guidanceScale;
78+
}
79+
80+
public void setGuidanceScale(Double guidanceScale) {
81+
this.guidanceScale = guidanceScale;
82+
}
83+
84+
public String getSize() {
85+
return this.size;
86+
}
87+
88+
public void setSize(String size) {
89+
this.size = size;
90+
}
91+
92+
public Boolean getWatermark() {
93+
return this.watermark;
94+
}
95+
96+
public void setWatermark(Boolean watermark) {
97+
this.watermark = watermark;
98+
}
99+
100+
@Override
101+
public String toString() {
102+
return "GenerateImagesRequest{" +
103+
"model='" + model + '\'' +
104+
", prompt=" + prompt +
105+
", responseFormat=" + responseFormat +
106+
", seed=" + seed +
107+
", guidanceScale=" + guidanceScale +
108+
", size=" + size +
109+
", watermark=" + watermark +
110+
'}';
111+
}
112+
113+
public static GenerateImagesRequest.Builder builder() {
114+
return new GenerateImagesRequest.Builder();
115+
}
116+
117+
public static class Builder {
118+
private String model;
119+
private String prompt;
120+
private String responseFormat;
121+
private Integer seed;
122+
private Double guidanceScale;
123+
private String size;
124+
private Boolean watermark;
125+
126+
private Builder() {
127+
}
128+
129+
public GenerateImagesRequest.Builder model(String model) {
130+
this.model = model;
131+
return this;
132+
}
133+
134+
public GenerateImagesRequest.Builder prompt(String prompt) {
135+
this.prompt = prompt;
136+
return this;
137+
}
138+
139+
public GenerateImagesRequest.Builder responseFormat(String responseFormat) {
140+
this.responseFormat = responseFormat;
141+
return this;
142+
}
143+
144+
public GenerateImagesRequest.Builder seed(Integer seed) {
145+
this.seed = seed;
146+
return this;
147+
}
148+
149+
public GenerateImagesRequest.Builder guidanceScale(Double guidanceScale) {
150+
this.guidanceScale = guidanceScale;
151+
return this;
152+
}
153+
154+
public GenerateImagesRequest.Builder size(String size) {
155+
this.size = size;
156+
return this;
157+
}
158+
159+
public GenerateImagesRequest.Builder watermark(Boolean watermark) {
160+
this.watermark = watermark;
161+
return this;
162+
}
163+
164+
public GenerateImagesRequest build() {
165+
GenerateImagesRequest generateImagesRequest = new GenerateImagesRequest();
166+
generateImagesRequest.setModel(model);
167+
generateImagesRequest.setPrompt(prompt);
168+
generateImagesRequest.setResponseFormat(responseFormat);
169+
generateImagesRequest.setSeed(seed);
170+
generateImagesRequest.setGuidanceScale(guidanceScale);
171+
generateImagesRequest.setSize(size);
172+
generateImagesRequest.setWatermark(watermark);
173+
return generateImagesRequest;
174+
}
175+
}
176+
}

‎volcengine-java-sdk-ark-runtime/src/main/java/com/volcengine/ark/runtime/model/images/generation/ImagesResponse.java‎

Lines changed: 212 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,212 @@
1+
package com.volcengine.ark.runtime.model.images.generation;
2+
3+
import com.fasterxml.jackson.annotation.JsonIgnoreProperties;
4+
import com.fasterxml.jackson.annotation.JsonProperty;
5+
import com.volcengine.ark.runtime.model.content.generation.CreateContentGenerationTaskResult;
6+
7+
import java.util.List;
8+
9+
public class ImagesResponse {
10+
11+
@JsonProperty("model")
12+
private String model;
13+
14+
@JsonProperty("created")
15+
private Integer created;
16+
17+
@JsonProperty("data")
18+
private List<Image> data;
19+
20+
@JsonProperty("usage")
21+
private Usage usage;
22+
23+
@JsonProperty("error")
24+
private Error error;
25+
26+
public String getModel() {
27+
return model;
28+
}
29+
30+
public void setModel(String model) {
31+
this.model = model;
32+
}
33+
34+
public Integer getCreated() {
35+
return created;
36+
}
37+
38+
public void setCreated(Integer created) {
39+
this.created = created;
40+
}
41+
42+
public List<Image> getData() {
43+
return data;
44+
}
45+
46+
public void setData(List<Image> data) {
47+
this.data = data;
48+
}
49+
50+
public Usage getUsage() {
51+
return usage;
52+
}
53+
54+
public void setUsage(Usage usage) {
55+
this.usage = usage;
56+
}
57+
58+
public Error getError() {
59+
return error;
60+
}
61+
62+
public void setError(Error error) {
63+
this.error = error;
64+
}
65+
66+
@JsonIgnoreProperties(ignoreUnknown = true)
67+
public static class Image {
68+
69+
@JsonProperty("url")
70+
private String url;
71+
72+
@JsonProperty("b64_json")
73+
private String b64Json;
74+
75+
public String getUrl() {
76+
return url;
77+
}
78+
79+
public void setUrl(String url) {
80+
this.url = url;
81+
}
82+
83+
public String getB64Json() {
84+
return url;
85+
}
86+
87+
public void setB64Json(String b64Json) {
88+
this.b64Json = b64Json;
89+
}
90+
91+
@Override
92+
public String toString() {
93+
return "Image{" +
94+
"url=" + url +
95+
"b64Json=" + b64Json +
96+
'}';
97+
}
98+
}
99+
100+
@JsonIgnoreProperties(ignoreUnknown = true)
101+
public static class Usage {
102+
103+
@JsonProperty("generated_images")
104+
private Integer generatedImages;
105+
106+
public Integer getGeneratedImages() {
107+
return generatedImages;
108+
}
109+
110+
public void setGeneratedImages(Integer generatedImages) {
111+
this.generatedImages = generatedImages;
112+
}
113+
114+
@Override
115+
public String toString() {
116+
return "Usage{" +
117+
"generatedImages=" + generatedImages +
118+
'}';
119+
}
120+
}
121+
122+
@JsonIgnoreProperties(ignoreUnknown = true)
123+
public static class Error {
124+
125+
@JsonProperty("code")
126+
private String code;
127+
128+
@JsonProperty("message")
129+
private String message;
130+
131+
public String getCode() {
132+
return code;
133+
}
134+
135+
public void setCode(String code) {
136+
this.code = code;
137+
}
138+
139+
public String getMessage() {
140+
return message;
141+
}
142+
143+
public void setMessage(String message) {
144+
this.message = message;
145+
}
146+
147+
@Override
148+
public String toString() {
149+
return "Error{" +
150+
"code=" + code +
151+
"message=" + message +
152+
'}';
153+
}
154+
}
155+
156+
@Override
157+
public String toString() {
158+
return "ImagesResponse{" +
159+
"model='" + model + '\'' +
160+
"created=" + created +
161+
"data=" + data +
162+
"usage=" + usage +
163+
"error=" + error +
164+
'}';
165+
}
166+
167+
public static class Builder {
168+
private String model;
169+
private Integer created;
170+
private List<Image> data;
171+
private Usage usage;
172+
private Error error;
173+
174+
private Builder() {
175+
}
176+
177+
public ImagesResponse.Builder model(String model) {
178+
this.model = model;
179+
return this;
180+
}
181+
182+
public ImagesResponse.Builder created(Integer created) {
183+
this.created = created;
184+
return this;
185+
}
186+
187+
public ImagesResponse.Builder data(List<Image> data) {
188+
this.data = data;
189+
return this;
190+
}
191+
192+
public ImagesResponse.Builder usage(Usage usage) {
193+
this.usage = usage;
194+
return this;
195+
}
196+
197+
public ImagesResponse.Builder error(Error error) {
198+
this.error = error;
199+
return this;
200+
}
201+
202+
public ImagesResponse build() {
203+
ImagesResponse imagesResponse = new ImagesResponse();
204+
imagesResponse.setModel(model);
205+
imagesResponse.setCreated(created);
206+
imagesResponse.setData(data);
207+
imagesResponse.setUsage(usage);
208+
imagesResponse.setError(error);
209+
return imagesResponse;
210+
}
211+
}
212+
}

‎volcengine-java-sdk-ark-runtime/src/main/java/com/volcengine/ark/runtime/model/images/generation/ResponseFormat.java‎

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,6 @@
1+
package com.volcengine.ark.runtime.model.images.generation;
2+
3+
public class ResponseFormat {
4+
public static final String Base64 = "b64_json";
5+
public static final String Url = "url";
6+
}

0 commit comments

Comments
 (0)