From c02bff2911f720d54388f4abb6456ec1777f8a96 Mon Sep 17 00:00:00 2001
From: zhangxiao <898441624@qq.com>
Date: 星期五, 30 八月 2024 10:40:02 +0800
Subject: [PATCH] fix: 修改bug

---
 src/views/dmx/knowledgeLib/config.vue |  113 +++++++++++++++++++++++++++++++++-----------------------
 1 files changed, 67 insertions(+), 46 deletions(-)

diff --git a/src/views/dmx/knowledgeLib/config.vue b/src/views/dmx/knowledgeLib/config.vue
index c4f5387..ebec067 100644
--- a/src/views/dmx/knowledgeLib/config.vue
+++ b/src/views/dmx/knowledgeLib/config.vue
@@ -35,7 +35,7 @@
 <!--          </a-form-item>-->
           <a-form-item field="embd_id" label="宓屽叆妯″瀷">
             <a-select :size="'large'" v-model="form.embd_id" placeholder="璇烽�夋嫨 ..." :disabled="kbObj.chunk_num && kbObj.token_num">
-              <a-optgroup  :label="index" v-for="(item,index) in modelList" :key=index>
+              <!-- <a-optgroup  :label="index" v-for="(item,index) in mobileData" :key=index>
                 <a-option
                   v-for="(obj) in item"
                   :key='obj.fid'
@@ -43,11 +43,18 @@
                 >
                   {{obj.llm_name}}
                 </a-option>
-              </a-optgroup>
+              </a-optgroup> -->
+              <a-option
+                  v-for="(item,index) in mobileData"
+                  :key='item.name'
+                  :value="item.name"
+                >
+                  {{item.name}}
+                </a-option>
             </a-select>
           </a-form-item>
           <a-form-item field="parser_id" label="瑙f瀽鏂规硶">
-            <a-select v-model="form.parser_id" placeholder="璇烽�夋嫨" :disabled="kbObj.chunk_num  && kbObj.token_num " >
+            <a-select v-model="form.parser_id" placeholder="璇烽�夋嫨" :disabled="!!(kbObj.chunk_num  && kbObj.token_num)" >
               <a-option v-for="item in parser_ids" :key="item.value" :label="item.name" :value="item.value"></a-option>
             </a-select>
           </a-form-item>
@@ -150,9 +157,9 @@
 </template>
 
 <script lang="ts" setup>
-import { onMounted, onBeforeMount, reactive, ref } from "vue";
+import { onMounted, onBeforeMount, reactive, ref, nextTick } from "vue";
 import { Message } from "@arco-design/web-vue";
-import { queryModelList,kbUpdate } from "@/api/kbList";
+import { queryModelList,kbUpdate,getKnowledgeBaseModel } from "@/api/kbList";
 import message from "@arco-design/web-vue/es/message";
 import useLoading from "@/hooks/loading";
 const { loading,setLoading } = useLoading(true);
@@ -187,7 +194,7 @@
   description: "",
   language: "English",
   permission: "",
-  embd_id: "BAAI/bge-large-zh-v1.5",
+  embd_id: "",
   parser_id: '',
   layout_recognize: true
 });
@@ -219,8 +226,9 @@
   ],
 }
 let visible = ref(false);
-const formRef = ref(null);
+const formRef = ref();
 const modelList = ref({});
+const mobileData= ref<any>([])
 const file = ref();
 
 const uploadAction = '/api/v1/llm/upload'; // 鏇挎崲涓轰綘鐨勪笂浼燗PI
@@ -314,14 +322,25 @@
   Object.assign(form,{
 
   });
-  formRef.value.resetFields();
+  // formRef.value.resetFields();
 }
 
 const queryModel = async (params) => {
   try {
-    const data = await queryModelList(params);
+    const data = await getKnowledgeBaseModel(params);
+    // const res =await queryModelList(params)
     // console.log(data.data, '澶фā鍨嬪垪琛�');
     modelList.value = data.data
+    for(let key in data.data) {
+      if (data.data.hasOwnProperty(key)) { // 纭繚鏄璞¤嚜鏈夊睘鎬э紝涓嶆槸鍘熷瀷閾句笂鐨勫睘鎬�
+        if(data.data[key].llm.length > 0){
+          mobileData.value= data.data[key].llm.filter(item => item.type == "embedding")
+        }
+      }  
+  
+    }
+
+   
   } catch (err) {
     // you can report use errorHandler or other
   } finally {
@@ -330,43 +349,45 @@
 };
 
 const changekbObj = (value) => {
-  formRef.value.resetFields();
-  kbObj = value;
-  console.log(kbObj, 'kbObj');
-  let configObj = {
-    name: kbObj.name,
-    avatar: kbObj.avatar,
-    pages:  kbObj.parser_config.pages,
-    chunk_token_num:  kbObj.parser_config.chunk_token_num,
-    description: kbObj.description,
-    language: "English",
-    permission: kbObj.permission,
-    embd_id: props.kbtenantInfo.embd_id || kbObj.embd_id,
-    parser_id: kbObj.parser_id,
-    layout_recognize: true,
-  }
-  let raptor = {}
-  if(kbObj.parser_config && kbObj.parser_config.raptor){
-    raptor = kbObj.parser_config.raptor;
-    Object.assign(form,{
-      ...configObj,
-      ...raptor,
-      use_raptor: kbObj.parser_config.raptor.use_raptor,
-    });
-  }else{
-    Object.assign(form,{
-      ...configObj,
-      use_raptor: false,
-    });
-  }
-  imageUrls.value = [kbObj.avatar];
-  console.log(form,'閰嶇疆椤甸潰琛ㄥ崟淇℃伅');
-  avatarShow.value = false;
-  setTimeout(() => {
-    avatarShow.value = true;
-  },100);
-  scrollContainer.value.scrollTop = 0;
-
+  
+   formRef.value.resetFields();
+  nextTick(()=>{
+    kbObj = value;
+    console.log(kbObj, 'kbObj');
+    let configObj = {
+      name: kbObj.name,
+      avatar: kbObj.avatar,
+      pages:  kbObj.parser_config.pages,
+      chunk_token_num:  kbObj.parser_config.chunk_token_num,
+      description: kbObj.description,
+      language: "English",
+      permission: kbObj.permission,
+      embd_id: kbObj.embd_id || props.kbtenantInfo.embd_id,
+      parser_id: kbObj.parser_id,
+      layout_recognize: true,
+    }
+    let raptor = {}
+    if(kbObj.parser_config && kbObj.parser_config.raptor){
+      raptor = kbObj.parser_config.raptor;
+      Object.assign(form,{
+        ...configObj,
+        ...raptor,
+        use_raptor: kbObj.parser_config.raptor.use_raptor,
+      });
+    }else{
+      Object.assign(form,{
+        ...configObj,
+        use_raptor: false,
+      });
+    }
+    imageUrls.value = [kbObj.avatar];
+    console.log(form,'閰嶇疆椤甸潰琛ㄥ崟淇℃伅');
+    avatarShow.value = false;
+    setTimeout(() => {
+      avatarShow.value = true;
+    },100);
+    scrollContainer.value.scrollTop = 0;
+  })
 }
 
 defineExpose({

--
Gitblit v1.8.0