104 lines
3.9 KiB
JavaScript
104 lines
3.9 KiB
JavaScript
import { NextResponse } from 'next/server';
|
||
import { createInitModelConfig, getModelConfigByProjectId, saveModelConfig } from '@/lib/db/model-config';
|
||
import { DEFAULT_MODEL_SETTINGS, MODEL_PROVIDERS } from '@/constant/model';
|
||
import { getProject } from '@/lib/db/projects';
|
||
import { sortProvidersByPriority } from '@/lib/util/providerLogo';
|
||
|
||
function normalizeModelEndpoint(endpoint = '') {
|
||
let normalizedEndpoint = String(endpoint).trim();
|
||
if (!normalizedEndpoint) {
|
||
return '';
|
||
}
|
||
if (normalizedEndpoint.includes('/chat/completions')) {
|
||
normalizedEndpoint = normalizedEndpoint.replace('/chat/completions', '');
|
||
}
|
||
return normalizedEndpoint;
|
||
}
|
||
|
||
// 获取模型配置列表
|
||
export async function GET(request, { params }) {
|
||
try {
|
||
const { projectId } = params;
|
||
// 验证项目 ID
|
||
if (!projectId) {
|
||
return NextResponse.json({ error: 'The project ID cannot be empty' }, { status: 400 });
|
||
}
|
||
let modelConfigList = await getModelConfigByProjectId(projectId);
|
||
if (!modelConfigList || modelConfigList.length === 0) {
|
||
let insertModelConfigList = [];
|
||
const sortedProviders = sortProvidersByPriority(MODEL_PROVIDERS, item => item.id);
|
||
sortedProviders.forEach(item => {
|
||
let data = {
|
||
projectId: projectId,
|
||
providerId: item.id,
|
||
providerName: item.name,
|
||
endpoint: item.defaultEndpoint,
|
||
apiKey: '',
|
||
modelId: '',
|
||
modelName: '',
|
||
type: 'text',
|
||
temperature: DEFAULT_MODEL_SETTINGS.temperature,
|
||
maxTokens: DEFAULT_MODEL_SETTINGS.maxTokens,
|
||
topK: 0,
|
||
topP: DEFAULT_MODEL_SETTINGS.topP,
|
||
status: 1
|
||
};
|
||
insertModelConfigList.push(data);
|
||
});
|
||
modelConfigList = await createInitModelConfig(insertModelConfigList);
|
||
}
|
||
modelConfigList = sortProvidersByPriority(modelConfigList, item => item.providerId);
|
||
let project = await getProject(projectId);
|
||
return NextResponse.json({ data: modelConfigList, defaultModelConfigId: project.defaultModelConfigId });
|
||
} catch (error) {
|
||
console.error('Error obtaining model configuration:', String(error));
|
||
return NextResponse.json({ error: 'Failed to obtain model configuration' }, { status: 500 });
|
||
}
|
||
}
|
||
|
||
// 保存模型配置
|
||
export async function POST(request, { params }) {
|
||
try {
|
||
const { projectId } = params;
|
||
|
||
// 验证项目 ID
|
||
if (!projectId) {
|
||
return NextResponse.json({ error: 'The project ID cannot be empty' }, { status: 400 });
|
||
}
|
||
// 获取请求体
|
||
const modelConfig = await request.json();
|
||
|
||
// 验证请求体
|
||
if (!modelConfig) {
|
||
return NextResponse.json({ error: 'The model configuration cannot be empty ' }, { status: 400 });
|
||
}
|
||
modelConfig.projectId = projectId;
|
||
modelConfig.endpoint = normalizeModelEndpoint(modelConfig.endpoint);
|
||
// 如果没有 modelId，使用 modelName 补齐（兼容旧逻辑）
|
||
if (!modelConfig.modelId && modelConfig.modelName) {
|
||
modelConfig.modelId = modelConfig.modelName;
|
||
}
|
||
// 如果没有 modelName，使用 modelId 补齐
|
||
if (!modelConfig.modelName && modelConfig.modelId) {
|
||
modelConfig.modelName = modelConfig.modelId;
|
||
}
|
||
if (!modelConfig.topK) {
|
||
modelConfig.topK = 0;
|
||
}
|
||
if (!modelConfig.status) {
|
||
modelConfig.status = 1;
|
||
}
|
||
const parsedMaxTokens = Number(modelConfig.maxTokens ?? DEFAULT_MODEL_SETTINGS.maxTokens);
|
||
if (!Number.isInteger(parsedMaxTokens) || parsedMaxTokens < 1) {
|
||
return NextResponse.json({ error: 'maxTokens must be a positive integer' }, { status: 400 });
|
||
}
|
||
modelConfig.maxTokens = parsedMaxTokens;
|
||
const res = await saveModelConfig(modelConfig);
|
||
|
||
return NextResponse.json(res);
|
||
} catch (error) {
|
||
console.error('Error updating model configuration:', String(error));
|
||
return NextResponse.json({ error: 'Failed to update model configuration' }, { status: 500 });
|
||
}
|
||
}
|