@shun-js/aibaiban-server 1.0.1 → 1.0.3

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@shun-js/aibaiban-server",
-  "version": "1.0.1",
+  "version": "1.0.3",
   "description": "aibaiban.com server",
   "keywords": [
     "ai aibaiban"
@@ -44,5 +44,5 @@
     "access": "public",
     "registry": "https://registry.npmjs.org/"
   },
-  "gitHead": "620060d4440a59999fdaf310790434eba9c81d3c"
+  "gitHead": "6882b0a600e3ad421b67cee58134371ec478ce93"
 }

package/server/service/LLMService.js

@@ -28,6 +28,9 @@ exports.drawAgent = async (req, res) => {
   chatFeishuMsg(req, `userPrompt-${input}`);
 
   try {
+    const startTime = Date.now();
+    let stepStart = startTime;
+
     // start
     res.streaming(`data: ${JSON.stringify({ step: 'router', status: 'start' })}\n\n`);
     req.logger.info(methodName, 'step: router');
@@ -35,9 +38,10 @@ exports.drawAgent = async (req, res) => {
     // intent
     const intentResult = await callLLMForJSON(prompts.ROUTER_PROMPT.replace('{input}', input), res, 'router');
     const intent = intentResult.intent;
-    req.logger.info(methodName, 'intent', intent);
+    const routerTime = Date.now() - stepStart;
+    req.logger.info(methodName, 'intent', intent, `${routerTime}ms`);
     chatFeishuMsg(req, `intent-${intent}`);
-    res.streaming(`data: ${JSON.stringify({ step: 'router', intent })}\n\n`);
+    res.streaming(`data: ${JSON.stringify({ step: 'router', intent, duration: routerTime })}\n\n`);
 
     // 非白板请求
     if (intent === 'irrelevant') {
@@ -49,15 +53,18 @@ exports.drawAgent = async (req, res) => {
     }
 
     // 2. classify - 分类图表类型
+    stepStart = Date.now();
     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);
+    const classifyTime = Date.now() - stepStart;
+    req.logger.info(methodName, 'diagramType', diagramType, `${classifyTime}ms`);
     chatFeishuMsg(req, `diagramType-${diagramType}`);
-    res.streaming(`data: ${JSON.stringify({ step: 'classify', diagramType })}\n\n`);
+    res.streaming(`data: ${JSON.stringify({ step: 'classify', diagramType, duration: classifyTime })}\n\n`);
 
     // 3. elaborate - 细化内容
+    stepStart = Date.now();
     res.streaming(`data: ${JSON.stringify({ step: 'elaborate', status: 'start' })}\n\n`);
     req.logger.info(methodName, 'step: elaborate');
     const elaboration = await callLLM(
@@ -65,11 +72,13 @@ exports.drawAgent = async (req, res) => {
       res,
       'elaborate',
     );
-    req.logger.info(methodName, 'elaboration', elaboration.slice(0, 100) + '...');
+    const elaborateTime = Date.now() - stepStart;
+    req.logger.info(methodName, 'elaboration', elaboration.slice(0, 100) + '...', `${elaborateTime}ms`);
     chatFeishuMsg(req, `elaboration-${elaboration}`);
-    res.streaming(`data: ${JSON.stringify({ step: 'elaborate', done: true })}\n\n`);
+    res.streaming(`data: ${JSON.stringify({ step: 'elaborate', done: true, duration: elaborateTime })}\n\n`);
 
     // 4. review - 质量检查
+    stepStart = Date.now();
     res.streaming(`data: ${JSON.stringify({ step: 'review', status: 'start' })}\n\n`);
     req.logger.info(methodName, 'step: review');
     const reviewResult = await callLLMForJSON(
@@ -79,9 +88,10 @@ exports.drawAgent = async (req, res) => {
       res,
       'review',
     );
-    req.logger.info(methodName, 'reviewResult', reviewResult);
+    const reviewTime = Date.now() - stepStart;
+    req.logger.info(methodName, 'reviewResult', reviewResult, `${reviewTime}ms`);
     chatFeishuMsg(req, `reviewResult-${reviewResult}`);
-    res.streaming(`data: ${JSON.stringify({ step: 'review', result: reviewResult.result })}\n\n`);
+    res.streaming(`data: ${JSON.stringify({ step: 'review', result: reviewResult.result, duration: reviewTime })}\n\n`);
 
     // 信息不足，追问用户
     if (reviewResult.result === 'need_more_info') {
@@ -97,6 +107,7 @@ exports.drawAgent = async (req, res) => {
     }
 
     // 5. generate - 生成 Mermaid
+    stepStart = Date.now();
     res.streaming(`data: ${JSON.stringify({ step: 'generate', status: 'start' })}\n\n`);
     req.logger.info(methodName, 'step: generate');
     const mermaidCode = await callLLM(
@@ -104,11 +115,21 @@ exports.drawAgent = async (req, res) => {
       res,
       'generate',
     );
-    req.logger.info(methodName, 'mermaidCode', mermaidCode.slice(0, 100) + '...');
+    const generateTime = Date.now() - stepStart;
+    const totalTime = Date.now() - startTime;
+    req.logger.info(
+      methodName,
+      'mermaidCode',
+      mermaidCode.slice(0, 100) + '...',
+      `${generateTime}ms`,
+      `total: ${totalTime}ms`,
+    );
     chatFeishuMsg(req, `mermaidCode-${mermaidCode}`);
 
     // 返回最终结果
-    res.streaming(`data: ${JSON.stringify({ step: 'generate', mermaidCode })}\n\n`);
+    res.streaming(
+      `data: ${JSON.stringify({ step: 'generate', mermaidCode, duration: generateTime, totalDuration: totalTime })}\n\n`,
+    );
     res.streamingEnd();
   } catch (error) {
     req.logger.error(methodName, 'error', error);

package/server/util/llm-agent.js

@@ -38,38 +38,26 @@ function extractJSON(text) {
 }
 
 /**
- * 流式调用 LLM 并解析 JSON
- * 不发送中间 delta，只返回解析后的结果
+ * 调用 LLM 并解析 JSON
  */
 exports.callLLMForJSON = async (prompt) => {
-  // 非流式调用，等待完整响应
   const response = await llm.chat({
     model: finalLLMConfig.modelName,
     messages: [{ role: 'user', content: prompt }],
   });
 
   const fullContent = response.content || '';
-  // 不发送 delta，直接返回结果让调用方处理
   return extractJSON(fullContent);
 };
 
 /**
- * 流式调用 LLM（普通文本）
- * 不发送中间 delta，只返回完整内容
+ * 调用 LLM（普通文本）
  */
 exports.callLLM = async (prompt) => {
-  let fullContent = '';
-
-  await llm.chatWithStreaming(
-    { model: finalLLMConfig.modelName, messages: [{ role: 'user', content: prompt }] },
-    {
-      contentCallback: (chunk) => {
-        fullContent += chunk;
-        // 不发送中间 delta，避免显示问题
-      },
-    },
-  );
+  const response = await llm.chat({
+    model: finalLLMConfig.modelName,
+    messages: [{ role: 'user', content: prompt }],
+  });
 
-  // 返回完整内容（用于后续处理）
-  return fullContent;
+  return response.content || '';
 };

package/server/util/prompt-agent.js

@@ -86,9 +86,10 @@ module.exports = {
 
 ⚠️ 重要语法限制:
 1. 禁止使用以下 Mermaid 保留字作为节点名称: start, end, stop, pause, resume, done, state, choice, fork, join
-2. 如果流程有"开始"或"结束"，使用"开始"((开始))、"结束"((结束)) 或 图标如((●))、((○)) 代替
-3. 节点名称避免使用特殊符号如括号()[]{}，可以用中文双引号""包裹
-4. 连接线避免使用特殊字符，用中文描述如"是"、"否"、"到"
+2. 如果流程有"开始"或"结束"，使用 node_start((开始))、node_end((结束)) 这样的自定义ID
+3. 节点文本和连接线文本中禁止使用括号()[]{}，这些是 Mermaid 的形状语法符号，会导致解析错误
+4. 连接线标签示例: -->|是| 、-->|否| 、-->|提交|，不要写 -->|否(注册)| 或 -->|是(通过)|
+5. 如需补充说明，用斜杠/顿号/破折号代替括号，如"否-注册"、"否/注册"
 
 要求:
 1. 生成合法的 Mermaid 语法

package/views/index.html

@@ -89,50 +89,52 @@
         ]
       }
     </script>
-    <!-- Microsoft Clarity -->
+    <!-- Microsoft Clarity (production only) -->
     <script type="text/javascript">
-      (function (c, l, a, r, i, t, y) {
-        c[a] =
-          c[a] ||
-          function () {
-            (c[a].q = c[a].q || []).push(arguments);
-          };
-        t = l.createElement(r);
-        t.async = 1;
-        t.src = 'https://www.clarity.ms/tag/' + i;
-        y = l.getElementsByTagName(r)[0];
-        y.parentNode.insertBefore(t, y);
-      })(window, document, 'clarity', 'script', 't5b230u2zp');
+      if (location.hostname.includes('aibaiban.com')) {
+        (function (c, l, a, r, i, t, y) {
+          c[a] =
+            c[a] ||
+            function () {
+              (c[a].q = c[a].q || []).push(arguments);
+            };
+          t = l.createElement(r);
+          t.async = 1;
+          t.src = 'https://www.clarity.ms/tag/' + i;
+          y = l.getElementsByTagName(r)[0];
+          y.parentNode.insertBefore(t, y);
+        })(window, document, 'clarity', 'script', 't5b230u2zp');
+      }
     </script>
     <script
       type="module"
       crossorigin
-      src="https://static-small.vincentqiao.com/aibaiban/static/index-CsqtP3ZT.js"
+      src="https://static-small.vincentqiao.com/aibaiban/static/index-CYIYkLbz.js"
     ></script>
     <link
       rel="modulepreload"
       crossorigin
-      href="https://static-small.vincentqiao.com/aibaiban/static/chunks/react-vendor-Cy8ztC7t.js"
+      href="https://static-small.vincentqiao.com/aibaiban/static/chunks/react-vendor-zXWpQZWf.js"
     />
     <link
       rel="modulepreload"
       crossorigin
-      href="https://static-small.vincentqiao.com/aibaiban/static/chunks/antd-base-eH03AryP.js"
+      href="https://static-small.vincentqiao.com/aibaiban/static/chunks/antd-base-BkT6tJ3N.js"
     />
     <link
       rel="modulepreload"
       crossorigin
-      href="https://static-small.vincentqiao.com/aibaiban/static/chunks/antd-icons-YV1DFW0X.js"
+      href="https://static-small.vincentqiao.com/aibaiban/static/chunks/antd-icons-AALiXfxf.js"
     />
     <link
       rel="modulepreload"
       crossorigin
-      href="https://static-small.vincentqiao.com/aibaiban/static/chunks/antd-x-DyGKn05l.js"
+      href="https://static-small.vincentqiao.com/aibaiban/static/chunks/antd-x-DKY-WOLC.js"
     />
     <link
       rel="modulepreload"
       crossorigin
-      href="https://static-small.vincentqiao.com/aibaiban/static/chunks/excalidraw-CaW1wDJE.js"
+      href="https://static-small.vincentqiao.com/aibaiban/static/chunks/excalidraw-WA-IihWf.js"
     />
     <link
       rel="stylesheet"