Skip to content

Commit 73f5bf8

Browse files
committed
feat: prefill
1 parent bf05f3f commit 73f5bf8

File tree

2 files changed

+39
-2
lines changed

2 files changed

+39
-2
lines changed

volcengine-java-sdk-ark-runtime/src/main/java/com/volcengine/ark/runtime/model/responses/item/ItemEasyMessage.java

Lines changed: 20 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -12,6 +12,9 @@ public class ItemEasyMessage extends BaseMessageItem implements InputItem {
1212
@JsonProperty("content")
1313
private MessageContent content;
1414

15+
@JsonProperty("partial")
16+
private Boolean partial;
17+
1518
public MessageContent getContent() {
1619
return content;
1720
}
@@ -20,13 +23,22 @@ public void setContent(MessageContent content) {
2023
this.content = content;
2124
}
2225

26+
public Boolean getPartial() {
27+
return partial;
28+
}
29+
30+
public void setPartial(Boolean partial) {
31+
this.partial = partial;
32+
}
33+
2334
@Override
2435
public String toString() {
2536
return "ItemEasyMessage{" +
2637
"type='" + getType() + '\'' +
2738
", role='" + getRole() + '\'' +
2839
", content=" + content +
2940
", id='" + getId() + '\'' +
41+
", partial=" + partial +
3042
'}';
3143
}
3244

@@ -39,6 +51,8 @@ public static class Builder {
3951
private MessageContent content;
4052
private String id;
4153

54+
private Boolean partial;
55+
4256
public Builder role(String role) {
4357
this.role = role;
4458
return this;
@@ -54,11 +68,17 @@ public Builder id(String id) {
5468
return this;
5569
}
5670

71+
public Builder partial(Boolean partial) {
72+
this.partial = partial;
73+
return this;
74+
}
75+
5776
public ItemEasyMessage build() {
5877
ItemEasyMessage itemEasyMessage = new ItemEasyMessage();
5978
itemEasyMessage.setRole(role);
6079
itemEasyMessage.setContent(content);
6180
itemEasyMessage.setId(id);
81+
itemEasyMessage.setPartial(partial);
6282
return itemEasyMessage;
6383
}
6484
}

volcengine-java-sdk-ark-runtime/src/main/java/com/volcengine/ark/runtime/model/responses/item/ItemOutputMessage.java

Lines changed: 19 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -3,9 +3,7 @@
33
import com.fasterxml.jackson.annotation.JsonIgnoreProperties;
44
import com.fasterxml.jackson.annotation.JsonProperty;
55
import com.fasterxml.jackson.annotation.JsonTypeInfo;
6-
import com.fasterxml.jackson.databind.JsonDeserializer;
76
import com.fasterxml.jackson.databind.annotation.JsonDeserialize;
8-
import com.volcengine.ark.runtime.model.responses.constant.ResponsesConstants;
97
import com.volcengine.ark.runtime.model.responses.content.OutputContentItem;
108

119
import java.util.List;
@@ -20,6 +18,9 @@ public class ItemOutputMessage extends BaseMessageItem implements OutputItem {
2018
@JsonProperty("status")
2119
private String status;
2220

21+
@JsonProperty("partial")
22+
private Boolean partial;
23+
2324
public List<OutputContentItem> getContent() {
2425
return content;
2526
}
@@ -36,6 +37,14 @@ public void setStatus(String status) {
3637
this.status = status;
3738
}
3839

40+
public Boolean getPartial() {
41+
return partial;
42+
}
43+
44+
public void setPartial(Boolean partial) {
45+
this.partial = partial;
46+
}
47+
3948
@Override
4049
public String toString() {
4150
return "ItemOutputMessage{" +
@@ -44,6 +53,7 @@ public String toString() {
4453
", content=" + content +
4554
", status='" + status + '\'' +
4655
", id='" + getId() + '\'' +
56+
", partial=" + partial +
4757
'}';
4858
}
4959

@@ -56,6 +66,7 @@ public static class Builder {
5666
private List<OutputContentItem> content;
5767
private String status;
5868
private String id;
69+
private Boolean partial;
5970

6071
public Builder role(String role) {
6172
this.role = role;
@@ -77,12 +88,18 @@ public Builder id(String id) {
7788
return this;
7889
}
7990

91+
public Builder partial(Boolean partial) {
92+
this.partial = partial;
93+
return this;
94+
}
95+
8096
public ItemOutputMessage build() {
8197
ItemOutputMessage itemOutputMessage = new ItemOutputMessage();
8298
itemOutputMessage.setRole(role);
8399
itemOutputMessage.setContent(content);
84100
itemOutputMessage.setStatus(status);
85101
itemOutputMessage.setId(id);
102+
itemOutputMessage.setPartial(partial);
86103
return itemOutputMessage;
87104
}
88105
}

0 commit comments

Comments
 (0)