iflow-feishu 1.1.2 → 1.1.3

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "iflow-feishu",
-  "version": "1.1.2",
+  "version": "1.1.3",
   "description": "iFlow CLI 飞书插件 - 将 iFlow AI 助手接入飞书机器人",
   "main": "src/index.js",
   "bin": {

package/src/core/stream-handler.js

@@ -30,8 +30,16 @@ class StreamHandler {
     }
     
     const maxTokens = this.getMaxTokens(modelName);
+    
+    // 首先尝试从 Execution Info 提取
     let contentLeftPercent = this.extractTokenUsage(text, maxTokens);
     
+    // 如果没有，尝试从回复文本中提取（三元会在回复中报告上下文剩余）
+    if (contentLeftPercent === null) {
+      contentLeftPercent = this.extractContextFromContent(text);
+    }
+    
+    // 最后尝试从日志读取
     if (contentLeftPercent === null && chatId) {
       contentLeftPercent = this.getTokenUsageFromLogs(modelName);
     }
@@ -43,6 +51,31 @@ class StreamHandler {
     return { reasoning, content: content || '(无法提取响应)', contentLeftPercent };
   }
 
+  /**
+   * 从回复内容中提取上下文剩余百分比
+   * 匹配格式: "📊 剩余 ~77%", "剩余 ~77%", "剩余 77%" 等
+   */
+  extractContextFromContent(text) {
+    // 匹配多种格式: 剩余 ~77%, 剩余 77%, 📊 剩余 ~77%
+    const patterns = [
+      /剩余\s*~?(\d+)%/i,
+      /📊\s*剩余\s*~?(\d+)%/i,
+      /remaining[:\s]*~?(\d+)%/i
+    ];
+    
+    for (const pattern of patterns) {
+      const match = text.match(pattern);
+      if (match) {
+        const percent = parseInt(match[1], 10);
+        if (percent >= 0 && percent <= 100) {
+          logger.info(`📊 从回复内容提取上下文剩余: ${percent}%`);
+          return percent;
+        }
+      }
+    }
+    return null;
+  }
+
   /**
    * 获取模型最大 token 数
    */