@shun-js/aibaiban-server 1.4.1 → 1.4.2

package/server/service/LLMService.js

@@ -1,9 +1,10 @@
 // llm
-const { OpenAIAPI, LibLibAPI, runAgents } = require('viho-llm');
-const prompts = require('../util/prompt-agent.js');
+const { OpenAIAPI, LibLibAPI, runAgents } = require("viho-llm");
+const prompts = require("../util/prompt-agent.js");
 
 // util
-const { chatFeishuMsg, errorFeishuMsg } = require('../util/feishu.js');
+const { chatFeishuMsg, errorFeishuMsg } = require("../util/feishu.js");
+const { sendPromptViaRelay } = require("../util/relay-client.js");
 
 // LLM 配置
 const llmConfig = global.QZ_CONFIG.llm;
@@ -20,9 +21,9 @@ const liblib = LibLibAPI(imageGenConfig);
  * convertToolCallToSkeleton - 将单个 tool call 转为 ExcalidrawElementSkeleton
  */
 function convertToolCallToSkeleton(name, args) {
-  if (name === 'draw_shape') {
+  if (name === "draw_shape") {
     const skeleton = {
-      type: args.shape || 'rectangle',
+      type: args.shape || "rectangle",
       x: args.x || 0,
       y: args.y || 0,
     };
@@ -43,8 +44,8 @@ function convertToolCallToSkeleton(name, args) {
     return [skeleton];
   }
 
-  if (name === 'draw_arrow') {
-    const skeleton = { type: 'arrow' };
+  if (name === "draw_arrow") {
+    const skeleton = { type: "arrow" };
     // 绑定
     if (args.startId) skeleton.start = { id: args.startId };
     if (args.endId) skeleton.end = { id: args.endId };
@@ -57,9 +58,15 @@ function convertToolCallToSkeleton(name, args) {
     if (args.label) skeleton.label = { text: args.label };
     // 箭头样式
     if (args.startArrowhead) skeleton.startArrowhead = args.startArrowhead;
-    if (args.endArrowhead !== undefined) skeleton.endArrowhead = args.endArrowhead;
+    if (args.endArrowhead !== undefined)
+      skeleton.endArrowhead = args.endArrowhead;
     // 如果没有绑定，用坐标计算 points
-    if (!args.startId && !args.endId && args.endX !== undefined && args.endY !== undefined) {
+    if (
+      !args.startId &&
+      !args.endId &&
+      args.endX !== undefined &&
+      args.endY !== undefined
+    ) {
       skeleton.width = args.endX - skeleton.x;
       skeleton.height = args.endY - skeleton.y;
     }
@@ -71,13 +78,16 @@ function convertToolCallToSkeleton(name, args) {
     return [skeleton];
   }
 
-  if (name === 'draw_line') {
+  if (name === "draw_line") {
     if (!args.points || args.points.length < 2) return [];
     const skeleton = {
-      type: 'line',
+      type: "line",
       x: args.points[0][0] || 0,
       y: args.points[0][1] || 0,
-      points: args.points.map((p) => [p[0] - (args.points[0][0] || 0), p[1] - (args.points[0][1] || 0)]),
+      points: args.points.map((p) => [
+        p[0] - (args.points[0][0] || 0),
+        p[1] - (args.points[0][1] || 0),
+      ]),
     };
     if (args.strokeColor) skeleton.strokeColor = args.strokeColor;
     if (args.strokeWidth) skeleton.strokeWidth = args.strokeWidth;
@@ -85,12 +95,12 @@ function convertToolCallToSkeleton(name, args) {
     return [skeleton];
   }
 
-  if (name === 'draw_text') {
+  if (name === "draw_text") {
     const skeleton = {
-      type: 'text',
+      type: "text",
       x: args.x || 0,
       y: args.y || 0,
-      text: args.text || '',
+      text: args.text || "",
     };
     if (args.id) skeleton.id = args.id;
     if (args.fontSize) skeleton.fontSize = args.fontSize;
@@ -100,9 +110,9 @@ function convertToolCallToSkeleton(name, args) {
     return [skeleton];
   }
 
-  if (name === 'draw_frame') {
+  if (name === "draw_frame") {
     const skeleton = {
-      type: 'frame',
+      type: "frame",
       children: args.childIds || [],
     };
     if (args.name) skeleton.name = args.name;
@@ -122,16 +132,20 @@ function convertToolCallToSkeleton(name, args) {
 function convertToolCallsToSkeletons(toolCalls) {
   const skeletons = [];
   for (const tc of toolCalls) {
-    if (tc.function.name === 'clear_canvas') continue;
+    if (tc.function.name === "clear_canvas") continue;
     try {
       const args = JSON.parse(tc.function.arguments);
-      if (tc.function.name === 'draw_group') {
+      if (tc.function.name === "draw_group") {
         // draw_group: 递归转换子元素，统一加 groupIds
-        const groupId = 'group_' + Date.now() + '_' + Math.random().toString(36).slice(2, 8);
+        const groupId =
+          "group_" + Date.now() + "_" + Math.random().toString(36).slice(2, 8);
         const children = args.elements || [];
         for (const child of children) {
-          const toolName = 'draw_' + child.tool;
-          const childSkeletons = convertToolCallToSkeleton(toolName, child.args || {});
+          const toolName = "draw_" + child.tool;
+          const childSkeletons = convertToolCallToSkeleton(
+            toolName,
+            child.args || {},
+          );
           for (const s of childSkeletons) {
             s.groupIds = [groupId];
             skeletons.push(s);
@@ -141,7 +155,7 @@ function convertToolCallsToSkeletons(toolCalls) {
         const result = convertToolCallToSkeleton(tc.function.name, args);
         skeletons.push(...result);
       }
-    } catch (e) {
+    } catch {
       // JSON 解析失败，跳过该 tool call
     }
   }
@@ -153,11 +167,11 @@ function convertToolCallsToSkeletons(toolCalls) {
  * 1 次决策 LLM → reply / generate(elaborate+generate) / canvas / irrelevant
  */
 exports.drawAgent = async (req, res) => {
-  const methodName = 'drawAgent';
+  const methodName = "drawAgent";
   const messages = req.body.messages;
 
   if (!messages?.length) {
-    const msg = 'need messages';
+    const msg = "need messages";
     req.logger.error(methodName, msg);
     res.jsonFail(msg);
     return;
@@ -166,14 +180,15 @@ exports.drawAgent = async (req, res) => {
   // 启动 SSE
   res.streamingStart();
 
-  const lastUserMsg = messages.filter((m) => m.role === 'user').pop()?.content || '';
-  req.logger.info(methodName, 'lastUserMsg', lastUserMsg);
+  const lastUserMsg =
+    messages.filter((m) => m.role === "user").pop()?.content || "";
+  req.logger.info(methodName, "lastUserMsg", lastUserMsg);
   chatFeishuMsg(req, `${lastUserMsg}`);
 
   // 清洗对话历史：将前端确认文案替换为简短标记，避免 LLM 模仿自然语言回复
   const cleanedMessages = messages.map((m) => {
-    if (m.role === 'assistant' && /^[✅❌]/.test(m.content)) {
-      return { ...m, content: '[已执行]' };
+    if (m.role === "assistant" && /^[✅❌]/.test(m.content)) {
+      return { ...m, content: "[已执行]" };
     }
     return m;
   });
@@ -181,65 +196,18 @@ exports.drawAgent = async (req, res) => {
   try {
     const startTime = Date.now();
 
-    // 1. Agent 决策
-    res.streaming(`data: ${JSON.stringify({ type: 'status', step: 'thinking' })}\n\n`);
-
-    let decision = null;
-    try {
-      await runAgents([
-        {
-          agentStartCallback: () => {
-            req.logger.info(methodName, 'step: agent decision');
-          },
-          agentRequestOptions: {
-            llm,
-            modelName,
-            thinking,
-            isJson: true,
-            messages: [{ role: 'system', content: prompts.AGENT_PROMPT }, ...cleanedMessages],
-          },
-          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: 'welcome' })}\n\n`);
-    } else if (decision.action === 'generate') {
-      let elaboration = '';
+    // LLM Mermaid 管线函数
+    const runLLMMermaidPipeline = async () => {
+      let elaboration = "";
 
       await runAgents([
         // elaborate
         {
           agentStartCallback: () => {
-            res.streaming(`data: ${JSON.stringify({ type: 'status', step: 'elaborate' })}\n\n`);
-            req.logger.info(methodName, 'step: elaborate');
+            res.streaming(
+              `data: ${JSON.stringify({ type: "status", step: "elaborate" })}\n\n`,
+            );
+            req.logger.info(methodName, "step: elaborate");
           },
           agentRequestOptions: {
             llm,
@@ -248,25 +216,31 @@ exports.drawAgent = async (req, res) => {
             get messages() {
               return [
                 {
-                  role: 'user',
-                  content: prompts.ELABORATE_PROMPT.replace('{input}', decision.description).replace(
-                    '{diagramType}',
-                    decision.diagramType,
-                  ),
+                  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) + '...');
+            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');
+            res.streaming(
+              `data: ${JSON.stringify({ type: "status", step: "generate" })}\n\n`,
+            );
+            req.logger.info(methodName, "step: generate");
           },
           agentRequestOptions: {
             llm,
@@ -275,11 +249,11 @@ exports.drawAgent = async (req, res) => {
             get messages() {
               return [
                 {
-                  role: 'user',
-                  content: prompts.GENERATE_PROMPT.replace('{diagramType}', decision.diagramType).replace(
-                    '{elaboration}',
-                    elaboration,
-                  ),
+                  role: "user",
+                  content: prompts.GENERATE_PROMPT.replace(
+                    "{diagramType}",
+                    decision.diagramType,
+                  ).replace("{elaboration}", elaboration),
                 },
               ];
             },
@@ -289,64 +263,251 @@ exports.drawAgent = async (req, res) => {
             const duration = Date.now() - startTime;
             req.logger.info(
               methodName,
-              'mermaidCode',
-              String(mermaidCode).slice(0, 100) + '...',
+              "mermaidCode",
+              String(mermaidCode).slice(0, 100) + "...",
               `total: ${duration}ms`,
             );
             chatFeishuMsg(req, `mermaid-${mermaidCode}`);
-            res.streaming(`data: ${JSON.stringify({ type: 'mermaid', code: mermaidCode, duration })}\n\n`);
+            res.streaming(
+              `data: ${JSON.stringify({ type: "mermaid", code: mermaidCode, duration })}\n\n`,
+            );
           },
         },
       ]);
-    } else if (decision.action === 'canvas') {
+    };
+
+    // 1. Agent 决策
+    res.streaming(
+      `data: ${JSON.stringify({ type: "status", step: "thinking" })}\n\n`,
+    );
+
+    let decision = null;
+    try {
+      await runAgents([
+        {
+          agentStartCallback: () => {
+            req.logger.info(methodName, "step: agent decision");
+          },
+          agentRequestOptions: {
+            llm,
+            modelName,
+            thinking,
+            isJson: true,
+            messages: [
+              { role: "system", content: prompts.AGENT_PROMPT },
+              ...cleanedMessages,
+            ],
+          },
+          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: "welcome" })}\n\n`);
+    } else if (decision.action === "generate") {
+      // Docker 模式优先，降级到 LLM Mermaid
+      const relayConfig = global.QZ_CONFIG.relay;
+      const useDocker = relayConfig?.wsUrl;
+
+      if (useDocker) {
+        try {
+          req.logger.info(methodName, "using Docker mode via relay");
+          res.streaming(
+            `data: ${JSON.stringify({ type: "status", step: "docker" })}\n\n`,
+          );
+
+          // SSE keepalive 防超时
+          const keepalive = setInterval(() => {
+            res.streaming(": keepalive\n\n");
+          }, 15000);
+
+          const result = await sendPromptViaRelay({
+            wsUrl: relayConfig.wsUrl,
+            authSecret: relayConfig.authSecret,
+            userId: req.user?.id || "guest",
+            prompt: JSON.stringify({ messages }),
+            timeout: relayConfig.timeout || 600000,
+          });
+
+          clearInterval(keepalive);
+
+          // 解析结果
+          if (result.startsWith("[IRRELEVANT]")) {
+            res.streaming(`data: ${JSON.stringify({ type: "welcome" })}\n\n`);
+          } else if (result.startsWith("[REPLY]")) {
+            const replyText = result.slice(7).trim();
+            res.streaming(
+              `data: ${JSON.stringify({ type: "message", content: replyText })}\n\n`,
+            );
+          } else {
+            // 尝试解析为 excalidraw JSON
+            try {
+              const data = JSON.parse(result);
+              if (data.type === "excalidraw" && data.elements) {
+                const duration = Date.now() - startTime;
+                req.logger.info(
+                  methodName,
+                  "excalidraw elements",
+                  data.elements.length,
+                  `${duration}ms`,
+                );
+                chatFeishuMsg(
+                  req,
+                  `docker-excalidraw-${data.elements.length}-elements`,
+                );
+                res.streaming(
+                  `data: ${JSON.stringify({ type: "draw", elements: data.elements, duration })}\n\n`,
+                );
+              } else {
+                // 未知格式，当文字回复
+                res.streaming(
+                  `data: ${JSON.stringify({ type: "message", content: result })}\n\n`,
+                );
+              }
+            } catch {
+              // JSON 解析失败，当文字回复
+              res.streaming(
+                `data: ${JSON.stringify({ type: "message", content: result })}\n\n`,
+              );
+            }
+          }
+        } catch (dockerErr) {
+          req.logger.error(
+            methodName,
+            "Docker mode failed, fallback to LLM",
+            dockerErr.message,
+          );
+          // 降级到 LLM Mermaid 管线
+          await runLLMMermaidPipeline();
+        }
+      } else {
+        // 无 relay 配置，直接走 LLM Mermaid
+        await runLLMMermaidPipeline();
+      }
+    } else if (decision.action === "canvas") {
       // 3. Canvas - Function Calling 直接画图
-      res.streaming(`data: ${JSON.stringify({ type: 'status', step: 'drawing' })}\n\n`);
-      req.logger.info(methodName, 'step: canvas function calling');
+      res.streaming(
+        `data: ${JSON.stringify({ type: "status", step: "drawing" })}\n\n`,
+      );
+      req.logger.info(methodName, "step: canvas function calling");
 
       const canvasResponse = await llm.chat({
         model: modelName,
         messages: [
-          { role: 'system', content: prompts.CANVAS_SYSTEM_PROMPT },
-          { role: 'user', content: decision.description },
+          { role: "system", content: prompts.CANVAS_SYSTEM_PROMPT },
+          { role: "user", content: decision.description },
         ],
         tools: prompts.CANVAS_TOOLS,
-        tool_choice: 'auto',
+        tool_choice: "auto",
       });
 
       if (canvasResponse.tool_calls?.length) {
-        req.logger.info(methodName, 'tool_calls', canvasResponse.tool_calls.map((tc) => tc.function.name).join(', '));
+        req.logger.info(
+          methodName,
+          "tool_calls",
+          canvasResponse.tool_calls.map((tc) => tc.function.name).join(", "),
+        );
 
         // 检查是否有 clear_canvas
-        const hasClear = canvasResponse.tool_calls.some((tc) => tc.function.name === 'clear_canvas');
+        const hasClear = canvasResponse.tool_calls.some(
+          (tc) => tc.function.name === "clear_canvas",
+        );
         if (hasClear) {
-          chatFeishuMsg(req, 'canvas-clear');
-          res.streaming(`data: ${JSON.stringify({ type: 'clear' })}\n\n`);
+          chatFeishuMsg(req, "canvas-clear");
+          res.streaming(`data: ${JSON.stringify({ type: "clear" })}\n\n`);
         }
 
         // 转换为 skeletons
-        const skeletons = convertToolCallsToSkeletons(canvasResponse.tool_calls);
+        const skeletons = convertToolCallsToSkeletons(
+          canvasResponse.tool_calls,
+        );
         if (skeletons.length > 0) {
           const duration = Date.now() - startTime;
-          req.logger.info(methodName, 'canvas elements', skeletons.length, `${duration}ms`);
+          req.logger.info(
+            methodName,
+            "canvas elements",
+            skeletons.length,
+            `${duration}ms`,
+          );
           chatFeishuMsg(req, `canvas-${skeletons.length}-elements`);
-          res.streaming(`data: ${JSON.stringify({ type: 'draw', elements: skeletons, duration })}\n\n`);
+          res.streaming(
+            `data: ${JSON.stringify({ type: "draw", elements: skeletons, duration })}\n\n`,
+          );
         }
       } else if (canvasResponse.content) {
         // LLM 选择文字回复而非调用函数
-        res.streaming(`data: ${JSON.stringify({ type: 'message', content: canvasResponse.content })}\n\n`);
+        res.streaming(
+          `data: ${JSON.stringify({ type: "message", content: canvasResponse.content })}\n\n`,
+        );
       }
-    } else if (decision.action === 'image') {
+    } else if (decision.action === "image") {
       // 4. Image - 调用 LibLib 图片生成 API
-      res.streaming(`data: ${JSON.stringify({ type: 'status', step: 'image' })}\n\n`);
-      req.logger.info(methodName, 'step: image generation', decision.prompt);
+      res.streaming(
+        `data: ${JSON.stringify({ type: "status", step: "image" })}\n\n`,
+      );
+      req.logger.info(methodName, "step: image generation", decision.prompt);
       chatFeishuMsg(req, `image-${decision.prompt}`);
 
-      const submitRes = await liblib.text2img(decision.prompt, { aspectRatio: 'square' });
+      const submitRes = await liblib.text2img(decision.prompt, {
+        aspectRatio: "square",
+      });
       const generateUuid = submitRes.data?.generateUuid;
 
       if (!generateUuid) {
-        req.logger.error(methodName, 'liblib submit failed', JSON.stringify(submitRes));
-        res.streaming(`data: ${JSON.stringify({ type: 'message', content: '图片生成失败，请稍后重试。' })}\n\n`);
+        req.logger.error(
+          methodName,
+          "liblib submit failed",
+          JSON.stringify(submitRes),
+        );
+        res.streaming(
+          `data: ${JSON.stringify({ type: "message", content: "图片生成失败，请稍后重试。" })}\n\n`,
+        );
       } else {
         const result = await liblib.waitForResult(generateUuid);
         const imageUrl = result.images?.[0]?.imageUrl;
@@ -355,39 +516,52 @@ exports.drawAgent = async (req, res) => {
           // 下载图片转 base64
           const imgResponse = await fetch(imageUrl);
           const imgBuffer = Buffer.from(await imgResponse.arrayBuffer());
-          const contentType = imgResponse.headers.get('content-type') || 'image/jpeg';
-          const dataURL = `data:${contentType};base64,${imgBuffer.toString('base64')}`;
+          const contentType =
+            imgResponse.headers.get("content-type") || "image/jpeg";
+          const dataURL = `data:${contentType};base64,${imgBuffer.toString("base64")}`;
 
           const duration = Date.now() - startTime;
-          req.logger.info(methodName, 'image generated', `${duration}ms`);
-          res.streaming(`data: ${JSON.stringify({ type: 'image', dataURL, duration })}\n\n`);
+          req.logger.info(methodName, "image generated", `${duration}ms`);
+          res.streaming(
+            `data: ${JSON.stringify({ type: "image", dataURL, duration })}\n\n`,
+          );
         } else {
-          req.logger.error(methodName, 'liblib no image url', JSON.stringify(result));
-          res.streaming(`data: ${JSON.stringify({ type: 'message', content: '图片生成失败，请稍后重试。' })}\n\n`);
+          req.logger.error(
+            methodName,
+            "liblib no image url",
+            JSON.stringify(result),
+          );
+          res.streaming(
+            `data: ${JSON.stringify({ type: "message", content: "图片生成失败，请稍后重试。" })}\n\n`,
+          );
         }
       }
-    } else if (decision.action === 'sketch') {
+    } else if (decision.action === "sketch") {
       // 5. Sketch - 通知前端执行线稿提取（处理在前端完成）
       const duration = Date.now() - startTime;
-      req.logger.info(methodName, 'step: sketch', `${duration}ms`);
-      chatFeishuMsg(req, 'sketch');
-      res.streaming(`data: ${JSON.stringify({ type: 'sketch', duration })}\n\n`);
+      req.logger.info(methodName, "step: sketch", `${duration}ms`);
+      chatFeishuMsg(req, "sketch");
+      res.streaming(
+        `data: ${JSON.stringify({ type: "sketch", duration })}\n\n`,
+      );
     } else {
       // 未知 action fallback
       res.streaming(
-        `data: ${JSON.stringify({ type: 'message', content: decision.message || '可以再说一次吗？' })}\n\n`,
+        `data: ${JSON.stringify({ type: "message", content: decision.message || "可以再说一次吗？" })}\n\n`,
       );
     }
 
     res.streamingEnd();
   } catch (error) {
-    req.logger.error(methodName, 'error', error);
+    req.logger.error(methodName, "error", error);
     errorFeishuMsg(req, error.message);
     // 对用户隐藏内部错误细节
-    const userMessage = error.message?.includes('Cannot parse JSON')
-      ? '请求处理失败，请重新描述你的需求'
-      : '服务暂时出错，请稍后重试';
-    res.streaming(`data: ${JSON.stringify({ type: 'error', message: userMessage })}\n\n`);
+    const userMessage = error.message?.includes("Cannot parse JSON")
+      ? "请求处理失败，请重新描述你的需求"
+      : "服务暂时出错，请稍后重试";
+    res.streaming(
+      `data: ${JSON.stringify({ type: "error", message: userMessage })}\n\n`,
+    );
     res.streamingEnd();
   }
 };