diff --git a/src/main/java/com/supervision/chat/controller/TestController.java b/src/main/java/com/supervision/chat/controller/TestController.java index 4b61716..c1b594f 100644 --- a/src/main/java/com/supervision/chat/controller/TestController.java +++ b/src/main/java/com/supervision/chat/controller/TestController.java @@ -1,6 +1,7 @@ package com.supervision.chat.controller; import cn.hutool.json.JSONUtil; +import com.supervision.chat.UploadParamEnum; import com.supervision.chat.client.CustomMultipartFile; import com.supervision.chat.client.LangChainChatService; import com.supervision.chat.client.dto.CreateBaseDTO; @@ -35,13 +36,13 @@ public class TestController { LangChainChatRes langChainChatRes = langChainChatClient.uploadFile("11111111", mockMultipartFile, "问讯笔录", - "true", - "false", - "false", - 250, - 50, - "false", - "{\"test.txt\":[{\"page_content\":\"custom doc\",\"metadata\":{},\"type\":\"Document\"}]}"); + UploadParamEnum.to_vector_store.getBooleanValue(), + UploadParamEnum.override.getBooleanValue(), + UploadParamEnum.not_refresh_vs_cache.getBooleanValue(), + UploadParamEnum.chunk_size.getIntValue(), + UploadParamEnum.chunk_overlap.getIntValue(), + UploadParamEnum.zh_title_enhance.getBooleanValue(), + UploadParamEnum.docs.getStrValue()); log.info(JSONUtil.toJsonStr(langChainChatRes)); }