@shun-js/aibaiban-server 1.2.4 → 1.2.6

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@shun-js/aibaiban-server",
-  "version": "1.2.4",
+  "version": "1.2.6",
   "description": "aibaiban.com server",
   "keywords": [
     "ai aibaiban"
@@ -45,5 +45,5 @@
     "access": "public",
     "registry": "https://registry.npmjs.org/"
   },
-  "gitHead": "d9255a37dbd25bf32a5c64b0230440eb34c6de2d"
+  "gitHead": "66c933a20f3e0b055d9e414c0651acd821e6a320"
 }

package/server/service/LLMService.js

@@ -171,32 +171,52 @@ exports.drawAgent = async (req, res) => {
     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, `decision-${decision.action}`);
+    try {
+      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, `decision-${decision.action}`);
+          },
         },
-      },
-    ]);
+      ]);
+    } catch (jsonErr) {
+      // LLM 未返回 JSON，兜底为 reply
+      req.logger.warn(methodName, 'JSON parse fallback', jsonErr.message);
+      const rawText = jsonErr.message.replace('Cannot parse JSON from LLM response:', '').trim();
+      req.logger.info(methodName, 'LLM raw response', rawText);
+
+      // 尝试从原始文本中提取 JSON（LLM 可能在 JSON 前后加了多余文字）
+      const jsonMatch = rawText.match(/\{[\s\S]*\}/);
+      if (jsonMatch) {
+        try {
+          decision = JSON.parse(jsonMatch[0]);
+          req.logger.info(methodName, 'extracted JSON from raw text', JSON.stringify(decision));
+        } catch {
+          decision = { action: 'reply', message: '好的，请告诉我你想画什么图表？' };
+        }
+      } else {
+        decision = { action: 'reply', message: '好的，请告诉我你想画什么图表？' };
+      }
+    }
 
     // 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`);
+      res.streaming(`data: ${JSON.stringify({ type: 'welcome' })}\n\n`);
     } else if (decision.action === 'generate') {
       let elaboration = '';
 
@@ -312,7 +332,11 @@ exports.drawAgent = async (req, res) => {
   } catch (error) {
     req.logger.error(methodName, 'error', error);
     errorFeishuMsg(req, error.message);
-    res.streaming(`data: ${JSON.stringify({ type: 'error', message: error.message })}\n\n`);
+    // 对用户隐藏内部错误细节
+    const userMessage = error.message?.includes('Cannot parse JSON')
+      ? '请求处理失败，请重新描述你的需求'
+      : '服务暂时出错，请稍后重试';
+    res.streaming(`data: ${JSON.stringify({ type: 'error', message: userMessage })}\n\n`);
     res.streamingEnd();
   }
 };

package/server/util/prompt-agent.js

@@ -45,7 +45,10 @@ module.exports = {
 - generate 时 description 要尽可能详细完整
 - canvas 时 description 要包含形状、颜色、位置等用户提到的所有细节
 
-只回复 JSON。`,
+严格要求：
+1. 你的回复必须是且仅是一个合法的 JSON 对象，不要包含任何其他文字、解释或 markdown 标记。
+2. 每条用户消息都是独立的新请求，即使对话历史中有之前生成过图表的记录，也要重新判断当前消息的意图。
+3. 永远不要回复"已生成"、"已完成"之类的文字，你只负责决策，不负责执行。`,
 
   /**
    * 图表内容细化
@@ -62,7 +65,7 @@ module.exports = {
 - classDiagram: {"classes": [{"name": "", "attributes": [...], "methods": [...]}], "relations": [...]}
 - erDiagram: {"entities": [{"name": "", "attributes": [...]}], "relations": [...]}
 
-只回复 JSON。`,
+严格要求：你的回复必须是且仅是一个合法的 JSON 对象，不要包含任何其他文字、解释或 markdown 标记。`,
 
   /**
    * Mermaid 生成
@@ -104,12 +107,6 @@ flowchart TD
 
 只回复 Mermaid 代码，不要包含 \`\`\`mermaid 标记。`,
 
-  /**
-   * 非白板请求的固定回复
-   */
-  FIXED_REPLY:
-    '你好！👋 我是 AI 白板助手，支持以下功能：\n\n📊 流程图 — 例如"画一个用户注册登录的流程图"\n🔄 时序图 — 例如"画一个用户下单支付的时序图"\n🏗️ 类图 — 例如"画一个电商系统的类图"\n🗄️ ER图 — 例如"画一个博客系统的ER图"\n🎨 自由绘图 — 例如"画一个红色的圆"、"画两个矩形用箭头连接"\n\n💡 试试直接输入你的需求吧！',
-
   /**
    * Canvas Function Calling 系统提示词
    */

package/views/index.html

@@ -105,7 +105,7 @@
     <script
       type="module"
       crossorigin
-      src="https://static-small.vincentqiao.com/aibaiban/static/index-Fo5F7gnm.js"
+      src="https://static-small.vincentqiao.com/aibaiban/static/index-DCJwq4XN.js"
     ></script>
     <link
       rel="modulepreload"