@shun-js/aibaiban-server 0.8.1 → 0.8.3

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@shun-js/aibaiban-server",
-  "version": "0.8.1",
+  "version": "0.8.3",
   "description": "aibaiban.com server",
   "keywords": [
     "ai aibaiban"
@@ -44,5 +44,5 @@
     "access": "public",
     "registry": "https://registry.npmjs.org/"
   },
-  "gitHead": "fcc8c9fb781e34771e75f1df5615d912a50ff63e"
+  "gitHead": "331bc8dc9ccfa3de0ea2c3c7326301f660cab528"
 }

package/server/controller/LLMController.js

@@ -14,4 +14,9 @@ module.exports = (app) => {
   app.post('/drawWithTools', (req, res) => {
     service.drawWithTools(req, res);
   });
+
+  // draw/agent (流式 Agent)
+  app.post('/draw/agent', (req, res) => {
+    service.drawAgent(req, res);
+  });
 };

package/server/service/LLMService.js

@@ -7,6 +7,10 @@ const { generateFlowchartWithTools } = require('../util/llm-toolcall.js');
 // util
 const { chatFeishuMsg, chatResFeishuMsg, errorFeishuMsg } = require('../util/feishu.js');
 
+// llm agent
+const { callLLMForJSON, callLLM } = require('../util/llm-agent.js');
+const prompts = require('../util/prompt-agent.js');
+
 /**
  * intent
  * @param {*} req
@@ -83,3 +87,104 @@ exports.drawWithTools = async (req, res) => {
     res.jsonFail(msg);
   }
 };
+
+/**
+ * drawAgent - 流式 Agent 接口
+ * 流程：router -> classify -> elaborate -> review -> generate
+ */
+exports.drawAgent = async (req, res) => {
+  const methodName = 'drawAgent';
+
+  // check
+  if (!req.body.userPrompt) {
+    const msg = 'need userPrompt';
+    req.logger.error(methodName, msg);
+    res.jsonFail(msg);
+    return;
+  }
+
+  // 启动 SSE
+  res.streamingStart();
+
+  const input = decodeURIComponent(req.body.userPrompt);
+  req.logger.info(methodName, 'userPrompt', input);
+
+  try {
+    // 1. router - 判断意图
+    res.streaming(`data: ${JSON.stringify({ step: 'router', status: 'start' })}\n\n`);
+    req.logger.info(methodName, 'step: router');
+    const intentResult = await callLLMForJSON(prompts.ROUTER_PROMPT.replace('{input}', input), res, 'router');
+    const intent = intentResult.intent;
+    req.logger.info(methodName, 'intent', intent);
+
+    // 非白板请求
+    if (intent === 'irrelevant') {
+      res.streaming(
+        `data: ${JSON.stringify({ step: 'router', result: 'irrelevant', response: prompts.FIXED_REPLY })}\n\n`,
+      );
+      res.streamingEnd();
+      return;
+    }
+
+    // 2. classify - 分类图表类型
+    res.streaming(`data: ${JSON.stringify({ step: 'classify', status: 'start' })}\n\n`);
+    req.logger.info(methodName, 'step: classify');
+    const classifyResult = await callLLMForJSON(prompts.CLASSIFY_PROMPT.replace('{input}', input), res, 'classify');
+    const diagramType = classifyResult.diagramType;
+    req.logger.info(methodName, 'diagramType', diagramType);
+
+    // 3. elaborate - 细化内容
+    res.streaming(`data: ${JSON.stringify({ step: 'elaborate', status: 'start' })}\n\n`);
+    req.logger.info(methodName, 'step: elaborate');
+    const elaboration = await callLLM(
+      prompts.ELABORATE_PROMPT.replace('{input}', input).replace('{diagramType}', diagramType),
+      res,
+      'elaborate',
+    );
+    req.logger.info(methodName, 'elaboration', elaboration.slice(0, 100) + '...');
+
+    // 4. review - 质量检查
+    res.streaming(`data: ${JSON.stringify({ step: 'review', status: 'start' })}\n\n`);
+    req.logger.info(methodName, 'step: review');
+    const reviewResult = await callLLMForJSON(
+      prompts.REVIEW_PROMPT.replace('{input}', input)
+        .replace('{diagramType}', diagramType)
+        .replace('{elaboration}', elaboration),
+      res,
+      'review',
+    );
+    req.logger.info(methodName, 'reviewResult', reviewResult);
+
+    // 信息不足，追问用户
+    if (reviewResult.result === 'need_more_info') {
+      res.streaming(
+        `data: ${JSON.stringify({
+          step: 'review',
+          result: 'need_more_info',
+          questions: reviewResult.questions,
+        })}\n\n`,
+      );
+      res.streamingEnd();
+      return;
+    }
+
+    // 5. generate - 生成 Mermaid
+    res.streaming(`data: ${JSON.stringify({ step: 'generate', status: 'start' })}\n\n`);
+    req.logger.info(methodName, 'step: generate');
+    const mermaidCode = await callLLM(
+      prompts.GENERATE_PROMPT.replace('{diagramType}', diagramType).replace('{elaboration}', elaboration),
+      res,
+      'generate',
+    );
+
+    req.logger.info(methodName, 'mermaidCode', mermaidCode.slice(0, 100) + '...');
+
+    // 返回最终结果
+    res.streaming(`data: ${JSON.stringify({ step: 'generate', mermaidCode })}\n\n`);
+    res.streamingEnd();
+  } catch (error) {
+    req.logger.error(methodName, 'error', error);
+    res.streaming(`data: ${JSON.stringify({ step: 'error', message: error.message })}\n\n`);
+    res.streamingEnd();
+  }
+};

