Skip to content

Commit b1af507

Browse files
author
BitsAdmin
committed
Merge branch 'fix/0603-java-sdk-oncall' into 'integration_2025-06-05_933333271298'
feat: [development task] ark runtime (1295221) See merge request iaasng/volcengine-java-sdk!519
2 parents 6d29cd5 + 361a813 commit b1af507

File tree

4 files changed

+56
-0
lines changed

4 files changed

+56
-0
lines changed

volcengine-java-sdk-ark-runtime/src/main/java/com/volcengine/ark/runtime/model/bot/completion/chat/usage/BotActionUsage.java

Lines changed: 24 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -26,6 +26,12 @@ public class BotActionUsage {
2626
@JsonProperty("search_count")
2727
private Integer searchCount;
2828

29+
@JsonProperty("action_name")
30+
private String actionName;
31+
32+
@JsonProperty("count")
33+
private Integer count;
34+
2935
public String getName() {
3036
return name;
3137
}
@@ -66,6 +72,22 @@ public void setSearchCount(Integer searchCount) {
6672
this.searchCount = searchCount;
6773
}
6874

75+
public String getActionName() {
76+
return actionName;
77+
}
78+
79+
public void setActionName(String actionName) {
80+
this.actionName = actionName;
81+
}
82+
83+
public Integer getCount() {
84+
return count;
85+
}
86+
87+
public void setCount(Integer count) {
88+
this.count = count;
89+
}
90+
6991
@Override
7092
public String toString() {
7193
return "BotActionUsage{" +
@@ -74,6 +96,8 @@ public String toString() {
7496
", completionTokens=" + completionTokens +
7597
", totalTokens=" + totalTokens +
7698
", searchCount=" + searchCount +
99+
", actionName='" + actionName + '\'' +
100+
", count=" + count +
77101
'}';
78102
}
79103
}

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/src/main/java/com/volcengine/ark/runtime/model/content/generation/ListContentGenerationTasksResponse.java

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -82,6 +82,12 @@ public void setStatus(String status) {
8282
this.status = status;
8383
}
8484

85+
public ContentGenerationError getError() {
86+
return error;
87+
}
88+
89+
public void setError(ContentGenerationError error) { this.error = error; }
90+
8591
public Content getContent() {
8692
return content;
8793
}

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)