feat(agent): add executor gatekeeper hook

src/main/java/com/superbiz/agent/hook/VerifierInputHook.java

@@ -8,6 +8,7 @@ import com.alibaba.cloud.ai.graph.agent.hook.messages.MessagesModelHook;
 import com.fasterxml.jackson.core.type.TypeReference;
 import com.fasterxml.jackson.databind.JsonNode;
 import com.fasterxml.jackson.databind.ObjectMapper;
+import com.superbiz.agent.service.ExecutorGatekeeperService;
 import com.superbiz.agent.service.ToolTraceSummaryService;
 import com.superbiz.agent.util.SessionContextHolder;
 import com.superbiz.agent.util.VerifierContextHolder;
@@ -28,12 +29,19 @@ import java.util.Map;
 public class VerifierInputHook extends MessagesModelHook {
 
     private final ToolTraceSummaryService toolTraceSummaryService;
+    private final ExecutorGatekeeperService executorGatekeeperService;
     private final ObjectMapper objectMapper = new ObjectMapper();
     private static final TypeReference<Map<String, Object>> MAP_TYPE = new TypeReference<>() {
     };
 
     public VerifierInputHook(ToolTraceSummaryService toolTraceSummaryService) {
+        this(toolTraceSummaryService, null);
+    }
+
+    public VerifierInputHook(ToolTraceSummaryService toolTraceSummaryService,
+                             ExecutorGatekeeperService executorGatekeeperService) {
         this.toolTraceSummaryService = toolTraceSummaryService;
+        this.executorGatekeeperService = executorGatekeeperService;
     }
 
     @Override
@@ -60,12 +68,16 @@ public class VerifierInputHook extends MessagesModelHook {
                     toolTraceSummaryService.buildVerifierTraceSummary(sessionId, executorFinalAnswer);
             VerifierContextHolder.setToolTraceSummary(toolTraceSummary);
 
+            Map<String, Object> gatekeeperResult = runGatekeeper(sessionId, parseResult);
+            VerifierContextHolder.setGatekeeperResult(gatekeeperResult);
+
             Map<String, Object> verifierInput = new LinkedHashMap<>();
             verifierInput.put("original_query", VerifierContextHolder.getOriginalQuery());
             verifierInput.put("executor_final_answer", executorFinalAnswer);
             verifierInput.put("executor_structured_output", parseResult.structuredOutput());
             verifierInput.put("executor_output_parse_status", parseResult.status());
             verifierInput.put("tool_trace_summary", toolTraceSummary);
+            verifierInput.put("gatekeeper_result", gatekeeperResult);
             verifierInput.put("retry_context", VerifierContextHolder.getRetryContext());
 
             String payload = objectMapper.writerWithDefaultPrettyPrinter().writeValueAsString(verifierInput);
@@ -76,6 +88,29 @@ public class VerifierInputHook extends MessagesModelHook {
         }
     }
 
+    private Map<String, Object> runGatekeeper(String sessionId, ExecutorOutputParseResult parseResult) {
+        if (executorGatekeeperService == null) {
+            return passGatekeeperResult();
+        }
+        try {
+            return executorGatekeeperService.validate(sessionId, parseResult.structuredOutput(), parseResult.status());
+        } catch (Exception e) {
+            log.error("Gatekeeper validation failed unexpectedly", e);
+            return executorGatekeeperService.fail("gatekeeper.internal_error",
+                    "gatekeeper",
+                    e.getMessage() == null ? "gatekeeper validation failed" : e.getMessage());
+        }
+    }
+
+    private Map<String, Object> passGatekeeperResult() {
+        Map<String, Object> result = new LinkedHashMap<>();
+        result.put("status", "pass");
+        result.put("failed_rules", List.of());
+        result.put("warnings", List.of());
+        result.put("errors", List.of());
+        return result;
+    }
+
     private ExecutorOutputParseResult parseExecutorOutput(String executorFinalAnswer) {
         if (executorFinalAnswer == null || executorFinalAnswer.isBlank()) {
             return new ExecutorOutputParseResult(null, status("missing", "executor_final_answer is blank"));