From bc1efcba5c8efd07f62c832d1f38c9c7ad88e57a Mon Sep 17 00:00:00 2001
From: liu <liujiatong112@163.com>
Date: Thu, 8 Aug 2024 16:35:23 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BC=98=E5=8C=96=E7=AC=94=E5=BD=95=E6=8B=86?=
 =?UTF-8?q?=E5=88=86=E9=80=BB=E8=BE=91?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../chat/client/LangChainChatService.java            | 12 +++++++-----
 .../chat/client/dto/LangChainChatRes.java            |  4 ++--
 .../police/service/impl/NoteRecordServiceImpl.java   |  4 ++--
 3 files changed, 11 insertions(+), 9 deletions(-)

diff --git a/src/main/java/com/supervision/chat/client/LangChainChatService.java b/src/main/java/com/supervision/chat/client/LangChainChatService.java
index da5b252..ad03c7f 100644
--- a/src/main/java/com/supervision/chat/client/LangChainChatService.java
+++ b/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);
 
 
 }
diff --git a/src/main/java/com/supervision/chat/client/dto/LangChainChatRes.java b/src/main/java/com/supervision/chat/client/dto/LangChainChatRes.java
index 7e9f614..5406e72 100644
--- a/src/main/java/com/supervision/chat/client/dto/LangChainChatRes.java
+++ b/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;
 }
diff --git a/src/main/java/com/supervision/police/service/impl/NoteRecordServiceImpl.java b/src/main/java/com/supervision/police/service/impl/NoteRecordServiceImpl.java
index 3517f48..15242c6 100644
--- a/src/main/java/com/supervision/police/service/impl/NoteRecordServiceImpl.java
+++ b/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(",")) {