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} +