@shun-js/aibaiban-server 1.0.0 → 1.0.2

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@shun-js/aibaiban-server",
-  "version": "1.0.0",
+  "version": "1.0.2",
   "description": "aibaiban.com server",
   "keywords": [
     "ai aibaiban"
@@ -23,7 +23,7 @@
   ],
   "scripts": {
     "pack": "cp ./static/index.html ./views/index.html",
-    "upload": "qcos fo ../../configs/aitubiao-web.json ./static aibaiban/static"
+    "upload": "qcos fo ../../configs/aibaiban-web.json ./static aibaiban/static"
   },
   "dependencies": {
     "@shun-js/shun-config": "^0.3.1",
@@ -44,5 +44,5 @@
     "access": "public",
     "registry": "https://registry.npmjs.org/"
   },
-  "gitHead": "5e3d563170fee894ac2192d67f0b68fbcfa50949"
+  "gitHead": "46030f263f7b99057edf9c9b0944260b9e48d759"
 }

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/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-DlsFnNe0.js"
+      src="https://static-small.vincentqiao.com/aibaiban/static/index-Oeeo1mmm.js"
     ></script>
     <link
       rel="modulepreload"
       crossorigin
-      href="https://static-small.vincentqiao.com/aibaiban/static/chunks/react-vendor-sruPWfIs.js"
+      href="https://static-small.vincentqiao.com/aibaiban/static/chunks/react-vendor-Cy8ztC7t.js"
     />
     <link
       rel="modulepreload"
       crossorigin
-      href="https://static-small.vincentqiao.com/aibaiban/static/chunks/antd-base-CUHU3t6W.js"
+      href="https://static-small.vincentqiao.com/aibaiban/static/chunks/antd-base-D2IRhyJe.js"
     />
     <link
       rel="modulepreload"
       crossorigin
-      href="https://static-small.vincentqiao.com/aibaiban/static/chunks/antd-icons-C5ujIBy3.js"
+      href="https://static-small.vincentqiao.com/aibaiban/static/chunks/antd-icons-CjM6MvPn.js"
     />
     <link
       rel="modulepreload"
       crossorigin
-      href="https://static-small.vincentqiao.com/aibaiban/static/chunks/antd-x-Ch3MovBd.js"
+      href="https://static-small.vincentqiao.com/aibaiban/static/chunks/antd-x-B6lj3020.js"
     />
     <link
       rel="modulepreload"
       crossorigin
-      href="https://static-small.vincentqiao.com/aibaiban/static/chunks/excalidraw-W2v_i_jq.js"
+      href="https://static-small.vincentqiao.com/aibaiban/static/chunks/excalidraw-D605XF9z.js"
     />
     <link
       rel="stylesheet"