Skip to content

Commit c3f8f48

Browse files
author
Lin Fangyuan
committed
feat: videogen return last frame
1 parent 8cd0fc8 commit c3f8f48

File tree

2 files changed

+33
-1
lines changed

2 files changed

+33
-1
lines changed

volcengine-java-sdk-ark-runtime/src/main/java/com/volcengine/ark/runtime/model/content/generation/CreateContentGenerationTaskRequest.java

Lines changed: 21 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -17,6 +17,9 @@ public class CreateContentGenerationTaskRequest {
1717
@JsonProperty("callback_url")
1818
private String callbackUrl;
1919

20+
@JsonProperty("return_last_frame")
21+
private Boolean returnLastFrame;
22+
2023
public CreateContentGenerationTaskRequest() {
2124
}
2225

@@ -25,10 +28,11 @@ public CreateContentGenerationTaskRequest(String model, List<Content> content) {
2528
this.content = content;
2629
}
2730

28-
public CreateContentGenerationTaskRequest(String model, List<Content> content, String callbackUrl) {
31+
public CreateContentGenerationTaskRequest(String model, List<Content> content, String callbackUrl, Boolean returnLastFrame) {
2932
this.model = model;
3033
this.content = content;
3134
this.callbackUrl = callbackUrl;
35+
this.returnLastFrame = returnLastFrame;
3236
}
3337

3438
public String getModel() {
@@ -55,12 +59,21 @@ public void setCallbackUrl(String callbackUrl) {
5559
this.callbackUrl = callbackUrl;
5660
}
5761

62+
public Boolean getReturnLastFrame() {
63+
return returnLastFrame;
64+
}
65+
66+
public void setReturnLastFrame(Boolean returnLastFrame) {
67+
this.returnLastFrame = returnLastFrame;
68+
}
69+
5870
@Override
5971
public String toString() {
6072
return "CreateContentGenerationTaskRequest{" +
6173
"model='" + model + '\'' +
6274
", content=" + content +
6375
", callbackUrl='" + callbackUrl + '\'' +
76+
", returnLastFrame=" + returnLastFrame +
6477
'}';
6578
}
6679

@@ -72,6 +85,7 @@ public static class Builder {
7285
private String model;
7386
private List<Content> content;
7487
private String callbackUrl;
88+
private Boolean returnLastFrame;
7589

7690
private Builder() {
7791
}
@@ -91,11 +105,17 @@ public Builder callbackUrl(String callbackUrl) {
91105
return this;
92106
}
93107

108+
public Builder returnLastFrame(Boolean returnLastFrame) {
109+
this.returnLastFrame = returnLastFrame;
110+
return this;
111+
}
112+
94113
public CreateContentGenerationTaskRequest build() {
95114
CreateContentGenerationTaskRequest createContentGenerationTaskRequest = new CreateContentGenerationTaskRequest();
96115
createContentGenerationTaskRequest.setModel(model);
97116
createContentGenerationTaskRequest.setContent(content);
98117
createContentGenerationTaskRequest.setCallbackUrl(callbackUrl);
118+
createContentGenerationTaskRequest.setReturnLastFrame(returnLastFrame);
99119
return createContentGenerationTaskRequest;
100120
}
101121
}

volcengine-java-sdk-ark-runtime/src/main/java/com/volcengine/ark/runtime/model/content/generation/GetContentGenerationTaskResponse.java

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -133,10 +133,22 @@ public void setVideoUrl(String videoUrl) {
133133
this.videoUrl = videoUrl;
134134
}
135135

136+
@JsonProperty("last_frame_url")
137+
private String lastFrameUrl;
138+
139+
public String getLastFrameUrl() {
140+
return lastFrameUrl;
141+
}
142+
143+
public void setLastFrameUrl(String lastFrameUrl) {
144+
this.lastFrameUrl = lastFrameUrl;
145+
}
146+
136147
@Override
137148
public String toString() {
138149
return "Content{" +
139150
"videoUrl='" + videoUrl + '\'' +
151+
", lastFrameUrl='" + lastFrameUrl + '\'' +
140152
'}';
141153
}
142154
}

0 commit comments

Comments
 (0)