Skip to content

Commit 8ca0224

Browse files
author
BitsAdmin
committed
Merge branch 'feat/seededit' into 'integration_2025-07-17_1012625778946'
feat: [development task] ark runtime (1462278) See merge request iaasng/volcengine-java-sdk!576
2 parents 5943f60 + 664d702 commit 8ca0224

File tree

3 files changed

+66
-9
lines changed

3 files changed

+66
-9
lines changed

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

Lines changed: 40 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,9 +1,6 @@
11
package com.volcengine.ark.runtime.model.images.generation;
22

33
import com.fasterxml.jackson.annotation.JsonProperty;
4-
import com.volcengine.ark.runtime.model.content.generation.CreateContentGenerationTaskRequest;
5-
6-
import java.util.List;
74

85
public 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
}
Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,5 @@
1+
package com.volcengine.ark.runtime.model.images.generation;
2+
3+
public class Size {
4+
public static final String Adaptive = "adaptive";
5+
}

volcengine-java-sdk-ark-runtime/test/java/com/volcengine/ark/runtime/ImageGenerationsExample.java

Lines changed: 21 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1,15 +1,13 @@
11
package com.volcengine.ark.runtime;
22

3-
import com.volcengine.ark.runtime.model.content.generation.*;
43
import com.volcengine.ark.runtime.model.images.generation.GenerateImagesRequest;
54
import com.volcengine.ark.runtime.model.images.generation.ImagesResponse;
65
import com.volcengine.ark.runtime.model.images.generation.ResponseFormat;
6+
import com.volcengine.ark.runtime.model.images.generation.Size;
77
import com.volcengine.ark.runtime.service.ArkService;
88
import okhttp3.ConnectionPool;
99
import okhttp3.Dispatcher;
1010

11-
import java.util.ArrayList;
12-
import java.util.List;
1311
import java.util.concurrent.TimeUnit;
1412

1513
public class ImageGenerationsExample {
@@ -30,10 +28,10 @@ public class ImageGenerationsExample {
3028
public static void main(String[] args) {
3129
String model = "YOUR_ENDPOINT_ID";
3230

33-
System.out.println("\n----- Generate Images Request -----");
31+
System.out.println("\n----- [Seedream] Generate Images Request -----");
3432

3533
GenerateImagesRequest generateRequest = GenerateImagesRequest.builder()
36-
.model(model)
34+
.model(model) // Replace with your Seedream endpoint ID
3735
.prompt("龙与地下城女骑士背景是起伏的平原,目光从镜头转向平原")
3836
.responseFormat(ResponseFormat.Url)
3937
.seed(1234567890)
@@ -47,6 +45,24 @@ public static void main(String[] args) {
4745
ImagesResponse imagesResponse = service.generateImages(generateRequest);
4846
System.out.println(imagesResponse.getData().get(0).getUrl());
4947

48+
System.out.println("\n----- [Seededit] Generate Images Request -----");
49+
50+
generateRequest = GenerateImagesRequest.builder()
51+
.model(model) // Replace with your Seededit endpoint ID
52+
.prompt("龙与地下城女骑士背景是起伏的平原,目光从镜头转向平原")
53+
.image("https://an-test-imgs.tos-cn-beijing.volces.com/avi/9m_001.jpg") // Replace with your input image URL
54+
.responseFormat(ResponseFormat.Url)
55+
.seed(1234567890)
56+
.watermark(true)
57+
.size(Size.Adaptive)
58+
.guidanceScale(2.5)
59+
.build();
60+
61+
System.out.println(generateRequest.toString());
62+
// send create request
63+
imagesResponse = service.generateImages(generateRequest);
64+
System.out.println(imagesResponse.getData().get(0).getUrl());
65+
5066
service.shutdownExecutor();
5167
}
5268
}

0 commit comments

Comments
 (0)