Skip to content

phsl / admin

Go to a project
Commit c2a6b02b by YunaiV
Options

【功能新增】AI：新增知识库分段的新增

parent 45ceacdc
Showing with 344 additions and 7 deletions
src/api/ai/knowledge/segment/index.ts
import request from '@/config/axios' import request from '@/config/axios'
// AI 知识库分片 VO // AI 知识库分段 VO
export interface AiKnowledgeSegmentRespVO { export interface KnowledgeSegmentVO {
id: number // 编号 id: number // 编号
documentId: number // 文档编号 documentId: number // 文档编号
knowledgeId: number // 知识库编号 knowledgeId: number // 知识库编号
...@@ -14,23 +14,41 @@ export interface AiKnowledgeSegmentRespVO { ...@@ -14,23 +14,41 @@ export interface AiKnowledgeSegmentRespVO {
createTime: number // 创建时间 createTime: number // 创建时间
} }
// AI 知识库分片 API // AI 知识库分段 API
export const KnowledgeSegmentApi = { export const KnowledgeSegmentApi = {
// 查询知识库分片分页 // 查询知识库分段分页
getKnowledgeSegmentPage: async (params: any) => { getKnowledgeSegmentPage: async (params: any) => {
return await request.get({ url: `/ai/knowledge/segment/page`, params }) return await request.get({ url: `/ai/knowledge/segment/page`, params })
}, },
// 查询知识库分片详情 // 查询知识库分段详情
getKnowledgeSegment: async (id: number) => { getKnowledgeSegment: async (id: number) => {
return await request.get({ url: `/ai/knowledge/segment/get?id=` + id }) return await request.get({ url: `/ai/knowledge/segment/get?id=` + id })
}, },
// 删除知识库分片 // 删除知识库分段
deleteKnowledgeSegment: async (id: number) => { deleteKnowledgeSegment: async (id: number) => {
return await request.delete({ url: `/ai/knowledge/segment/delete?id=` + id }) return await request.delete({ url: `/ai/knowledge/segment/delete?id=` + id })
}, },
// 新增知识库分段
createKnowledgeSegment: async (data: KnowledgeSegmentVO) => {
return await request.post({ url: `/ai/knowledge/segment/create`, data })
},
// 修改知识库分段
updateKnowledgeSegment: async (data: KnowledgeSegmentVO) => {
return await request.put({ url: `/ai/knowledge/segment/update`, data })
},
// 修改知识库分段状态
updateKnowledgeSegmentStatus: async (data: any) => {
return await request.put({
url: `/ai/knowledge/segment/update-status`,
data
})
},
// 切片内容 // 切片内容
splitContent: async (url: string, segmentMaxTokens: number) => { splitContent: async (url: string, segmentMaxTokens: number) => {
return await request.get({ return await request.get({
...@@ -47,7 +65,7 @@ export const KnowledgeSegmentApi = { ...@@ -47,7 +65,7 @@ export const KnowledgeSegmentApi = {
}) })
}, },
// 搜索知识库分片 // 搜索知识库分段
searchKnowledgeSegment: async (params: any) => { searchKnowledgeSegment: async (params: any) => {
return await request.get({ return await request.get({
url: `/ai/knowledge/segment/search`, url: `/ai/knowledge/segment/search`,
......
src/router/modules/remaining.ts
...@@ -667,6 +667,18 @@ const remainingRouter: AppRouteRecordRaw[] = [ ...@@ -667,6 +667,18 @@ const remainingRouter: AppRouteRecordRaw[] = [
hidden: true, hidden: true,
activeMenu: '/ai/knowledge' activeMenu: '/ai/knowledge'
} }
},
{
path: 'knowledge/segment',
component: () => import('@/views/ai/knowledge/segment/index.vue'),
name: 'AiKnowledgeSegment',
meta: {
title: '知识库分段',
icon: 'ep:tickets',
noCache: true,
hidden: true,
activeMenu: '/ai/knowledge'
}
} }
] ]
}, },
......
src/views/ai/knowledge/document/index.vue
...@@ -81,6 +81,14 @@ ...@@ -81,6 +81,14 @@
</el-button> </el-button>
<el-button <el-button
link link
type="primary"
@click="handleSegment(scope.row.id)"
v-hasPermi="['ai:knowledge:query']"
>
分段
</el-button>
<el-button
link
type="danger" type="danger"
@click="handleDelete(scope.row.id)" @click="handleDelete(scope.row.id)"
v-hasPermi="['ai:knowledge:delete']" v-hasPermi="['ai:knowledge:delete']"
...@@ -203,6 +211,14 @@ const handleStatusChange = async (row: KnowledgeDocumentVO) => { ...@@ -203,6 +211,14 @@ const handleStatusChange = async (row: KnowledgeDocumentVO) => {
} }
} }
/** 跳转到知识库分段页面 */
const handleSegment = (id: number) => {
router.push({
name: 'AiKnowledgeSegment',
query: { documentId: id }
})
}
/** 初始化 **/ /** 初始化 **/
onMounted(() => { onMounted(() => {
// 如果知识库 ID 不存在，显示错误提示并关闭页面 // 如果知识库 ID 不存在，显示错误提示并关闭页面
......
src/views/ai/knowledge/segment/KnowledgeSegmentForm.vue 0 → 100644
<template>
<Dialog :title="dialogTitle" v-model="dialogVisible">
<el-form
ref="formRef"
:model="formData"
:rules="formRules"
label-width="100px"
v-loading="formLoading"
>
<el-form-item label="切片内容" prop="content">
<el-input
v-model="formData.content"
type="textarea"
:rows="6"
placeholder="请输入切片内容"
/>
</el-form-item>
</el-form>
<template #footer>
<el-button @click="submitForm" type="primary" :disabled="formLoading">确 定</el-button>
<el-button @click="dialogVisible = false">取 消</el-button>
</template>
</Dialog>
</template>
<script setup lang="ts">
import { KnowledgeSegmentApi, KnowledgeSegmentVO } from '@/api/ai/knowledge/segment'
/** AI 知识库分段表单 */
defineOptions({ name: 'KnowledgeSegmentForm' })
const { t } = useI18n() // 国际化
const message = useMessage() // 消息弹窗
const dialogVisible = ref(false) // 弹窗的是否展示
const dialogTitle = ref('') // 弹窗的标题
const formLoading = ref(false) // 表单的加载中：1）修改时的数据加载；2）提交的按钮禁用
const formType = ref('') // 表单的类型：create - 新增；update - 修改
const formData = ref({
id: undefined,
documentId: undefined,
content: undefined
})
const formRules = reactive({
content: [{ required: true, message: '切片内容不能为空', trigger: 'blur' }]
})
const formRef = ref() // 表单 Ref
/** 打开弹窗 */
const open = async (type: string, id?: number, documentId?: any) => {
dialogVisible.value = true
dialogTitle.value = t('action.' + type)
formType.value = type
resetForm()
formData.value.documentId = documentId as any
// 修改时，设置数据
if (id) {
formLoading.value = true
try {
formData.value = await KnowledgeSegmentApi.getKnowledgeSegment(id)
} finally {
formLoading.value = false
}
}
}
defineExpose({ open }) // 提供 open 方法，用于打开弹窗
/** 提交表单 */
const emit = defineEmits(['success']) // 定义 success 事件，用于操作成功后的回调
const submitForm = async () => {
// 校验表单
await formRef.value.validate()
// 提交请求
formLoading.value = true
try {
const data = formData.value as unknown as KnowledgeSegmentVO
if (formType.value === 'create') {
await KnowledgeSegmentApi.createKnowledgeSegment(data)
message.success(t('common.createSuccess'))
} else {
await KnowledgeSegmentApi.updateKnowledgeSegment(data)
message.success(t('common.updateSuccess'))
}
dialogVisible.value = false
// 发送操作成功的事件
emit('success')
} finally {
formLoading.value = false
}
}
/** 重置表单 */
const resetForm = () => {
formData.value = {
id: undefined,
documentId: undefined,
content: undefined
}
formRef.value?.resetFields()
}
</script>
src/views/ai/knowledge/segment/index.vue 0 → 100644
<template>
<ContentWrap>
<!-- 搜索工作栏 -->
<el-form
class="-mb-15px"
:model="queryParams"
ref="queryFormRef"
:inline="true"
label-width="68px"
>
<el-form-item label="文档编号" prop="documentId">
<el-input
v-model="queryParams.documentId"
placeholder="请输入文档编号"
clearable
@keyup.enter="handleQuery"
class="!w-240px"
/>
</el-form-item>
<el-form-item label="是否启用" prop="status">
<el-select
v-model="queryParams.status"
placeholder="请选择是否启用"
clearable
class="!w-240px"
>
<el-option
v-for="dict in getIntDictOptions(DICT_TYPE.COMMON_STATUS)"
:key="dict.value"
:label="dict.label"
:value="dict.value"
/>
</el-select>
</el-form-item>
<el-form-item>
<el-button @click="handleQuery"><Icon icon="ep:search" class="mr-5px" /> 搜索</el-button>
<el-button @click="resetQuery"><Icon icon="ep:refresh" class="mr-5px" /> 重置</el-button>
<el-button
type="primary"
plain
@click="openForm('create')"
v-hasPermi="['ai:knowledge:create']"
>
<Icon icon="ep:plus" class="mr-5px" /> 新增
</el-button>
</el-form-item>
</el-form>
</ContentWrap>
<!-- 列表 -->
<ContentWrap>
<el-table v-loading="loading" :data="list" :stripe="true" :show-overflow-tooltip="true">
<el-table-column label="编号" align="center" prop="id" />
<el-table-column label="知识库编号" align="center" prop="knowledgeId" />
<el-table-column label="文档编号" align="center" prop="documentId" />
<el-table-column label="切片内容" align="center" prop="content" />
<el-table-column label="字符数" align="center" prop="contentLength" />
<el-table-column label="向量库的id" align="center" prop="vectorId" />
<el-table-column label="token数量" align="center" prop="tokens" />
<el-table-column label="召回次数" align="center" prop="retrievalCount" />
<el-table-column label="是否启用" align="center" prop="status">
<template #default="scope">
<dict-tag :type="DICT_TYPE.COMMON_STATUS" :value="scope.row.status" />
</template>
</el-table-column>
<el-table-column
label="创建时间"
align="center"
prop="createTime"
:formatter="dateFormatter"
width="180px"
/>
<el-table-column label="操作" align="center" min-width="120px">
<template #default="scope">
<el-button
link
type="primary"
@click="openForm('update', scope.row.id)"
v-hasPermi="['ai:knowledge:update']"
>
编辑
</el-button>
<el-button
link
type="danger"
@click="handleDelete(scope.row.id)"
v-hasPermi="['ai:knowledge:delete']"
>
删除
</el-button>
</template>
</el-table-column>
</el-table>
<!-- 分页 -->
<Pagination
:total="total"
v-model:page="queryParams.pageNo"
v-model:limit="queryParams.pageSize"
@pagination="getList"
/>
</ContentWrap>
<!-- 表单弹窗：添加/修改 -->
<KnowledgeSegmentForm ref="formRef" @success="getList" />
</template>
<script setup lang="ts">
import { getIntDictOptions, DICT_TYPE } from '@/utils/dict'
import { dateFormatter } from '@/utils/formatTime'
import { KnowledgeSegmentApi, KnowledgeSegmentVO } from '@/api/ai/knowledge/segment'
import KnowledgeSegmentForm from './KnowledgeSegmentForm.vue'
/** AI 知识库分段 列表 */
defineOptions({ name: 'KnowledgeSegment' })
const message = useMessage() // 消息弹窗
const router = useRouter() // 路由
const route = useRoute() // 路由
const { t } = useI18n() // 国际化
const loading = ref(true) // 列表的加载中
const list = ref<KnowledgeSegmentVO[]>([]) // 列表的数据
const total = ref(0) // 列表的总页数
const queryParams = reactive({
pageNo: 1,
pageSize: 10,
documentId: undefined,
content: undefined,
status: undefined
})
const queryFormRef = ref() // 搜索的表单
/** 查询列表 */
const getList = async () => {
loading.value = true
try {
const data = await KnowledgeSegmentApi.getKnowledgeSegmentPage(queryParams)
list.value = data.list
total.value = data.total
} finally {
loading.value = false
}
}
/** 搜索按钮操作 */
const handleQuery = () => {
queryParams.pageNo = 1
getList()
}
/** 重置按钮操作 */
const resetQuery = () => {
queryFormRef.value.resetFields()
handleQuery()
}
/** 添加/修改操作 */
const formRef = ref()
const openForm = (type: string, id?: number) => {
formRef.value.open(type, id, queryParams.documentId)
}
/** 删除按钮操作 */
const handleDelete = async (id: number) => {
try {
// 删除的二次确认
await message.delConfirm()
// 发起删除
await KnowledgeSegmentApi.deleteKnowledgeSegment(id)
message.success(t('common.delSuccess'))
// 刷新列表
await getList()
} catch {}
}
/** 初始化 **/
onMounted(() => {
// 如果文档 ID 不存在，显示错误提示并关闭页面
if (!route.query.documentId) {
message.error('文档 ID 不存在，无法查看分段列表')
// 关闭当前路由，返回到文档列表页面
router.push({ name: 'AiKnowledgeDocument' })
return
}
// 从路由参数中获取文档 ID
queryParams.documentId = route.query.documentId as any
getList()
})
</script>
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or sign in to comment