Skip to content

Commit 80226f5

Browse files
author
hks
committed
feat: support reasoning effort
1 parent 1a415b1 commit 80226f5

File tree

1 file changed

+21
-2
lines changed

1 file changed

+21
-2
lines changed

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

Lines changed: 21 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -2,8 +2,6 @@
22

33
import com.fasterxml.jackson.annotation.JsonProperty;
44
import com.fasterxml.jackson.databind.JsonNode;
5-
import com.volcengine.ark.runtime.model.bot.completion.chat.BotChatCompletionRequest;
6-
import com.volcengine.ark.runtime.utils.JacksonUtil;
75

86
import java.util.List;
97
import java.util.Map;
@@ -167,6 +165,10 @@ public class ChatCompletionRequest {
167165
@JsonProperty("thinking")
168166
ChatCompletionRequestThinking thinking;
169167

168+
@JsonProperty("reasoning_effort")
169+
String reasoningEffort;
170+
171+
170172
@Override
171173
public String toString() {
172174
return "ChatCompletionRequest{" +
@@ -194,10 +196,19 @@ public String toString() {
194196
", toolChoice=" + toolChoice +
195197
", responseFormat=" + responseFormat +
196198
", thinking=" + thinking +
199+
", reasoningEffort='" + reasoningEffort + '\'' +
197200
'}';
198201
}
199202

200203

204+
public String getReasoningEffort() {
205+
return reasoningEffort;
206+
}
207+
208+
public void setReasoningEffort(String reasoningEffort) {
209+
this.reasoningEffort = reasoningEffort;
210+
}
211+
201212
public Integer getMaxCompletionTokens() {
202213
return maxCompletionTokens;
203214
}
@@ -573,6 +584,7 @@ public static class Builder {
573584
private ChatCompletionRequestResponseFormat responseFormat;
574585
private ChatCompletionRequestThinking thinking;
575586
private Integer maxCompletionTokens;
587+
private String reasoningEffort;
576588

577589
public ChatCompletionRequest.Builder model(String model) {
578590
this.model = model;
@@ -698,6 +710,12 @@ public ChatCompletionRequest.Builder thinking(ChatCompletionRequestThinking thin
698710
this.thinking = thinking;
699711
return this;
700712
}
713+
714+
public ChatCompletionRequest.Builder reasoningEffort(String reasoningEffort) {
715+
this.reasoningEffort = reasoningEffort;
716+
return this;
717+
}
718+
701719
public ChatCompletionRequest build() {
702720
ChatCompletionRequest chatCompletionRequest = new ChatCompletionRequest();
703721
chatCompletionRequest.setModel(model);
@@ -724,6 +742,7 @@ public ChatCompletionRequest build() {
724742
chatCompletionRequest.setResponseFormat(responseFormat);
725743
chatCompletionRequest.setThinking(thinking);
726744
chatCompletionRequest.setMaxCompletionTokens(maxCompletionTokens);
745+
chatCompletionRequest.setReasoningEffort(reasoningEffort);
727746
return chatCompletionRequest;
728747
}
729748
}

0 commit comments

Comments
 (0)