byt-lingxiao-ai 0.2.8 → 0.3.0

package/components/utils/StreamParser.js

@@ -1,11 +1,3 @@
-/**
- * SSE (Server-Sent Events) 流解析器
- * 优化点：
- * 1. 使用状态机模式处理标签解析
- * 2. 批量更新减少DOM操作
- * 3. 内存优化：避免频繁字符串拼接
- * 4. 可配置的更新频率
- */
 export class StreamParser {
   constructor(options = {}) {
     this.options = {
@@ -46,6 +38,23 @@ export class StreamParser {
     this.metrics.chunks++;
     this.buffer += chunk;
 
+    if (this.options.debug) {
+      console.log('[StreamParser] 收到chunk:', chunk.substring(0, 100));
+    }
+
+    // 尝试解析为 SSE 格式
+    if (this.buffer.includes('data:')) {
+      this.processSSEFormat(callback);
+    } else {
+      // 直接处理纯文本流
+      this.processPlainText(callback);
+    }
+  }
+
+  /**
+   * 处理标准 SSE 格式
+   */
+  processSSEFormat(callback) {
     // SSE 格式：事件由双换行符分隔
     const events = this.buffer.split('\n\n');
     
@@ -56,15 +65,33 @@ export class StreamParser {
     for (const event of events) {
       if (event.trim()) {
         this.metrics.events++;
-        this.processEvent(event, callback);
+        this.processSSEEvent(event, callback);
+      }
+    }
+  }
+
+  /**
+   * 处理纯文本流格式
+   */
+  processPlainText(callback) {
+    const content = this.buffer;
+    this.buffer = ''; // 清空缓冲区
+    
+    if (content) {
+      this.metrics.chars += content.length;
+      
+      if (this.options.debug) {
+        console.log('[StreamParser] 处理纯文本:', content);
       }
+      
+      this.parseContent(content, callback);
     }
   }
 
   /**
    * 处理单个SSE事件
    */
-  processEvent(eventStr, callback) {
+  processSSEEvent(eventStr, callback) {
     const lines = eventStr.split('\n');
     
     for (const line of lines) {
@@ -83,11 +110,16 @@ export class StreamParser {
           
           if (content) {
             this.metrics.chars += content.length;
+            
+            if (this.options.debug) {
+              console.log('[StreamParser] 解析SSE内容:', content);
+            }
+            
             this.parseContent(content, callback);
           }
         } catch (error) {
           if (this.options.debug) {
-            console.warn('[StreamParser] JSON解析失败:', data);
+            console.warn('[StreamParser] JSON解析失败:', data, error);
           }
         }
       }
@@ -158,15 +190,16 @@ export class StreamParser {
   handleTag(tag) {
     const tagName = tag.toLowerCase();
     
+    if (this.options.debug) {
+      console.log('[StreamParser] 处理标签:', tag);
+    }
+    
     if (tagName === '<think>') {
       this.status = 'thinking';
     } else if (tagName === '</think>') {
       this.status = 'output';
     }
     // 可扩展：支持更多标签类型
-    // else if (tagName.startsWith('<code')) {
-    //   this.status = 'code';
-    // }
   }
 
   /**
@@ -198,7 +231,10 @@ export class StreamParser {
    * 立即刷新缓冲区
    */
   flush(callback) {
-    if (!callback) return;
+    if (!callback) {
+      console.warn('[StreamParser] flush: callback 为空');
+      return;
+    }
 
     const hasThinking = this.thinkingBuffer.length > 0;
     const hasContent = this.contentBuffer.length > 0;
@@ -212,12 +248,26 @@ export class StreamParser {
       status: this.status
     };
 
+    if (this.options.debug) {
+      console.log('[StreamParser] 刷新缓冲区:', {
+        thinking: result.thinking?.length || 0,
+        content: result.content?.length || 0,
+        status: result.status,
+        thinkingPreview: result.thinking?.substring(0, 50),
+        contentPreview: result.content?.substring(0, 50)
+      });
+    }
+
     // 清空缓冲区
     this.thinkingBuffer = [];
     this.contentBuffer = [];
 
     // 回调更新
-    callback(result);
+    try {
+      callback(result);
+    } catch (error) {
+      console.error('[StreamParser] 回调执行错误:', error);
+    }
   }
 
   /**
@@ -252,26 +302,4 @@ export class StreamParser {
     }
     this.reset();
   }
-}
-
-/**
- * 使用示例：
- * 
- * const parser = new StreamParser({ debug: true });
- * 
- * // 处理流
- * for await (const chunk of stream) {
- *   parser.processChunk(chunk, (result) => {
- *     if (result.thinking) {
- *       message.thinking += result.thinking;
- *     }
- *     if (result.content) {
- *       message.content += result.content;
- *     }
- *     this.$forceUpdate();
- *   });
- * }
- * 
- * // 完成
- * parser.finish();
- */
+}