Skip to content

Commit 361a813

Browse files
Merge 'feat/ark-aag-callback' into 'fix/0603-java-sdk-oncall'
feat: add aag callback example See merge request: !513
2 parents 490e2d8 + 882048d commit 361a813

File tree

2 files changed

+26
-0
lines changed

2 files changed

+26
-0
lines changed

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

Lines changed: 25 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("callback_url")
18+
private String callbackUrl;
19+
1720
public CreateContentGenerationTaskRequest() {
1821
}
1922

@@ -22,6 +25,12 @@ public CreateContentGenerationTaskRequest(String model, List<Content> content) {
2225
this.content = content;
2326
}
2427

28+
public CreateContentGenerationTaskRequest(String model, List<Content> content, String callbackUrl) {
29+
this.model = model;
30+
this.content = content;
31+
this.callbackUrl = callbackUrl;
32+
}
33+
2534
public String getModel() {
2635
return model;
2736
}
@@ -38,11 +47,20 @@ public void setContent(List<Content> content) {
3847
this.content = content;
3948
}
4049

50+
public String getCallbackUrl() {
51+
return callbackUrl;
52+
}
53+
54+
public void setCallbackUrl(String callbackUrl) {
55+
this.callbackUrl = callbackUrl;
56+
}
57+
4158
@Override
4259
public String toString() {
4360
return "CreateContentGenerationTaskRequest{" +
4461
"model='" + model + '\'' +
4562
", content=" + content +
63+
", callbackUrl='" + callbackUrl + '\'' +
4664
'}';
4765
}
4866

@@ -53,6 +71,7 @@ public static CreateContentGenerationTaskRequest.Builder builder() {
5371
public static class Builder {
5472
private String model;
5573
private List<Content> content;
74+
private String callbackUrl;
5675

5776
private Builder() {
5877
}
@@ -67,10 +86,16 @@ public Builder content(List<Content> content) {
6786
return this;
6887
}
6988

89+
public Builder callbackUrl(String callbackUrl) {
90+
this.callbackUrl = callbackUrl;
91+
return this;
92+
}
93+
7094
public CreateContentGenerationTaskRequest build() {
7195
CreateContentGenerationTaskRequest createContentGenerationTaskRequest = new CreateContentGenerationTaskRequest();
7296
createContentGenerationTaskRequest.setModel(model);
7397
createContentGenerationTaskRequest.setContent(content);
98+
createContentGenerationTaskRequest.setCallbackUrl(callbackUrl);
7499
return createContentGenerationTaskRequest;
75100
}
76101
}

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

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -50,6 +50,7 @@ public static void main(String[] args) {
5050
CreateContentGenerationTaskRequest createRequest = CreateContentGenerationTaskRequest.builder()
5151
.model(model)
5252
.content(contents)
53+
// .callbackUrl("YOUR CALLBACK URL")
5354
.build();
5455

5556
// send create request

0 commit comments

Comments
 (0)