Skip to content

Commit fbf3841

Browse files
author
BitsAdmin
committed
Merge branch 'feat/add_max_completion_tokens' into 'integration_2025-06-19_956533420802'
feat: [development task] ark runtime (1361622) See merge request iaasng/volcengine-java-sdk!547
2 parents 080aed9 + f7714d2 commit fbf3841

File tree

2 files changed

+31
-0
lines changed

2 files changed

+31
-0
lines changed

volcengine-java-sdk-ark-runtime/src/main/java/com/volcengine/ark/runtime/model/completion/chat/ChatCompletionRequest.java

Lines changed: 23 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -79,6 +79,12 @@ public class ChatCompletionRequest {
7979
@JsonProperty("max_tokens")
8080
Integer maxTokens;
8181

82+
/**
83+
* The maximum number of tokens allowed for the generated answer, including reasoning tokens.
84+
*/
85+
@JsonProperty("max_completion_tokens")
86+
Integer maxCompletionTokens;
87+
8288
/**
8389
* Number between -2.0 and 2.0. Positive values penalize new tokens based on whether they appear in the text so far,
8490
* increasing the model's likelihood to talk about new topics.
@@ -173,6 +179,7 @@ public String toString() {
173179
", serviceTier='" + serviceTier + '\'' +
174180
", stop=" + stop +
175181
", maxTokens=" + maxTokens +
182+
", maxCompletionTokens=" + maxCompletionTokens +
176183
", presencePenalty=" + presencePenalty +
177184
", frequencyPenalty=" + frequencyPenalty +
178185
", logitBias=" + logitBias +
@@ -190,6 +197,15 @@ public String toString() {
190197
'}';
191198
}
192199

200+
201+
public Integer getMaxCompletionTokens() {
202+
return maxCompletionTokens;
203+
}
204+
205+
public void setMaxCompletionTokens(Integer maxCompletionTokens) {
206+
this.maxCompletionTokens = maxCompletionTokens;
207+
}
208+
193209
public ChatCompletionRequestThinking getThinking() {
194210
return thinking;
195211
}
@@ -556,6 +572,7 @@ public static class Builder {
556572
private Object toolChoice;
557573
private ChatCompletionRequestResponseFormat responseFormat;
558574
private ChatCompletionRequestThinking thinking;
575+
private Integer maxCompletionTokens;
559576

560577
public ChatCompletionRequest.Builder model(String model) {
561578
this.model = model;
@@ -602,6 +619,11 @@ public ChatCompletionRequest.Builder maxTokens(Integer maxTokens) {
602619
return this;
603620
}
604621

622+
public ChatCompletionRequest.Builder maxCompletionTokens(Integer maxCompletionTokens) {
623+
this.maxCompletionTokens = maxCompletionTokens;
624+
return this;
625+
}
626+
605627
public ChatCompletionRequest.Builder presencePenalty(Double presencePenalty) {
606628
this.presencePenalty = presencePenalty;
607629
return this;
@@ -701,6 +723,7 @@ public ChatCompletionRequest build() {
701723
chatCompletionRequest.setToolChoice(toolChoice);
702724
chatCompletionRequest.setResponseFormat(responseFormat);
703725
chatCompletionRequest.setThinking(thinking);
726+
chatCompletionRequest.setMaxCompletionTokens(maxCompletionTokens);
704727
return chatCompletionRequest;
705728
}
706729
}

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

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -57,6 +57,14 @@ public void setStatus(String status) {
5757
this.status = status;
5858
}
5959

60+
public ContentGenerationError getError() {
61+
return error;
62+
}
63+
64+
public void setError(ContentGenerationError error) {
65+
this.error = error;
66+
}
67+
6068
public Content getContent() {
6169
return content;
6270
}

0 commit comments

Comments
 (0)