提示词配置代码优化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

@@ -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绑定的分类信息
             // 首先查询已经有的,如果都存在,就不变,如果数据库有,前端没有,就删除,如果前端有,数据库没有,就新增
@@ -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,12 +89,13 @@ 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);
@@ -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;