From 8c5ab8eac4c2864a6ccda55c346c04d3d969bf43 Mon Sep 17 00:00:00 2001 From: sion123 <450702724@qq.com> Date: Thu, 12 Mar 2026 23:56:47 +0800 Subject: [PATCH] =?UTF-8?q?feat:=20=E4=BC=98=E5=8C=96?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../tik/dify/service/DifyServiceImpl.java | 26 +++++++++++++++++-- .../service/AiUsageStatsServiceImpl.java | 1 + 2 files changed, 25 insertions(+), 2 deletions(-) diff --git a/yudao-module-tik/src/main/java/cn/iocoder/yudao/module/tik/dify/service/DifyServiceImpl.java b/yudao-module-tik/src/main/java/cn/iocoder/yudao/module/tik/dify/service/DifyServiceImpl.java index e2fcfcbdcd..16c22d1ae0 100644 --- a/yudao-module-tik/src/main/java/cn/iocoder/yudao/module/tik/dify/service/DifyServiceImpl.java +++ b/yudao-module-tik/src/main/java/cn/iocoder/yudao/module/tik/dify/service/DifyServiceImpl.java @@ -669,7 +669,18 @@ public class DifyServiceImpl implements DifyService { // Dify 用户标识 String difyUserId = "user-" + userId; - return difyClient.getConversations(config.getApiKey(), difyUserId, lastId, limit); + DifyConversationListRespVO result = difyClient.getConversations(config.getApiKey(), difyUserId, lastId, limit); + + // 过滤掉 inputs 中的敏感字段(如 sysPrompt) + if (result != null && result.getData() != null) { + result.getData().forEach(conv -> { + if (conv.getInputs() != null) { + conv.getInputs().remove("sysPrompt"); + } + }); + } + + return result; } @Override @@ -688,7 +699,18 @@ public class DifyServiceImpl implements DifyService { // Dify 用户标识 String difyUserId = "user-" + userId; - return difyClient.getMessages(config.getApiKey(), conversationId, difyUserId, firstId, limit); + DifyMessageListRespVO result = difyClient.getMessages(config.getApiKey(), conversationId, difyUserId, firstId, limit); + + // 过滤掉 inputs 中的敏感字段(如 sysPrompt) + if (result != null && result.getData() != null) { + result.getData().forEach(msg -> { + if (msg.getInputs() != null) { + msg.getInputs().remove("sysPrompt"); + } + }); + } + + return result; } } diff --git a/yudao-module-tik/src/main/java/cn/iocoder/yudao/module/tik/muye/aiusagestats/service/AiUsageStatsServiceImpl.java b/yudao-module-tik/src/main/java/cn/iocoder/yudao/module/tik/muye/aiusagestats/service/AiUsageStatsServiceImpl.java index 8ace06c3ea..fbbea449ee 100644 --- a/yudao-module-tik/src/main/java/cn/iocoder/yudao/module/tik/muye/aiusagestats/service/AiUsageStatsServiceImpl.java +++ b/yudao-module-tik/src/main/java/cn/iocoder/yudao/module/tik/muye/aiusagestats/service/AiUsageStatsServiceImpl.java @@ -2,6 +2,7 @@ package cn.iocoder.yudao.module.tik.muye.aiusagestats.service; import cn.iocoder.yudao.framework.common.pojo.PageResult; import cn.iocoder.yudao.framework.mybatis.core.query.LambdaQueryWrapperX; +import cn.iocoder.yudao.framework.tenant.core.util.TenantUtils; import cn.iocoder.yudao.module.tik.muye.aiusagestats.vo.*; import cn.iocoder.yudao.module.tik.muye.aiserviceconfig.dal.AiServiceConfigDO; import cn.iocoder.yudao.module.tik.muye.aiserviceconfig.mapper.AiServiceConfigMapper;