提示词配置代码优化bugfix

src/main/java/com/supervision/police/domain/TripleInfo.java

@@ -45,6 +45,11 @@ public class TripleInfo implements Serializable {
      */
     private String recordSplitId;
 
+    /**
+     * 提交给大模型的提示词
+     */
+    private String submitPrompt;
+
     /**
      * 是否生成图谱
      */

src/main/java/com/supervision/police/service/impl/ModelRecordTypeServiceImpl.java

@@ -60,13 +60,13 @@ public class ModelRecordTypeServiceImpl extends ServiceImpl<ModelRecordTypeMappe
     @Override
     public List<TypeDTO> queryTypeListChoose() {
         List<ModelRecordType> list = modelRecordTypeMapper.selectByName(null);
-        return list.stream().map(e -> new TypeDTO(e.getId(),e.getRecordType())).collect(Collectors.toList());
+        return list.stream().map(e -> new TypeDTO(e.getId(), e.getRecordType())).collect(Collectors.toList());
     }
 
     @Override
     public List<ModelRecordType> queryType(String name, Integer page, Integer size) {
         List<ModelRecordType> list = modelRecordTypeMapper.selectByName(name);
-        Map<String, TypeDTO> modelRecordTypeMap = list.stream().map(e -> new TypeDTO(e.getId(),e.getRecordType())).collect(Collectors.toMap(TypeDTO::getId, Function.identity()));
+        Map<String, TypeDTO> modelRecordTypeMap = list.stream().map(e -> new TypeDTO(e.getId(), e.getRecordType())).collect(Collectors.toMap(TypeDTO::getId, Function.identity()));
         // 获取类型和三元组提示词的关系表
         List<NotePromptTypeRel> relList = notePromptTypeRelService.lambdaQuery().list();
         Map<String, List<NotePromptTypeRel>> typeIdMap = relList.stream().collect(Collectors.groupingBy(NotePromptTypeRel::getTypeId));
@@ -152,6 +152,8 @@ public class ModelRecordTypeServiceImpl extends ServiceImpl<ModelRecordTypeMappe
         int i = 0;
         boolean save;
         if (StringUtils.isEmpty(prompt.getId())) {
+            // 新增的时候,校验是否已经存在相同的三元组关系,如果已经存在了相同的三元组关系,不允许添加
+            checkHasSameTriple(prompt.getStartEntityType(), prompt.getRelType(),prompt.getEndEntityType(),null);
             save = notePromptService.save(prompt);
             // 新增prompt绑定的分类信息
             for (String typeId : typeList) {
@@ -161,6 +163,7 @@ public class ModelRecordTypeServiceImpl extends ServiceImpl<ModelRecordTypeMappe
                 notePromptTypeRelService.save(rel);
             }
         } else {
+            checkHasSameTriple(prompt.getStartEntityType(), prompt.getRelType(),prompt.getEndEntityType(),prompt.getId());
             save = notePromptService.updateById(prompt);
             // 更新prompt绑定的分类信息
             // 首先查询已经有的,如果都存在,就不变,如果数据库有,前端没有,就删除,如果前端有,数据库没有,就新增
@@ -175,13 +178,13 @@ public class ModelRecordTypeServiceImpl extends ServiceImpl<ModelRecordTypeMappe
                     notePromptTypeRelService.lambdaUpdate().in(NotePromptTypeRel::getTypeId, deleteIdList).eq(NotePromptTypeRel::getPromptId, prompt.getId()).remove();
                 }
                 // 新增(前端有数据库没有的)
-                frontRelIdList.stream().filter(id -> !existTypeList.contains(id)).forEach(e ->{
+                frontRelIdList.stream().filter(id -> !existTypeList.contains(id)).forEach(e -> {
                     NotePromptTypeRel rel = new NotePromptTypeRel();
                     rel.setPromptId(prompt.getId());
                     rel.setTypeId(e);
                     notePromptTypeRelService.save(rel);
                 });
-            }else {
+            } else {
                 // 如果数据库里面没查到,直接新增,一般不会走这一步
                 for (String typeId : typeList) {
                     NotePromptTypeRel rel = new NotePromptTypeRel();
@@ -218,6 +221,22 @@ public class ModelRecordTypeServiceImpl extends ServiceImpl<ModelRecordTypeMappe
         }
     }
 
+    private void checkHasSameTriple(String startEntityType, String relType, String endEntityType,String promptId) {
+        List<NotePrompt> list = notePromptService.lambdaQuery().eq(NotePrompt::getStartEntityType, startEntityType)
+                .eq(NotePrompt::getRelType, relType).eq(NotePrompt::getEndEntityType, endEntityType).list();
+        if (CollUtil.isNotEmpty(list) ) {
+            if (StrUtil.isBlank(promptId)){
+                throw new RuntimeException("该三元组关系已经存在,请勿重复添加");
+            }else {
+                // 校验list查出来的是不是和promptId相等,如果不想等,也报错
+                if (!list.get(0).getId().equals(promptId)){
+                    throw new RuntimeException("该三元组关系已经存在,请勿重复添加");
+                }
+            }
+
+        }
+    }
+
     @Override
     @Transactional(transactionManager = "dataSourceTransactionManager", rollbackFor = Exception.class)
     public R<?> delPrompt(NotePrompt prompt) {

src/main/java/com/supervision/thread/RecordSplitTypeThread.java

@@ -87,7 +87,6 @@ public class RecordSplitTypeThread implements Callable<Boolean> {
             ---
             返回格式为json,字段名要严格一致:{"result":[{"type":"分类1","explain":"分类原因"},{"type":"分类2","explain":"分类原因"}]}
             """;
-
     private static final String TYPE_CONTEXT_TEMPLATE = "{分类type:{type},区别点(分类释义):{typeExt}}";
 
     @Override

src/main/java/com/supervision/thread/TripleExtractThread.java

@@ -52,25 +52,27 @@ public class TripleExtractThread implements Callable<TripleInfo> {
 
 
     /**
-     * 三元组提取任务:从给定对话中根据给定实体类型和关系提取对应关系的三元组。
+     * 信息抽取任务:从给定对话中根据给定实体类型和关系提取对应关系的三元组。
+     * 在提取三元组时，请务必严格遵循以下要求：
+     * 1. 精准理解需要分析的文本内容，确保提取的信息准确无误、合理恰当。
+     * 2. 只提取给定的实体类型和关系，不要提取给定关系和实体之外的三元组。
+     * 3. 尽量遵循常见的语义和逻辑规则，杜绝过度解读或不合理的关系推断。
+     * 4. 不要提取例子中的实体和关系，提取的结果一定来自需要分析的文本内容!!
+     * 5. 提取之后,再检查一遍,提取的关系和实体是否与给定关系和实体类型对应
+     *
      * 给定的头实体类型为"{headEntityType}";给定的尾实体类型为"{tailEntityType}",给定的关系为"{relation}"。
      * 请仔细分析以下的文本内容，精准找出符合给定关系且头尾实体类型相符的三元组，并进行提取。如果没有识别给定的三元组关系，请返回json:{"result":[]}。
      * ---
-     * 为您提供一个示例供学习:
+     * example:
-     * 给定三元组类型为:头实体类型:"行为人"关系:"伪造",尾实体类型:"合同"
+     * 需要分析提取的文本内容如下：
-     * 办案警官问:描述一下事情的经过。 行为人小明答:我做了一份假的购房合同。
+     * 办案警官问:描述一下事情的经过。 行为人小明答:我用合同款给我老婆买了法国的高档化妆品。
-     * 本示例中应提取给定关系为"伪造"的三元组,则最终应提取的三元组为{"result":[{"headEntity": {"type": "行为人","name":"小明"},"relation": "伪造","tailEntity": {"type": "合同","name": "假的购房合同"}}]}。
+     * 返回结果：{"result":[{"headEntity": {"type": "{headEntityType}","name":"小明"},"relation": "{relation}","tailEntity": {"type": "{tailEntityType}","name": "法国的高档化妆品"}}]}。
      * ---
-     * 需要分析提取的QA对如下：
+     * 需要分析提取文本内容如下：
      * {question}
      * {answer}
      * ---
-     * 在提取三元组时，请务必严格遵循以下要求：
+     *
-     * 1. 精准理解需要分析的QA文本的含义，确保提取的信息准确无误、合理恰当。
-     * 2. 只提取给定的实体类型和关系，不要提取给定关系和实体之外的三元组。
-     * 3. 尽量遵循常见的语义和逻辑规则，杜绝过度解读或不合理的关系推断。
-     * 4. 例子仅供参考,不要简单地返回示例中的结果。
-     * 5. 提取之后,再检查一遍,提取的关系和实体是否与给定关系和实体类型对应
      * 返回格式为必须为以下的json格式:
      * {"result":[{"headEntity": {"type": "{headEntityType}","name":"提取到的头实体内容1"},"relation": "{relation}","tailEntity": {"type": "{tailEntityType}","name": "提取到的尾实体内容1"}}]}
      */
@@ -87,19 +89,20 @@ public class TripleExtractThread implements Callable<TripleInfo> {
             paramMap.put("question", question);
             paramMap.put("answer", answer);
             String format = StrUtil.format(prompt.getPrompt(), paramMap);
-            // 对format进行切分,把前面两个---作为systemPrompt
+//            // 对format进行切分,把前面两个---作为systemPrompt
-            String[] split = format.split("---");
+//            String[] split = format.split("---");
-            SystemMessage systemMessage = new SystemMessage(split[0]);
+//            SystemMessage systemMessage = new SystemMessage(split[0]);
-            UserMessage userMessage = new UserMessage(split[1]);
+//            UserMessage userMessage = new UserMessage(split[1]);
-            Prompt ask = new Prompt(List.of(systemMessage, userMessage));
+//            Prompt ask = new Prompt(List.of(systemMessage, userMessage));
-            ChatResponse call = chatClient.call(ask);
+//            ChatResponse call = chatClient.call(ask);
+            ChatResponse call = chatClient.call(new Prompt(new UserMessage(format)));
             stopWatch.stop();
             String content = call.getResult().getOutput().getContent();
             log.info("耗时:{},分析的结果是:{}", stopWatch.getTotalTimeSeconds(), content);
             // 获取从提示词中提取到的三元组信息
             TripleExtractResult extractResult = JSONUtil.toBean(content, TripleExtractResult.class);
             if (ObjectUtil.isEmpty(extractResult) || CollUtil.isEmpty(extractResult.getResult())) {
-                log.info("提取三元组信息为空,忽略.提取的内容为:{}",content);
+                log.info("提取三元组信息为空,忽略.提取的内容为:{}", content);
                 return null;
             }
             for (TripleExtractNode tripleExtractNode : extractResult.getResult()) {
@@ -118,6 +121,7 @@ public class TripleExtractThread implements Callable<TripleInfo> {
                 tripleInfo.setCaseId(caseId);
                 tripleInfo.setRecordId(recordId);
                 tripleInfo.setRecordSplitId(recordSplitId);
+                tripleInfo.setSubmitPrompt(format);
                 tripleInfo.setStartNodeType(prompt.getStartEntityType());
                 tripleInfo.setEndNodeType(prompt.getEndEntityType());
                 return tripleInfo;