prompt
This commit is contained in:
ire
parent
db9925e1ce
commit
8b7e97b2f8
@ -8,7 +8,6 @@ import cn.iocoder.yudao.framework.common.util.object.BeanUtils;
|
||||
import cn.iocoder.yudao.framework.excel.core.util.ExcelUtils;
|
||||
import cn.iocoder.yudao.module.llm.controller.admin.prompttemplates.vo.*;
|
||||
import cn.iocoder.yudao.module.llm.controller.admin.prompttemplatesbackup.vo.PromptTemplatesBackupSaveReqVO;
|
||||
import cn.iocoder.yudao.module.llm.dal.dataobject.application.ApplicationDO;
|
||||
import cn.iocoder.yudao.module.llm.dal.dataobject.label.LabelDO;
|
||||
import cn.iocoder.yudao.module.llm.dal.dataobject.prompttemplates.PromptTemplatesDO;
|
||||
import cn.iocoder.yudao.module.llm.dal.dataobject.prompttemplatesapplications.PromptTemplatesApplicationsDO;
|
||||
@ -68,25 +67,24 @@ public class PromptTemplatesServiceImpl implements PromptTemplatesService {
|
||||
this.isNameDuplicate(createReqVO.getName());
|
||||
PromptTemplatesDO promptTemplates = BeanUtils.toBean(createReqVO, PromptTemplatesDO.class);
|
||||
//
|
||||
promptTemplates.setTemplateType(2);
|
||||
String categoryName = this.dictDataApi.getDictDataLabel("llm_prompt_template_framework",Integer.valueOf(createReqVO
|
||||
.getTemplateFramework()));
|
||||
switch (categoryName) {
|
||||
case "简介框架":
|
||||
BriefFramework briefFramework = createReqVO.getBriefFramework();
|
||||
promptTemplates.setTemplateText(JSON.toJSONString(briefFramework));
|
||||
break;
|
||||
case "角色能力框架":
|
||||
AbilityFramework abilityFramework = createReqVO.getAbilityFramework();
|
||||
promptTemplates.setTemplateText(JSON.toJSONString(abilityFramework));
|
||||
break;
|
||||
case "样例提示框架":
|
||||
SampleFramework sampleFramework = createReqVO.getSampleFramework();
|
||||
promptTemplates.setTemplateText(JSON.toJSONString(sampleFramework));
|
||||
break;
|
||||
default:
|
||||
break;
|
||||
}
|
||||
promptTemplates.setTemplateType(2);
|
||||
String categoryName = this.dictDataApi.getDictDataLabel("llm_prompt_template_framework",Integer.valueOf(createReqVO.getTemplateFramework()));
|
||||
switch (categoryName) {
|
||||
case "简介框架":
|
||||
BriefFramework briefFramework = createReqVO.getBriefFramework();
|
||||
promptTemplates.setTemplateText(JSON.toJSONString(briefFramework));
|
||||
break;
|
||||
case "角色能力框架":
|
||||
AbilityFramework abilityFramework = createReqVO.getAbilityFramework();
|
||||
promptTemplates.setTemplateText(JSON.toJSONString(abilityFramework));
|
||||
break;
|
||||
case "样例提示框架":
|
||||
SampleFramework sampleFramework = createReqVO.getSampleFramework();
|
||||
promptTemplates.setTemplateText(JSON.toJSONString(sampleFramework));
|
||||
break;
|
||||
default:
|
||||
break;
|
||||
}
|
||||
promptTemplatesMapper.insert(promptTemplates);
|
||||
List<String> tagIds = createReqVO.getTagIds();
|
||||
if (tagIds != null && !tagIds.isEmpty()) {
|
||||
@ -237,24 +235,42 @@ public class PromptTemplatesServiceImpl implements PromptTemplatesService {
|
||||
|
||||
String templateText = templatesDO.getTemplateText();
|
||||
|
||||
String append = "";
|
||||
String categoryName = this.dictDataApi.getDictDataLabel("llm_prompt_template_framework",Integer.valueOf(templatesDO
|
||||
.getTemplateFramework()));
|
||||
switch (categoryName) {
|
||||
case "简介框架":
|
||||
BriefFramework briefFramework = JSONObject.parseObject(templateText, BriefFramework.class);
|
||||
promptTemplates.setBriefFramework(briefFramework);
|
||||
append = "指令:" + briefFramework.getOrderInfo() + "/n" +
|
||||
"背景信息" + briefFramework.getBgInfo() + "/n" +
|
||||
"补充数据" + briefFramework.getPlusInfo() + "/n" +
|
||||
"输出格式" + briefFramework.getOutputFormat() + "/n";
|
||||
break;
|
||||
case "角色能力框架":
|
||||
AbilityFramework abilityFramework = JSONObject.parseObject(templateText, AbilityFramework.class);
|
||||
promptTemplates.setAbilityFramework(abilityFramework);
|
||||
append = "指令:" + abilityFramework.getOrderInfo() + "/n" +
|
||||
"背景信息" + abilityFramework.getBgInfo() + "/n" +
|
||||
"能力与角色" + abilityFramework.getAbilityInfo() + "/n" +
|
||||
"输出格式" + abilityFramework.getOutputFormat() + "/n";
|
||||
break;
|
||||
case "样例提示框架":
|
||||
SampleFramework sampleFramework = JSONObject.parseObject(templateText, SampleFramework.class);
|
||||
promptTemplates.setSampleFramework(sampleFramework);
|
||||
append = "背景信息" + sampleFramework.getBgInfo() + "/n" +
|
||||
"样例列表:" + "/n";
|
||||
List<SampleFrameworkItem> itemList = sampleFramework.getItemList();
|
||||
if(itemList != null && itemList.size() > 0){
|
||||
for (SampleFrameworkItem sampleFrameworkItem : itemList) {
|
||||
append += "输入:" + sampleFrameworkItem.getInput() + "," + "输出:" + sampleFrameworkItem.getOutput() + "/n";
|
||||
}
|
||||
}
|
||||
break;
|
||||
default:
|
||||
break;
|
||||
}
|
||||
promptTemplates.setTemplateText(append);
|
||||
|
||||
// 标签
|
||||
getPromptTemplatesLabelResult(id, promptTemplates);
|
||||
@ -395,6 +411,42 @@ public class PromptTemplatesServiceImpl implements PromptTemplatesService {
|
||||
} catch (Exception e) {
|
||||
respVO.setIsBackup(false);
|
||||
}
|
||||
|
||||
String append = "";
|
||||
String categoryName = this.dictDataApi.getDictDataLabel("llm_prompt_template_framework",Integer.valueOf(respVO
|
||||
.getTemplateFramework()));
|
||||
switch (categoryName) {
|
||||
case "简介框架":
|
||||
BriefFramework briefFramework = JSONObject.parseObject(respVO.getTemplateText(), BriefFramework.class);
|
||||
respVO.setBriefFramework(briefFramework);
|
||||
append = "指令:" + briefFramework.getOrderInfo() + "/n" +
|
||||
"背景信息" + briefFramework.getBgInfo() + "/n" +
|
||||
"补充数据" + briefFramework.getPlusInfo() + "/n" +
|
||||
"输出格式" + briefFramework.getOutputFormat() + "/n";
|
||||
break;
|
||||
case "角色能力框架":
|
||||
AbilityFramework abilityFramework = JSONObject.parseObject(respVO.getTemplateText(), AbilityFramework.class);
|
||||
respVO.setAbilityFramework(abilityFramework);
|
||||
append = "指令:" + abilityFramework.getOrderInfo() + "/n" +
|
||||
"背景信息" + abilityFramework.getBgInfo() + "/n" +
|
||||
"能力与角色" + abilityFramework.getAbilityInfo() + "/n" +
|
||||
"输出格式" + abilityFramework.getOutputFormat() + "/n";
|
||||
break;
|
||||
case "样例提示框架":
|
||||
SampleFramework sampleFramework = JSONObject.parseObject(respVO.getTemplateText(), SampleFramework.class);
|
||||
respVO.setSampleFramework(sampleFramework);
|
||||
append = "背景信息" + sampleFramework.getBgInfo() + "/n" +
|
||||
"样例列表:" + "/n";
|
||||
List<SampleFrameworkItem> itemList = sampleFramework.getItemList();
|
||||
if(itemList != null && itemList.size() > 0){
|
||||
for (SampleFrameworkItem sampleFrameworkItem : itemList) {
|
||||
append += "输入:" + sampleFrameworkItem.getInput() + "," + "输出:" + sampleFrameworkItem.getOutput() + "/n";
|
||||
}
|
||||
}
|
||||
break;
|
||||
default:
|
||||
break;
|
||||
}
|
||||
}
|
||||
return pageRespList;
|
||||
}
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user