131 lines
2.8 KiB
TypeScript
131 lines
2.8 KiB
TypeScript
import type { VbenFormSchema } from '#/adapter/form';
|
|
import type { VxeTableGridOptions } from '#/adapter/vxe-table';
|
|
import type { AiKnowledgeSegmentApi } from '#/api/ai/knowledge/segment';
|
|
|
|
import { CommonStatusEnum, DICT_TYPE } from '@vben/constants';
|
|
import { getDictOptions } from '@vben/hooks';
|
|
|
|
/** 新增/修改的表单 */
|
|
export function useFormSchema(): VbenFormSchema[] {
|
|
return [
|
|
{
|
|
component: 'Input',
|
|
fieldName: 'id',
|
|
dependencies: {
|
|
triggerFields: [''],
|
|
show: () => false,
|
|
},
|
|
},
|
|
{
|
|
component: 'Input',
|
|
fieldName: 'documentId',
|
|
dependencies: {
|
|
triggerFields: [''],
|
|
show: () => false,
|
|
},
|
|
},
|
|
{
|
|
fieldName: 'content',
|
|
label: '切片内容',
|
|
component: 'Textarea',
|
|
componentProps: {
|
|
placeholder: '请输入切片内容',
|
|
rows: 6,
|
|
showCount: true,
|
|
},
|
|
rules: 'required',
|
|
},
|
|
];
|
|
}
|
|
/** 列表的搜索表单 */
|
|
export function useGridFormSchema(): VbenFormSchema[] {
|
|
return [
|
|
{
|
|
fieldName: 'documentId',
|
|
label: '文档编号',
|
|
component: 'Input',
|
|
componentProps: {
|
|
placeholder: '请输入文档编号',
|
|
allowClear: true,
|
|
},
|
|
},
|
|
{
|
|
fieldName: 'status',
|
|
label: '是否启用',
|
|
component: 'Select',
|
|
componentProps: {
|
|
placeholder: '请选择是否启用',
|
|
allowClear: true,
|
|
options: getDictOptions(DICT_TYPE.COMMON_STATUS, 'number'),
|
|
},
|
|
},
|
|
];
|
|
}
|
|
|
|
/** 列表的字段 */
|
|
export function useGridColumns(
|
|
onStatusChange?: (
|
|
newStatus: number,
|
|
row: AiKnowledgeSegmentApi.KnowledgeSegment,
|
|
) => PromiseLike<boolean | undefined>,
|
|
): VxeTableGridOptions['columns'] {
|
|
return [
|
|
{
|
|
field: 'id',
|
|
title: '分段编号',
|
|
minWidth: 100,
|
|
},
|
|
{
|
|
type: 'expand',
|
|
width: 40,
|
|
slots: { content: 'expand_content' },
|
|
},
|
|
{
|
|
field: 'content',
|
|
title: '切片内容',
|
|
minWidth: 250,
|
|
},
|
|
{
|
|
field: 'contentLength',
|
|
title: '字符数',
|
|
minWidth: 100,
|
|
},
|
|
{
|
|
field: 'tokens',
|
|
title: 'token 数量',
|
|
minWidth: 120,
|
|
},
|
|
{
|
|
field: 'retrievalCount',
|
|
title: '召回次数',
|
|
minWidth: 100,
|
|
},
|
|
{
|
|
field: 'status',
|
|
title: '状态',
|
|
minWidth: 100,
|
|
align: 'center',
|
|
cellRender: {
|
|
attrs: { beforeChange: onStatusChange },
|
|
name: 'CellSwitch',
|
|
props: {
|
|
checkedValue: CommonStatusEnum.ENABLE,
|
|
unCheckedValue: CommonStatusEnum.DISABLE,
|
|
},
|
|
},
|
|
},
|
|
{
|
|
field: 'createTime',
|
|
title: '创建时间',
|
|
minWidth: 180,
|
|
formatter: 'formatDateTime',
|
|
},
|
|
{
|
|
title: '操作',
|
|
width: 150,
|
|
fixed: 'right',
|
|
slots: { default: 'actions' },
|
|
},
|
|
];
|
|
}
|