优化笔录拆分逻辑

src/main/java/com/supervision/chat/client/LangChainChatService.java

@@ -11,6 +11,8 @@ import org.springframework.web.service.annotation.GetExchange;
 import org.springframework.web.service.annotation.HttpExchange;
 import org.springframework.web.service.annotation.PostExchange;
 
+import java.util.List;
+
 @HttpExchange
 public interface LangChainChatService {
 
@@ -20,7 +22,7 @@ public interface LangChainChatService {
      * @return 结果
      */
     @PostExchange(url = "create_knowledge_base", contentType = MediaType.APPLICATION_JSON_VALUE)
-    LangChainChatRes createBase(@RequestBody CreateBaseDTO createBaseDTO);
+    LangChainChatRes<Object> createBase(@RequestBody CreateBaseDTO createBaseDTO);
 
     /**
      * 上传文档接口
@@ -37,7 +39,7 @@ public interface LangChainChatService {
      * @return 调用的结果
      */
     @PostExchange(url = "upload_docs", contentType = MediaType.MULTIPART_FORM_DATA_VALUE)
-    LangChainChatRes uploadFile(@RequestPart String knowledge_base_name,
+    LangChainChatRes<Object> uploadFile(@RequestPart String knowledge_base_name,
                                 @RequestPart MultipartFile files,
                                 @RequestPart String text_splitter_type,
                                 @RequestPart Boolean to_vector_store,
@@ -54,13 +56,13 @@ public interface LangChainChatService {
      * @return 返回结果
      */
     @PostExchange(url = "delete_docs", contentType = MediaType.APPLICATION_JSON_VALUE)
-    LangChainChatRes deleteFile(@RequestBody DeleteFileDTO deleteFileDTO);
+    LangChainChatRes<Object> deleteFile(@RequestBody DeleteFileDTO deleteFileDTO);
 
     @GetExchange(url = "list_files")
-    LangChainChatRes queryFileList(@RequestParam String knowledge_base_name);
+    LangChainChatRes<List<String>> queryFileList(@RequestParam String knowledge_base_name);
 
     @PostExchange(url = "/delete_knowledge_base", contentType = MediaType.APPLICATION_JSON_VALUE)
-    LangChainChatRes deleteBase(@RequestBody String knowledge_base_name);
+    LangChainChatRes<Object> deleteBase(@RequestBody String knowledge_base_name);
 
 
 }

src/main/java/com/supervision/chat/client/dto/LangChainChatRes.java

@@ -3,11 +3,11 @@ package com.supervision.chat.client.dto;
 import lombok.Data;
 
 @Data
-public class LangChainChatRes {
+public class LangChainChatRes <T> {
 
     private Integer code;
 
     private String msg;
 
-    private Object data;
+    private T data;
 }

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

@@ -51,11 +51,11 @@ public class NoteRecordServiceImpl extends ServiceImpl<NoteRecordMapper, NoteRec
             return;
         }
         // 查询知识库中已经存在的文件
-        LangChainChatRes queryFileListRes = langChainChatService.queryFileList(modelCase.getCaseNo());
+        LangChainChatRes<List<String>> queryFileListRes = langChainChatService.queryFileList(modelCase.getCaseNo());
         if (!queryFileListRes.getCode().equals(200)) {
             log.info("获取知识库已存在的文件失败");
         }
-        HashSet<String> knowledgeFileNameSet = new HashSet<>(JSONUtil.toList((String) queryFileListRes.getData(), String.class));
+        HashSet<String> knowledgeFileNameSet = new HashSet<>(queryFileListRes.getData());
         Set<String> databaseFileNameSet = new HashSet<>();
         for (NoteRecord record : recordList) {
             for (String fileId : record.getFileIds().split(",")) {