Skip to content

Commit cf69b08

Browse files
author
BitsAdmin
committed
Merge branch 'feat/arkruntime_bot_addheaders' into 'integration_2024-09-26_456257596674'
feat: [development task] ark-runtime-manual-Java (793783) See merge request iaasng/volcengine-java-sdk!260
2 parents 4bce459 + 2544b7d commit cf69b08

File tree

1 file changed

+10
-0
lines changed
  • volcengine-java-sdk-ark-runtime/src/main/java/com/volcengine/ark/runtime/service

1 file changed

+10
-0
lines changed

volcengine-java-sdk-ark-runtime/src/main/java/com/volcengine/ark/runtime/service/ArkService.java

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -195,12 +195,22 @@ public BotChatCompletionResult createBotChatCompletion(BotChatCompletionRequest
195195
return execute(api.createBotChatCompletion(request, request.getModel(), new HashMap<>()));
196196
}
197197

198+
public BotChatCompletionResult createBotChatCompletion(BotChatCompletionRequest request, Map<String, String> customHeaders) {
199+
return execute(api.createBotChatCompletion(request, request.getModel(), customHeaders));
200+
}
201+
202+
198203
@Override
199204
public Flowable<BotChatCompletionChunk> streamBotChatCompletion(BotChatCompletionRequest request) {
200205
request.setStream(true);
201206
return stream(api.createBotChatCompletionStream(request, request.getModel(), new HashMap<>()), BotChatCompletionChunk.class);
202207
}
203208

209+
public Flowable<BotChatCompletionChunk> streamBotChatCompletion(BotChatCompletionRequest request, Map<String, String> customHeaders) {
210+
request.setStream(true);
211+
return stream(api.createBotChatCompletionStream(request, request.getModel(), customHeaders), BotChatCompletionChunk.class);
212+
}
213+
204214
public TokenizationResult createTokenization(TokenizationRequest request) {
205215
return execute(api.createTokenization(request, request.getModel(), new HashMap<>()));
206216
}

0 commit comments

Comments
 (0)