【功能新增】AI:聊天角色,新增 knowledge 的引用绑定

This commit is contained in:
YunaiV 2025-03-09 17:38:44 +08:00
parent d4a90b7dd5
commit 47e1cd9c1d
4 changed files with 30 additions and 2 deletions

View File

@ -35,5 +35,10 @@ export const KnowledgeApi = {
// 删除知识库
deleteKnowledge: async (id: number) => {
return await request.delete({ url: `/ai/knowledge/delete?id=` + id })
},
// 获取知识库简单列表
getSimpleKnowledgeList: async () => {
return await request.get({ url: `/ai/knowledge/simple-list` })
}
}

View File

@ -13,6 +13,7 @@ export interface ChatRoleVO {
welcomeMessage: string // 角色设定
publicStatus: boolean // 是否公开
status: number // 状态
knowledgeIds?: number[] // 引用的知识库ID列表
}
// AI 聊天角色 分页请求 vo

View File

@ -32,6 +32,16 @@
<el-form-item label="角色设定" prop="systemMessage">
<el-input type="textarea" v-model="formData.systemMessage" placeholder="请输入角色设定" />
</el-form-item>
<el-form-item label="引用知识库" prop="knowledgeIds">
<el-select v-model="formData.knowledgeIds" placeholder="请选择知识库" clearable multiple>
<el-option
v-for="item in knowledgeList"
:key="item.id"
:label="item.name"
:value="item.id"
/>
</el-select>
</el-form-item>
<el-form-item label="是否公开" prop="publicStatus" v-if="!isUser">
<el-radio-group v-model="formData.publicStatus">
<el-radio
@ -71,6 +81,7 @@ import { CommonStatusEnum } from '@/utils/constants'
import { ModelApi, ModelVO } from '@/api/ai/model/model'
import { FormRules } from 'element-plus'
import { AiModelTypeEnum } from '@/views/ai/utils/constants'
import { KnowledgeApi, KnowledgeVO } from '@/api/ai/knowledge/knowledge'
/** AI 聊天角色 表单 */
defineOptions({ name: 'ChatRoleForm' })
@ -92,10 +103,12 @@ const formData = ref({
description: undefined,
systemMessage: undefined,
publicStatus: true,
status: CommonStatusEnum.ENABLE
status: CommonStatusEnum.ENABLE,
knowledgeIds: [] as number[]
})
const formRef = ref() // Ref
const models = ref([] as ModelVO[]) //
const knowledgeList = ref([] as KnowledgeVO[]) //
/** 是否【我】自己创建,私有角色 */
const isUser = computed(() => {
@ -130,6 +143,8 @@ const open = async (type: string, id?: number, title?: string) => {
}
//
models.value = await ModelApi.getModelSimpleList(AiModelTypeEnum.CHAT)
//
knowledgeList.value = await KnowledgeApi.getSimpleKnowledgeList()
}
defineExpose({ open }) // open
@ -177,7 +192,8 @@ const resetForm = () => {
description: undefined,
systemMessage: undefined,
publicStatus: true,
status: CommonStatusEnum.ENABLE
status: CommonStatusEnum.ENABLE,
knowledgeIds: []
}
formRef.value?.resetFields()
}

View File

@ -69,6 +69,12 @@
<el-table-column label="角色类别" align="center" prop="category" />
<el-table-column label="角色描述" align="center" prop="description" />
<el-table-column label="角色设定" align="center" prop="systemMessage" />
<el-table-column label="知识库" align="center" prop="knowledgeIds">
<template #default="scope">
<span v-if="!scope.row.knowledgeIds || scope.row.knowledgeIds.length === 0">-</span>
<span v-else>引用 {{ scope.row.knowledgeIds.length }} </span>
</template>
</el-table-column>
<el-table-column label="是否公开" align="center" prop="publicStatus">
<template #default="scope">
<dict-tag :type="DICT_TYPE.INFRA_BOOLEAN_STRING" :value="scope.row.publicStatus" />