st-comp 0.0.251 → 0.0.253

package/es/aiTools.js

@@ -30,40 +30,80 @@ export const sendToBaiLianAppNonStreaming = async ({ appId, apiKey, value }) =>
 // 流式返回
 export const sendToBaiLianAppStreaming = async ({ appId, apiKey, value, callback }) => {
   try {
-    const res = await fetch(`https://dashscope.aliyuncs.com/api/v1/apps/${appId}/completion`, {
+    const response = await fetch(`https://dashscope.aliyuncs.com/api/v1/apps/${appId}/completion`, {
       method: "POST",
       body: JSON.stringify({
         input: { prompt: value },
-        parameters: {
-          incremental_output: "true", // 流式返回
-        },
+        parameters: { incremental_output: "true" },
         debug: {},
       }),
       headers: {
         Authorization: `Bearer ${apiKey}`,
         "Content-Type": "application/json",
-        "X-DashScope-SSE": "enable", // 流式输出
+        "X-DashScope-SSE": "enable",
       },
     });
-    if (!res.ok || !res.body) throw new Error("请求失败");
-
-    const reader = res.body.getReader();
+    
+    if (!response.ok) throw new Error(`HTTP ${response.status}`);
+    
+    const reader = response.body.getReader();
     const decoder = new TextDecoder();
-
+    let buffer = "";
+    
     while (true) {
       const { done, value } = await reader.read();
-      if (done && callback) {
+      if (done) {
+        // 处理最后可能遗留的数据
+        if (buffer.trim()) {
+          tryProcessBuffer(buffer, callback);
+        }
         callback("finish", "");
         break;
       }
-      try {
-        const data = decoder.decode(value, { stream: true });
-        const resData = JSON.parse(data.split("\n")[3].substr(5));
-        const resText = resData?.output?.text;
-        if (resText && callback) callback("message", resText);
-      } catch (error) {}
+      
+      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);
+        
+        if (line.startsWith("data:")) {
+          const data = line.substring(5).trim();
+          if (data && data !== "[DONE]") {
+            try {
+              const parsed = JSON.parse(data);
+              const text = parsed?.output?.text;
+              if (text && callback) {
+                callback("message", text);
+              }
+            } catch (e) {
+              // 可能是部分数据，继续等待
+              console.debug("等待完整数据...");
+            }
+          }
+        }
+      }
     }
   } catch (error) {
-    console.error(error);
+    console.error("流式请求失败:", error);
+    callback("error", error.message);
   }
 };
+
+// 辅助函数：处理缓冲区剩余数据
+function tryProcessBuffer(buffer, callback) {
+  const lines = buffer.split("\n");
+  for (const line of lines) {
+    if (line.startsWith("data:")) {
+      const data = line.substring(5).trim();
+      if (data && data !== "[DONE]") {
+        try {
+          const parsed = JSON.parse(data);
+          const text = parsed?.output?.text;
+          if (text && callback) callback("message", text);
+        } catch (e) {}
+      }
+    }
+  }
+}