From d1cfd849c6a61800dcd86ffb45f80882ae20b2e7 Mon Sep 17 00:00:00 2001
From: zhizhijie <zhizhijie@users.noreply.gitee.com>
Date: 星期一, 14 九月 2026 22:32:36 +0800
Subject: [PATCH] security: 接口鉴权 + 跨域收紧 + AI 数据脱敏 + 日志脱敏

---
 traffic-audit-server/src/main/java/com/trafficaudit/llmintegration/service/LLMService.java |    8 +++++++-
 1 files changed, 7 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 69101c4..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 缁撴灉

--
Gitblit v1.9.1