package/server/util/llm-agent.js

@@ -0,0 +1,81 @@
+// llm
+const { OpenAIAPI } = require('viho-llm');
+
+/**
+ * LLM 配置 - 使用 moonshot 平台的 kimi-k2.5
+ */
+const llmConfig = {
+  apiKey: global.QZ_CONFIG.kimi.apiKey,
+  baseURL: global.QZ_CONFIG.kimi.baseURL,
+  modelName: global.QZ_CONFIG.kimi.modelID,
+};
+
+const llm = OpenAIAPI(llmConfig);
+
+/**
+ * 从文本中提取 JSON
+ */
+function extractJSON(text) {
+  const cleaned = text.replace(/\{\{/g, '{').replace(/\}\}/g, '}');
+  try {
+    return JSON.parse(cleaned);
+  } catch (e) {
+    // try next
+  }
+  const codeBlock = cleaned.match(/```(?:json)?\s*([\s\S]*?)```/);
+  if (codeBlock) {
+    try {
+      return JSON.parse(codeBlock[1].trim());
+    } catch (e) {
+      // try next
+    }
+  }
+  const match = cleaned.match(/(\{[\s\S]*\}|\[[\s\S]*\])/);
+  if (match) {
+    try {
+      return JSON.parse(match[1]);
+    } catch (e) {
+      // try next
+    }
+  }
+  throw new Error(`Cannot extract JSON from: ${text.slice(0, 200)}`);
+}
+
+/**
+ * 流式调用 LLM 并解析 JSON
+ * 使用非流式 API，但过程流式输出给客户端
+ */
+exports.callLLMForJSON = async (prompt, res, step) => {
+  // 非流式调用，等待完整响应
+  const response = await llm.chat({
+    model: llmConfig.modelName,
+    messages: [{ role: 'user', content: prompt }],
+  });
+
+  const fullContent = response.content || '';
+  // 流式输出给客户端
+  res.streaming(`data: ${JSON.stringify({ step, delta: fullContent })}\n\n`);
+
+  return extractJSON(fullContent);
+};
+
+/**
+ * 流式调用 LLM（普通文本）
+ */
+exports.callLLM = async (prompt, res, step) => {
+  let fullContent = '';
+
+  await llm.chatWithStreaming(
+    { model: llmConfig.modelName, messages: [{ role: 'user', content: prompt }] },
+    {
+      contentCallback: (chunk) => {
+        fullContent += chunk;
+        // 直接流式返回给客户端
+        res.streaming(`data: ${JSON.stringify({ step, delta: chunk })}\n\n`);
+      },
+    },
+  );
+
+  // 返回完整内容（用于后续处理）
+  return fullContent;
+};

