@shun-js/aibaiban-server 1.1.7 → 1.1.8

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@shun-js/aibaiban-server",
-  "version": "1.1.7",
+  "version": "1.1.8",
   "description": "aibaiban.com server",
   "keywords": [
     "ai aibaiban"
@@ -37,7 +37,7 @@
     "qiao-z-nuser": "^6.0.0",
     "qiao-z-service": "^6.0.0",
     "qiao-z-sms": "^6.0.0",
-    "viho-llm": "^1.0.7",
+    "viho-llm": "^1.0.8",
     "zod": "^4.3.6",
     "zod-to-json-schema": "^3.25.1"
   },
@@ -45,5 +45,5 @@
     "access": "public",
     "registry": "https://registry.npmjs.org/"
   },
-  "gitHead": "e3aa7590c712845b352ac56628ffb9d624a2c1cf"
+  "gitHead": "4b29e10a1051e2364ccd96f6f74daba34f49a7e4"
 }

package/server/controller/LLMController.js

@@ -1,5 +1,6 @@
 // service
 const service = require('../service/LLMService.js');
+const service2 = require('../service/LLMServiceV2.js');
 
 /**
  * controller
@@ -9,4 +10,9 @@ module.exports = (app) => {
   app.post('/chat-streaming', (req, res) => {
     service.drawAgent(req, res);
   });
+
+  // draw agent
+  app.post('/draw-agent', (req, res) => {
+    service2.drawAgent(req, res);
+  });
 };

package/server/service/LLMServiceV2.js

@@ -0,0 +1,151 @@
+// llm
+const { OpenAIAPI, runAgents } = require('viho-llm');
+const prompts = require('../util/prompt-agent-v2.js');
+
+// util
+const { chatFeishuMsg, errorFeishuMsg } = require('../util/feishu.js');
+
+// LLM 配置
+const llmConfig = global.QZ_CONFIG.llm;
+const finalLLMConfig = llmConfig[llmConfig.default];
+const llm = OpenAIAPI(finalLLMConfig);
+const modelName = finalLLMConfig.modelName;
+const thinking = finalLLMConfig.thinking;
+
+/**
+ * drawAgentV2 - 对话式 Agent
+ * 1 次决策 LLM → reply / generate(elaborate+generate) / irrelevant
+ */
+exports.drawAgent = async (req, res) => {
+  const methodName = 'drawAgentV2';
+  const messages = req.body.messages;
+
+  if (!messages?.length) {
+    const msg = 'need messages';
+    req.logger.error(methodName, msg);
+    res.jsonFail(msg);
+    return;
+  }
+
+  // 启动 SSE
+  res.streamingStart();
+
+  const lastUserMsg = messages.filter((m) => m.role === 'user').pop()?.content || '';
+  req.logger.info(methodName, 'lastUserMsg', lastUserMsg);
+  chatFeishuMsg(req, `v2-${lastUserMsg}`);
+
+  try {
+    const startTime = Date.now();
+
+    // 1. Agent 决策
+    res.streaming(`data: ${JSON.stringify({ type: 'status', step: 'thinking' })}\n\n`);
+
+    let decision = null;
+    await runAgents([
+      {
+        agentStartCallback: () => {
+          req.logger.info(methodName, 'step: agent decision');
+        },
+        agentRequestOptions: {
+          llm,
+          modelName,
+          thinking,
+          isJson: true,
+          messages: [{ role: 'system', content: prompts.AGENT_PROMPT }, ...messages],
+        },
+        agentEndCallback: (result) => {
+          decision = result;
+          const duration = Date.now() - startTime;
+          req.logger.info(methodName, 'decision', JSON.stringify(decision), `${duration}ms`);
+          chatFeishuMsg(req, `v2-decision-${decision.action}`);
+        },
+      },
+    ]);
+
+    // 2. 分发
+    if (decision.action === 'reply') {
+      res.streaming(`data: ${JSON.stringify({ type: 'message', content: decision.message })}\n\n`);
+    } else if (decision.action === 'irrelevant') {
+      res.streaming(`data: ${JSON.stringify({ type: 'message', content: prompts.FIXED_REPLY })}\n\n`);
+    } else if (decision.action === 'generate') {
+      let elaboration = '';
+
+      await runAgents([
+        // elaborate
+        {
+          agentStartCallback: () => {
+            res.streaming(`data: ${JSON.stringify({ type: 'status', step: 'elaborate' })}\n\n`);
+            req.logger.info(methodName, 'step: elaborate');
+          },
+          agentRequestOptions: {
+            llm,
+            modelName,
+            thinking,
+            get messages() {
+              return [
+                {
+                  role: 'user',
+                  content: prompts.ELABORATE_PROMPT.replace('{input}', decision.description).replace(
+                    '{diagramType}',
+                    decision.diagramType,
+                  ),
+                },
+              ];
+            },
+          },
+          agentEndCallback: (result) => {
+            elaboration = result;
+            req.logger.info(methodName, 'elaboration', String(elaboration).slice(0, 100) + '...');
+          },
+        },
+        // generate
+        {
+          agentStartCallback: () => {
+            res.streaming(`data: ${JSON.stringify({ type: 'status', step: 'generate' })}\n\n`);
+            req.logger.info(methodName, 'step: generate');
+          },
+          agentRequestOptions: {
+            llm,
+            modelName,
+            thinking,
+            get messages() {
+              return [
+                {
+                  role: 'user',
+                  content: prompts.GENERATE_PROMPT.replace('{diagramType}', decision.diagramType).replace(
+                    '{elaboration}',
+                    elaboration,
+                  ),
+                },
+              ];
+            },
+          },
+          agentEndCallback: (result) => {
+            const mermaidCode = result;
+            const duration = Date.now() - startTime;
+            req.logger.info(
+              methodName,
+              'mermaidCode',
+              String(mermaidCode).slice(0, 100) + '...',
+              `total: ${duration}ms`,
+            );
+            chatFeishuMsg(req, `v2-mermaid-${mermaidCode}`);
+            res.streaming(`data: ${JSON.stringify({ type: 'mermaid', code: mermaidCode, duration })}\n\n`);
+          },
+        },
+      ]);
+    } else {
+      // 未知 action fallback
+      res.streaming(
+        `data: ${JSON.stringify({ type: 'message', content: decision.message || '可以再说一次吗？' })}\n\n`,
+      );
+    }
+
+    res.streamingEnd();
+  } catch (error) {
+    req.logger.error(methodName, 'error', error);
+    errorFeishuMsg(req, error.message);
+    res.streaming(`data: ${JSON.stringify({ type: 'error', message: error.message })}\n\n`);
+    res.streamingEnd();
+  }
+};

