From c3e34118307e1c70260647a7ebd90c99d4407ec4 Mon Sep 17 00:00:00 2001
From: liu <liujiatong112@163.com>
Date: Wed, 7 Aug 2024 15:33:35 +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/controller/TestController.java           | 15 ++++++++-------
 1 file changed, 8 insertions(+), 7 deletions(-)

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));
     }