guanghulab/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 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
};