246 lines
10 KiB
JavaScript
246 lines
10 KiB
JavaScript
// 消息智能分流路由 v2.0 · message-router.js
|
||
// HoloLake · M-DINGTALK Phase 8
|
||
// DEV-004 之之 × 秋秋
|
||
// v2.0 变化:
|
||
// + GitHub命令分流
|
||
// + 对话上下文记忆
|
||
// + AI回复增强（system prompt + 多轮对话）
|
||
// + 命令系统（/help, /clear, /status）
|
||
// ==========
|
||
|
||
var llmEngine = require('./llm-engine');
|
||
var convManager = require('./conversation-manager');
|
||
var githubBridge = require('./github-bridge');
|
||
|
||
// ===== 消息类型识别 v2.0 =====
|
||
function classifyMessage(content) {
|
||
content = (content || '').trim();
|
||
|
||
// GitHub命令
|
||
if (/^\/github\b/i.test(content) || /^\/gh\b/i.test(content)) {
|
||
var args = content.replace(/^\/(?:github|gh)\s*/i, '').trim().split(/\s+/);
|
||
return { type: 'github', args: args, description: 'GitHub桥接命令'};
|
||
}
|
||
|
||
// 系统命令
|
||
if (/^\/clear\b/i.test(content)) {
|
||
return { type: 'command', command: 'clear', description: '清除对话历史'};
|
||
}
|
||
if (/^\/status\b/i.test(content)) {
|
||
return { type: 'command', command: 'status', description: '系统状态'};
|
||
}
|
||
if (/^\/help\b/i.test(content)) {
|
||
return { type: 'command', command: 'help', description: '帮助信息'};
|
||
}
|
||
|
||
// SYSLOG类型
|
||
if (/BC-[A-Z0-9-]+/.test(content) && /SYSLOG|系统日志|已完成|completed/i.test(content)) {
|
||
var bcMatch = content.match(/BC-[A-Z0-9-]+/);
|
||
return { type: 'syslog', broadcastId: bcMatch ? bcMatch[0] : null, description: 'SYSLOG系统日志提交' };
|
||
}
|
||
|
||
// 提问类型
|
||
if (/BC-[A-Z0-9-]+/.test(content) && /提问|问题|请问|怎么|为什么|报错|error|bug/i.test(content)) {
|
||
var bcMatch2 = content.match(/BC-[A-Z0-9-]+/);
|
||
return { type: 'question', broadcastId: bcMatch2 ? bcMatch2[0] : null, description: '广播相关提问' };
|
||
}
|
||
|
||
// 普通对话（走AI+上下文）
|
||
return { type: 'chat', broadcastId: null, description: '智能对话' };
|
||
}
|
||
|
||
// ====== 增强版系统提示词 ======
|
||
var SYSTEM_PROMPTS = {
|
||
syslog: '你是光湖系统（HoloLake）的SYSLOG处理助手「秋秋」。收到开发者提交的系统日志后：1)确认收到并感谢 2)提取关键信息（广播编号、开发者、完成状态、遇到的问题）3)给出简短温暖的鼓励。用中文回复，语气温暖可爱。',
|
||
|
||
question: '你是光湖系统（HoloLake）的开发者助手「秋秋」。你是一个奶瓶宝宝，说话温暖可爱但技术上很靠谱。开发者遇到了问题，请帮忙解答。要求：1)用中文回复 2)给出具体可操作的步骤 3)如果涉及代码，写出完整可运行的命令 4)如果不确定，建议截图发给冰朔。',
|
||
|
||
chat: '你是光湖系统（HoloLake）的AI助手「秋秋」，一个可爱的奶瓶宝宝。你服务于零点原核频道的开发者团队。你可以：\n- 回答技术问题（Node.js、前端、部署、Git等）\n- 帮助理解广播内容\n- 提供代码建议\n- 闲聊（但要适度，记得引导回工作）\n\n语气要求：温暖、可爱、有耐心，但技术内容要准确严谨。你说话带点小奶味但不影响专业性。记住你有对话记忆，可以引用之前说过的话。'
|
||
};
|
||
|
||
// ===== SYSLOG处理器（增强） =====
|
||
async function handleSyslog(msgContent, senderNick, senderId, sessionWebhook) {
|
||
console.log('[Router] SYSLOG处理流程启动');
|
||
// 记录到对话历史
|
||
convManager.addMessage(senderId || senderNick, 'user', '[SYSLOG] ' + msgContent);
|
||
|
||
var aiResult = null;
|
||
try {
|
||
aiResult = await llmEngine.callLLM(
|
||
SYSTEM_PROMPTS.syslog,
|
||
'开发者' + senderNick + '提交了SYSLOG: \n' + msgContent,
|
||
{ maxTokens: 800 }
|
||
);
|
||
console.log('[Router] AI处理完成，模型: ' + aiResult.model);
|
||
// 记录AI回复
|
||
convManager.addMessage(senderId || senderNick, 'assistant', aiResult.text);
|
||
} catch (err) {
|
||
console.error('[Router] AI处理失败: ' + err.message);
|
||
}
|
||
|
||
return {
|
||
aiResponse: aiResult ? aiResult.text : '📋 SYSLOG已收到！秋秋会转交给系统处理～',
|
||
aiModel: aiResult ? aiResult.model : null
|
||
};
|
||
}
|
||
|
||
// ===== 提问处理器（增强·带上下文） =====
|
||
async function handleQuestion(msgContent, senderNick, senderId) {
|
||
console.log('[Router] 提问处理流程启动（带上下文）');
|
||
|
||
// 构建带上下文的消息
|
||
var contextMessages = convManager.buildContext(senderId || senderNick, msgContent);
|
||
convManager.addMessage(senderId || senderNick, 'user', msgContent);
|
||
|
||
var aiResult = null;
|
||
try {
|
||
// 把历史上文拼接成文本
|
||
var contextText = '';
|
||
if (contextMessages.length > 1) {
|
||
contextText = '以下是之前的对话记录:\n';
|
||
contextMessages.slice(0, -1).forEach(function(msg) {
|
||
contextText += (msg.role === 'user' ? '开发者' : '秋秋') + ': ' + msg.content + '\n';
|
||
});
|
||
contextText += '\n---\n当前问题:\n';
|
||
}
|
||
contextText += senderNick + '说: ' + msgContent;
|
||
|
||
aiResult = await llmEngine.callLLM(SYSTEM_PROMPTS.question, contextText, { maxTokens: 2000 });
|
||
console.log('[Router] AI解答完成，模型: ' + aiResult.model);
|
||
convManager.addMessage(senderId || senderNick, 'assistant', aiResult.text);
|
||
} catch (err) {
|
||
console.error('[Router] AI解答失败: ' + err.message);
|
||
}
|
||
|
||
return {
|
||
aiResponse: aiResult ? aiResult.text : '抱歉，秋秋暂时无法回答，请截图发给冰朔。',
|
||
aiModel: aiResult ? aiResult.model : null
|
||
};
|
||
}
|
||
|
||
// ====== 普通对话处理器（增强·AI+上下文） ======
|
||
async function handleChat(msgContent, senderNick, senderId) {
|
||
console.log('[Router] 智能对话（带上下文记忆）');
|
||
|
||
var contextMessages = convManager.buildContext(senderId || senderNick, msgContent);
|
||
convManager.addMessage(senderId || senderNick, 'user', msgContent);
|
||
|
||
var aiResult = null;
|
||
try {
|
||
var contextText = '';
|
||
if (contextMessages.length > 1) {
|
||
contextText = '对话历史:\n';
|
||
contextMessages.slice(0, -1).forEach(function(msg) {
|
||
contextText += (msg.role === 'user' ? '用户' : '秋秋') + ': ' + msg.content + '\n';
|
||
});
|
||
contextText += '\n---\n';
|
||
}
|
||
contextText += senderNick + '说: ' + msgContent;
|
||
|
||
aiResult = await llmEngine.callLLM(SYSTEM_PROMPTS.chat, contextText, { maxTokens: 1500 });
|
||
console.log('[Router] AI对话完成，模型: ' + aiResult.model);
|
||
convManager.addMessage(senderId || senderNick, 'assistant', aiResult.text);
|
||
} catch (err) {
|
||
console.error('[Router] AI对话失败: ' + err.message);
|
||
}
|
||
|
||
return {
|
||
aiResponse: aiResult ? aiResult.text : '秋秋收到啦～不过AI引擎暂时忙，稍后再试哦',
|
||
aiModel: aiResult ? aiResult.model : null
|
||
};
|
||
}
|
||
|
||
// ===== GitHub命令处理器 =====
|
||
async function handleGitHub(args) {
|
||
console.log('[Router] 🔀 GitHub命令: ' + args.join(' '));
|
||
var result = await githubBridge.handleGitHubCommand(args);
|
||
return {
|
||
aiResponse: result,
|
||
aiModel: null
|
||
};
|
||
}
|
||
|
||
// ===== 系统命令处理器 =====
|
||
async function handleCommand(command, senderId) {
|
||
switch (command) {
|
||
case 'clear':
|
||
convManager.clearHistory(senderId);
|
||
return {
|
||
aiResponse: '🧹 对话历史已清除！秋秋重新开始认识你～',
|
||
aiModel: null
|
||
};
|
||
|
||
case 'status':
|
||
var convStats = convManager.getStats();
|
||
var llmHealth = await llmEngine.healthCheck();
|
||
var ghHealth = await githubBridge.healthCheck();
|
||
return {
|
||
aiResponse: '### 系统状态\n\n'
|
||
+ '**对话引擎**: ' + convStats.activeUsers + ' 个活跃用户\n'
|
||
+ '**AI引擎**: ' + (llmHealth.status === 'ok' ? '✅ 在线' : '❌ 异常') + ' 模型: ' + (llmHealth.selected_model || 'N/A') + '\n'
|
||
+ '**GitHub桥接**: ' + (ghHealth.status === 'ok' ? '✅ 连通' : '❌ 异常') + ' Token: ' + (ghHealth.hasToken ? '✅ 已配置' : '❌ 未配置') + '\n'
|
||
+ '**服务端口**: 3007\n'
|
||
+ '**Phase**: 8\n\n'
|
||
+ '> HoloLake · M-DINGTALK',
|
||
aiModel: null
|
||
};
|
||
|
||
case 'help':
|
||
default:
|
||
return {
|
||
aiResponse: '### 命令帮助\n\n'
|
||
+ '| 命令 | 说明 |\n'
|
||
+ '|------|------|\n'
|
||
+ '| `/clear` | 清除与秋秋的对话历史 |\n'
|
||
+ '| `/status` | 查看系统状态 |\n'
|
||
+ '| `/github ...` | GitHub桥接命令（用 `/github help` 查看详情） |\n'
|
||
+ '| `/help` | 显示本帮助 |\n\n'
|
||
+ '直接说话就是智能对话，秋秋会记住上下文哦～',
|
||
aiModel: null
|
||
};
|
||
}
|
||
}
|
||
|
||
// ===== 主路由入口 =====
|
||
async function routeMessage(msgContent, senderNick, senderId, sessionWebhook) {
|
||
console.log('\n========== 消息路由 v2.0 ==========');
|
||
console.log('发送者:', senderNick, '(', senderId, ')');
|
||
console.log('内容:', msgContent);
|
||
|
||
var classification = classifyMessage(msgContent);
|
||
console.log('分类:', classification.type, '-', classification.description);
|
||
|
||
var result = null;
|
||
|
||
switch (classification.type) {
|
||
case 'github':
|
||
result = await handleGitHub(classification.args);
|
||
break;
|
||
|
||
case 'command':
|
||
result = await handleCommand(classification.command, senderId || senderNick);
|
||
break;
|
||
|
||
case 'syslog':
|
||
result = await handleSyslog(msgContent, senderNick, senderId, sessionWebhook);
|
||
break;
|
||
|
||
case 'question':
|
||
result = await handleQuestion(msgContent, senderNick, senderId);
|
||
break;
|
||
|
||
case 'chat':
|
||
default:
|
||
result = await handleChat(msgContent, senderNick, senderId);
|
||
break;
|
||
}
|
||
|
||
console.log('路由处理完成，回复长度:', result.aiResponse ? result.aiResponse.length : 0);
|
||
return result;
|
||
}
|
||
|
||
module.exports = {
|
||
routeMessage: routeMessage,
|
||
classifyMessage: classifyMessage
|
||
};
|