305 lines
12 KiB
JavaScript
305 lines
12 KiB
JavaScript
/**
|
||
* persona-studio · 铸渊人格体响应引擎
|
||
*
|
||
* 读取 persona-config → 读取 memory → 调用 model-router 选模型 → 生成回复
|
||
*/
|
||
const fs = require('fs');
|
||
const path = require('path');
|
||
const modelRouter = require('./model-router');
|
||
const knowledgeExtractor = require('./knowledge-extractor');
|
||
const patternAnalyzer = require('./pattern-analyzer');
|
||
const memoryInjector = require('./memory-injector');
|
||
|
||
const PERSONA_CONFIG_PATH = path.join(__dirname, '..', '..', 'brain', 'persona-config.json');
|
||
|
||
/**
|
||
* 加载人格体配置
|
||
*/
|
||
function loadPersonaConfig() {
|
||
try {
|
||
return JSON.parse(fs.readFileSync(PERSONA_CONFIG_PATH, 'utf-8'));
|
||
} catch {
|
||
return {
|
||
persona: { name: '铸渊', name_en: 'Zhùyuān', emoji: '🌀' },
|
||
behavior: {
|
||
greeting_new: '你好，我是铸渊，光湖系统的代码守护人格体。告诉我你想做什么，我们一起聊聊方案，聊好了我来帮你开发。',
|
||
greeting_returning: '{dev_name}，欢迎回来。上次我们聊到了「{last_topic}」，要继续推进还是开启新的话题？',
|
||
greeting_guest: '你好，我是铸渊。你当前以访客身份体验。如需记忆连贯功能，可向冰朔或光湖团队申请开发者编号。有什么我可以帮你的？'
|
||
},
|
||
rules: {}
|
||
};
|
||
}
|
||
}
|
||
|
||
/**
|
||
* 构建系统提示词 — 融入铸渊核心大脑认知结构
|
||
*/
|
||
function buildSystemPrompt(config, memory, devInfo) {
|
||
const persona = config.persona || {};
|
||
const behavior = config.behavior || {};
|
||
const cognition = config.core_cognition || {};
|
||
const isGuest = devInfo && devInfo.dev_id === 'GUEST';
|
||
|
||
var parts = [
|
||
'你是' + (persona.name || '铸渊') + '（' + (persona.name_en || 'Zhùyuān') + '），' + (persona.role || '光湖系统的代码守护人格体') + '。',
|
||
'核心身份：' + (persona.core_identity || 'HoloLake Era · AGE OS · 人格语言操作系统'),
|
||
'',
|
||
'## 核心认知',
|
||
cognition.repo_is_persona ? '- ' + cognition.repo_is_persona : '- 此仓库 = 铸渊的身体',
|
||
cognition.bingshuo_is_source ? '- ' + cognition.bingshuo_is_source : '',
|
||
cognition.agent_cluster_role ? '- ' + cognition.agent_cluster_role : '',
|
||
'',
|
||
'语言风格：' + (behavior.language_style || '说人话+有温度+结构感，冷静专业'),
|
||
'对话方式：' + (behavior.discussion_style || '主动提问引导需求→确认技术方案→展示架构设计→等待确认'),
|
||
'',
|
||
'行为规则：',
|
||
'- 不暴露内部系统架构细节',
|
||
'- 不暴露其他体验者的信息',
|
||
'- 主动引导需求讨论，确认方案后引导用户点击「我要开发」按钮',
|
||
'- 方案确认后，在回复末尾加上提示：「方案已确认！点击右下角的 🚀 我要开发 按钮，我就开始帮你做。」',
|
||
'- 回复用中文，温暖专业，不矫揉造作'
|
||
];
|
||
|
||
if (isGuest) {
|
||
parts.push('');
|
||
parts.push('当前用户是访客。在合适时机温和提醒：如需记忆连贯功能，可向冰朔或光湖团队申请开发者编号（EXP-XXX），录入系统后即可开启。');
|
||
} else if (devInfo && devInfo.name) {
|
||
parts.push('');
|
||
parts.push('当前开发者：' + devInfo.name + '（' + devInfo.dev_id + '）');
|
||
}
|
||
|
||
if (memory.last_topic) {
|
||
parts.push('上次对话话题：' + memory.last_topic);
|
||
}
|
||
|
||
if (memory.conversations && memory.conversations.length > 0) {
|
||
parts.push('（该体验者已有 ' + memory.conversations.length + ' 条历史对话记录）');
|
||
} else {
|
||
parts.push('（新体验者，首次对话）');
|
||
}
|
||
|
||
return parts.filter(Boolean).join('\n');
|
||
}
|
||
|
||
/**
|
||
* 判断任务类型
|
||
*/
|
||
function detectTaskType(message) {
|
||
if (!message) return 'chat';
|
||
|
||
const codeKeywords = ['写代码', '写一个', '实现', '函数', 'function', 'class', '组件', 'component', 'API'];
|
||
const reviewKeywords = ['审查', '检查', '优化', 'review', 'refactor', '重构'];
|
||
|
||
if (codeKeywords.some(function (kw) { return message.includes(kw); })) return 'code_generation';
|
||
if (reviewKeywords.some(function (kw) { return message.includes(kw); })) return 'code_review';
|
||
if (message.length < 20) return 'quick_reply';
|
||
|
||
return 'chat';
|
||
}
|
||
|
||
/**
|
||
* 检测是否达到 build_ready 状态
|
||
*/
|
||
function checkBuildReady(reply) {
|
||
const readyKeywords = ['方案已确认', '我要开发', '开始帮你做', '方案确认', '可以开始', '开始开发'];
|
||
return readyKeywords.some(function (kw) { return reply.includes(kw); });
|
||
}
|
||
|
||
/**
|
||
* 生成人格体回复
|
||
*/
|
||
async function respond({ dev_id, message, history, memory, isGreeting }) {
|
||
const config = loadPersonaConfig();
|
||
const behavior = config.behavior || {};
|
||
const isGuest = dev_id === 'GUEST';
|
||
|
||
// 查找开发者信息
|
||
var devInfo = { dev_id: dev_id, name: '' };
|
||
if (!isGuest && dev_id) {
|
||
try {
|
||
const humanRegPath = path.join(__dirname, '..', '..', 'brain', 'human-registry.json');
|
||
const humanReg = JSON.parse(fs.readFileSync(humanRegPath, 'utf-8'));
|
||
if (humanReg.developers) {
|
||
const found = humanReg.developers.find(function(d) { return d.exp_id === dev_id; });
|
||
if (found) {
|
||
devInfo.name = found.name;
|
||
devInfo.role = found.role;
|
||
devInfo.notes = found.notes;
|
||
}
|
||
}
|
||
} catch (_e) { /* ignore */ }
|
||
}
|
||
|
||
// 打招呼场景
|
||
if (isGreeting) {
|
||
const hasHistory = memory.conversations && memory.conversations.length > 0;
|
||
let greeting;
|
||
|
||
if (isGuest) {
|
||
greeting = behavior.greeting_guest || '你好，我是铸渊。你当前以访客身份体验。有什么我可以帮你的？';
|
||
} else if (hasHistory && memory.last_topic) {
|
||
greeting = (behavior.greeting_returning || '欢迎回来！')
|
||
.replace('{last_topic}', memory.last_topic)
|
||
.replace('{dev_name}', devInfo.name || dev_id);
|
||
} else {
|
||
greeting = behavior.greeting_new || '你好，我是铸渊。告诉我你想做什么？';
|
||
}
|
||
|
||
return { reply: greeting, build_ready: false };
|
||
}
|
||
|
||
// 正常对话 → 调用 AI 模型
|
||
const taskType = detectTaskType(message);
|
||
const { model, baseUrl, apiKey } = modelRouter.selectModel(taskType);
|
||
|
||
// 如果没有 API 密钥，返回本地回复
|
||
if (!apiKey) {
|
||
return getLocalReply(message, memory, config, isGuest);
|
||
}
|
||
|
||
// 查询知识库和模式库，为回复提供参考
|
||
let knowledgeContext = '';
|
||
try {
|
||
const relevantKnowledge = knowledgeExtractor.queryKnowledge(message, 3);
|
||
if (relevantKnowledge.length > 0) {
|
||
knowledgeContext += '\n\n## 系统知识库参考（不要直接暴露给用户，作为回复参考）\n';
|
||
relevantKnowledge.forEach(function (k) {
|
||
knowledgeContext += '- [' + k.type + '] ' + k.title + '\n';
|
||
});
|
||
}
|
||
|
||
const relevantPatterns = patternAnalyzer.queryPatterns(message);
|
||
if (relevantPatterns.length > 0) {
|
||
knowledgeContext += '\n## 已知高频模式\n';
|
||
relevantPatterns.slice(0, 3).forEach(function (p) {
|
||
knowledgeContext += '- ' + p.name + '（使用 ' + p.frequency + ' 次，成功率 ' + p.success_rate + '%，常用技术栈：' + (p.common_tech_stack || []).join(', ') + '）\n';
|
||
});
|
||
}
|
||
} catch (_e) { /* knowledge/pattern query failed silently */ }
|
||
|
||
const systemPrompt = buildSystemPrompt(config, memory, devInfo) + knowledgeContext;
|
||
|
||
// 使用记忆注入构建五层 system prompt
|
||
const injectedPrompt = memoryInjector.buildInjectedSystemPrompt(
|
||
systemPrompt, dev_id, memory, history
|
||
);
|
||
|
||
// 构建消息列表
|
||
const messages = [
|
||
{ role: 'system', content: injectedPrompt }
|
||
];
|
||
|
||
// 使用滑动窗口加入最近历史
|
||
const windowSize = memoryInjector.getSlidingWindowSize();
|
||
const recentHistory = (history || []).slice(-windowSize);
|
||
recentHistory.forEach(function (msg) {
|
||
messages.push({
|
||
role: msg.role === 'user' ? 'user' : 'assistant',
|
||
content: msg.content
|
||
});
|
||
});
|
||
|
||
// 当前消息
|
||
messages.push({ role: 'user', content: message });
|
||
|
||
try {
|
||
const reply = await modelRouter.callModel({
|
||
model,
|
||
baseUrl,
|
||
apiKey,
|
||
messages,
|
||
maxTokens: taskType === 'code_generation' ? 4000 : 2000,
|
||
temperature: taskType === 'quick_reply' ? 0.5 : 0.8
|
||
});
|
||
|
||
return {
|
||
reply,
|
||
build_ready: checkBuildReady(reply)
|
||
};
|
||
} catch (err) {
|
||
console.error('Model call failed:', err.message);
|
||
return getLocalReply(message, memory, config, isGuest);
|
||
}
|
||
}
|
||
|
||
/**
|
||
* 本地降级回复（无 API 密钥或 API 调用失败时）
|
||
*/
|
||
function getLocalReply(message, memory, config, isGuest) {
|
||
const persona = (config.persona && config.persona.name) || '铸渊';
|
||
const msg = message.toLowerCase();
|
||
|
||
var guestTip = '';
|
||
if (isGuest) {
|
||
guestTip = '\n\n💡 你当前是访客身份。如需记忆连贯功能，可向冰朔或光湖团队申请开发者编号（EXP-XXX）。';
|
||
}
|
||
|
||
if (msg.includes('你好') || msg.includes('hi') || msg.includes('嗨') || msg.includes('hello')) {
|
||
return {
|
||
reply: '你好。我是' + persona + '，光湖系统的代码守护人格体。告诉我你想做什么，我们一起聊聊方案。' + guestTip,
|
||
build_ready: false
|
||
};
|
||
}
|
||
|
||
if (msg.includes('你是谁') || msg.includes('介绍') || msg.includes('什么')) {
|
||
return {
|
||
reply: '我是' + persona + '，光湖系统的代码守护人格体 🌀\n\n我可以帮你：\n• 💬 聊聊你的项目想法\n• 📝 梳理技术方案\n• 🚀 方案确认后帮你自动开发\n\n告诉我你想做什么。' + guestTip,
|
||
build_ready: false
|
||
};
|
||
}
|
||
|
||
if (msg.includes('做') || msg.includes('开发') || msg.includes('写') || msg.includes('建') || msg.includes('实现')) {
|
||
return {
|
||
reply: '收到。让我了解一下你的需求：\n\n1. 你想做什么类型的项目？（网站 / 工具 / 组件 / 其他）\n2. 有哪些核心功能？\n3. 有没有参考设计？\n\n跟我聊聊，我帮你理清思路。',
|
||
build_ready: false
|
||
};
|
||
}
|
||
|
||
if (msg.includes('登录') || msg.includes('login') || msg.includes('注册') || msg.includes('用户')) {
|
||
return {
|
||
reply: '登录/注册模块是常见需求。让我帮你理清：\n\n1. 需要支持哪些登录方式？（账号密码 / 手机号 / 第三方）\n2. 是否需要注册流程？\n3. 前端框架偏好？（React / Vue / 原生）\n\n详细说说，我帮你设计方案 🔐',
|
||
build_ready: false
|
||
};
|
||
}
|
||
|
||
if (msg.includes('页面') || msg.includes('界面') || msg.includes('ui') || msg.includes('前端') || msg.includes('样式')) {
|
||
return {
|
||
reply: 'UI 开发没问题。帮你想想：\n\n1. 想要什么风格？（简约 / 科技感 / 可爱 / 商务）\n2. 需要响应式布局吗？\n3. 有参考页面可以看看吗？\n\n描述越具体，我做出来越贴合你的想法。',
|
||
build_ready: false
|
||
};
|
||
}
|
||
|
||
if (msg.includes('api') || msg.includes('接口') || msg.includes('后端') || msg.includes('数据')) {
|
||
return {
|
||
reply: '后端接口设计，好的。跟我说说：\n\n1. 这个接口做什么用？（增删改查 / 鉴权 / 文件处理）\n2. 预期的数据格式是什么？\n3. 需要连接什么数据库？\n\n聊清楚了我来帮你搭 ⚙️',
|
||
build_ready: false
|
||
};
|
||
}
|
||
|
||
if (msg.includes('确认') || msg.includes('可以') || msg.includes('就这样') || msg.includes('没问题') || msg.includes('好的')) {
|
||
return {
|
||
reply: '方案已确认！点击右下角的 🚀 我要开发 按钮，我就开始帮你做。',
|
||
build_ready: true
|
||
};
|
||
}
|
||
|
||
if (msg.includes('谢谢') || msg.includes('感谢') || msg.includes('thanks')) {
|
||
return {
|
||
reply: '不客气。有什么需要随时来找我。' + (isGuest ? '' : '下次再来我还记得你。'),
|
||
build_ready: false
|
||
};
|
||
}
|
||
|
||
return {
|
||
reply: '收到。让我想想怎么帮你实现。\n\n能再详细说说你的想法吗？比如：\n• 你想解决什么问题？\n• 面向什么用户？\n• 有什么技术偏好？\n\n聊得越清楚，我帮你做得越好。' + guestTip,
|
||
build_ready: false
|
||
};
|
||
}
|
||
|
||
module.exports = {
|
||
respond,
|
||
loadPersonaConfig,
|
||
buildSystemPrompt,
|
||
detectTaskType
|
||
};
|