st-comp 0.0.253 → 0.0.255

package/es/aiTools.js

@@ -27,7 +27,7 @@ export const sendToBaiLianAppNonStreaming = async ({ appId, apiKey, value }) =>
     throw error;
   }
 };
-// 流式返回
+// 流式返回 - 支持错误处理
 export const sendToBaiLianAppStreaming = async ({ appId, apiKey, value, callback }) => {
   try {
     const response = await fetch(`https://dashscope.aliyuncs.com/api/v1/apps/${appId}/completion`, {
@@ -43,13 +43,17 @@ export const sendToBaiLianAppStreaming = async ({ appId, apiKey, value, callback
         "X-DashScope-SSE": "enable",
       },
     });
-    
-    if (!response.ok) throw new Error(`HTTP ${response.status}`);
-    
+
+    if (!response.ok) {
+      callback("error", `HTTP ${response.status}: ${response.statusText}`);
+      return;
+    }
+
     const reader = response.body.getReader();
     const decoder = new TextDecoder();
     let buffer = "";
-    
+    let currentEvent = null; // 记录当前 event 类型
+
     while (true) {
       const { done, value } = await reader.read();
       if (done) {
@@ -60,28 +64,48 @@ export const sendToBaiLianAppStreaming = async ({ appId, apiKey, value, callback
         callback("finish", "");
         break;
       }
-      
+
       buffer += decoder.decode(value, { stream: true });
-      
+
       let newlineIndex;
       while ((newlineIndex = buffer.indexOf("\n")) !== -1) {
         const line = buffer.substring(0, newlineIndex).trim();
         buffer = buffer.substring(newlineIndex + 1);
-        
+
+        // 处理 event: 行（错误类型）
+        if (line.startsWith("event:")) {
+          currentEvent = line.substring(6).trim();
+          continue;
+        }
+
+        // 处理 data: 行
         if (line.startsWith("data:")) {
-          const data = line.substring(5).trim();
-          if (data && data !== "[DONE]") {
+          const dataStr = line.substring(5).trim();
+
+          // 检查是否是错误响应
+          if (currentEvent === "error" || (dataStr && dataStr.startsWith("{"))) {
             try {
-              const parsed = JSON.parse(data);
+              const parsed = JSON.parse(dataStr);
+
+              // 处理错误响应
+              if (parsed.code || parsed.message || currentEvent === "error") {
+                const errorMsg = parsed.message || parsed.code || "未知错误";
+                callback("error", `百炼服务端错误: ${errorMsg}`);
+                return; // 终止流式处理
+              }
+
+              // 正常响应：提取 text
               const text = parsed?.output?.text;
               if (text && callback) {
                 callback("message", text);
               }
             } catch (e) {
-              // 可能是部分数据，继续等待
-              console.debug("等待完整数据...");
+              console.debug("JSON 解析失败:", dataStr);
             }
           }
+
+          // 重置 event 类型
+          currentEvent = null;
         }
       }
     }
@@ -92,18 +116,35 @@ export const sendToBaiLianAppStreaming = async ({ appId, apiKey, value, callback
 };
 
 // 辅助函数：处理缓冲区剩余数据
-function tryProcessBuffer(buffer, callback) {
+const tryProcessBuffer = (buffer, callback) => {
   const lines = buffer.split("\n");
+  let currentEvent = null;
+
   for (const line of lines) {
+    if (line.startsWith("event:")) {
+      currentEvent = line.substring(6).trim();
+      continue;
+    }
+
     if (line.startsWith("data:")) {
-      const data = line.substring(5).trim();
-      if (data && data !== "[DONE]") {
+      const dataStr = line.substring(5).trim();
+      if (dataStr && dataStr !== "[DONE]") {
         try {
-          const parsed = JSON.parse(data);
+          const parsed = JSON.parse(dataStr);
+
+          // 错误响应
+          if (currentEvent === "error" || parsed.code || parsed.message) {
+            const errorMsg = parsed.message || parsed.code || "未知错误";
+            callback("error", `百炼服务端错误: ${errorMsg}`);
+            return;
+          }
+
+          // 正常响应
           const text = parsed?.output?.text;
           if (text && callback) callback("message", text);
         } catch (e) {}
       }
+      currentEvent = null;
     }
   }
-}
+};