|
1 | 1 | package com.volcengine.ark.runtime; |
2 | 2 |
|
3 | 3 | import com.fasterxml.jackson.databind.ObjectMapper; |
| 4 | +import com.volcengine.ark.runtime.model.responses.common.ResponsesCaching; |
4 | 5 | import com.volcengine.ark.runtime.model.responses.common.ResponsesThinking; |
5 | 6 | import com.volcengine.ark.runtime.model.responses.constant.ResponsesConstants; |
6 | 7 | import com.volcengine.ark.runtime.model.responses.content.*; |
@@ -293,6 +294,46 @@ public static void main(String[] args) { |
293 | 294 | System.err.println("Create Response 6 Error " + e.getMessage()); |
294 | 295 | } |
295 | 296 |
|
| 297 | + System.out.println("\n----- [Request with Caching] Request 7-----"); |
| 298 | + String longPrompt = "你是豆包,你必须用4个字回答我的问题"; |
| 299 | + for (int i = 0; i < 1000; i++) { |
| 300 | + longPrompt += "你是豆包,你必须用4个字回答我的问题"; |
| 301 | + } |
| 302 | + CreateResponsesRequest request7 = CreateResponsesRequest.builder() |
| 303 | + .model(modelName) |
| 304 | + .stream(false) |
| 305 | + .caching(ResponsesCaching.builder().type(ResponsesConstants.CACHE_TYPE_ENABLED).prefix(true).build()) |
| 306 | + .input(ResponsesInput.builder().stringValue(longPrompt).build()) |
| 307 | + .thinking(ResponsesThinking.builder().type(ResponsesConstants.THINKING_TYPE_DISABLED).build()) |
| 308 | + .build(); |
| 309 | + |
| 310 | + String cacheResponseId = null; |
| 311 | + |
| 312 | + try { |
| 313 | + ResponseObject cachePrefix = service.createResponse(request7); |
| 314 | + System.out.println("=== cache prefix response ==="); |
| 315 | + printResponseObject(cachePrefix); |
| 316 | + cacheResponseId = cachePrefix.getId(); |
| 317 | + } catch (Exception e) { |
| 318 | + System.err.println("Create Response 7 Error " + e.getMessage()); |
| 319 | + } |
| 320 | + |
| 321 | + CreateResponsesRequest cachedRequest = CreateResponsesRequest.builder() |
| 322 | + .model(modelName) |
| 323 | + .stream(false) |
| 324 | + .previousResponseId(cacheResponseId) |
| 325 | + .input(ResponsesInput.builder().stringValue("你好").build()) |
| 326 | + .thinking(ResponsesThinking.builder().type(ResponsesConstants.THINKING_TYPE_DISABLED).build()) |
| 327 | + .build(); |
| 328 | + |
| 329 | + try { |
| 330 | + ResponseObject cacheHitResponse = service.createResponse(cachedRequest); |
| 331 | + System.out.println("=== cache hit response ==="); |
| 332 | + printResponseObject(cacheHitResponse); |
| 333 | + } catch (Exception e) { |
| 334 | + System.err.println("Create Cached Response Error " + e.getMessage()); |
| 335 | + } |
| 336 | + |
296 | 337 | service.shutdownExecutor(); |
297 | 338 | } |
298 | 339 |
|
|
0 commit comments