From b0239d5884d378e67b928a55cf096f49e0bdbaa0 Mon Sep 17 00:00:00 2001 From: Liuyang <2746366019@qq.com> Date: Mon, 17 Mar 2025 12:44:38 +0800 Subject: [PATCH] =?UTF-8?q?feat(llm):=20=E6=9B=B4=E6=96=B0=E6=8F=90?= =?UTF-8?q?=E7=A4=BA=E6=A8=A1=E6=9D=BF=E4=BD=BF=E7=94=A8=E8=AE=A1=E6=95=B0?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit - 在 ConversationServiceImpl 类中的处理逻辑中增加了对提示模板使用计数的更新 - 当应用的 promptId 存在时，获取对应的 PromptTemplatesRespVO 对象 - 将使用计数增加 1，并调用 updatePromptTemplates 方法更新提示模板 --- .../llm/service/conversation/ConversationServiceImpl.java | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/conversation/ConversationServiceImpl.java b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/conversation/ConversationServiceImpl.java index 4b14b1afb..8de90b3b0 100644 --- a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/conversation/ConversationServiceImpl.java +++ b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/conversation/ConversationServiceImpl.java @@ -16,6 +16,8 @@ import cn.iocoder.yudao.module.llm.controller.admin.datarefluxdata.vo.DataReflux import cn.iocoder.yudao.module.llm.controller.admin.knowledgebase.vo.KnowledgeBaseSaveReqVO; import cn.iocoder.yudao.module.llm.controller.admin.knowledgebase.vo.KnowledgeHitRateTestReqVO; import cn.iocoder.yudao.module.llm.controller.admin.knowledgebase.vo.KnowledgeHitRateTestResultVO; +import cn.iocoder.yudao.module.llm.controller.admin.prompttemplates.vo.PromptTemplatesRespVO; +import cn.iocoder.yudao.module.llm.controller.admin.prompttemplates.vo.PromptTemplatesSaveReqVO; import cn.iocoder.yudao.module.llm.dal.dataobject.basemodel.BaseModelDO; import cn.iocoder.yudao.module.llm.dal.dataobject.conversation.ConversationDO; import cn.iocoder.yudao.module.llm.dal.dataobject.knowledgedocuments.KnowledgeDocumentsDO; @@ -326,6 +328,12 @@ public class ConversationServiceImpl implements ConversationService { } } + Optional.ofNullable(application.getPromptId()).ifPresent(promptId -> { + PromptTemplatesRespVO promptTemplates = promptTemplatesService.getPromptTemplates(promptId); + promptTemplates.setUseCount(Math.max(1, Optional.ofNullable(promptTemplates.getUseCount()).orElse(0) + 1)); + promptTemplatesService.updatePromptTemplates(BeanUtil.toBean(promptTemplates, PromptTemplatesSaveReqVO.class)); + }); + log.info("已更新系统提示信息为: {}", chatReqVO.getSystemPrompt()); } }