Merge pull request #1204 from edeandrea/enhance-json-guardrail

Allow customization of invalid json reprompt
quarkiverse · Jan 9, 2025 · 7d10993 · 7d10993
2 parents 7cf8fd4 + b03f26d
commit 7d10993
Showing 1 changed file with 13 additions and 3 deletions.
diff --git a/...main/java/io/quarkiverse/langchain4j/guardrails/AbstractJsonExtractorOutputGuardrail.java b/...main/java/io/quarkiverse/langchain4j/guardrails/AbstractJsonExtractorOutputGuardrail.java
@@ -40,9 +40,19 @@ public OutputGuardrailResult validate(AiMessage responseFromLLM) {
             }
         }
 
-        return reprompt("Invalid JSON",
-                "Make sure you return a valid JSON object following "
-                        + "the specified format");
+        return invokeInvalidJson(responseFromLLM, json);
+    }
+
+    protected OutputGuardrailResult invokeInvalidJson(AiMessage aiMessage, String json) {
+        return reprompt(getInvalidJsonMessage(aiMessage, json), getInvalidJsonReprompt(aiMessage, json));
+    }
+
+    protected String getInvalidJsonMessage(AiMessage aiMessage, String json) {
+        return "Invalid JSON";
+    }
+
+    protected String getInvalidJsonReprompt(AiMessage aiMessage, String json) {
+        return "Make sure you return a valid JSON object following the specified format";
     }
 
     protected Object deserialize(String llmResponse) {