<template>
|
<div class="flex flex-col h100">
|
<div class="mb-[10px] flex flex-shrink-0 items-center">
|
<el-button style="margin-left: 8px; width: 40px" link @click="handleExitFlow">
|
<el-icon style="font-size: 24px !important">
|
<ArrowLeft />
|
</el-icon>
|
</el-button>
|
<span class="text-[24px] text-[#26244c] font-[700]">{{ state.knowledgeInfo.knowledge_title }}</span>
|
</div>
|
<div class="set-table-height" v-show="!state.showKnowledgeForm">
|
<!-- 查询、重置、排序、增加表单 -->
|
<el-form :inline="true" :model="graphQueryParams">
|
<el-form-item label="名称" prop="title">
|
<el-input v-model="graphQueryParams.title" style="width: 226.4px" placeholder="文件名称" clearable></el-input>
|
</el-form-item>
|
<el-form-item>
|
<el-button type="primary" icon="ele-Search" @click="handleQueryTable"> 查询 </el-button>
|
<el-button icon="ele-Refresh" @click="resetQuery">重置 </el-button>
|
<el-button icon="ele-Plus" @click="getImportantFileTreeData()" type="primary"> 导入数据 </el-button>
|
</el-form-item>
|
</el-form>
|
<div class="flex-auto flex-column h-full">
|
<el-table
|
ref="viewKnowledgeTableRef"
|
border
|
row-key="id"
|
class="flex-auto"
|
:header-cell-style="{ textAlign: 'center' }"
|
:data="displayTableData"
|
highlight-current-row
|
>
|
<el-table-column prop="name" label="文件名称" fixed="left" show-overflow-tooltip align="left" />
|
<el-table-column prop="type" width="120" label="文件类型" show-overflow-tooltip align="center" />
|
<el-table-column prop="time" label="上传时间" show-overflow-tooltip width="280" align="center"></el-table-column>
|
<el-table-column label="操作" width="80" fixed="right" show-overflow-tooltip align="center">
|
<template #default="scope">
|
<el-tooltip effect="dark" content="删除" placement="top">
|
<i
|
class="ywifont ywicon-shanchu !text-[17px] text-red-400 cursor-pointer"
|
@click="deleteKnowledgeFileData(scope.row)"
|
></i>
|
</el-tooltip>
|
</template>
|
</el-table-column>
|
</el-table>
|
</div>
|
</div>
|
<div class="set-form-height" v-show="state.showKnowledgeForm">
|
<el-form :model="state.knowledgeFile" label-width="120px" label-position="left" class="h100" ref="knowledgeFileRef">
|
<el-divider content-position="left"><span class="text-[18px] font-[500]">选择文件</span></el-divider>
|
<div class="flex items-start gap-[16px] max-h-[686px] min-h-[360px] set-file-height">
|
<div class="min-h-[360px] h100 set_file_left">
|
<div class="font-[500] mb-2">请选择文档</div>
|
<div class="left_content">
|
<div class="file_menu">
|
<LeftTreeByMgr
|
title-name="类目"
|
ref="leftTreeRef"
|
:treedata="state.knowledgeBaseData"
|
:current-node-key="currentListID"
|
:defaultProps="{
|
children: 'Children',
|
label: 'title',
|
id: 'id',
|
}"
|
@click="handleClickNode"
|
>
|
</LeftTreeByMgr>
|
</div>
|
<div class="file_table">
|
<el-table
|
:data="state.fileData"
|
border
|
@select="handleSelectItem"
|
highlight-current-row
|
ref="multipleTableRef"
|
@select-all="
|
(selection) => {
|
handleSelectAll(selection, state.fileData);
|
}
|
"
|
>
|
<el-table-column type="selection" width="55" />
|
<el-table-column prop="name" label="文件名称" />
|
</el-table>
|
</div>
|
</div>
|
</div>
|
<div class="set_file_right">
|
<div class="h100">
|
<div class="font-[500] mb-[12px]">已选文档{{ multipleSelection.length }}/50个</div>
|
<div class="selected_file">
|
<div v-for="(item, index) in multipleSelection" :key="index" class="set_file_item">
|
<span class="set_file_name">{{ item.name }}</span>
|
<span class="ywifont ywicon-guanbi set_file_close" @click="handleRemoveItem(item)"></span>
|
</div>
|
</div>
|
</div>
|
</div>
|
</div>
|
<el-divider content-position="left"><span class="text-[18px]">数据处理</span></el-divider>
|
<el-form-item label="文档切分chunk:">
|
<div class="flex-auto flex flex-col">
|
<div class="flex">
|
<div v-for="(item, index) in state.segmentationList" :key="index">
|
<div
|
class="bg-[#fff] border-[1px] border-solid border-[#d8d9e6] py-[12px] w-[215px] mr-[10px] px-[16px] rounded-lg cursor-pointer"
|
:class="{ activeColor: state.knowledgeForm.segmentationMode === item.ID }"
|
@click="activeDataProcessType(item.ID)"
|
>
|
<div class="flex items-center">
|
<div class="data_left">
|
<img :src="item.ImageURL" class="w-[40px] h-[40px] mr-[10px]" />
|
</div>
|
<div class="data_right">
|
<el-radio v-model="state.knowledgeForm.segmentationMode" size="large" :label="item.ID">
|
<span class="font-[500] text[16px]">{{ item.Name }}</span>
|
</el-radio>
|
|
<el-tooltip :content="item.DemoDesc" placement="top" effect="light" popper-class="set_tooltip_demo">
|
<div class="text-[#878aab] text-[12px] mx-0 mt-[2px] mb-0 set-desc">
|
{{ item.DemoDesc }}
|
</div>
|
</el-tooltip>
|
</div>
|
</div>
|
</div>
|
</div>
|
</div>
|
</div>
|
</el-form-item>
|
</el-form>
|
</div>
|
</div>
|
</template>
|
|
<script setup lang="ts">
|
import { ElMessage, ElMessageBox } from 'element-plus';
|
import { computed, nextTick, onMounted, reactive, ref } from 'vue';
|
import { useRoute, useRouter } from 'vue-router';
|
import { delete_docvector_file, get_docvector_file_list } from '/@/api/knowledge/docvector';
|
import { get_knowledge_group_list, list_knowledge_file } from '/@/api/knowledge/group';
|
import LeftTreeByMgr from '/@/components/tree/leftTreeByMgr.vue';
|
import { useQueryTable } from '/@/hooks/useQueryTable';
|
import { convertListToTree } from '/@/utils/util';
|
const viewKnowledgeTableRef = ref();
|
const router = useRouter();
|
const route = useRoute();
|
const knowledgeTableData = ref([]);
|
// 定义变量内容
|
const state = reactive({
|
knowledgeInfo: {
|
knowledge_title: '',
|
knowledge_id: '',
|
} as any,
|
knowledgeForm: {
|
segmentationMode: 1,
|
},
|
segmentationList: [
|
{
|
ID: 1,
|
Name: '智能切分',
|
ImageURL: 'static/images/knowledge/data_type_3.png',
|
DemoDesc: '在通用文档上的最优chunk切分方法，经过评测可在多数文档上获得最佳的检索效果',
|
},
|
{
|
ID: 3,
|
Name: '自定义切分',
|
ImageURL: 'static/images/knowledge/data_type_4.png',
|
DemoDesc: '完全开放的chunk切分配置，按照实际文档情况自由配置，通过调试获得更好的检索效果',
|
},
|
],
|
showKnowledgeForm: false,
|
knowledgeFile: {},
|
knowledgeBaseData: [], //知识库类目
|
fileData: [], //文件数据源
|
isCreateIndex: false, //是否调用创建的索引的字段
|
});
|
//返回
|
const handleExitFlow = () => {
|
//是否显示返回
|
router.back();
|
state.showKnowledgeForm = true;
|
};
|
//#region ====================== 搜索表格，对表格排序 ======================
|
const graphQueryParams = ref({
|
title: '',
|
});
|
const { resetQuery, handleQueryTable, displayTableData } = useQueryTable(knowledgeTableData, graphQueryParams, () => {
|
displayTableData.value = knowledgeTableData.value;
|
});
|
//#endregion
|
//#region ====================== 对知识库进行操作 ======================
|
const getKnowledgeFileData = async () => {
|
const res = await get_docvector_file_list({
|
knowlg_id: state.knowledgeInfo.knowledge_id,
|
});
|
if (res.json_ok) {
|
knowledgeTableData.value = res.values;
|
}
|
};
|
//删除知识库的文件列表
|
const deleteKnowledgeFileData = (row: any) => {
|
ElMessageBox.confirm(`确定删除文档列表：【${row.name}】?`, '提示', {
|
confirmButtonText: '确定',
|
cancelButtonText: '取消',
|
type: 'warning',
|
}).then(async () => {
|
const res = await delete_docvector_file({
|
knowlg_id: state.knowledgeInfo.knowledge_id,
|
file_id: row.id,
|
});
|
|
if (res.json_ok) {
|
ElMessage.success('删除文档列表成功');
|
const index = knowledgeTableData.value.findIndex((d) => d.id === row.id);
|
knowledgeTableData.value.splice(index, 1);
|
} else {
|
ElMessage.error('删除文档列表失败' + (res?.json_msg ? `，${JSON.stringify(res.json_msg)}` : ''));
|
}
|
});
|
};
|
//#endregion
|
|
//#region ====================== 知识库文件上传 ==========
|
const treeLoading = ref(false);
|
const currentTreeNode = ref(null);
|
const leftTreeRef = ref(null);
|
const currentListID = computed(() => currentTreeNode.value?.id);
|
//获取文件列表
|
const getImportantFileTreeData = async (selectFirst = false) => {
|
state.showKnowledgeForm = false;
|
treeLoading.value = true;
|
const res = await get_knowledge_group_list().finally(() => {
|
treeLoading.value = false;
|
});
|
if (res?.json_ok) {
|
const resData = (res.values || []) as [];
|
state.knowledgeBaseData = convertListToTree(resData, {
|
ID: 'id',
|
Children: 'Children',
|
ParentID: 'parent',
|
});
|
if (selectFirst) {
|
const firstListTreeNode = state.knowledgeBaseData[0];
|
if (firstListTreeNode) {
|
handleClickNode(firstListTreeNode);
|
} else {
|
state.fileData = [];
|
multipleSelection.value = [];
|
}
|
} else {
|
currentTreeNode.value && handleClickNode(currentTreeNode.value);
|
}
|
}
|
};
|
const handleClickNode = (data: any) => {
|
multipleSelection.value = [];
|
nextTick(() => {
|
leftTreeRef.value?.treeRef.setCurrentKey(data.id);
|
});
|
currentTreeNode.value = data;
|
getFileTableData();
|
};
|
//获取文件表格列表
|
const getFileTableData = async () => {
|
const res = await list_knowledge_file({
|
group_id: currentListID.value,
|
}).finally(() => {});
|
if (res?.json_ok) {
|
const resData = (res.values || []) as [];
|
state.fileData = resData;
|
} else {
|
ElMessage.error('获取文档列表失败' + (res?.json_msg ? `，${JSON.stringify(res.json_msg)}` : ''));
|
}
|
};
|
let multipleSelection = ref([]);
|
const multipleTableRef = ref(null);
|
//表格单选和多选
|
const handleSelectAll = (selection: any[], pageSelectionData: any[]) => {
|
let checked = selection.length ? true : false; // selection为空数组时代表取消全选
|
if (checked) {
|
let mIds = multipleSelection.value.map((v) => v.id);
|
let filterData = pageSelectionData.filter((v) => {
|
// 筛选出非重复项
|
return !mIds.includes(v.id);
|
});
|
multipleSelection.value = multipleSelection.value.concat(filterData);
|
} else {
|
let pIds = pageSelectionData.map((v) => v.id);
|
multipleSelection.value = multipleSelection.value.filter((v) => !pIds.includes(v.id));
|
}
|
};
|
|
const handleSelectItem = (selection: any[], row) => {
|
multipleSelection.value = selection;
|
};
|
// 移除已选文件
|
const handleRemoveItem = (item) => {
|
multipleSelection.value = multipleSelection.value.filter((v) => v.id !== item.id);
|
multipleTableRef.value.toggleRowSelection(item, false);
|
};
|
//选择数据文档切分
|
const activeDataProcessType = (id: number) => {
|
state.knowledgeForm.segmentationMode = id;
|
};
|
//#endregion
|
onMounted(() => {
|
const { knowledge_id, knowledge_title } = route.query;
|
state.knowledgeInfo.knowledge_title = knowledge_title;
|
state.knowledgeInfo.knowledge_id = knowledge_id;
|
getKnowledgeFileData();
|
});
|
</script>
|
<style scoped lang="scss">
|
.set-table-height {
|
background: #fff;
|
border-radius: 16px;
|
height: calc(100% - 80px);
|
margin-top: 12px;
|
overflow-y: auto;
|
padding: 20px 24px;
|
box-sizing: border-box;
|
}
|
</style>
|
import { get_knowledge_group_list, list_knowledge_file } from '/@/api/knowledge/group'; import { convertListToTree } from
|
'/@/utils/util';
|
