From 226cc856b504a0982fae8008f47acc02abab90a3 Mon Sep 17 00:00:00 2001 From: Eric Deandrea Date: Mon, 6 Jan 2025 15:11:40 -0500 Subject: [PATCH] Fix isSuccessful --- .../guardrails/OutputGuardrailResultAssert.java | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/testing/core/src/main/java/io/quarkiverse/langchain4j/guardrails/OutputGuardrailResultAssert.java b/testing/core/src/main/java/io/quarkiverse/langchain4j/guardrails/OutputGuardrailResultAssert.java index f49c613c5..05bf5e19c 100644 --- a/testing/core/src/main/java/io/quarkiverse/langchain4j/guardrails/OutputGuardrailResultAssert.java +++ b/testing/core/src/main/java/io/quarkiverse/langchain4j/guardrails/OutputGuardrailResultAssert.java @@ -55,16 +55,17 @@ public OutputGuardrailResultAssert hasResult(Result result) { /** * Asserts that the actual {@code OutputGuardrailResult} represents a successful state. - * A successful state is determined by having a {@link Result} of {@link Result#SUCCESS} - * and being equal to {@link OutputGuardrailResult#success()}. + * A successful state is determined by having {@link OutputGuardrailResult#isSuccess()}. * * @return this assertion object for method chaining * @throws AssertionError if the actual result is not successful as per the aforementioned criteria */ public OutputGuardrailResultAssert isSuccessful() { isNotNull(); - hasResult(Result.SUCCESS); - isEqualTo(OutputGuardrailResult.success()); + + if (!actual.isSuccess()) { + throw failure("Expected result to be successful but was <%s>", actual.getResult()); + } return this; }