From bcabde8bfb8f16671fa5937e62668f99210e0720 Mon Sep 17 00:00:00 2001
From: Junjie <fallin.jie@qq.com>
Date: 星期四, 30 四月 2026 08:56:45 +0800
Subject: [PATCH] #Agent自动调参
---
src/main/java/com/zy/system/controller/DashboardController.java | 32 ++++++++++++++++++++++++++++++++
1 files changed, 32 insertions(+), 0 deletions(-)
diff --git a/src/main/java/com/zy/system/controller/DashboardController.java b/src/main/java/com/zy/system/controller/DashboardController.java
index de5cdc9..989f8c7 100644
--- a/src/main/java/com/zy/system/controller/DashboardController.java
+++ b/src/main/java/com/zy/system/controller/DashboardController.java
@@ -3,10 +3,13 @@
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.core.annotations.ManagerAuth;
import com.core.common.R;
+import com.zy.ai.entity.AiAutoTuneJob;
import com.zy.ai.entity.AiChatSession;
import com.zy.ai.entity.LlmCallLog;
import com.zy.ai.entity.LlmRouteConfig;
+import com.zy.ai.enums.AiPromptScene;
import com.zy.ai.mapper.AiChatSessionMapper;
+import com.zy.ai.service.AiAutoTuneJobService;
import com.zy.ai.service.LlmCallLogService;
import com.zy.ai.service.LlmRouteConfigService;
import com.zy.asrs.entity.*;
@@ -58,6 +61,8 @@
private LlmRouteConfigService llmRouteConfigService;
@Autowired
private LlmCallLogService llmCallLogService;
+ @Autowired
+ private AiAutoTuneJobService aiAutoTuneJobService;
@Autowired
private AiChatSessionMapper aiChatSessionMapper;
@Autowired
@@ -327,6 +332,9 @@
long completionTokenTotal = 0L;
long askCount = 0L;
long sessionCount = 0L;
+ long autoTunePromptTokenTotal = 0L;
+ long autoTuneCompletionTokenTotal = 0L;
+ long autoTuneTokenTotal = 0L;
try {
List<AiChatSession> sessions = aiChatSessionMapper.selectList(new QueryWrapper<AiChatSession>()
.select("id", "sum_prompt_tokens", "sum_completion_tokens", "sum_total_tokens", "ask_count"));
@@ -341,6 +349,27 @@
}
} catch (Exception e) {
log.warn("dashboard ai session stats load failed: {}", safeMessage(e));
+ }
+
+ try {
+ List<Map<String, Object>> autoTuneRows = aiAutoTuneJobService.listMaps(new QueryWrapper<AiAutoTuneJob>()
+ .select("COALESCE(SUM(prompt_tokens), 0) AS prompt_token_total",
+ "COALESCE(SUM(completion_tokens), 0) AS completion_token_total",
+ "COALESCE(SUM(total_tokens), 0) AS token_total")
+ .eq("prompt_scene_code", AiPromptScene.AUTO_TUNE_DISPATCH.getCode()));
+ Map<String, Object> autoTuneRow = autoTuneRows == null || autoTuneRows.isEmpty()
+ ? Collections.emptyMap()
+ : autoTuneRows.get(0);
+ autoTunePromptTokenTotal = toLong(autoTuneRow.get("prompt_token_total"));
+ autoTuneCompletionTokenTotal = toLong(autoTuneRow.get("completion_token_total"));
+ autoTuneTokenTotal = toLong(autoTuneRow.get("token_total"));
+
+ // Agent 鑷姩璋冨弬涓嶇敓鎴� sys_ai_chat_session锛屼細鍗曠嫭钀藉埌 sys_ai_auto_tune_job銆�
+ promptTokenTotal += autoTunePromptTokenTotal;
+ completionTokenTotal += autoTuneCompletionTokenTotal;
+ tokenTotal += autoTuneTokenTotal;
+ } catch (Exception e) {
+ log.warn("dashboard ai auto tune token stats load failed: {}", safeMessage(e));
}
List<LlmRouteConfig> routes = Collections.emptyList();
@@ -415,6 +444,9 @@
overview.put("tokenTotal", tokenTotal);
overview.put("promptTokenTotal", promptTokenTotal);
overview.put("completionTokenTotal", completionTokenTotal);
+ overview.put("autoTuneTokenTotal", autoTuneTokenTotal);
+ overview.put("autoTunePromptTokenTotal", autoTunePromptTokenTotal);
+ overview.put("autoTuneCompletionTokenTotal", autoTuneCompletionTokenTotal);
overview.put("askCount", askCount);
overview.put("sessionCount", sessionCount);
overview.put("routeTotal", routeTotal);
--
Gitblit v1.9.1