DATA_MIDDLE_PLATFORM/Data_Middle_Platform_Server
3
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

feat(llm): 更新提示模板使用计数

Browse Source 
- 在 ConversationServiceImpl 类中的处理逻辑中增加了对提示模板使用计数的更新
- 当应用的 promptId 存在时，获取对应的 PromptTemplatesRespVO 对象
- 将使用计数增加 1，并调用 updatePromptTemplates 方法更新提示模板
This commit is contained in:
Liuyang 2025-03-17 12:44:38 +08:00
parent fe6bbbc633
commit b0239d5884
1 changed files with 8 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/conversation/ConversationServiceImpl.java
View File

@ -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());
}
}