From 71a77e721619e956cab116be691cc3ab58e18093 Mon Sep 17 00:00:00 2001
From: xyc <jc_xiong@hotmail.com>
Date: 星期二, 22 九月 2026 15:35:09 +0800
Subject: [PATCH] Merge remote-tracking branch 'origin/main'
---
traffic-audit-server/src/main/java/com/trafficaudit/llmintegration/service/LLMService.java | 14 +++++++++++++-
1 files changed, 13 insertions(+), 1 deletions(-)
diff --git a/traffic-audit-server/src/main/java/com/trafficaudit/llmintegration/service/LLMService.java b/traffic-audit-server/src/main/java/com/trafficaudit/llmintegration/service/LLMService.java
index cb88562..bf66012 100644
--- a/traffic-audit-server/src/main/java/com/trafficaudit/llmintegration/service/LLMService.java
+++ b/traffic-audit-server/src/main/java/com/trafficaudit/llmintegration/service/LLMService.java
@@ -24,6 +24,8 @@
private DeepSeekConfig config;
@Resource
private DbQueryExecutor dbQueryExecutor;
+ @Resource
+ private DesensitizeService desensitizeService;
public String chat(String prompt) {
JSONObject body = JSONUtil.createObj()
@@ -84,7 +86,11 @@
JSONArray toolCalls = message.getJSONArray("tool_calls");
if (toolCalls == null || toolCalls.isEmpty()) {
String content = message.getStr("content");
- return content == null ? "锛圓I 鏈繑鍥炲唴瀹癸級" : content;
+ if (content == null) {
+ return "锛圓I 鏈繑鍥炲唴瀹癸級";
+ }
+ // 鍥炲～缁欏墠绔墠锛屾妸浼佷笟鍚嶇О鍒悕杩樺師涓虹湡瀹炲悕绉�
+ return desensitizeService.unmaskAll(content);
}
// 鍥炰紶 assistant 娑堟伅锛堝惈宸ュ叿璋冪敤锛夛紝鍐嶉�愪釜鎵ц骞跺洖濉� tool 缁撴灉
@@ -159,6 +165,12 @@
if (ctx.get("reportId") != null) {
sb.append("鍏宠仈涓婃姤璁板綍ID: ").append(ctx.get("reportId")).append("\n");
}
+ if (ctx.get("reportType") != null) {
+ sb.append("鍏宠仈涓氬姟绫诲瀷: ").append(ctx.get("reportType")).append("\n");
+ }
+ if (ctx.get("tableName") != null) {
+ sb.append("鍏宠仈鏁版嵁琛�: ").append(ctx.get("tableName")).append("锛堟煡璇㈣浼佷笟瑙ｉ噴瀵瑰簲鏁版嵁鏃朵紭鍏堟煡姝よ〃锛屾寜 id=").append(ctx.get("reportId")).append(" 杩囨护锛塡n");
+ }
}
sb.append("\n浣犲彲浠ヨ皟鐢� query_database 宸ュ叿鏌ヨ鏁版嵁搴撹幏鍙栫湡瀹炴暟鎹紝濡備紒涓氳溅杈嗘暟銆佸惃浣嶃�佽揣杩愰噺銆佸懆杞噺銆佽建杩归噷绋嬨�佽涓婅涓嬫媶鍒嗐�佸幓骞村悓鏈熺瓑銆俓n\n");
sb.append("鏁版嵁搴撹〃缁撴瀯濡備笅锛堝垪鍚嶄竴寰嬩娇鐢� camelCase锛屼緥濡� enterpriseName銆乿ehicleTotal銆乫reightTotal銆乼urnoverTotal锛沠ilters 鐨勯敭鍚屾牱浣跨敤 camelCase锛夛細\n");
--
Gitblit v1.9.1