feat: add chat verifier agent

2026-07-03 10:54:33 +08:00
parent 4f5316d473
commit 9050487307
28 changed files with 3200 additions and 209 deletions
@@ -0,0 +1,105 @@
+package com.superbiz.agent.hook;
+
+import com.alibaba.cloud.ai.graph.RunnableConfig;
+import com.alibaba.cloud.ai.graph.agent.hook.HookPosition;
+import com.alibaba.cloud.ai.graph.agent.hook.HookPositions;
+import com.alibaba.cloud.ai.graph.agent.hook.messages.AgentCommand;
+import com.alibaba.cloud.ai.graph.agent.hook.messages.MessagesModelHook;
+import com.fasterxml.jackson.databind.ObjectMapper;
+import com.superbiz.agent.service.ToolTraceSummaryService;
+import com.superbiz.agent.util.SessionContextHolder;
+import com.superbiz.agent.util.VerifierContextHolder;
+import lombok.extern.slf4j.Slf4j;
+import org.springframework.ai.chat.messages.AssistantMessage;
+import org.springframework.ai.chat.messages.Message;
+import org.springframework.ai.chat.messages.UserMessage;
+
+import java.util.LinkedHashMap;
+import java.util.List;
+import java.util.Map;
+
+/**
+ * Replaces verifier history with an explicit structured payload.
+ */
+@Slf4j
+@HookPositions(HookPosition.BEFORE_MODEL)
+public class VerifierInputHook extends MessagesModelHook {
+
+    private final ToolTraceSummaryService toolTraceSummaryService;
+    private final ObjectMapper objectMapper = new ObjectMapper();
+
+    public VerifierInputHook(ToolTraceSummaryService toolTraceSummaryService) {
+        this.toolTraceSummaryService = toolTraceSummaryService;
+    }
+
+    @Override
+    public String getName() {
+        return "verifier_input_hook";
+    }
+
+    @Override
+    public AgentCommand beforeModel(List<Message> previousMessages, RunnableConfig config) {
+        try {
+            String sessionId = config.metadata("sessionId")
+                    .map(Object::toString)
+                    .orElseGet(SessionContextHolder::getSessionId);
+            String executorFinalAnswer = VerifierContextHolder.getExecutorFinalAnswer();
+            if (executorFinalAnswer == null || executorFinalAnswer.isBlank()) {
+                executorFinalAnswer = extractLastAssistantText(previousMessages);
+            }
+
+            List<Map<String, Object>> toolTraceSummary =
+                    toolTraceSummaryService.buildVerifierTraceSummary(sessionId, executorFinalAnswer);
+            VerifierContextHolder.setToolTraceSummary(toolTraceSummary);
+
+            Map<String, Object> verifierInput = new LinkedHashMap<>();
+            verifierInput.put("original_query", VerifierContextHolder.getOriginalQuery());
+            verifierInput.put("executor_final_answer", executorFinalAnswer);
+            verifierInput.put("tool_trace_summary", toolTraceSummary);
+            verifierInput.put("retry_context", VerifierContextHolder.getRetryContext());
+
+            String payload = objectMapper.writerWithDefaultPrettyPrinter().writeValueAsString(verifierInput);
+            return new AgentCommand(List.of(new UserMessage(payload)));
+        } catch (Exception e) {
+            log.error("Failed to build verifier input, fallback to original messages", e);
+            return new AgentCommand(previousMessages);
+        }
+    }
+
+    private String extractLastAssistantText(List<Message> previousMessages) {
+        for (int i = previousMessages.size() - 1; i >= 0; i--) {
+            if (previousMessages.get(i) instanceof AssistantMessage assistantMessage) {
+                String text = extractTextContent(assistantMessage);
+                if (text != null && !text.isBlank()) {
+                    return text;
+                }
+            }
+        }
+        return "";
+    }
+
+    private String extractTextContent(AssistantMessage message) {
+        try {
+            try {
+                return message.getText();
+            } catch (Exception ignore) {
+                // Fallback for older implementations.
+            }
+
+            for (String methodName : List.of("getText", "getContent")) {
+                try {
+                    var method = message.getClass().getMethod(methodName);
+                    Object value = method.invoke(message);
+                    if (value != null) {
+                        return value.toString();
+                    }
+                } catch (NoSuchMethodException ignore) {
+                    // continue
+                }
+            }
+        } catch (Exception e) {
+            log.debug("Failed to extract verifier assistant text", e);
+        }
+        return message.toString();
+    }
+}