fix failure IT

Signed-off-by: zane-neo <[email protected]>
zane-neo · Apr 12, 2024 · f0f3a5b · f0f3a5b
1 parent e138d83
commit f0f3a5b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ml-algorithms/src/main/java/org/opensearch/ml/engine/algorithms/remote/ConnectorUtils.java b/ml-algorithms/src/main/java/org/opensearch/ml/engine/algorithms/remote/ConnectorUtils.java
@@ -192,7 +192,7 @@ public static ModelTensors processOutput(
         if (modelResponse == null) {
             throw new IllegalArgumentException("model response is null");
         }
-        if (mlGuard != null && mlGuard.validate(modelResponse, MLGuard.Type.OUTPUT)) {
+        if (mlGuard != null && !mlGuard.validate(modelResponse, MLGuard.Type.OUTPUT)) {
             throw new IllegalArgumentException("guardrails triggered for LLM output");
         }
         List<ModelTensor> modelTensors = new ArrayList<>();