From 5ff5e947cbd5cc0d0819be8da5bbc18df8965a06 Mon Sep 17 00:00:00 2001 From: liudong <liudong> Date: 星期四, 01 八月 2024 11:06:36 +0800 Subject: [PATCH] 知识库文件的下载 --- src/views/dmx/knowledgeLib/index.vue | 168 +++++++++++++++++++++++++++++++++++++++++-------------- 1 files changed, 125 insertions(+), 43 deletions(-) diff --git a/src/views/dmx/knowledgeLib/index.vue b/src/views/dmx/knowledgeLib/index.vue index 12287ee..7da5b5e 100644 --- a/src/views/dmx/knowledgeLib/index.vue +++ b/src/views/dmx/knowledgeLib/index.vue @@ -26,7 +26,6 @@ <a-popconfirm content="纭畾鍒犻櫎鍚�?" @ok="deleteKnowledge(tab.id)" type="warning"> <span style="cursor: pointer;color: #0960bd;font-size: 12px"><icon-delete />鍒犻櫎</span> </a-popconfirm> - </template> </a-popover> </div> @@ -37,7 +36,7 @@ <a-tabs default-active-key="1"> <a-tab-pane key="1" :title= "t('dmx.list.DataSet')"> <div class="rt-container-main"> - <a-card ref="account" class="general-card" :title="$t('')" style="padding-top: 20px"> + <a-card ref="account" class="general-card" title="" style="padding-top: 20px"> <a-row> <a-col :flex="1" :span="12"> <a-form @@ -51,9 +50,11 @@ <a-input-search :placeholder="$t('cardList.searchInput.placeholder')" style="width: 200px;margin-right: 10px" + v-model="keywords" + @change="search" /> - <!-- 鏂板缓--> - <add /> + <!--鏂板缓--> + <add :kbobj="kbobj" @changeFetchData="changeFetchData"/> </a-col> </a-row> </a-form> @@ -89,7 +90,7 @@ <!-- {{ $t('searchTable.operation.download') }}--> <!-- </a-button>--> <a-tooltip :content="$t('searchTable.actions.refresh')"> - <div class="action-icon" @click="search" + <div class="action-icon" ><icon-refresh size="18" /></div> </a-tooltip> @@ -166,8 +167,11 @@ <template #create_time="{ record }"> <span>{{parseTime(record.create_time)}}</span> </template> +<!-- <template #parser_id="{ record }">--> +<!-- <span>{{ parser_ids[record.parser_id]}}</span>--> +<!-- </template>--> <template #status="{ record }"> - <a-switch v-model="record.status" checked-value="0" unchecked-value="1"/> + <a-switch v-model="record.status" size="small" checked-value="1" unchecked-value="0" @change="handleChangeStatus(record)"/> </template> <template #run="{ record }"> <div style="display: flex;align-items: center;"> @@ -178,13 +182,13 @@ <p>Here is the text content</p> </template> </a-popover> - <a-popover title="Title" v-if="record.run == '2'"> + <a-popover title="Title" v-if="record.run == '1'"> <a-tag :loading="record.loading" :color="'blue'" border>{{$t('dmx.list.complete')}}</a-tag> <template #content> <p>Here is the text content</p> </template> </a-popover> - <a-popover title="Title" v-if="record.run == '1'"> + <a-popover title="Title" v-if="record.run == '2'"> <a-tag :loading="record.loading" :color="'gold'" border>{{$t('dmx.list.cancel')}}</a-tag> <template #content> <p>Here is the text content</p> @@ -210,18 +214,23 @@ </div> </template> <template #action="{ record }" > - <tool/> + <tool :kbtenantInfo="kbtenantInfo" :item="record" :kbdetail="kbdetail" /> <edit :item="record" @upTabdateItem="upTabdateItem"/> - <a-popconfirm :content="t('纭畾鍒犻櫎鍚�')" type="warning" @ok="deleteItem(record)"> + <a-popconfirm :content="'纭畾鍒犻櫎鍚�'" type="warning" @ok="deleteItem(record)"> <a-button type="text" size="small"> <template #icon> <icon-delete /> </template> </a-button> </a-popconfirm> - <a-button type="text" size="small"> + <a-button type="text" size="small" @click="onDownloadDocument(record)"> <template #icon> <icon-download /> + </template> + </a-button> + <a-button type="text" size="small" @click="seeload(record)"> + <template #icon> + <icon-eye /> </template> </a-button> </template> @@ -231,12 +240,12 @@ </a-tab-pane> <a-tab-pane key="2" :title="t('dmx.list.test')"> <div class="rt-container-main"> - <test ></test> + <test ref="testForm"></test> </div> </a-tab-pane> <a-tab-pane key="3" :title="t('dmx.list.disposition')"> <div class="rt-container-main"> - <config ></config> + <config ref="configForm" :kbtenantInfo="kbtenantInfo" :kbobj="kbobj"></config> </div> </a-tab-pane> </a-tabs> @@ -282,7 +291,7 @@ </template> <script lang="ts" setup> -import { computed, ref, reactive, watch, nextTick, onBeforeMount } from "vue"; +import { computed, ref, reactive, watch, nextTick, onBeforeMount, onMounted } from "vue"; import { useI18n } from 'vue-i18n'; import useLoading from '@/hooks/loading'; import { @@ -292,7 +301,7 @@ queryKbDocumentList, deleteKnow, queryKbdetail, - queryKbtenantInfo, kbdocumentrm, kbdocumentchangeparser + queryKbtenantInfo, kbdocumentrm, kbdocumentchangeparser, kbdocumentrun, kbdocumentchangeStatus,downloadFile } from "@/api/kbList"; import { Pagination } from '@/types/global'; import type { SelectOptionData } from '@arco-design/web-vue/es/select/interface'; @@ -306,9 +315,10 @@ import test from '@/views/dmx/knowledgeLib/test.vue' import tool from '@/views/dmx/knowledgeLib/tool.vue' import addDetails from '@/views/dmx/knowledgeLib/addDetails.vue' - import { deleteAccount } from "@/api/account"; -import { Message } from "@arco-design/web-vue"; -import { parseTime } from "@/utils"; + import { Message } from "@arco-design/web-vue"; + import { parseTime } from "@/utils"; + import { usePinia } from "@/store"; + const store = usePinia(); type SizeProps = 'mini' | 'small' | 'medium' | 'large'; type Column = TableColumnData & { checked?: true }; const account = ref(null) @@ -331,15 +341,38 @@ const size = ref<SizeProps>('medium'); let visible = ref(false) + let keywords = ref('') let detailsShow = ref(false) let selectedTab = ref(0) + let testForm = ref(null) + let configForm = ref(null) + let kbId = ref('') + let kbobj = reactive({}) + let kbdetail = reactive({}) + let parser_ids = reactive({}) + let kbtenantInfo = reactive({ + "asr_id": "paraformer-realtime-8k-v1", + "embd_id": "BAAI/bge-large-zh-v1.5", + "img2txt_id": "qwen-vl-max", + "llm_id": "qwen-plus", + "name": "wanghao鈥榮 Kingdom", + "parser_ids": "naive:General,qa:Q&A,resume:Resume,manual:Manual,table:Table,paper:Paper,book:Book,laws:Laws,presentation:Presentation,picture:Picture,one:One", + "rerank_id": "BAAI/bge-reranker-v2-m3", + "role": "owner", + "tenant_id": "948fc6fa41ab11ef8fb80242ac120004", + parser_idObj:{} + }) let tabs = ref([]) const selectTab = (index,item) => { selectedTab.value= index; - console.log(item.id); + // console.log(item.id); + kbobj= item; + kbId = kbobj.id; + testForm.value.changekbid(kbId); + configForm.value.changekbObj(kbobj); fetchData({ - kb_id: item.id, + kb_id: kbobj.id, page: 1, page_size: 20 }) @@ -395,7 +428,6 @@ title: t('鍚敤'), dataIndex: 'status', slotName: 'status', - slotName: 'status', }, { title: t('瑙f瀽鐘舵��'), @@ -406,8 +438,8 @@ title: t('dmx.column.Operate'), dataIndex: 'action', slotName: 'action', - width: 150, - minWidth: 150, + width: 200, + minWidth: 200, }, ]); const contentTypeOptions = computed<SelectOptionData[]>(() => [ @@ -450,23 +482,31 @@ const data = await queryKbDocumentList(params); console.log(data, 'data'); if(data.code=='0'){ + // 璇︽儏 + // const kbtenantInfo = await queryKbtenantInfo(); + parser_ids = kbtenantInfo.parser_ids.split(',').reduce((acc, pair) => { + const [key, value] = pair.split(':'); + acc[key] = value; + return acc; + }, {}); + // console.log(parser_ids, 'parser_ids'); + kbtenantInfo.parser_idObj = parser_ids; renderData.value = data.data.docs||[]; renderData.value = renderData.value.map((item) => { return { ...item, loading: false, + parser_id: parser_ids[item.parser_id], }; }); - console.log(renderData.value, 'renderData.value'); + console.log(renderData.value, 'renderData'); pagination.current = params.page; pagination.total = data.data.total; // 鏌ヨ鐭ヨ瘑搴撹鎯� - const kbdetail = await queryKbdetail({ + kbdetail = await queryKbdetail({ kb_id:params.kb_id, }); - // 璇︽儏 - // const kbtenantInfo = await queryKbtenantInfo(); @@ -480,11 +520,12 @@ }; const search = () => { - alert(1) - // fetchData({ - // ...basePagination, - // ...formModel.value, - // }); + fetchData({ + kb_id: kbobj.id, + page: 1, + page_size: 20, + keywords:keywords.value + }) }; const handleClick = ()=>{ @@ -505,8 +546,12 @@ tabs.value = data; console.log(tabs.value, 'tabs'); if(tabs.value.length>0 && tabs.value[0]){ + kbobj = tabs.value[0] + kbId = kbobj.id; + testForm.value.changekbid(kbId); + configForm.value.changekbObj(kbobj); fetchData({ - kb_id: tabs.value[0].id, + kb_id: kbobj.id, page: 1, page_size: 20 }) @@ -523,10 +568,6 @@ }; const deleteKnowledge = async (id) => { - - - - let data = await deleteKnow({ kb_id:id }) @@ -615,17 +656,49 @@ let data = await kbdocumentrm({doc_id: row.id}) if(data.code == 0){ Message.success('鍒犻櫎鎴愬姛'); - knowledgeData() + console.log(kbobj, 'kbobj'); + fetchData({ + kb_id: kbobj.id, + page: 1, + page_size: 20 + }) } } + const onDownloadDocument = async (record)=>{ + console.log(record); + downloadFile({ + url: `/api/v1/document/get/${record.id}`, + filename: record.name, + }); + } +const seeload = async (row)=>{ + console.log(row); +} + +const handleChangeStatus = async (row)=>{ + console.log(row); + let data = await kbdocumentchangeStatus({ + doc_id: row.id, + status: row.status + }) + if(data.code == 0){ + Message.success('鎿嶄綔鎴愬姛'); + } +} + const run = async (row)=>{ console.log(row); row.loading = true; - let data = await kbdocumentchangeparser({ - parser_id: row.parser_id, - doc_id: row.id, - parser_config: row.parser_config, + let run = '1' + if(row.run!='2'){ + run = '1' + }else { + run = '2' + } + let data = await kbdocumentrun({ + doc_ids: [row.id], + run: run }) if(data.code == 0){ Message.success('瑙f瀽鎴愬姛'); @@ -637,7 +710,13 @@ row.loading = false; } - +const changeFetchData = async (row)=>{ + fetchData({ + kb_id: kbobj.id, + page: 1, + page_size: 20 + }) +} watch( () => columns.value, (val) => { @@ -652,6 +731,9 @@ ); onBeforeMount(()=>{ + + }) + onMounted(()=>{ knowledgeData() }) -- Gitblit v1.8.0