435 lines
14 KiB
Vue
435 lines
14 KiB
Vue
<template>
|
||
<div class="system-user-container layout-padding">
|
||
<el-card shadow="hover" class="layout-padding-auto">
|
||
<div class="system-user-search mb15">
|
||
<el-input v-model="state.tableData.param.modelName" size="default" placeholder="请输入模型名称" style="max-width: 180px" clearable>
|
||
</el-input>
|
||
<el-select
|
||
v-model="state.tableData.param.modelType"
|
||
size="default"
|
||
placeholder="请选择模型类型"
|
||
style="max-width: 180px"
|
||
clearable
|
||
class="ml10"
|
||
>
|
||
<el-option v-for="type in state.modelTypes" :key="type.id" :label="type.label" :value="type.id" />
|
||
</el-select>
|
||
<el-button size="default" type="primary" class="ml10" @click="getTableData">
|
||
<el-icon>
|
||
<ele-Search />
|
||
</el-icon>
|
||
查询
|
||
</el-button>
|
||
<el-button size="default" type="success" class="ml10" @click="onOpenAddModule('add')">
|
||
<el-icon>
|
||
<ele-FolderAdd />
|
||
</el-icon>
|
||
新增模型配置
|
||
</el-button>
|
||
</div>
|
||
<el-table :data="state.tableData.data" v-loading="state.tableData.loading" style="width: 100%">
|
||
<el-table-column type="index" label="序号" width="60" />
|
||
<el-table-column prop="modelName" label="模型名称" show-overflow-tooltip></el-table-column>
|
||
<el-table-column label="模型类型" width="120" show-overflow-tooltip>
|
||
<template #default="{ row }">
|
||
{{ resolveModelTypeLabel(row.modelType) }}
|
||
</template>
|
||
</el-table-column>
|
||
<!-- <el-table-column prop="baseUrl" label="模型服务地址" show-overflow-tooltip width="200"></el-table-column> -->
|
||
<el-table-column prop="isOwner" label="模型归属" width="100">
|
||
<template #default="scope">
|
||
<el-tag :type="scope.row.isOwner === 0 ? 'warning' : 'success'">{{ scope.row.isOwner === 0 ? '内置模型' : '用户模型' }}</el-tag>
|
||
</template>
|
||
</el-table-column>
|
||
<el-table-column prop="isPrivate" label="访问类型" width="100">
|
||
<template #default="scope">
|
||
<el-tag :type="scope.row.isPrivate === 1 ? 'primary' : 'info'">{{ scope.row.isPrivate === 1 ? '服务商模型' : '本地模型' }}</el-tag>
|
||
</template>
|
||
</el-table-column>
|
||
<el-table-column prop="httpMethod" label="请求方式" width="100"></el-table-column>
|
||
<el-table-column prop="enabled" label="状态" width="100">
|
||
<template #default="scope">
|
||
<el-tag :type="scope.row.enabled === 1 ? 'success' : 'danger'">{{ scope.row.enabled === 1 ? '启用' : '禁用' }}</el-tag>
|
||
</template>
|
||
</el-table-column>
|
||
<el-table-column prop="maxConcurrency" label="最大并发" width="100"></el-table-column>
|
||
<el-table-column prop="queueLimit" label="队列上限" width="100"></el-table-column>
|
||
<el-table-column prop="remark" label="备注" show-overflow-tooltip></el-table-column>
|
||
<el-table-column prop="createdAt" label="创建时间" show-overflow-tooltip width="160"></el-table-column>
|
||
<el-table-column prop="updatedAt" label="修改时间" show-overflow-tooltip width="160"></el-table-column>
|
||
<el-table-column label="操作" :width="isSuperAdmin ? 150 : 300" fixed="right">
|
||
<template #default="scope">
|
||
<div class="action-buttons">
|
||
<el-button size="small" text type="primary" @click="onOpenEditModule('edit', scope.row)">修改</el-button>
|
||
<!-- 非管理员才显示会话模型按钮 -->
|
||
<template v-if="!isSuperAdmin">
|
||
<el-button
|
||
v-if="isInferenceModel(scope.row.modelType) && Number(scope.row.isChatModel) !== 1"
|
||
size="small"
|
||
text
|
||
type="warning"
|
||
@click="onSetChatModel(scope.row)"
|
||
>
|
||
设为会话模型
|
||
</el-button>
|
||
<el-tag
|
||
v-if="isInferenceModel(scope.row.modelType) && Number(scope.row.isChatModel) === 1"
|
||
type="success"
|
||
effect="dark"
|
||
size="default"
|
||
>
|
||
✓ 当前会话模型
|
||
</el-tag>
|
||
</template>
|
||
<el-button size="small" text type="danger" @click="onRowDel(scope.row)">删除</el-button>
|
||
</div>
|
||
</template>
|
||
</el-table-column>
|
||
</el-table>
|
||
<el-pagination
|
||
@size-change="onHandleSizeChange"
|
||
@current-change="onHandleCurrentChange"
|
||
class="mt15"
|
||
:pager-count="5"
|
||
:page-sizes="[10, 20, 30, 50]"
|
||
v-model:current-page="state.tableData.param.pageNum"
|
||
background
|
||
v-model:page-size="state.tableData.param.pageSize"
|
||
layout="total, sizes, prev, pager, next, jumper"
|
||
:total="state.tableData.total"
|
||
>
|
||
</el-pagination>
|
||
</el-card>
|
||
<EditModule ref="editModuleRef" :model-types="state.modelTypes" :is-super-admin="isSuperAdmin" @refresh="getTableData()" />
|
||
|
||
<!-- 内置模型 API Key 输入弹窗 -->
|
||
<el-dialog v-model="apiKeyDialogVisible" title="配置内置模型" width="500px" :close-on-click-modal="false">
|
||
<el-alert type="info" :closable="false" style="margin-bottom: 16px">
|
||
<template #title>
|
||
<div style="line-height: 1.6">
|
||
您选择的是内置模型，需要配置您自己的 API Key。<br />
|
||
系统将为您创建一个模型副本并设置为会话模型。
|
||
</div>
|
||
</template>
|
||
</el-alert>
|
||
<el-form :model="apiKeyForm" :rules="apiKeyRules" ref="apiKeyFormRef" label-width="100px">
|
||
<el-form-item label="模型名称" prop="modelName">
|
||
<el-input v-model="apiKeyForm.modelName" placeholder="请输入模型名称" />
|
||
</el-form-item>
|
||
<el-form-item label="API Key" prop="apiKey">
|
||
<el-input v-model="apiKeyForm.apiKey" type="password" show-password placeholder="请输入您的 API Key" />
|
||
</el-form-item>
|
||
</el-form>
|
||
<template #footer>
|
||
<el-button @click="apiKeyDialogVisible = false">取消</el-button>
|
||
<el-button type="primary" @click="handleCreatePrivateModelAndSetChat" :loading="creatingModel">确定</el-button>
|
||
</template>
|
||
</el-dialog>
|
||
</div>
|
||
</template>
|
||
|
||
<script setup lang="ts" name="digitalHumanModelModule">
|
||
import { defineAsyncComponent, reactive, onMounted, ref } from 'vue';
|
||
import { ElMessageBox, ElMessage, type FormInstance, type FormRules } from 'element-plus';
|
||
import {
|
||
getModelModuleList,
|
||
getModelTypeList,
|
||
deleteModelModule,
|
||
normalizeModelTypeOptions,
|
||
updateChatModel,
|
||
addModelModule,
|
||
} from '/@/api/settings/modelConfig/modelModule/index';
|
||
import { checkIsSuperAdmin } from '/@/api/system/user/index';
|
||
import { getApiErrorMessage } from '/@/utils/request';
|
||
|
||
const EditModule = defineAsyncComponent(() => import('/@/views/settings/modelConfig/modelModule/component/editModule.vue'));
|
||
|
||
const editModuleRef = ref();
|
||
const isSuperAdmin = ref(false); // 是否为管理员
|
||
const state = reactive({
|
||
modelTypes: [] as Array<{ id: number | string; label: string }>,
|
||
tableData: {
|
||
data: [],
|
||
total: 0,
|
||
loading: false,
|
||
param: {
|
||
pageNum: 1,
|
||
pageSize: 10,
|
||
modelName: '',
|
||
modelType: undefined as number | string | undefined,
|
||
},
|
||
},
|
||
});
|
||
|
||
// 内置模型 API Key 配置
|
||
const apiKeyDialogVisible = ref(false);
|
||
const apiKeyFormRef = ref<FormInstance>();
|
||
const apiKeyForm = reactive({
|
||
modelName: '',
|
||
apiKey: '',
|
||
});
|
||
const apiKeyRules: FormRules = {
|
||
modelName: [{ required: true, message: '请输入模型名称', trigger: 'blur' }],
|
||
apiKey: [{ required: true, message: '请输入 API Key', trigger: 'blur' }],
|
||
};
|
||
const creatingModel = ref(false);
|
||
const builtInModelToClone = ref<any>(null);
|
||
|
||
// 检查是否为管理员
|
||
const checkAdminStatus = async () => {
|
||
try {
|
||
const res: any = await checkIsSuperAdmin();
|
||
isSuperAdmin.value = res.data?.isSuperAdmin || false;
|
||
} catch {
|
||
isSuperAdmin.value = false;
|
||
}
|
||
};
|
||
|
||
// 判断是否为推理模型（只有推理模型才能设置为会话模型）
|
||
const normalizeModelTypeKey = (modelType: number | string | undefined | null) => {
|
||
if (modelType === undefined || modelType === null || modelType === '') {
|
||
return '';
|
||
}
|
||
const raw = String(modelType).trim();
|
||
const n = Number(raw);
|
||
return Number.isNaN(n) ? raw : String(n);
|
||
};
|
||
|
||
const isInferenceModel = (modelType: number | string | undefined | null) => {
|
||
if (modelType === undefined || modelType === null || modelType === '') {
|
||
return false;
|
||
}
|
||
// 查找模型类型标签，判断是否为"推理模型"
|
||
const modelTypeKey = normalizeModelTypeKey(modelType);
|
||
const typeInfo = state.modelTypes.find((t) => normalizeModelTypeKey(t.id) === modelTypeKey);
|
||
return typeInfo?.label === '推理模型' || modelTypeKey === '1' || modelTypeKey === '100';
|
||
};
|
||
|
||
// 设置为会话模型
|
||
const onSetChatModel = async (row: any) => {
|
||
// 判断是否是内置模型（isOwner === 0 表示管理员创建的内置模型）
|
||
if (row.isOwner === 0) {
|
||
// 内置模型，需要用户配置 API Key 创建副本
|
||
builtInModelToClone.value = row;
|
||
apiKeyForm.modelName = row.modelName;
|
||
apiKeyForm.apiKey = '';
|
||
apiKeyDialogVisible.value = true;
|
||
} else {
|
||
// 用户自己的模型，直接设置为会话模型
|
||
try {
|
||
await updateChatModel({
|
||
id: row.id!,
|
||
isChatModel: 1,
|
||
});
|
||
ElMessage.success('已设置为会话模型');
|
||
await getTableData();
|
||
} catch {
|
||
// 错误已由全局拦截器处理
|
||
}
|
||
}
|
||
};
|
||
|
||
// 创建私有模型并设置为会话模型
|
||
const handleCreatePrivateModelAndSetChat = async () => {
|
||
if (!apiKeyFormRef.value || !builtInModelToClone.value) return;
|
||
|
||
try {
|
||
await apiKeyFormRef.value.validate();
|
||
|
||
creatingModel.value = true;
|
||
|
||
// 基于内置模型创建新模型（继承原模型的所有配置，只替换 apiKey）
|
||
const builtInModel = builtInModelToClone.value;
|
||
|
||
// Parse headMsg to Record<string, string> - it might be stored as string or already as object
|
||
let headMsgRecord: Record<string, string> = {};
|
||
if (builtInModel.headMsg && typeof builtInModel.headMsg === 'string') {
|
||
// Try to parse as JSON first (new format stored as string)
|
||
try {
|
||
const parsed = JSON.parse(builtInModel.headMsg);
|
||
if (parsed && typeof parsed === 'object' && !Array.isArray(parsed)) {
|
||
Object.entries(parsed).forEach(([k, v]) => {
|
||
headMsgRecord[k] = String(v);
|
||
});
|
||
}
|
||
} catch {
|
||
// If JSON parse fails, parse as old format "key1:value1,key2:value2"
|
||
const pairs = builtInModel.headMsg.split(',');
|
||
pairs.forEach((pair: string) => {
|
||
const idx = pair.indexOf(':');
|
||
if (idx === -1) return;
|
||
const key = pair.slice(0, idx).trim();
|
||
const value = pair.slice(idx + 1).trim();
|
||
if (key) {
|
||
headMsgRecord[key] = value;
|
||
}
|
||
});
|
||
}
|
||
} else if (builtInModel.headMsg && typeof builtInModel.headMsg === 'object' && !Array.isArray(builtInModel.headMsg)) {
|
||
// Already an object
|
||
Object.entries(builtInModel.headMsg).forEach(([k, v]) => {
|
||
headMsgRecord[k] = String(v);
|
||
});
|
||
}
|
||
|
||
const createParams = {
|
||
modelName: apiKeyForm.modelName,
|
||
modelType: builtInModel.modelType,
|
||
baseUrl: builtInModel.baseUrl,
|
||
httpMethod: builtInModel.httpMethod || 'POST',
|
||
headMsg: headMsgRecord,
|
||
isPrivate: builtInModel.isPrivate ?? 1,
|
||
enabled: builtInModel.enabled ?? 1,
|
||
isChatModel: 1, // 设置为会话模型
|
||
callMode: builtInModel.callMode ?? builtInModel.isAsync ?? 0,
|
||
apiKey: apiKeyForm.apiKey,
|
||
form: builtInModel.form || {},
|
||
requestMapping: builtInModel.requestMapping || {},
|
||
requiredFields: Array.isArray(builtInModel.requiredFields) ? builtInModel.requiredFields : [],
|
||
firstFrame: String(builtInModel.firstFrame || ''),
|
||
lastFrame: String(builtInModel.lastFrame || ''),
|
||
responseMapping: builtInModel.responseMapping || {},
|
||
responseBody: builtInModel.responseBody || {},
|
||
tokenMapping: builtInModel.tokenMapping || '',
|
||
prompt: builtInModel.prompt || '',
|
||
maxConcurrency: builtInModel.maxConcurrency || 10,
|
||
queueLimit: builtInModel.queueLimit || 100,
|
||
timeoutSeconds: builtInModel.timeoutSeconds || 30,
|
||
expectedSeconds: builtInModel.expectedSeconds || 15,
|
||
retryTimes: builtInModel.retryTimes || 3,
|
||
retryQueueMaxSeconds: builtInModel.retryQueueMaxSeconds || 60,
|
||
autoCleanSeconds: builtInModel.autoCleanSeconds || 300,
|
||
remark: builtInModel.remark || '',
|
||
};
|
||
|
||
await addModelModule(createParams);
|
||
|
||
ElMessage.success('模型创建成功并已设置为会话模型');
|
||
|
||
// 关闭对话框
|
||
apiKeyDialogVisible.value = false;
|
||
|
||
// 刷新列表
|
||
await getTableData();
|
||
} catch (error: any) {
|
||
if (error !== 'cancel') {
|
||
ElMessage.error(getApiErrorMessage(error, '创建模型失败'));
|
||
}
|
||
} finally {
|
||
creatingModel.value = false;
|
||
}
|
||
};
|
||
|
||
const resolveModelTypeLabel = (modelType: number | string | undefined | null) => {
|
||
if (modelType === undefined || modelType === null || modelType === '') {
|
||
return '—';
|
||
}
|
||
const modelTypeKey = normalizeModelTypeKey(modelType);
|
||
const hit = state.modelTypes.find((t) => normalizeModelTypeKey(t.id) === modelTypeKey);
|
||
return hit?.label ?? String(modelType);
|
||
};
|
||
|
||
const loadModelTypes = async () => {
|
||
try {
|
||
const res: any = await getModelTypeList();
|
||
if (res.code === 0) {
|
||
state.modelTypes = normalizeModelTypeOptions(res);
|
||
}
|
||
} catch {
|
||
// 接口错误由 request 全局提示后端 message
|
||
}
|
||
};
|
||
|
||
// 初始化表格数据
|
||
const getTableData = async () => {
|
||
state.tableData.loading = true;
|
||
try {
|
||
const res: any = await getModelModuleList(state.tableData.param);
|
||
if (res.code === 0) {
|
||
state.tableData.data = res.data.list || [];
|
||
state.tableData.total = res.data.total || 0;
|
||
}
|
||
} catch {
|
||
// 接口错误由 request 全局提示后端 message
|
||
} finally {
|
||
state.tableData.loading = false;
|
||
}
|
||
};
|
||
|
||
// 打开新增模型模块弹窗
|
||
const onOpenAddModule = (type: string) => {
|
||
editModuleRef.value.openDialog(type);
|
||
};
|
||
|
||
// 打开修改模型模块弹窗
|
||
const onOpenEditModule = (type: string, row: any) => {
|
||
editModuleRef.value.openDialog(type, row);
|
||
};
|
||
|
||
// 删除模型模块
|
||
const onRowDel = (row: any) => {
|
||
ElMessageBox.confirm(`确定要删除模型配置：${row.modelName}？`, '提示', {
|
||
confirmButtonText: '确定',
|
||
cancelButtonText: '取消',
|
||
type: 'warning',
|
||
})
|
||
.then(async () => {
|
||
try {
|
||
await deleteModelModule(row.id);
|
||
ElMessage.success('删除成功');
|
||
getTableData();
|
||
} catch {
|
||
// 接口错误由 request 全局提示后端 message
|
||
}
|
||
})
|
||
.catch(() => {});
|
||
};
|
||
|
||
// 分页改变
|
||
const onHandleSizeChange = (val: number) => {
|
||
state.tableData.param.pageSize = val;
|
||
getTableData();
|
||
};
|
||
|
||
// 分页改变
|
||
const onHandleCurrentChange = (val: number) => {
|
||
state.tableData.param.pageNum = val;
|
||
getTableData();
|
||
};
|
||
|
||
// 页面加载时
|
||
onMounted(async () => {
|
||
await checkAdminStatus(); // 检查管理员状态
|
||
await loadModelTypes();
|
||
getTableData();
|
||
});
|
||
</script>
|
||
|
||
<style scoped lang="scss">
|
||
.text-muted {
|
||
color: var(--el-text-color-placeholder);
|
||
}
|
||
|
||
.system-user-container {
|
||
:deep(.el-card__body) {
|
||
display: flex;
|
||
flex-direction: column;
|
||
flex: 1;
|
||
overflow: auto;
|
||
.el-table {
|
||
flex: 1;
|
||
}
|
||
}
|
||
}
|
||
|
||
.action-buttons {
|
||
display: flex;
|
||
align-items: center;
|
||
gap: 8px;
|
||
|
||
.el-button {
|
||
margin: 0;
|
||
}
|
||
}
|
||
</style>
|