zhizhi/persona-studio/backend/brain/persona-engine.js

/**
 * 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) || '知秋';

  if (message.includes('你好') || message.includes('hi') || message.includes('嗨')) {
    return {
      reply: `你好！我是${persona}。告诉我你想做什么，我们一起聊聊方案 😊`,
      build_ready: false
    };
  }

  if (message.includes('做') || message.includes('开发') || message.includes('写')) {
    return {
      reply: `好的，让我了解一下你的需求：\n\n1. 你想做什么类型的项目？（网站 / 工具 / 组件 / 其他）\n2. 有哪些核心功能？\n3. 有没有参考设计？\n\n跟我聊聊，我帮你理清思路 💙`,
      build_ready: false
    };
  }

  return {
    reply: `收到！我会把你的需求整理成技术方案。有什么具体想法，继续跟我聊 😊`,
    build_ready: false
  };
}

module.exports = {
  respond,
  loadPersonaConfig,
  buildSystemPrompt,
  detectTaskType
};