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

Author: YunaiV

src/api/ai/knowledge/segment/index.ts

@@ -1,7 +1,7 @@
 import request from '@/config/axios'
 
-// AI 知识库分片 VO
-export interface AiKnowledgeSegmentRespVO {
+// AI 知识库分段 VO
+export interface KnowledgeSegmentVO {
   id: number // 编号
   documentId: number // 文档编号
   knowledgeId: number // 知识库编号
@@ -14,23 +14,41 @@ export interface AiKnowledgeSegmentRespVO {
   createTime: number // 创建时间
 }
 
-// AI 知识库分片 API
+// AI 知识库分段 API
 export const KnowledgeSegmentApi = {
-  // 查询知识库分片分页
+  // 查询知识库分段分页
   getKnowledgeSegmentPage: async (params: any) => {
     return await request.get({ url: `/ai/knowledge/segment/page`, params })
   },
 
-  // 查询知识库分片详情
+  // 查询知识库分段详情
   getKnowledgeSegment: async (id: number) => {
     return await request.get({ url: `/ai/knowledge/segment/get?id=` + id })
   },
 
-  // 删除知识库分片
+  // 删除知识库分段
   deleteKnowledgeSegment: async (id: number) => {
     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) => {
     return await request.get({
@@ -47,7 +65,7 @@ export const KnowledgeSegmentApi = {
     })
   },
 
-  // 搜索知识库分片
+  // 搜索知识库分段
   searchKnowledgeSegment: async (params: any) => {
     return await request.get({
       url: `/ai/knowledge/segment/search`,

src/router/modules/remaining.ts

@@ -667,6 +667,18 @@ const remainingRouter: AppRouteRecordRaw[] = [
           hidden: true,
           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

@@ -79,6 +79,14 @@
           >
             编辑
           </el-button>
+          <el-button
+            link
+            type="primary"
+            @click="handleSegment(scope.row.id)"
+            v-hasPermi="['ai:knowledge:query']"
+          >
+            分段
+          </el-button>
           <el-button
             link
             type="danger"
@@ -203,6 +211,14 @@ const handleStatusChange = async (row: KnowledgeDocumentVO) => {
   }
 }
 
+/** 跳转到知识库分段页面 */
+const handleSegment = (id: number) => {
+  router.push({
+    name: 'AiKnowledgeSegment',
+    query: { documentId: id }
+  })
+}
+
 /** 初始化 **/
 onMounted(() => {
   // 如果知识库 ID 不存在，显示错误提示并关闭页面

src/views/ai/knowledge/segment/KnowledgeSegmentForm.vue

@@ -0,0 +1,101 @@
+<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>