194 lines
4.7 KiB
TypeScript
194 lines
4.7 KiB
TypeScript
import type { VxeTableGridOptions } from '@vben/plugins/vxe-table';
|
||
|
||
import type { VbenFormSchema } from '#/adapter/form';
|
||
import type { OnActionClickFn } from '#/adapter/vxe-table';
|
||
import type { AiAIApiKeyApi } from '#/models/ai/ai_api_key';
|
||
|
||
import { z } from '#/adapter/form';
|
||
import { $t } from '#/locales';
|
||
import { op } from '#/utils/permission';
|
||
|
||
/**
|
||
* 获取编辑表单的字段配置
|
||
*/
|
||
export function useSchema(): VbenFormSchema[] {
|
||
return [
|
||
{
|
||
component: 'Input',
|
||
fieldName: 'name',
|
||
label: '名称',
|
||
rules: z
|
||
.string()
|
||
.min(1, $t('ui.formRules.required', ['名称']))
|
||
.max(100, $t('ui.formRules.maxLength', ['名称', 100])),
|
||
},
|
||
{
|
||
component: 'Select',
|
||
fieldName: 'platform',
|
||
label: '平台',
|
||
componentProps: {
|
||
options: PLATFORM_OPTIONS,
|
||
class: 'w-full',
|
||
placeholder: '请选择',
|
||
},
|
||
rules: z.string(),
|
||
},
|
||
{
|
||
component: 'Input',
|
||
fieldName: 'api_key',
|
||
label: '密钥',
|
||
rules: z
|
||
.string()
|
||
.min(1, $t('ui.formRules.required', ['密钥']))
|
||
.max(100, $t('ui.formRules.maxLength', ['密钥', 100])),
|
||
},
|
||
{
|
||
component: 'Input',
|
||
fieldName: 'url',
|
||
label: '自定义 API 地址',
|
||
},
|
||
{
|
||
component: 'RadioGroup',
|
||
componentProps: {
|
||
buttonStyle: 'solid',
|
||
options: [
|
||
{ label: $t('common.enabled'), value: 1 },
|
||
{ label: $t('common.disabled'), value: 0 },
|
||
],
|
||
optionType: 'button',
|
||
},
|
||
defaultValue: 1,
|
||
fieldName: 'status',
|
||
label: $t('system.status'),
|
||
},
|
||
{
|
||
component: 'Input',
|
||
fieldName: 'remark',
|
||
label: $t('system.remark'),
|
||
},
|
||
];
|
||
}
|
||
|
||
/**
|
||
* 获取编辑表单的字段配置
|
||
*/
|
||
export function useGridFormSchema(): VbenFormSchema[] {
|
||
return [
|
||
{
|
||
component: 'Input',
|
||
fieldName: 'name',
|
||
label: '名称',
|
||
},
|
||
{
|
||
component: 'Select',
|
||
fieldName: 'platform',
|
||
label: '平台',
|
||
componentProps: {
|
||
allowClear: true,
|
||
options: PLATFORM_OPTIONS,
|
||
},
|
||
},
|
||
{
|
||
component: 'Select',
|
||
fieldName: 'status',
|
||
label: '状态',
|
||
componentProps: {
|
||
allowClear: true,
|
||
options: [
|
||
{ label: '启用', value: 1 },
|
||
{ label: '禁用', value: 0 },
|
||
],
|
||
},
|
||
},
|
||
];
|
||
}
|
||
|
||
/**
|
||
* 获取表格列配置
|
||
* @description 使用函数的形式返回列数据而不是直接export一个Array常量，是为了响应语言切换时重新翻译表头
|
||
* @param onActionClick 表格操作按钮点击事件
|
||
*/
|
||
export function useColumns(
|
||
onActionClick?: OnActionClickFn<AiAIApiKeyApi.AiAIApiKey>,
|
||
): VxeTableGridOptions<AiAIApiKeyApi.AiAIApiKey>['columns'] {
|
||
// 平台映射表
|
||
return [
|
||
{
|
||
field: 'id',
|
||
title: 'ID',
|
||
},
|
||
{
|
||
field: 'name',
|
||
title: '名称',
|
||
},
|
||
{
|
||
cellRender: {
|
||
name: 'CellTag',
|
||
},
|
||
field: 'platform',
|
||
title: '平台',
|
||
formatter: ({ cellValue }: { cellValue: string }) => {
|
||
return PLATFORM_MAP[String(cellValue)] || String(cellValue);
|
||
},
|
||
},
|
||
{
|
||
field: 'url',
|
||
title: '自定义 API 地址',
|
||
},
|
||
{
|
||
cellRender: {
|
||
name: 'CellTag',
|
||
},
|
||
field: 'status',
|
||
title: '状态',
|
||
},
|
||
{
|
||
field: 'remark',
|
||
title: '备注',
|
||
},
|
||
{
|
||
align: 'center',
|
||
cellRender: {
|
||
attrs: {
|
||
nameField: 'name',
|
||
nameTitle: $t('ai.api_key.name'),
|
||
onClick: onActionClick,
|
||
},
|
||
name: 'CellOperation',
|
||
options: [
|
||
op('ai:ai_api_key:edit', 'edit'),
|
||
op('ai:ai_api_key:delete', 'delete'),
|
||
],
|
||
},
|
||
field: 'action',
|
||
fixed: 'right',
|
||
title: '操作',
|
||
width: 120,
|
||
},
|
||
];
|
||
}
|
||
|
||
// 平台选项和映射常量
|
||
export const PLATFORM_OPTIONS = [
|
||
{ label: 'OpenAI 微软', value: 'AzureOpenAI' },
|
||
{ label: 'OpenAI', value: 'OpenAI' },
|
||
{ label: 'Ollama', value: 'Ollama' },
|
||
{ label: '文心一言', value: 'YiYan' },
|
||
{ label: '讯飞星火', value: 'XingHuo' },
|
||
{ label: '通义千问', value: 'TongYi' },
|
||
{ label: 'StableDiffusion', value: 'StableDiffusion' },
|
||
{ label: 'Midjourney', value: 'Midjourney' },
|
||
{ label: 'Suno', value: 'Suno' },
|
||
{ label: 'DeepSeek', value: 'DeepSeek' },
|
||
{ label: '字节豆包', value: 'DouBao' },
|
||
{ label: '腾讯混元', value: 'HunYuan' },
|
||
{ label: '硅基流动', value: 'SiliconFlow' },
|
||
{ label: '智谱', value: 'ZhiPu' },
|
||
{ label: 'MiniMax', value: 'MiniMax' },
|
||
{ label: '月之暗灭', value: 'Moonshot' },
|
||
{ label: '百川智能', value: 'BaiChuan' },
|
||
];
|
||
export const PLATFORM_MAP: Record<string, string> = Object.fromEntries(
|
||
PLATFORM_OPTIONS.map((opt) => [opt.value, opt.label]),
|
||
);
|