Browse Source

add docs embedding

tycoding 1 year ago
parent
commit
ebc1ba06d3

+ 16 - 0
langchat-ui/src/api/aigc/embedding.ts

@@ -24,3 +24,19 @@ export function embeddingDocs(
     onUploadProgress,
   });
 }
+
+export function embeddingExcel(
+  knowledgeId: string,
+  data: any,
+  onUploadProgress?: (progressEvent: AxiosProgressEvent) => void
+) {
+  return http.request({
+    url: `/aigc/embedding/excel/${knowledgeId}`,
+    method: 'post',
+    data,
+    headers: {
+      'Content-Type': 'multipart/form-data',
+    },
+    onUploadProgress,
+  });
+}

+ 1 - 11
langchat-ui/src/views/aigc/knowledge/components/ImportFile/components/DocImport.vue

@@ -9,16 +9,7 @@
   const message = useMessage();
   const fileList = ref<any[]>([]);
 
-  const handleImport = ({
-    file,
-    data,
-    headers,
-    withCredentials,
-    action,
-    onFinish,
-    onError,
-    onProgress,
-  }: UploadCustomRequestOptions) => {
+  const handleImport = ({ file, onFinish, onError, onProgress }: UploadCustomRequestOptions) => {
     const kbId = router.currentRoute.value.params.id;
     embeddingDocs(
       String(kbId),
@@ -32,7 +23,6 @@
       }
     )
       .then((res) => {
-        console.log(res);
         fileList.value.push(res);
         message.success('上传成功');
         onFinish();

+ 3 - 13
langchat-ui/src/views/aigc/knowledge/components/ImportFile/components/ExcelImport.vue

@@ -1,7 +1,7 @@
 <script setup lang="ts">
   import { DownloadOutline } from '@vicons/ionicons5';
   import { UploadCustomRequestOptions, useMessage } from 'naive-ui';
-  import { embeddingDocs } from '@/api/aigc/embedding';
+  import { embeddingExcel } from '@/api/aigc/embedding';
   import { useRouter } from 'vue-router';
   import { ref } from 'vue';
 
@@ -9,18 +9,9 @@
   const message = useMessage();
   const fileList = ref<any[]>([]);
 
-  const handleImport = ({
-    file,
-    data,
-    headers,
-    withCredentials,
-    action,
-    onFinish,
-    onError,
-    onProgress,
-  }: UploadCustomRequestOptions) => {
+  const handleImport = ({ file, onFinish, onError, onProgress }: UploadCustomRequestOptions) => {
     const kbId = router.currentRoute.value.params.id;
-    embeddingDocs(
+    embeddingExcel(
       String(kbId),
       {
         file: file.file,
@@ -32,7 +23,6 @@
       }
     )
       .then((res) => {
-        console.log(res);
         fileList.value.push(res);
         message.success('上传成功');
         onFinish();