diff --git a/volcengine-java-sdk-ark-runtime/src/main/java/com/volcengine/ark/runtime/model/completion/chat/ChatCompletionRequest.java b/volcengine-java-sdk-ark-runtime/src/main/java/com/volcengine/ark/runtime/model/completion/chat/ChatCompletionRequest.java index d2e8c96c7..6dd1a338b 100644 --- a/volcengine-java-sdk-ark-runtime/src/main/java/com/volcengine/ark/runtime/model/completion/chat/ChatCompletionRequest.java +++ b/volcengine-java-sdk-ark-runtime/src/main/java/com/volcengine/ark/runtime/model/completion/chat/ChatCompletionRequest.java @@ -696,6 +696,11 @@ public ChatCompletionRequest.Builder toolChoice(String toolChoice) { return this; } + public ChatCompletionRequest.Builder toolChoice(Object toolChoice) { + this.toolChoice = toolChoice; + return this; + } + public ChatCompletionRequest.Builder toolChoice(ChatCompletionRequestToolChoice toolChoice) { this.toolChoice = toolChoice; return this; diff --git a/volcengine-java-sdk-ark-runtime/src/main/java/com/volcengine/ark/runtime/model/completion/chat/ChatMessage.java b/volcengine-java-sdk-ark-runtime/src/main/java/com/volcengine/ark/runtime/model/completion/chat/ChatMessage.java index 88904964a..47fcc1424 100644 --- a/volcengine-java-sdk-ark-runtime/src/main/java/com/volcengine/ark/runtime/model/completion/chat/ChatMessage.java +++ b/volcengine-java-sdk-ark-runtime/src/main/java/com/volcengine/ark/runtime/model/completion/chat/ChatMessage.java @@ -138,6 +138,11 @@ public ChatMessage.Builder content(String content) { return this; } + public ChatMessage.Builder content(Object content) { + this.content = content; + return this; + } + public ChatMessage.Builder multiContent(List multiContent) { this.content = multiContent; return this;