Skip to content

Commit 91c75f5

Browse files
sonianuj287ylwu-amzn
authored andcommitted
Test Fix: Adjust Bedrock Claude error expectation in DefaultLlmImplTests
Signed-off-by: Anuj Soni <[email protected]>
1 parent 7dbd4fa commit 91c75f5

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

search-processors/src/test/java/org/opensearch/searchpipelines/questionanswering/generative/llm/DefaultLlmImplTests.java

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -629,7 +629,7 @@ public void testChatCompletionBedrockThrowingError() throws Exception {
629629
DefaultLlmImpl connector = new DefaultLlmImpl("model_id", client);
630630
connector.setMlClient(mlClient);
631631

632-
String errorMessage = "throttled";
632+
String errorMessage = "Unsupported Claude response format";
633633
Map<String, String> messageMap = Map.of("message", errorMessage);
634634
Map<String, ?> dataAsMap = Map.of("error", messageMap);
635635
ModelTensor tensor = new ModelTensor("tensor", new Number[0], new long[0], MLResultDataType.STRING, null, null, dataAsMap);
@@ -657,7 +657,7 @@ public void testChatCompletionBedrockThrowingError() throws Exception {
657657
@Override
658658
public void onResponse(ChatCompletionOutput output) {
659659
assertTrue(output.isErrorOccurred());
660-
assertEquals(errorMessage, output.getErrors().get(0));
660+
assertTrue(output.getErrors().get(0).startsWith(errorMessage));
661661
}
662662

663663
@Override

0 commit comments

Comments
 (0)