Upload 2 files

src/index.js

@@ -137,15 +137,70 @@ const makeRequest = async (session_id, requestModel, messages) => {
   }
 }
 
+// 用于存储未完成的数据块
+let dataBuffer = '';
+
 const parseStreamData = (data) => {
   try {
     if (!data) return null
-    const cleanData = data.replace("data: ", '').trim()
-    if (!cleanData || cleanData === '[DONE]') return null
-    return JSON.parse(cleanData)
+    
+    // 将新数据添加到缓冲区
+    dataBuffer += data;
+    
+    // 查找完整的JSON对象
+    const matches = dataBuffer.match(/data: ({[^}]+})/g);
+    if (!matches) return null;
+    
+    // 处理找到的每个完整JSON对象
+    for (const match of matches) {
+      try {
+        const cleanData = match.replace("data: ", '').trim();
+        if (cleanData === '[DONE]') return null;
+        
+        // 尝试解析JSON
+        const parsed = JSON.parse(cleanData);
+        
+        // 从缓冲区移除已处理的数据
+        dataBuffer = dataBuffer.slice(dataBuffer.indexOf(match) + match.length);
+        
+        // 返回解析后的对象
+        return parsed;
+      } catch (parseError) {
+        // 如果解析失败，尝试提取字段
+        const messageIdMatch = cleanData.match(/"message_id":\s*"([^"]+)"/);
+        const fieldNameMatch = cleanData.match(/"field_name":\s*"([^"]+)"/);
+        const fieldValueMatch = cleanData.match(/"field_value":\s*"([^]*?)(?:(?<!\\)",|$)/);
+        
+        if (messageIdMatch && fieldNameMatch) {
+          const fieldValue = fieldValueMatch ? fieldValueMatch[1] : '';
+          return {
+            message_id: messageIdMatch[1],
+            field_name: fieldNameMatch[1],
+            field_value: fieldValue,
+            type: 'text'
+          };
+        }
+      }
+    }
+    
+    // 如果缓冲区过大，清理它
+    if (dataBuffer.length > 10000) {
+      console.warn('缓冲区过大，正在清理...');
+      dataBuffer = dataBuffer.slice(-5000);
+    }
+    
+    return {
+      field_name: 'content',
+      field_value: '',
+      type: 'text'
+    };
   } catch (error) {
-    console.error('解析数据出错:', error, '原始数据:', data)
-    return null
+    console.error('解析数据出错:', error, '原始数据:', data);
+    return {
+      field_name: 'content',
+      field_value: '',
+      type: 'text'
+    };
   }
 }
 
@@ -235,27 +290,34 @@ const handleChatCompletions = async (ctx) => {
           if (!content) continue
 
           if (stream == "true" || stream == true) {
-            if (content.delta) {
+            // 处理流式响应
+            const delta = content.delta || 
+                         (content.field_name === 'content' ? content.field_value : null)
+            
+            if (delta) {
               try {
-                ctx.res.write(`data: ${JSON.stringify({
+                const chunk = {
                   "id": `chatcmpl-${messageId}`,
                   "choices": [
                     {
                       "index": 0,
                       "delta": {
-                        "content": content.delta
+                        "content": delta
                       }
                     }
                   ],
                   "created": Math.floor(Date.now() / 1000),
                   "model": model,
                   "object": "chat.completion.chunk"
-                })}\n\n`)
+                };
+                
+                ctx.res.write(`data: ${JSON.stringify(chunk)}\n\n`)
               } catch (e) {
-                console.error('写入流数据失败:', e)
+                console.error('写入流数据失败:', e, '数据:', delta)
               }
             }
           } else {
+            // 处理非流式响应
             if (content.field_value && 
                 content.field_name !== 'session_state.answer_is_finished' && 
                 content.field_name !== 'content' &&