From d454971b8f6f936f4b07a72e62a9e80860f2bcfa Mon Sep 17 00:00:00 2001 From: Liuyang <2746366019@qq.com> Date: Mon, 17 Mar 2025 13:08:45 +0800 Subject: [PATCH] =?UTF-8?q?refactor(module-llm):=E4=BC=98=E5=8C=96=20Promp?= =?UTF-8?q?t=20=E6=A8=A1=E6=9D=BF=E4=BD=BF=E7=94=A8=E6=AC=A1=E6=95=B0?= =?UTF-8?q?=E6=9B=B4=E6=96=B0=E9=80=BB=E8=BE=91?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit - 在 ConversationServiceImpl 中，将 PromptTemplatesSaveReqVO 的使用次数更新逻辑改为调用新的 updatePromptTemplatesUseCount 方法 - 在 PromptTemplatesApplicationsMapper 中添加 updatePromptTemplatesUseCount 方法，用于更新 Prompt 模板的使用次数 - 在 PromptTemplatesApplicationsMapper.xml 中添加对应的 SQL 语句，用于更新 Prompt 模板的使用次数- 在 PromptTemplatesService 中添加 updatePromptTemplatesUseCount 方法的接口定义 - 在 PromptTemplatesServiceImpl 中实现 --- .../PromptTemplatesApplicationsMapper.java | 3 +++ .../llm/service/conversation/ConversationServiceImpl.java | 4 ++-- .../llm/service/prompttemplates/PromptTemplatesService.java | 2 +- .../service/prompttemplates/PromptTemplatesServiceImpl.java | 6 ++++++ .../PromptTemplatesApplicationsMapper.xml | 3 +++ 5 files changed, 15 insertions(+), 3 deletions(-) diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/dal/mysql/prompttemplatesapplications/PromptTemplatesApplicationsMapper.java b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/dal/mysql/prompttemplatesapplications/PromptTemplatesApplicationsMapper.java index ae6fbf319..165254c71 100644 --- a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/dal/mysql/prompttemplatesapplications/PromptTemplatesApplicationsMapper.java +++ b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/dal/mysql/prompttemplatesapplications/PromptTemplatesApplicationsMapper.java @@ -8,6 +8,7 @@ import cn.iocoder.yudao.framework.mybatis.core.mapper.BaseMapperX; import cn.iocoder.yudao.module.llm.dal.dataobject.prompttemplatesapplications.PromptTemplatesApplicationsDO; import org.apache.ibatis.annotations.Mapper; import cn.iocoder.yudao.module.llm.controller.admin.prompttemplatesapplications.vo.*; +import org.apache.ibatis.annotations.Param; /** * Prompt 模板 应用关系表，记录模板和应用之间的多对多关系 Mapper @@ -25,4 +26,6 @@ public interface PromptTemplatesApplicationsMapper extends BaseMapperX { PromptTemplatesRespVO promptTemplates = promptTemplatesService.getPromptTemplates(promptId); - promptTemplates.setUseCount(Math.max(1, Optional.ofNullable(promptTemplates.getUseCount()).orElse(0) + 1)); - promptTemplatesService.updatePromptTemplates(BeanUtil.toBean(promptTemplates, PromptTemplatesSaveReqVO.class)); + int count = Math.max(1, Optional.ofNullable(promptTemplates.getUseCount()).orElse(0) + 1); + promptTemplatesService.updatePromptTemplatesUseCount(Math.toIntExact(promptId),count); }); log.info("已更新系统提示信息为: {}", chatReqVO.getSystemPrompt()); diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/prompttemplates/PromptTemplatesService.java b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/prompttemplates/PromptTemplatesService.java index c43a423d8..26b67c563 100644 --- a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/prompttemplates/PromptTemplatesService.java +++ b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/prompttemplates/PromptTemplatesService.java @@ -15,7 +15,7 @@ public interface PromptTemplatesService { Long createPromptTemplates (@Valid PromptTemplatesSaveReqVO createReqVO); void updatePromptTemplates (@Valid PromptTemplatesSaveReqVO updateReqVO); - + void updatePromptTemplatesUseCount (Integer promptId,Integer count); /** * 删除 Prompt模板 * diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/prompttemplates/PromptTemplatesServiceImpl.java b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/prompttemplates/PromptTemplatesServiceImpl.java index dbf4fcbeb..f9f521614 100644 --- a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/prompttemplates/PromptTemplatesServiceImpl.java +++ b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/prompttemplates/PromptTemplatesServiceImpl.java @@ -184,6 +184,12 @@ public class PromptTemplatesServiceImpl implements PromptTemplatesService { } } + @Override + public void updatePromptTemplatesUseCount (Integer promptId,Integer count) { +// + this.promptTemplatesApplicationsMapper.updatePromptTemplatesUseCount(promptId,count); + } + /** * 删除 Prompt模板 * diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/resources/mapper/prompttemplatesapplications/PromptTemplatesApplicationsMapper.xml b/yudao-module-llm/yudao-module-llm-biz/src/main/resources/mapper/prompttemplatesapplications/PromptTemplatesApplicationsMapper.xml index 36ad28841..d9e51c205 100644 --- a/yudao-module-llm/yudao-module-llm-biz/src/main/resources/mapper/prompttemplatesapplications/PromptTemplatesApplicationsMapper.xml +++ b/yudao-module-llm/yudao-module-llm-biz/src/main/resources/mapper/prompttemplatesapplications/PromptTemplatesApplicationsMapper.xml @@ -9,4 +9,7 @@ 文档可见：https://www.iocoder.cn/MyBatis/x-plugins/ --> + + UPDATE llm_prompt_templates t SET t.use_count = #{count} WHERE t.id = #{promptId} +