Skip to content

Commit b4ec0b7

Browse files
jizhongBaojizhongBao
committed
feat(ark-runtime): add safety identifier
1 parent e08591c commit b4ec0b7

4 files changed

Lines changed: 67 additions & 0 deletions

File tree

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

Lines changed: 24 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -14,6 +14,9 @@ public class CreateContentGenerationTaskRequest {
1414
@JsonProperty("content")
1515
private List<Content> content;
1616

17+
@JsonProperty("safety_identifier")
18+
private String safetyIdentifier;
19+
1720
@JsonProperty("callback_url")
1821
private String callbackUrl;
1922

@@ -72,8 +75,13 @@ public CreateContentGenerationTaskRequest(String model, List<Content> content, S
7275
}
7376

7477
public CreateContentGenerationTaskRequest(String model, List<Content> content, String callbackUrl, Boolean returnLastFrame, String serviceTier, Long executionExpiresAfter, Boolean generateAudio, Boolean cameraFixed, Boolean watermark, Long seed, String resolution, String ratio, Long duration, Long frames, Boolean draft, List<ContentGenerationTool> tools ) {
78+
this(model, content, null, callbackUrl, returnLastFrame, serviceTier, executionExpiresAfter, generateAudio, cameraFixed, watermark, seed, resolution, ratio, duration, frames, draft, tools);
79+
}
80+
81+
public CreateContentGenerationTaskRequest(String model, List<Content> content, String safetyIdentifier, String callbackUrl, Boolean returnLastFrame, String serviceTier, Long executionExpiresAfter, Boolean generateAudio, Boolean cameraFixed, Boolean watermark, Long seed, String resolution, String ratio, Long duration, Long frames, Boolean draft, List<ContentGenerationTool> tools ) {
7582
this.model = model;
7683
this.content = content;
84+
this.safetyIdentifier = safetyIdentifier;
7785
this.callbackUrl = callbackUrl;
7886
this.returnLastFrame = returnLastFrame;
7987
this.serviceTier = serviceTier;
@@ -106,6 +114,14 @@ public void setContent(List<Content> content) {
106114
this.content = content;
107115
}
108116

117+
public String getSafetyIdentifier() {
118+
return safetyIdentifier;
119+
}
120+
121+
public void setSafetyIdentifier(String safetyIdentifier) {
122+
this.safetyIdentifier = safetyIdentifier;
123+
}
124+
109125
public String getCallbackUrl() {
110126
return callbackUrl;
111127
}
@@ -217,6 +233,7 @@ public String toString() {
217233
return "CreateContentGenerationTaskRequest{" +
218234
"model='" + model + '\'' +
219235
", content=" + content +
236+
", safetyIdentifier='" + safetyIdentifier + '\'' +
220237
", callbackUrl='" + callbackUrl + '\'' +
221238
", returnLastFrame=" + returnLastFrame +
222239
", serviceTier='" + serviceTier + '\'' +
@@ -241,6 +258,7 @@ public static CreateContentGenerationTaskRequest.Builder builder() {
241258
public static class Builder {
242259
private String model;
243260
private List<Content> content;
261+
private String safetyIdentifier;
244262
private String callbackUrl;
245263
private Boolean returnLastFrame;
246264
private String serviceTier;
@@ -269,6 +287,11 @@ public Builder content(List<Content> content) {
269287
return this;
270288
}
271289

290+
public Builder safetyIdentifier(String safetyIdentifier) {
291+
this.safetyIdentifier = safetyIdentifier;
292+
return this;
293+
}
294+
272295
public Builder callbackUrl(String callbackUrl) {
273296
this.callbackUrl = callbackUrl;
274297
return this;
@@ -343,6 +366,7 @@ public CreateContentGenerationTaskRequest build() {
343366
CreateContentGenerationTaskRequest createContentGenerationTaskRequest = new CreateContentGenerationTaskRequest();
344367
createContentGenerationTaskRequest.setModel(model);
345368
createContentGenerationTaskRequest.setContent(content);
369+
createContentGenerationTaskRequest.setSafetyIdentifier(safetyIdentifier);
346370
createContentGenerationTaskRequest.setCallbackUrl(callbackUrl);
347371
createContentGenerationTaskRequest.setReturnLastFrame(returnLastFrame);
348372
createContentGenerationTaskRequest.setServiceTier(serviceTier);

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

Lines changed: 19 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -9,6 +9,9 @@ public class CreateContentGenerationTaskResult {
99
@JsonProperty("id")
1010
private String id;
1111

12+
@JsonProperty("safety_identifier")
13+
private String safetyIdentifier;
14+
1215
public CreateContentGenerationTaskResult() {
1316
}
1417

@@ -24,10 +27,19 @@ public void setId(String id) {
2427
this.id = id;
2528
}
2629

30+
public String getSafetyIdentifier() {
31+
return safetyIdentifier;
32+
}
33+
34+
public void setSafetyIdentifier(String safetyIdentifier) {
35+
this.safetyIdentifier = safetyIdentifier;
36+
}
37+
2738
@Override
2839
public String toString() {
2940
return "CreateContentGenerationTaskResult{" +
3041
"id='" + id + '\'' +
42+
", safetyIdentifier='" + safetyIdentifier + '\'' +
3143
'}';
3244
}
3345

@@ -37,6 +49,7 @@ public static CreateContentGenerationTaskResult.Builder builder() {
3749

3850
public static class Builder {
3951
private String id;
52+
private String safetyIdentifier;
4053

4154
private Builder() {
4255
}
@@ -46,9 +59,15 @@ public Builder id(String id) {
4659
return this;
4760
}
4861

62+
public Builder safetyIdentifier(String safetyIdentifier) {
63+
this.safetyIdentifier = safetyIdentifier;
64+
return this;
65+
}
66+
4967
public CreateContentGenerationTaskResult build() {
5068
CreateContentGenerationTaskResult createContentGenerationTaskResult = new CreateContentGenerationTaskResult();
5169
createContentGenerationTaskResult.setId(id);
70+
createContentGenerationTaskResult.setSafetyIdentifier(safetyIdentifier);
5271
return createContentGenerationTaskResult;
5372
}
5473
}

‎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
@@ -14,6 +14,9 @@ public class GetContentGenerationTaskResponse {
1414
@JsonProperty("model")
1515
private String model;
1616

17+
@JsonProperty("safety_identifier")
18+
private String safetyIdentifier;
19+
1720
@JsonProperty("status")
1821
private String status;
1922

@@ -93,6 +96,14 @@ public void setModel(String model) {
9396
this.model = model;
9497
}
9598

99+
public String getSafetyIdentifier() {
100+
return safetyIdentifier;
101+
}
102+
103+
public void setSafetyIdentifier(String safetyIdentifier) {
104+
this.safetyIdentifier = safetyIdentifier;
105+
}
106+
96107
public String getStatus() {
97108
return status;
98109
}
@@ -360,6 +371,7 @@ public String toString() {
360371
return "GetContentGenerationTaskResponse{" +
361372
"id='" + id + '\'' +
362373
", model='" + model + '\'' +
374+
", safetyIdentifier='" + safetyIdentifier + '\'' +
363375
", status='" + status + '\'' +
364376
", error='" + error + '\'' +
365377
", content=" + (content != null ? content.toString() : "null") +

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

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -40,6 +40,9 @@ public static class Item {
4040
@JsonProperty("model")
4141
private String model;
4242

43+
@JsonProperty("safety_identifier")
44+
private String safetyIdentifier;
45+
4346
@JsonProperty("status")
4447
private String status;
4548

@@ -119,6 +122,14 @@ public void setModel(String model) {
119122
this.model = model;
120123
}
121124

125+
public String getSafetyIdentifier() {
126+
return safetyIdentifier;
127+
}
128+
129+
public void setSafetyIdentifier(String safetyIdentifier) {
130+
this.safetyIdentifier = safetyIdentifier;
131+
}
132+
122133
public String getStatus() {
123134
return status;
124135
}
@@ -374,6 +385,7 @@ public String toString() {
374385
return "Item{" +
375386
"id='" + id + '\'' +
376387
", model='" + model + '\'' +
388+
", safetyIdentifier='" + safetyIdentifier + '\'' +
377389
", status='" + status + '\'' +
378390
", error='" + error + '\'' +
379391
", content=" + (content != null ? content.toString() : "null") +

0 commit comments

Comments
 (0)