From 266c704e37110366d0f7197859b79638ded6f598 Mon Sep 17 00:00:00 2001
From: liu <liujiatong112@163.com>
Date: Tue, 6 Aug 2024 17:44:15 +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

---
 .../chat/client/LangChainChatService.java      | 18 +++++++++++++++++-
 .../service/impl/ModelCaseServiceImpl.java     |  5 +++++
 2 files changed, 22 insertions(+), 1 deletion(-)

diff --git a/src/main/java/com/supervision/chat/client/LangChainChatService.java b/src/main/java/com/supervision/chat/client/LangChainChatService.java
index cafa4c3..d7e56bf 100644
--- a/src/main/java/com/supervision/chat/client/LangChainChatService.java
+++ b/src/main/java/com/supervision/chat/client/LangChainChatService.java
@@ -2,16 +2,32 @@ package com.supervision.chat.client;
 
 import com.supervision.chat.client.dto.CreateBaseDTO;
 import com.supervision.chat.client.dto.LangChainChatRes;
+import org.springframework.core.io.Resource;
+import org.springframework.http.MediaType;
+import org.springframework.web.bind.annotation.PostMapping;
 import org.springframework.web.bind.annotation.RequestBody;
+import org.springframework.web.bind.annotation.RequestPart;
+import org.springframework.web.multipart.MultipartFile;
 import org.springframework.web.service.annotation.HttpExchange;
 import org.springframework.web.service.annotation.PostExchange;
 
 @HttpExchange
 public interface LangChainChatService {
 
-    @PostExchange(url = "create_knowledge_base", contentType = "application/json")
+    @PostExchange(url = "create_knowledge_base", contentType = MediaType.APPLICATION_JSON_VALUE)
     LangChainChatRes chat(@RequestBody CreateBaseDTO createBaseDTO);
 
+    @PostExchange(url = "upload_docs", contentType = MediaType.MULTIPART_FORM_DATA_VALUE)
+    void uploadFile(@RequestPart String knowledge_base_name,
+                    @RequestPart Resource files,
+                    @RequestPart String to_vector_store,
+                    @RequestPart String override,
+                    @RequestPart String not_refresh_vs_cache,
+                    @RequestPart Integer chunk_size,
+                    @RequestPart Integer chunk_overlap,
+                    @RequestPart String zh_title_enhance,
+                    @RequestPart String text_splitter_type,
+                    @RequestPart String docs);
 
 
 }
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 84af73d..d8854ee 100644
--- a/src/main/java/com/supervision/police/service/impl/ModelCaseServiceImpl.java
+++ b/src/main/java/com/supervision/police/service/impl/ModelCaseServiceImpl.java
@@ -293,6 +293,11 @@ public class ModelCaseServiceImpl extends ServiceImpl<ModelCaseMapper, ModelCase
                     atomic.setRecord("无");
                 }
             }
+            record.setRecord("无");
+            // 遍历,atomic的record字段存在不为空,且不是无的,如果有,就是有
+            if (atomics.stream().anyMatch(atomic -> StrUtil.isNotBlank(atomic.getRecord()) && !StrUtil.equals("无", atomic.getRecord()))) {
+                record.setRecord("有");
+            }
             record.setChildren(atomics);
         }
         iPage.setRecords(records);