package/server/util/prompt-agent.js

@@ -0,0 +1,89 @@
+/**
+ * Agent Prompts
+ * 从 test-agent/src/prompts/index.ts 复制
+ */
+
+module.exports = {
+  /**
+   * 意图分类 - 判断是否与白板相关
+   */
+  ROUTER_PROMPT: `你是一个意图分类器。判断用户输入是否与"画图/白板/图表/流程图/甘特图/思维导图/时序图/PRD"相关。
+
+用户输入: {input}
+
+只回复 JSON: {"intent": "whiteboard"} 或 {"intent": "irrelevant"}`,
+
+  /**
+   * 图表类型分类
+   */
+  CLASSIFY_PROMPT: `你是一个图表类型分析专家。根据用户需求，判断最适合的图表类型。
+
+可选类型:
+- flowchart: 流程图、步骤图、决策流程
+- gantt: 甘特图、项目计划、时间线
+- prd: 产品需求文档结构图
+- mindmap: 思维导图、知识结构
+- sequence: 时序图、交互流程
+
+用户输入: {input}
+
+只回复 JSON: {"diagramType": "类型"}`,
+
+  /**
+   * 图表内容细化
+   */
+  ELABORATE_PROMPT: `你是一个图表内容规划专家。根据用户需求和图表类型，细化出具体的节点、关系和步骤。
+
+用户输入: {input}
+图表类型: {diagramType}
+
+请输出结构化的 JSON 描述，包含图表的所有节点、连接关系、标签等信息。
+根据图表类型调整结构:
+- flowchart: {"nodes": [...], "edges": [...]}
+- gantt: {"tasks": [{"name": "", "start": "", "duration": ""}]}
+- mindmap: {"root": "", "children": [...]}
+- sequence: {"participants": [...], "messages": [...]}
+- prd: {"modules": [...], "relations": [...]}
+
+只回复 JSON。`,
+
+  /**
+   * 质量检查
+   */
+  REVIEW_PROMPT: `你是一个质量检查专家。检查以下图表描述是否信息充足，能否生成完整的图表。
+
+用户原始输入: {input}
+图表类型: {diagramType}
+细化描述: {elaboration}
+
+判断标准:
+1. 节点/步骤是否足够（至少3个有意义的元素）
+2. 关系/连接是否清晰
+3. 是否有明显缺失的关键信息
+
+如果信息充足，回复: {"result": "pass"}
+如果信息不足，回复: {"result": "need_more_info", "questions": ["问题1", "问题2"]}
+
+只回复 JSON。`,
+
+  /**
+   * Mermaid 生成
+   */
+  GENERATE_PROMPT: `你是一个 Mermaid 图表生成专家。根据以下信息生成正确的 Mermaid 代码。
+
+图表类型: {diagramType}
+细化描述: {elaboration}
+
+要求:
+1. 生成合法的 Mermaid 语法
+2. 使用中文标签
+3. 结构清晰美观
+
+只回复 Mermaid 代码，不要包含 \`\`\`mermaid 标记。`,
+
+  /**
+   * 非白板请求的固定回复
+   */
+  FIXED_REPLY:
+    '我是一个白板助手，专门帮你生成各种图表。你可以让我画流程图、甘特图、思维导图、时序图等。请告诉我你想画什么图吧！',
+};