|
6 | 6 | import static com.box.sdkgen.internal.utils.UtilsManager.dateTimeFromString; |
7 | 7 | import static com.box.sdkgen.internal.utils.UtilsManager.delayInSeconds; |
8 | 8 | import static com.box.sdkgen.internal.utils.UtilsManager.getUuid; |
9 | | -import static com.box.sdkgen.internal.utils.UtilsManager.getValueFromObjectRawData; |
10 | 9 | import static com.box.sdkgen.internal.utils.UtilsManager.stringToByteStream; |
11 | 10 |
|
12 | 11 | import com.box.sdkgen.client.BoxClient; |
@@ -338,13 +337,12 @@ public void testAiExtractStructuredWithFields() { |
338 | 337 | .build())) |
339 | 338 | .aiAgent(agentIgnoringOverridingEmbeddingsModel) |
340 | 339 | .build()); |
341 | | - assert convertToString(getValueFromObjectRawData(response, "answer.hobby")) |
| 340 | + assert convertToString(response.getAnswer().get("hobby")) |
342 | 341 | .equals(convertToString(Arrays.asList("guitar"))); |
343 | | - assert convertToString(getValueFromObjectRawData(response, "answer.firstName")).equals("John"); |
344 | | - assert convertToString(getValueFromObjectRawData(response, "answer.lastName")).equals("Doe"); |
345 | | - assert convertToString(getValueFromObjectRawData(response, "answer.dateOfBirth")) |
346 | | - .equals("1990-07-04"); |
347 | | - assert convertToString(getValueFromObjectRawData(response, "answer.age")).equals("34"); |
| 342 | + assert convertToString(response.getAnswer().get("firstName")).equals("John"); |
| 343 | + assert convertToString(response.getAnswer().get("lastName")).equals("Doe"); |
| 344 | + assert convertToString(response.getAnswer().get("dateOfBirth")).equals("1990-07-04"); |
| 345 | + assert convertToString(response.getAnswer().get("age")).equals("34"); |
348 | 346 | assert response.getCompletionReason().equals("done"); |
349 | 347 | client.getFiles().deleteFileById(file.getId()); |
350 | 348 | } |
@@ -420,12 +418,11 @@ public void testAiExtractStructuredWithMetadataTemplate() { |
420 | 418 | .scope("enterprise") |
421 | 419 | .build()) |
422 | 420 | .build()); |
423 | | - assert convertToString(getValueFromObjectRawData(response, "answer.firstName")).equals("John"); |
424 | | - assert convertToString(getValueFromObjectRawData(response, "answer.lastName")).equals("Doe"); |
425 | | - assert convertToString(getValueFromObjectRawData(response, "answer.dateOfBirth")) |
426 | | - .equals("1990-07-04T00:00:00Z"); |
427 | | - assert convertToString(getValueFromObjectRawData(response, "answer.age")).equals("34"); |
428 | | - assert convertToString(getValueFromObjectRawData(response, "answer.hobby")) |
| 421 | + assert convertToString(response.getAnswer().get("firstName")).equals("John"); |
| 422 | + assert convertToString(response.getAnswer().get("lastName")).equals("Doe"); |
| 423 | + assert convertToString(response.getAnswer().get("dateOfBirth")).equals("1990-07-04T00:00:00Z"); |
| 424 | + assert convertToString(response.getAnswer().get("age")).equals("34"); |
| 425 | + assert convertToString(response.getAnswer().get("hobby")) |
429 | 426 | .equals(convertToString(Arrays.asList("guitar"))); |
430 | 427 | assert response.getCompletionReason().equals("done"); |
431 | 428 | client |
|
0 commit comments