Allow customization of invalid json reprompt

quarkiverse · Jan 8, 2025 · b03f26d · b03f26d
1 parent 4afa2f2
commit b03f26d
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) {