diff --git a/src/main/java/com/supervision/chat/UploadParamEnum.java b/src/main/java/com/supervision/chat/UploadParamEnum.java new file mode 100644 index 0000000..cf6856e --- /dev/null +++ b/src/main/java/com/supervision/chat/UploadParamEnum.java @@ -0,0 +1,37 @@ +package com.supervision.chat; + +import org.springframework.web.bind.annotation.RequestPart; + +public enum UploadParamEnum { + to_vector_store(null, true, null), + override(null, false, null), + not_refresh_vs_cache(null, false, null), + chunk_size(null, null, 250), + chunk_overlap(null, null, 50), + zh_title_enhance(null, false, null), + docs("{\"test.txt\":[{\"page_content\":\"custom doc\",\"metadata\":{},\"type\":\"Document\"}]}", null, null); + + private final String strValue; + + private final Boolean booleanValue; + + private final Integer intValue; + + UploadParamEnum(String strValue, Boolean booleanValue, Integer intValue) { + this.strValue = strValue; + this.booleanValue = booleanValue; + this.intValue = intValue; + } + + public String getStrValue() { + return strValue; + } + + public Boolean getBooleanValue() { + return booleanValue; + } + + public Integer getIntValue() { + return intValue; + } +} diff --git a/src/main/java/com/supervision/chat/client/LangChainChatService.java b/src/main/java/com/supervision/chat/client/LangChainChatService.java index 0f0f8fc..bb3ce29 100644 --- a/src/main/java/com/supervision/chat/client/LangChainChatService.java +++ b/src/main/java/com/supervision/chat/client/LangChainChatService.java @@ -41,12 +41,12 @@ public interface LangChainChatService { LangChainChatRes uploadFile(@RequestPart String knowledge_base_name, @RequestPart MultipartFile files, @RequestPart String text_splitter_type, - @RequestPart String to_vector_store, - @RequestPart String override, - @RequestPart String not_refresh_vs_cache, + @RequestPart Boolean to_vector_store, + @RequestPart Boolean override, + @RequestPart Boolean not_refresh_vs_cache, @RequestPart Integer chunk_size, @RequestPart Integer chunk_overlap, - @RequestPart String zh_title_enhance, + @RequestPart Boolean zh_title_enhance, @RequestPart String docs); /** diff --git a/src/main/java/com/supervision/chat/client/LangChainChatServiceImpl.java b/src/main/java/com/supervision/chat/client/LangChainChatServiceImpl.java deleted file mode 100644 index 17885d4..0000000 --- a/src/main/java/com/supervision/chat/client/LangChainChatServiceImpl.java +++ /dev/null @@ -1,32 +0,0 @@ -package com.supervision.chat.client; - -import cn.hutool.http.HttpUtil; -import cn.hutool.json.JSONUtil; -import com.supervision.chat.client.dto.CreateBaseDTO; -import com.supervision.chat.client.dto.LangChainChatRes; -import org.apache.hc.client5.http.classic.HttpClient; -import org.apache.hc.client5.http.classic.methods.HttpPost; -import org.apache.hc.core5.http.HttpEntity; -import org.apache.hc.core5.http.HttpResponse; -import org.apache.hc.core5.http.io.entity.EntityUtils; -import org.apache.hc.core5.http.io.entity.StringEntity; -import org.springframework.beans.factory.annotation.Value; -import org.springframework.stereotype.Service; - -import java.io.IOException; - -@Service -public class LangChainChatServiceImpl{ - - @Value("${langChain-chat.url}") - private String LangChainChatClientUrl; - - - public LangChainChatRes chat(CreateBaseDTO createBaseDTO) { - String post = HttpUtil.post(LangChainChatClientUrl + "create_knowledge_base", JSONUtil.toJsonStr(createBaseDTO)); - if (JSONUtil.isTypeJSON(post)) { - return JSONUtil.toBean(post, LangChainChatRes.class); - } - return null; - } -} 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 a85c9cf..ee33bb2 100644 --- a/src/main/java/com/supervision/police/service/impl/NoteRecordSplitServiceImpl.java +++ b/src/main/java/com/supervision/police/service/impl/NoteRecordSplitServiceImpl.java @@ -6,6 +6,7 @@ import cn.hutool.core.util.StrUtil; import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper; import com.baomidou.mybatisplus.core.toolkit.Wrappers; import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl; +import com.supervision.chat.UploadParamEnum; import com.supervision.chat.client.CustomMultipartFile; import com.supervision.chat.client.LangChainChatService; import com.supervision.chat.client.dto.DeleteFileDTO; @@ -116,18 +117,18 @@ public class NoteRecordSplitServiceImpl extends ServiceImpl