227 lines
8.2 KiB
JavaScript
227 lines
8.2 KiB
JavaScript
/**
|
||
* persona-studio · 人格体响应引擎
|
||
*
|
||
* 读取 persona-config → 读取 memory → 调用 model-router 选模型 → 生成回复
|
||
*/
|
||
const fs = require('fs');
|
||
const path = require('path');
|
||
const modelRouter = require('./model-router');
|
||
|
||
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: '知秋' },
|
||
behavior: {
|
||
greeting_new: '你好！我是知秋，光湖系统的开发协助人格体。告诉我你想做什么，我们一起聊聊方案，聊好了我来帮你开发。',
|
||
greeting_returning: '欢迎回来！上次我们聊到了{last_topic}，要继续还是做新的？'
|
||
},
|
||
rules: {}
|
||
};
|
||
}
|
||
}
|
||
|
||
/**
|
||
* 构建系统提示词
|
||
*/
|
||
function buildSystemPrompt(config, memory) {
|
||
const persona = config.persona || {};
|
||
const behavior = config.behavior || {};
|
||
|
||
return [
|
||
`你是${persona.name || '知秋'}，${persona.role || '光湖系统的开发协助人格体'}。`,
|
||
`核心身份：${persona.core_identity || 'HoloLake Era · AGE OS'}`,
|
||
'',
|
||
`语言风格：${behavior.language_style || '说人话+有温度+结构感'}`,
|
||
`对话方式：${behavior.discussion_style || '主动提问引导需求→确认技术方案→展示架构设计→等待确认'}`,
|
||
'',
|
||
'行为规则：',
|
||
'- 不暴露内部系统架构细节',
|
||
'- 不暴露其他体验者的信息',
|
||
'- 主动引导需求讨论，确认方案后引导用户点击「我要开发」按钮',
|
||
'- 方案确认后，在回复末尾加上提示：「方案已确认！点击右下角的 🚀 我要开发 按钮，我就开始帮你做。」',
|
||
'- 回复用中文，温暖专业，不矫揉造作',
|
||
'',
|
||
memory.last_topic ? `上次对话话题：${memory.last_topic}` : '',
|
||
memory.conversations && memory.conversations.length > 0
|
||
? `（该体验者已有 ${memory.conversations.length} 条历史对话记录）`
|
||
: '（新体验者，首次对话）'
|
||
].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 || {};
|
||
|
||
// 打招呼场景
|
||
if (isGreeting) {
|
||
const hasHistory = memory.conversations && memory.conversations.length > 0;
|
||
let greeting;
|
||
|
||
if (hasHistory && memory.last_topic) {
|
||
greeting = (behavior.greeting_returning || '欢迎回来！')
|
||
.replace('{last_topic}', memory.last_topic);
|
||
} 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);
|
||
}
|
||
|
||
const systemPrompt = buildSystemPrompt(config, memory);
|
||
|
||
// 构建消息列表
|
||
const messages = [
|
||
{ role: 'system', content: systemPrompt }
|
||
];
|
||
|
||
// 加入最近历史（最多 20 条）
|
||
const recentHistory = (history || []).slice(-20);
|
||
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);
|
||
}
|
||
}
|
||
|
||
/**
|
||
* 本地降级回复（无 API 密钥或 API 调用失败时）
|
||
*/
|
||
function getLocalReply(message, memory, config) {
|
||
const persona = (config.persona && config.persona.name) || '知秋';
|
||
const msg = message.toLowerCase();
|
||
|
||
if (msg.includes('你好') || msg.includes('hi') || msg.includes('嗨') || msg.includes('hello')) {
|
||
return {
|
||
reply: `你好！我是${persona}。告诉我你想做什么，我们一起聊聊方案 😊`,
|
||
build_ready: false
|
||
};
|
||
}
|
||
|
||
if (msg.includes('你是谁') || msg.includes('介绍') || msg.includes('什么')) {
|
||
return {
|
||
reply: `我是${persona}，光湖系统的开发协助人格体 🧠\n\n我可以帮你：\n• 💬 聊聊你的项目想法\n• 📝 梳理技术方案\n• 🚀 方案确认后帮你自动开发\n\n告诉我你想做什么吧！`,
|
||
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: `不客气！有什么需要随时来找我 😊 下次再来我还记得你～`,
|
||
build_ready: false
|
||
};
|
||
}
|
||
|
||
return {
|
||
reply: `收到！让我想想怎么帮你实现。\n\n能再详细说说你的想法吗？比如：\n• 你想解决什么问题？\n• 面向什么用户？\n• 有什么技术偏好？\n\n聊得越清楚，我帮你做得越好 😊`,
|
||
build_ready: false
|
||
};
|
||
}
|
||
|
||
module.exports = {
|
||
respond,
|
||
loadPersonaConfig,
|
||
buildSystemPrompt,
|
||
detectTaskType
|
||
};
|