From d75c2769026f982ac561f74e739a52ed5f041298 Mon Sep 17 00:00:00 2001
From: zhangxiao <898441624@qq.com>
Date: 星期一, 26 八月 2024 18:03:45 +0800
Subject: [PATCH] fix: 修改bug
---
src/api/kbList.ts | 7 +++
src/components/navbar/index.vue | 16 +++++++
src/views/dmx/knowledgeLib/index.vue | 24 +++++++----
src/views/dmx/knowledgeLib/config.vue | 29 ++++++++++++--
4 files changed, 61 insertions(+), 15 deletions(-)
diff --git a/src/api/kbList.ts b/src/api/kbList.ts
index c6ed8d2..d479996 100644
--- a/src/api/kbList.ts
+++ b/src/api/kbList.ts
@@ -117,6 +117,13 @@
});
}
+// 鑾峰彇鐭ヨ瘑搴撶鐞嗛厤缃祵鍏ユā鍨�
+export function getKnowledgeBaseModel(params) {
+ return axios.get('/api/v1/llm/my_llms', {
+ params,
+ });
+}
+
// 娴嬭瘯鎺ュ彛
export function kbretrievalTest(params) {
return axios.post('/api/v1/chunk/retrieval_test', params);
diff --git a/src/components/navbar/index.vue b/src/components/navbar/index.vue
index ff370a4..fca31ae 100644
--- a/src/components/navbar/index.vue
+++ b/src/components/navbar/index.vue
@@ -185,7 +185,21 @@
</template>
</a-dropdown>
</li>
- <li style="padding: 0"> {{ name }} </li>
+ <li style="padding: 0">
+ <a-dropdown trigger="click" @select="changeLocale as any">
+ <div> {{ name }}</div>
+ <template #content>
+ <a-doption>
+ <a-space @click="handleLogout">
+ <icon-export />
+ <span>
+ {{ $t('messageBox.logout') }}
+ </span>
+ </a-space>
+ </a-doption>
+ </template>
+ </a-dropdown>
+ </li>
</ul>
</div>
</template>
diff --git a/src/views/dmx/knowledgeLib/config.vue b/src/views/dmx/knowledgeLib/config.vue
index 088b890..52ed4e5 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,7 +43,14 @@
>
{{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="瑙ｆ瀽鏂规硶">
@@ -152,7 +159,7 @@
<script lang="ts" setup>
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
});
@@ -221,6 +228,7 @@
let visible = ref(false);
const formRef = ref(null);
const modelList = ref({});
+const mobileData= ref<any>([])
const file = ref();
const uploadAction = '/api/v1/llm/upload'; // 鏇挎崲涓轰綘鐨勪笂浼燗PI
@@ -319,9 +327,20 @@
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 {
diff --git a/src/views/dmx/knowledgeLib/index.vue b/src/views/dmx/knowledgeLib/index.vue
index 226808d..b77ffe6 100644
--- a/src/views/dmx/knowledgeLib/index.vue
+++ b/src/views/dmx/knowledgeLib/index.vue
@@ -34,6 +34,7 @@
<span
style="cursor: pointer; color: #4977ba; font-size: 12px"
@click="deleteKnowledge(tab.id)"
+ v-hasPermi="'/kb/delete'"
><icon-delete style="font-size: 14px" />鍒犻櫎</span
>
</template>
@@ -88,6 +89,7 @@
type="text"
class="button"
style="color: #2a2a2b"
+ v-hasPermi="'/kb/data_status'"
@click="enableDisable(1)"
>
<template #icon>
@@ -101,6 +103,7 @@
type="text"
class="button"
style="color: #2a2a2b"
+ v-hasPermi="'/kb/data_status'"
@click="enableDisable(0)"
>
<template #icon>
@@ -114,6 +117,7 @@
type="text"
class="button"
style="color: #2a2a2b"
+ v-hasPermi="'/kb/data_analysis'"
@click="addBlock(1)"
>
<template #icon>
@@ -127,6 +131,7 @@
type="text"
class="button"
style="color: #2a2a2b"
+ v-hasPermi="'/kb/data_analysis'"
@click="addBlock(2)"
>
<template #icon>
@@ -140,6 +145,7 @@
<a-button
type="text"
class="button"
+ v-hasPermi="'/kb/data_delete'"
style="color: #2a2a2b"
@click="addBlock(3)"
>
@@ -469,7 +475,7 @@
type="text"
:disabled="record.run == '1'"
@click="handleClick(record)"
- v-hasPermi="'/kb/data_update'"
+ 鏌�
size="small"
>
<template #icon>
@@ -513,8 +519,8 @@
</template>
</a-button>
</a-tooltip>
-
- <!-- <a-button
+ <!--
+ <a-button
type="text"
size="small"
:disabled="record.run == '1'"
@@ -599,12 +605,12 @@
<!-- </div>-->
</div>
</div>
-<!-- <a-modal title=" " v-model:visible="visible" :footer="false" fullscreen>-->
-<!-- <!– <docx previewSrc="http://192.168.20.116:1080/v1/document/get/405c3efa4d8c11ef97560242ac120006"></docx>–>-->
-<!-- <docx v-if="documenttype == 'docx'" :previewSrc="previewSrc"></docx>-->
-<!-- <excel v-if="documenttype == 'excel'" :previewSrc="previewSrc"></excel>-->
-<!-- <txtPdf v-if="documenttype == 'txtPdf'" :previewSrc="previewSrc"></txtPdf>-->
-<!-- </a-modal>-->
+ <!-- <a-modal title=" " v-model:visible="visible" :footer="false" fullscreen>-->
+ <!-- <!– <docx previewSrc="http://192.168.20.116:1080/v1/document/get/405c3efa4d8c11ef97560242ac120006"></docx>–>-->
+ <!-- <docx v-if="documenttype == 'docx'" :previewSrc="previewSrc"></docx>-->
+ <!-- <excel v-if="documenttype == 'excel'" :previewSrc="previewSrc"></excel>-->
+ <!-- <txtPdf v-if="documenttype == 'txtPdf'" :previewSrc="previewSrc"></txtPdf>-->
+ <!-- </a-modal>-->
<tool
ref="toolForm"
:kbtenantInfo="kbtenantInfo"
--
Gitblit v1.8.0