81 lines
1.9 KiB
JavaScript
81 lines
1.9 KiB
JavaScript
|
/**
|
|||
|
|
* connectors/model-router — 模型调用路由
|
|||
|
|
*
|
|||
|
|
* 职责：
|
|||
|
|
* - 提供统一的模型调用入口
|
|||
|
|
* - 支持多模型切换（通过 backend-integration/api-proxy.js）
|
|||
|
|
* - 任务分发至合适的模型端点
|
|||
|
|
*
|
|||
|
|
* 调用方式：
|
|||
|
|
* node connectors/model-router [status]
|
|||
|
|
*/
|
|||
|
|
|
|||
|
|
const path = require('path');
|
|||
|
|
const fs = require('fs');
|
|||
|
|
|
|||
|
|
const ROOT = path.resolve(__dirname, '../..');
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* 模型配置注册表
|
|||
|
|
*/
|
|||
|
|
const MODEL_REGISTRY = {
|
|||
|
|
'default': {
|
|||
|
|
name: 'default',
|
|||
|
|
endpoint: 'http://localhost:3721/api/chat',
|
|||
|
|
description: 'AI Chat API 代理（api-proxy.js）'
|
|||
|
|
},
|
|||
|
|
'persona': {
|
|||
|
|
name: 'persona',
|
|||
|
|
endpoint: 'http://localhost:3002/api/ps/chat',
|
|||
|
|
description: '人格工作室 API'
|
|||
|
|
}
|
|||
|
|
};
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* 获取模型配置
|
|||
|
|
*/
|
|||
|
|
function getModel(modelName = 'default') {
|
|||
|
|
return MODEL_REGISTRY[modelName] || MODEL_REGISTRY['default'];
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* 列出所有可用模型
|
|||
|
|
*/
|
|||
|
|
function listModels() {
|
|||
|
|
return Object.values(MODEL_REGISTRY);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* 检查模型路由状态
|
|||
|
|
*/
|
|||
|
|
function status() {
|
|||
|
|
console.log('🤖 模型路由状态:');
|
|||
|
|
console.log('═'.repeat(40));
|
|||
|
|
|
|||
|
|
const models = listModels();
|
|||
|
|
for (const model of models) {
|
|||
|
|
console.log(` 📌 ${model.name}`);
|
|||
|
|
console.log(` 端点: ${model.endpoint}`);
|
|||
|
|
console.log(` 说明: ${model.description}`);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 检查 api-proxy 配置
|
|||
|
|
const proxyPath = path.join(ROOT, 'backend-integration/api-proxy.js');
|
|||
|
|
const proxyExists = fs.existsSync(proxyPath);
|
|||
|
|
console.log(`\n ${proxyExists ? '✅' : '❌'} api-proxy.js 存在`);
|
|||
|
|
|
|||
|
|
// 检查 persona-studio 配置
|
|||
|
|
const psPath = path.join(ROOT, 'persona-studio/backend/server.js');
|
|||
|
|
const psExists = fs.existsSync(psPath);
|
|||
|
|
console.log(` ${psExists ? '✅' : '❌'} persona-studio server.js 存在`);
|
|||
|
|
|
|||
|
|
return { models, proxy: proxyExists, persona_studio: psExists };
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// CLI 入口
|
|||
|
|
if (require.main === module) {
|
|||
|
|
status();
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
module.exports = { getModel, listModels, status, MODEL_REGISTRY };
|