refactor(harness): remove legacy agent architecture

2026-07-22 18:02:01 +08:00
parent bc36248cd8
commit 8ee7cc0b70
148 changed files with 3091 additions and 13889 deletions
@@ -0,0 +1,71 @@
+package com.superbiz.agent.harness.audit;
+
+import com.alibaba.cloud.ai.graph.RunnableConfig;
+import com.fasterxml.jackson.databind.ObjectMapper;
+import com.superbiz.agent.domain.entity.AgentStep;
+import com.superbiz.agent.harness.agent.DiagnosisAgentFactory;
+import com.superbiz.agent.repository.AgentStepRepository;
+import org.junit.jupiter.api.Test;
+import org.mockito.ArgumentCaptor;
+import org.springframework.ai.chat.messages.AssistantMessage;
+import org.springframework.ai.chat.messages.UserMessage;
+
+import java.util.List;
+import java.util.Optional;
+
+import static org.junit.jupiter.api.Assertions.assertEquals;
+import static org.junit.jupiter.api.Assertions.assertFalse;
+import static org.junit.jupiter.api.Assertions.assertNull;
+import static org.mockito.ArgumentMatchers.any;
+import static org.mockito.Mockito.mock;
+import static org.mockito.Mockito.never;
+import static org.mockito.Mockito.verify;
+import static org.mockito.Mockito.when;
+
+class HarnessAgentAuditHookTest {
+
+    @Test
+    void persistsExactIdentityAndMetadataWithoutContentOrArguments() {
+        AgentStepRepository repository = mock(AgentStepRepository.class);
+        AgentStep persisted = AgentStep.builder().id(7L).build();
+        when(repository.save(any(AgentStep.class))).thenReturn(persisted);
+        when(repository.findById(7L)).thenReturn(Optional.of(persisted));
+        HarnessAgentAuditHook hook = new HarnessAgentAuditHook(
+                repository, new ObjectMapper(), DiagnosisAgentFactory.AGENT_NAME);
+        RunnableConfig config = RunnableConfig.builder()
+                .addMetadata("sessionId", "session-audit")
+                .addMetadata("runId", "run-audit")
+                .build();
+
+        hook.beforeModel(List.of(new UserMessage("secret-query")), config);
+        AssistantMessage response = AssistantMessage.builder()
+                .content("secret-model-output")
+                .toolCalls(List.of(new AssistantMessage.ToolCall(
+                        "call-1", "function", "query_logs", "{\"query\":\"secret-argument\"}")))
+                .build();
+        hook.afterModel(List.of(response), config);
+
+        ArgumentCaptor<AgentStep> captor = ArgumentCaptor.forClass(AgentStep.class);
+        verify(repository, org.mockito.Mockito.times(2)).save(captor.capture());
+        AgentStep started = captor.getAllValues().get(0);
+        AgentStep completed = captor.getAllValues().get(1);
+        assertEquals("session-audit", started.getSessionId());
+        assertEquals("run-audit", started.getRunId());
+        assertFalse(started.getModelInput().contains("secret-query"));
+        assertFalse(completed.getModelOutput().contains("secret-model-output"));
+        assertFalse(completed.getModelOutput().contains("secret-argument"));
+        assertEquals("{\"has_text\":true,\"tool_names\":[\"query_logs\"]}", completed.getModelOutput());
+        assertNull(completed.getThought());
+    }
+
+    @Test
+    void missingIdentitySkipsPersistence() {
+        AgentStepRepository repository = mock(AgentStepRepository.class);
+        HarnessAgentAuditHook hook = new HarnessAgentAuditHook(
+                repository, new ObjectMapper(), DiagnosisAgentFactory.AGENT_NAME);
+
+        hook.beforeModel(List.of(new UserMessage("secret")), RunnableConfig.builder().build());
+
+        verify(repository, never()).save(any());
+    }
+}
@@ -0,0 +1,41 @@
+package com.superbiz.agent.harness.audit;
+
+import com.fasterxml.jackson.databind.ObjectMapper;
+import com.superbiz.agent.domain.entity.ToolInvocation;
+import com.superbiz.agent.harness.contract.EvidenceStatus;
+import com.superbiz.agent.harness.contract.InvocationStatus;
+import com.superbiz.agent.repository.ToolInvocationRepository;
+import org.junit.jupiter.api.Test;
+import org.mockito.ArgumentCaptor;
+
+import static org.junit.jupiter.api.Assertions.assertEquals;
+import static org.junit.jupiter.api.Assertions.assertFalse;
+import static org.mockito.Mockito.mock;
+import static org.mockito.Mockito.verify;
+
+class JpaToolInvocationAuditSinkTest {
+
+    @Test
+    void persistsOnlyBoundedStableMetadata() {
+        ToolInvocationRepository repository = mock(ToolInvocationRepository.class);
+        JpaToolInvocationAuditSink sink = new JpaToolInvocationAuditSink(repository, new ObjectMapper());
+
+        sink.record(new ToolInvocationAuditEvent(
+                "session-1", "run-1", "call-1", "query_logs",
+                InvocationStatus.ERROR, EvidenceStatus.ERROR, "TOOL_EXECUTION_ERROR",
+                17, 83, 0));
+
+        ArgumentCaptor<ToolInvocation> captor = ArgumentCaptor.forClass(ToolInvocation.class);
+        verify(repository).save(captor.capture());
+        ToolInvocation saved = captor.getValue();
+        assertEquals("session-1", saved.getSessionId());
+        assertEquals("run-1", saved.getRunId());
+        assertEquals("query_logs", saved.getToolName());
+        assertEquals("{\"tool_call_id\":\"call-1\",\"request_bytes\":83}", saved.getInputParams());
+        assertEquals("status=ERROR,evidence_status=ERROR", saved.getOutputPreview());
+        assertEquals("TOOL_EXECUTION_ERROR", saved.getErrorMessage());
+        String serialized = saved.getInputParams() + saved.getOutputPreview() + saved.getRetrievalDetails();
+        assertFalse(serialized.contains("query"));
+        assertFalse(serialized.contains("raw_response"));
+    }
+}