From 85261f49959e39f7428097cb389608f87ba43604 Mon Sep 17 00:00:00 2001
From: liu <liujiatong112@163.com>
Date: Thu, 8 Aug 2024 10:09:26 +0800
Subject: [PATCH] =?UTF-8?q?=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

---
 .../supervision/chat/client/dto/CreateBaseDTO.java |  6 ++++++
 .../chat/controller/TestController.java            | 14 ++++++++++++++
 .../service/impl/NoteRecordSplitServiceImpl.java   |  2 +-
 .../service/impl/RecordSplitTypeServiceImpl.java   |  2 +-
 4 files changed, 22 insertions(+), 2 deletions(-)

diff --git a/src/main/java/com/supervision/chat/client/dto/CreateBaseDTO.java b/src/main/java/com/supervision/chat/client/dto/CreateBaseDTO.java
index e9807ec..02a3010 100644
--- a/src/main/java/com/supervision/chat/client/dto/CreateBaseDTO.java
+++ b/src/main/java/com/supervision/chat/client/dto/CreateBaseDTO.java
@@ -10,4 +10,10 @@ public class CreateBaseDTO {
     private String vector_store_type = "faiss";
 
     private String embed_model = "bge-large-zh";
+
+    public static CreateBaseDTO create(String knowledge_base_name){
+        CreateBaseDTO createBaseDTO = new CreateBaseDTO();
+        createBaseDTO.setKnowledge_base_name(knowledge_base_name);
+        return createBaseDTO;
+    }
 }
diff --git a/src/main/java/com/supervision/chat/controller/TestController.java b/src/main/java/com/supervision/chat/controller/TestController.java
index c1b594f..a6210d9 100644
--- a/src/main/java/com/supervision/chat/controller/TestController.java
+++ b/src/main/java/com/supervision/chat/controller/TestController.java
@@ -7,12 +7,15 @@ import com.supervision.chat.client.LangChainChatService;
 import com.supervision.chat.client.dto.CreateBaseDTO;
 import com.supervision.chat.client.dto.DeleteFileDTO;
 import com.supervision.chat.client.dto.LangChainChatRes;
+import com.supervision.police.domain.ModelCase;
+import com.supervision.police.service.ModelCaseService;
 import lombok.RequiredArgsConstructor;
 import lombok.extern.slf4j.Slf4j;
 import org.springframework.web.bind.annotation.*;
 import org.springframework.web.multipart.MultipartFile;
 
 import java.io.IOException;
+import java.util.List;
 
 @Slf4j
 @RequestMapping("chat/test/")
@@ -22,6 +25,8 @@ public class TestController {
 
     private final LangChainChatService langChainChatClient;
 
+    private final ModelCaseService modelCaseService;
+
     @GetMapping("test")
     public void test() {
         CreateBaseDTO createBaseDTO = new CreateBaseDTO();
@@ -30,6 +35,15 @@ public class TestController {
         log.info(JSONUtil.toJsonStr(chat));
     }
 
+    @GetMapping("createExistCase")
+    public void createExistCase(){
+        List<ModelCase> list = modelCaseService.list();
+        for (ModelCase modelCase : list) {
+            LangChainChatRes chat = langChainChatClient.createBase(CreateBaseDTO.create(modelCase.getCaseNo()));
+            log.info(JSONUtil.toJsonStr(chat));
+        }
+    }
+
     @PostMapping("uploadFile")
     public void testUploadFile(@RequestPart("file") MultipartFile file) throws IOException {
         CustomMultipartFile mockMultipartFile = new CustomMultipartFile(file.getOriginalFilename(), file.getInputStream());
diff --git a/src/main/java/com/supervision/police/service/impl/NoteRecordSplitServiceImpl.java b/src/main/java/com/supervision/police/service/impl/NoteRecordSplitServiceImpl.java
index 5579f47..e667602 100644
--- a/src/main/java/com/supervision/police/service/impl/NoteRecordSplitServiceImpl.java
+++ b/src/main/java/com/supervision/police/service/impl/NoteRecordSplitServiceImpl.java
@@ -148,7 +148,7 @@ public class NoteRecordSplitServiceImpl extends ServiceImpl<NoteRecordSplitMappe
                 }
                 // 创建任务之后,再开始
                 boolean taskStatus = saveRecordProcessTask(record.getCaseId(), record.getId());
-                // 如果校验结果为false,则说明需要进行分类以及三元组操作
+                // 如果校验结果为true,则说明需要进行分类以及三元组操作
                 if (taskStatus) {
                     // 对笔录进行分类,并对笔录进行提取三元组
                     recordSplitProcessService.process(allTypeList, splitList);
diff --git a/src/main/java/com/supervision/police/service/impl/RecordSplitTypeServiceImpl.java b/src/main/java/com/supervision/police/service/impl/RecordSplitTypeServiceImpl.java
index ec92b5c..ec95e0a 100644
--- a/src/main/java/com/supervision/police/service/impl/RecordSplitTypeServiceImpl.java
+++ b/src/main/java/com/supervision/police/service/impl/RecordSplitTypeServiceImpl.java
@@ -45,7 +45,7 @@ public class RecordSplitTypeServiceImpl implements RecordSplitTypeService {
             log.error("分类任务线程休眠失败");
         }
         List<Future<String>> futures = new ArrayList<>();
-
+        log.info("开始执行笔录分类任务");
         for (NoteRecordSplit recordSplit : splitList) {
             // 进行分类
             log.info("分类任务提交线程池进行分类");