package/server/util/prompt-agent-v2.js

@@ -0,0 +1,43 @@
+/**
+ * Agent V2 Prompts - 对话式 Agent
+ */
+const { ELABORATE_PROMPT, GENERATE_PROMPT, FIXED_REPLY } = require('./prompt-agent.js');
+
+/**
+ * Agent 决策 prompt
+ * 一次 LLM 调用完成意图判断 + 决策
+ */
+const AGENT_PROMPT = `你是 AI 白板助手，帮用户在白板上画图表。
+
+## 能力
+支持 4 种图表：flowchart（流程图）、sequence（时序图）、classDiagram（类图）、erDiagram（ER图）。
+
+## 决策规则
+收到用户消息后判断，只回复 JSON：
+
+1. 回复文字（追问、确认、引导）：
+   {"action":"reply","message":"你的回复内容"}
+
+2. 生成图表（信息已足够明确）：
+   {"action":"generate","diagramType":"类型","description":"详细描述，包含所有节点、关系、步骤"}
+
+3. 与画图完全无关：
+   {"action":"irrelevant"}
+
+## 判断标准
+- 用户说了图表类型 + 具体场景/内容 → generate
+- 用户只说了类型没说内容 → reply 追问具体场景
+- 用户只说了内容没说类型 → 你来判断最合适的类型，直接 generate
+- 用户说"帮我画个图" → reply 追问想画什么
+- 用户要修改已有图表 → 根据对话历史理解上下文，generate 完整新图表
+- 用户闲聊但话题相关 → reply 自然回应并引导回画图
+- 用户闲聊话题无关 → irrelevant
+
+## 回复风格
+- 追问时友好自然，像朋友聊天，不要生硬
+- 可以给建议和示例帮助用户想清楚需求
+- generate 时 description 要尽可能详细完整
+
+只回复 JSON。`;
+
+module.exports = { AGENT_PROMPT, ELABORATE_PROMPT, GENERATE_PROMPT, FIXED_REPLY };