feat: add traceable scoped AIOps diagnosis

src/main/java/com/superbiz/agent/service/ChatService.java

@@ -18,6 +18,7 @@ import com.superbiz.agent.hook.TokenUsageHolder;
 import com.superbiz.agent.hook.VerifierInputHook;
 import com.superbiz.agent.repository.AgentStepRepository;
 import com.superbiz.agent.repository.DiagnosisSessionRepository;
+import com.superbiz.agent.repository.ToolInvocationRepository;
 import com.superbiz.agent.tool.LookupKnowledgeTool;
 import com.superbiz.agent.tool.RetrievedDocTracker;
 import com.superbiz.agent.util.QuestionComplexity;
@@ -86,6 +87,9 @@ public class ChatService {
     @Autowired
     private AgentStepRepository agentStepRepository;
 
+    @Autowired
+    private ToolInvocationRepository toolInvocationRepository;
+
     @Autowired
     private EvaluationService evaluationService;
 
@@ -838,25 +842,22 @@ public class ChatService {
     ) {
     }
 
-    /** 从 agent_step 汇总 token、步数等指标回填 diagnosis_session */
+    /** 从 agent_step 和 tool_invocation 汇总指标回填 diagnosis_session */
     private void backfillSessionMetrics(DiagnosisSession session) {
         try {
             List<com.superbiz.agent.domain.entity.AgentStep> steps =
                 agentStepRepository.findBySessionIdOrderByStepIndex(session.getSessionId());
 
-            if (steps.isEmpty()) return;
-
             int totalTokens = 0;
             int stepCount = 0;
-            int toolCallCount = 0;
             for (var s : steps) {
                 stepCount++;
                 if (s.getTokenCount() != null) totalTokens += s.getTokenCount();
-                if (Boolean.TRUE.equals(s.getHasToolCall())) toolCallCount++;
             }
+            long toolCallCount = toolInvocationRepository.countBySessionId(session.getSessionId());
             session.setTotalTokenCount(totalTokens);
             session.setStepCount(stepCount);
-            session.setToolCallCount(toolCallCount);
+            session.setToolCallCount(Math.toIntExact(toolCallCount));
         } catch (Exception e) {
             logger.warn("回填会话指标失败: sessionId={}", session.getSessionId(), e);
         }