From 0e8acc7aa87a68e1ea59d1654c1e3e88fcf078ef Mon Sep 17 00:00:00 2001
From: liu <liujiatong112@163.com>
Date: Thu, 8 Aug 2024 14:03:38 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BB=A3=E7=A0=81=E6=8F=90=E4=BA=A4,=E4=BF=AE?=
 =?UTF-8?q?=E6=94=B9=E4=B8=8A=E4=BC=A0=E7=9F=A5=E8=AF=86=E5=BA=93=E9=80=BB?=
 =?UTF-8?q?=E8=BE=91?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../com/supervision/chat/client/LangChainChatService.java   | 3 +++
 .../com/supervision/chat/controller/TestController.java     | 6 ++++++
 .../police/service/impl/ModelCaseServiceImpl.java           | 5 +++++
 3 files changed, 14 insertions(+)

diff --git a/src/main/java/com/supervision/chat/client/LangChainChatService.java b/src/main/java/com/supervision/chat/client/LangChainChatService.java
index 47849a4..da5b252 100644
--- a/src/main/java/com/supervision/chat/client/LangChainChatService.java
+++ b/src/main/java/com/supervision/chat/client/LangChainChatService.java
@@ -59,5 +59,8 @@ public interface LangChainChatService {
     @GetExchange(url = "list_files")
     LangChainChatRes queryFileList(@RequestParam String knowledge_base_name);
 
+    @PostExchange(url = "/delete_knowledge_base", contentType = MediaType.APPLICATION_JSON_VALUE)
+    LangChainChatRes deleteBase(@RequestBody String knowledge_base_name);
+
 
 }
diff --git a/src/main/java/com/supervision/chat/controller/TestController.java b/src/main/java/com/supervision/chat/controller/TestController.java
index a6210d9..10adc2f 100644
--- a/src/main/java/com/supervision/chat/controller/TestController.java
+++ b/src/main/java/com/supervision/chat/controller/TestController.java
@@ -66,4 +66,10 @@ public class TestController {
         log.info(JSONUtil.toJsonStr(langChainChatRes));
     }
 
+    @GetMapping("deleteBase")
+    public void deleteBase(String knowledgeBaseName) {
+        LangChainChatRes langChainChatRes = langChainChatClient.deleteBase(knowledgeBaseName);
+        log.info(JSONUtil.toJsonStr(langChainChatRes));
+    }
+
 }
diff --git a/src/main/java/com/supervision/police/service/impl/ModelCaseServiceImpl.java b/src/main/java/com/supervision/police/service/impl/ModelCaseServiceImpl.java
index 88099e1..75526fd 100644
--- a/src/main/java/com/supervision/police/service/impl/ModelCaseServiceImpl.java
+++ b/src/main/java/com/supervision/police/service/impl/ModelCaseServiceImpl.java
@@ -175,7 +175,12 @@ public class ModelCaseServiceImpl extends ServiceImpl<ModelCaseMapper, ModelCase
 
     @Override
     public R<?> del(String id) {
+
         ModelCase modelCase = modelCaseMapper.selectById(id);
+        LangChainChatRes langChainChatRes = langChainChatService.deleteBase(modelCase.getCaseNo());
+        if (200 != langChainChatRes.getCode()){
+            log.info("删除知识库失败");
+        }
         modelCase.setDataStatus(StringUtils.getUUID());
         int i = modelCaseMapper.updateById(modelCase);
         if (i > 0) {