byt-lingxiao-ai 0.3.23 → 0.3.25

package/components/utils/StreamParser.js

@@ -1,135 +1,181 @@
 export class StreamParser {
   constructor(options = {}) {
     this.options = {
-      updateInterval: 16, // 每16ms刷新一次（约60fps）
-      debug: false,
+      updateInterval: 16,
       ...options
-    };
-    this.reset();
+    }
+    this.reset()
   }
 
-  // 重置解析器状态
   reset() {
-    this.buffer = '';
-    this.contentBuffer = [];
-    this.thinkingBuffer = [];
-    this.inTag = false;
-    this.tagBuffer = '';
-    this.updateTimer = null;
-    this.status = 'output'; // thinking | output
+    this.inTag = false
+    this.tagBuffer = ''
+    this.currentType = 'content'
+    this.currentEvent = null
+    this.updateTimer = null
   }
 
-  // 处理接收到的流数据块
-  processChunk(chunk, callback) {
-    if (!chunk) return;
-    this.parseContent(chunk, callback);
+  processChunk(chunk, emit) {
+    if (!chunk) return
+    this.parse(chunk, emit)
   }
 
-  // 核心内容解析，支持 <think> 标签
-  parseContent(content, callback) {
-    let i = 0;
+  parse(text, emit) {
+    let i = 0
 
-    while (i < content.length) {
+    while (i < text.length) {
       if (this.inTag) {
-        const endIndex = content.indexOf('>', i);
-        if (endIndex !== -1) {
-          this.tagBuffer += content.substring(i, endIndex + 1);
-          this.handleTag(this.tagBuffer);
-          this.tagBuffer = '';
-          this.inTag = false;
-          i = endIndex + 1;
+        const end = text.indexOf('>', i)
+        if (end !== -1) {
+          this.tagBuffer += text.slice(i, end + 1)
+          this.handleTag(this.tagBuffer, emit)
+          this.tagBuffer = ''
+          this.inTag = false
+          i = end + 1
         } else {
-          // 标签未闭合，超过50字符强制输出，防止阻塞
-          this.tagBuffer += content.substring(i);
-          if (this.tagBuffer.length > 50) {
-            this.appendText(this.tagBuffer);
-            this.tagBuffer = '';
-            this.inTag = false;
-          }
-          break;
+          this.tagBuffer += text.slice(i)
+          break
         }
       } else {
-        const startIndex = content.indexOf('<', i);
-        if (startIndex !== -1) {
-          if (startIndex > i) this.appendText(content.substring(i, startIndex));
-
-          const endIndex = content.indexOf('>', startIndex);
-          if (endIndex !== -1) {
-            const tag = content.substring(startIndex, endIndex + 1);
-            this.handleTag(tag);
-            i = endIndex + 1;
+        const lt = text.indexOf('<', i)
+        if (lt !== -1) {
+          if (lt > i) this.append(text.slice(i, lt), emit)
+
+          const gt = text.indexOf('>', lt)
+          if (gt !== -1) {
+            this.handleTag(text.slice(lt, gt + 1), emit)
+            i = gt + 1
           } else {
-            const nextChar = content[startIndex + 1];
-            if (!/[a-zA-Z/]/.test(nextChar)) {
-              // 很可能不是标签，直接当文本输出
-              this.appendText('<');
-              i = startIndex + 1;
-            } else {
-              this.inTag = true;
-              this.tagBuffer = content.substring(startIndex);
-              break;
-            }
+            this.inTag = true
+            this.tagBuffer = text.slice(lt)
+            break
           }
         } else {
-          this.appendText(content.substring(i));
-          break;
+          this.append(text.slice(i), emit)
+          break
         }
       }
     }
 
-    this.scheduleUpdate(callback);
+    this.scheduleFlush(emit)
   }
 
-  // 处理标签
-  handleTag(tag) {
-    const t = tag.toLowerCase();
-    if (t === '<think>') this.status = 'thinking';
-    else if (t === '</think>') this.status = 'output';
+  handleTag(tag, emit) {
+    const t = tag.toLowerCase().trim()
+
+    if (t.startsWith('<think')) {
+      this.startThinking(emit)
+    }
+    else if (t.startsWith('</think')) {
+      this.closeThinking()
+      this.currentType = 'content'
+      this.currentEvent = null
+    }
+    else if (t.startsWith('<tool_call')) {
+      const attrs = this.parseTagAttributes(tag)
+      this.switchType('tool_call', emit, attrs.name)
+    }
+    else if (t.startsWith('</tool_call')) {
+      this.switchType('content', emit)
+    }
+    else if (t.startsWith('<tool_result')) {
+      this.switchType('tool_result', emit)
+    }
+    else if (t.startsWith('</tool_result')) {
+      this.switchType('content', emit)
+    }
+    else {
+      this.append(tag, emit)
+    }
+  }
+
+
+  startThinking(emit) {
+    // 如果上一个 thinking 还没结算，先结算
+    this.closeThinking()
+
+    this.currentType = 'thinking'
+    this.currentEvent = {
+      id: this.uuid(),
+      type: 'thinking',
+      content: '',
+      startTime: Date.now(),
+      endTime: null,
+      duration: null
+    }
+
+    emit({ type: 'create', event: this.currentEvent })
   }
 
-  // 添加文本到缓冲区
-  appendText(text) {
-    if (!text) return;
-    if (this.status === 'thinking') this.thinkingBuffer.push(text);
-    else this.contentBuffer.push(text);
+  closeThinking() {
+    if (
+      this.currentEvent &&
+      this.currentEvent.type === 'thinking' &&
+      !this.currentEvent.endTime
+    ) {
+      this.currentEvent.endTime = Date.now()
+      this.currentEvent.duration = (
+        (this.currentEvent.endTime - this.currentEvent.startTime) / 1000
+      ).toFixed(2)
+      console.log(this.currentEvent.endTime - this.currentEvent.startTime)
+    }
   }
 
-  // 防抖刷新
-  scheduleUpdate(callback) {
-    if (this.updateTimer) return;
-    this.updateTimer = setTimeout(() => {
-      this.flush(callback);
-      this.updateTimer = null;
-    }, this.options.updateInterval);
+  switchType(type) {
+    // 如果是从 thinking 切走，结算时间
+    if (this.currentEvent?.type === 'thinking') {
+      this.closeThinking()
+    }
+
+    this.currentType = type
+    this.currentEvent = null
   }
 
-  // 刷新缓冲区
-  flush(callback) {
-    if (!callback) return;
-    const result = {
-      thinking: this.thinkingBuffer.length ? this.thinkingBuffer.join('') : null,
-      content: this.contentBuffer.length ? this.contentBuffer.join('') : null,
-      status: this.status
-    };
-    this.thinkingBuffer = [];
-    this.contentBuffer = [];
-    callback(result);
+  append(text, emit) {
+    if (!text) return
+
+    if (!this.currentEvent || this.currentEvent.type !== this.currentType) {
+      this.currentEvent = {
+        id: this.uuid(),
+        type: this.currentType,
+        content: ''
+      }
+      emit({ type: 'create', event: this.currentEvent })
+    }
+
+    this.currentEvent.content += text
+    emit({ type: 'append', event: this.currentEvent })
   }
 
-  // 完成解析
-  finish(callback) {
-    if (this.inTag && this.tagBuffer) {
-      this.appendText(this.tagBuffer);
-      this.tagBuffer = '';
-      this.inTag = false;
+  parseTagAttributes(tag) {
+    const attrRegex = /(\w+)=['"]([^'"]+)['"]/g
+    const attrs = {}
+    let match
+    
+    while ((match = attrRegex.exec(tag)) !== null) {
+      attrs[match[1]] = match[2]
     }
-    this.flush(callback);
-    if (this.updateTimer) clearTimeout(this.updateTimer);
+    
+    return attrs
+  }
+
+  scheduleFlush(emit) {
+    if (this.updateTimer) return
+    this.updateTimer = setTimeout(() => {
+      emit({ type: 'flush' })
+      this.updateTimer = null
+    }, this.options.updateInterval)
+  }
+
+  finish(emit) {
+    // 结束时如果还在 thinking，也要结算
+    this.closeThinking()
+    emit({ type: 'flush' })
+    if (this.updateTimer) clearTimeout(this.updateTimer)
+    this.reset()
   }
 
-  // 销毁解析器
-  destroy() {
-    if (this.updateTimer) clearTimeout(this.updateTimer);
-    this.reset();
+  uuid() {
+    return Math.random().toString(36).slice(2)
   }
-}
+}