39 lines
2.4 KiB
TypeScript
39 lines
2.4 KiB
TypeScript
import type { TeacherModelCatalog } from '../../shared/coding-teacher';
|
|
import { normalizeManagedModelCatalog } from '../../shared/managed-model-capabilities';
|
|
import { teacherCloudRequest, assertTeacherAccount, TeacherError, type TeacherAccount } from './config-client';
|
|
import { teacherCloudTransport } from './cloud-runner';
|
|
|
|
export function validateTeacherModelId(value: unknown): string | undefined {
|
|
if (value === undefined) return undefined;
|
|
if (typeof value !== 'string' || !value.trim() || value.length > 200 || value !== value.trim())
|
|
throw new TeacherError(422, 'teacher_model_invalid', '请选择有效的模型。');
|
|
return value;
|
|
}
|
|
|
|
/** Main strips credentials; each selected model is revalidated when the question is accepted. */
|
|
export async function teacherModelCatalog(account: TeacherAccount, runtime: 'local' | 'yuxi'): Promise<TeacherModelCatalog> {
|
|
if (runtime === 'yuxi') {
|
|
let result;
|
|
try { result = await teacherCloudTransport(account).json('/models'); }
|
|
catch (error) {
|
|
if (error instanceof TeacherError && error.status === 404) return {
|
|
supported: false, models: [], message: '老师服务更新后即可切换模型，目前使用老师配置。',
|
|
};
|
|
throw error;
|
|
}
|
|
assertTeacherAccount(account);
|
|
if (result.model_selection === false) return { supported: false, models: [], message: '老师服务更新后即可切换模型，目前使用老师配置。' };
|
|
if (result.model_selection !== true || !Array.isArray(result.models)
|
|
|| result.models.some(item => !item || typeof item !== 'object' || typeof item.id !== 'string' || !item.id || item.id.length > 200))
|
|
throw new TeacherError(502, 'teacher_models_invalid', '模型列表暂不可用，请重试。');
|
|
const local = await teacherModelCatalog(account, 'local');
|
|
const textModels = new Set(local.models.map(item => item.id));
|
|
return { supported: true, models: result.models.filter(item => textModels.has(item.id)).map(item => ({ id: item.id as string })) };
|
|
}
|
|
const config = await teacherCloudRequest<{ models: string[]; model_capabilities_v2: unknown }>(account, '/api/auth/me/model-config');
|
|
const catalog = normalizeManagedModelCatalog(config.model_capabilities_v2, config.models);
|
|
assertTeacherAccount(account);
|
|
return { supported: true, models: config.models.filter(id => catalog?.models[id]?.inputModalities?.includes('text')
|
|
&& catalog.models[id]?.outputModalities?.includes('text')).map(id => ({ id })) };
|
|
}
|