This commit is contained in:
ire 2024-12-31 20:07:20 +08:00
parent c98721246d
commit 3db2597240

View File

@ -50,8 +50,8 @@ public class PromptTemplatesServiceImpl implements PromptTemplatesService {
PromptTemplatesDO promptTemplates = BeanUtils.toBean(createReqVO, PromptTemplatesDO.class);
// 暂时先去掉这个逻辑演示用 -- zhangtao
// promptTemplates.setTemplateType(2);
String categoryName = this.dictDataApi.getDictDataLabel(Long.valueOf(createReqVO
.getTemplateCategory().toString()));
String categoryName = this.dictDataApi.getDictDataLabel("llm_prompt_template_framework",Integer.valueOf(createReqVO
.getTemplateCategory()));
switch (categoryName) {
case "简介框架":
BriefFramework briefFramework = createReqVO.getBriefFramework();
@ -102,8 +102,8 @@ public class PromptTemplatesServiceImpl implements PromptTemplatesService {
this.isNameDuplicate(updateReqVO.getName());
}
PromptTemplatesDO updateObj = BeanUtils.toBean(updateReqVO, PromptTemplatesDO.class);
String categoryName = this.dictDataApi.getDictDataLabel(Long.valueOf(updateReqVO
.getTemplateCategory().toString()));
String categoryName = this.dictDataApi.getDictDataLabel("llm_prompt_template_framework",Integer.valueOf(updateReqVO
.getTemplateCategory()));
switch (categoryName) {
case "简介框架":
BriefFramework briefFramework = updateReqVO.getBriefFramework();
@ -193,7 +193,7 @@ public class PromptTemplatesServiceImpl implements PromptTemplatesService {
.collect(Collectors.toList());
promptTemplates.setApplicationIds(applicationIds);
}
String categoryName = this.dictDataApi.getDictDataLabel(Long.valueOf(templatesDO.getTemplateCategory().toString()));
String categoryName = this.dictDataApi.getDictDataLabel("llm_prompt_template_framework",Integer.valueOf(templatesDO.getTemplateCategory()));
switch (categoryName) {
case "简介框架":
promptTemplates.setBriefFramework(JSON.parseObject(promptTemplates
@ -278,4 +278,4 @@ public class PromptTemplatesServiceImpl implements PromptTemplatesService {
ExcelUtils.write(response, "Prompt模板信息.xls", "数据",
PromptTemplatesExcelVO.class, excelList);
}
}
}