From 34dfc0bfc7280bfa983de89e0b6f31d1da45bdc6 Mon Sep 17 00:00:00 2001
From: liu <liujiatong112@163.com>
Date: Fri, 19 Jul 2024 09:32:13 +0800
Subject: [PATCH] =?UTF-8?q?=E5=8E=9F=E5=AD=90=E6=8C=87=E6=A0=87=E5=88=A4?=
 =?UTF-8?q?=E6=96=AD=E4=BB=A3=E7=A0=81=E6=8F=90=E4=BA=A4?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../service/impl/ModelRecordTypeServiceImpl.java       | 10 ++++++++--
 .../police/service/impl/ModelServiceImpl.java          |  2 +-
 2 files changed, 9 insertions(+), 3 deletions(-)

diff --git a/src/main/java/com/supervision/police/service/impl/ModelRecordTypeServiceImpl.java b/src/main/java/com/supervision/police/service/impl/ModelRecordTypeServiceImpl.java
index 14f4f1e..1a4ca83 100644
--- a/src/main/java/com/supervision/police/service/impl/ModelRecordTypeServiceImpl.java
+++ b/src/main/java/com/supervision/police/service/impl/ModelRecordTypeServiceImpl.java
@@ -127,10 +127,14 @@ public class ModelRecordTypeServiceImpl extends ServiceImpl<ModelRecordTypeMappe
     }
 
     private List<TripleInfo> extractTripleInfo(String caseId, String name, String recordId) {
-        List<NoteRecordSplit> records = noteRecordSplitMapper.selectRecord(caseId, name, recordId);
+        // 首先获取所有切分后的笔录
+        List<NoteRecordSplit> recordSplitList= noteRecordSplitMapper.selectRecord(caseId, name, recordId);
         List<TripleInfo> tripleInfos = new ArrayList<>();
-        for (NoteRecordSplit record : records) {
+        // 对切分后的笔录进行遍历
+        for (NoteRecordSplit record : recordSplitList) {
+            // 根据笔录类型找到所有的提取三元组的提示词
             List<NotePrompt> prompts = notePromptMapper.queryPrompt(record.getRecordTypeId());
+            // 遍历提示词进行提取
             for (NotePrompt prompt : prompts) {
                 if (StringUtils.isEmpty(prompt.getPrompt())) {
                     continue;
@@ -146,6 +150,7 @@ public class ModelRecordTypeServiceImpl extends ServiceImpl<ModelRecordTypeMappe
                     log.info("耗时:{}", stopWatch.getTotalTimeSeconds());
                     String content = call.getResult().getOutput().getContent();
                     log.info("分析的结果是:{}", content);
+                    // 获取从提示词中提取到的三元组信息
                     JSONObject jsonObject = new JSONObject(content);
                     JSONArray threeInfo = jsonObject.getJSONArray("result");
                     for (int i = 0; i < threeInfo.length(); i++) {
@@ -159,6 +164,7 @@ public class ModelRecordTypeServiceImpl extends ServiceImpl<ModelRecordTypeMappe
                         if (StrUtil.hasEmpty(startNodeType, entity, endNodeType, property, value)) {
                             continue;
                         }
+                        // 将三元组信息进行保存操作
                         TripleInfo tripleInfo = new TripleInfo(entity, property, value, record.getId(), LocalDateTime.now(), startNodeType, endNodeType);
                         tripleInfoMapper.insert(tripleInfo);
                         tripleInfos.add(tripleInfo);
diff --git a/src/main/java/com/supervision/police/service/impl/ModelServiceImpl.java b/src/main/java/com/supervision/police/service/impl/ModelServiceImpl.java
index 42bb802..0ffb492 100644
--- a/src/main/java/com/supervision/police/service/impl/ModelServiceImpl.java
+++ b/src/main/java/com/supervision/police/service/impl/ModelServiceImpl.java
@@ -67,7 +67,7 @@ public class ModelServiceImpl implements ModelService {
             //原子指标结果表
             try {
                 // index_source==1
-                List<ModelAtomicIndex> list = analyseCaseDTO.getAtomicIndexList();
+                // List<ModelAtomicIndex> list = analyseCaseDTO.getAtomicIndexList();
                 //index_source==3
 
                 //查询图谱  index_source: 1人工定义  2数据库查询	3图谱生成	  4大模型