iflow-feishu 1.1.1 → 1.1.3

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "iflow-feishu",
-  "version": "1.1.1",
+  "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 数
    */
@@ -59,14 +92,19 @@ class StreamHandler {
     if (execInfoMatch) {
       try {
         const execInfo = JSON.parse(execInfoMatch[1]);
+        logger.info(`📊 解析到 Execution Info: ${JSON.stringify(execInfo.tokenUsage)}`);
         if (execInfo.tokenUsage && execInfo.tokenUsage.total > 0) {
           const used = execInfo.tokenUsage.total;
           const remaining = Math.max(0, maxTokens - used);
-          return Math.round((remaining / maxTokens) * 100);
+          const percent = Math.round((remaining / maxTokens) * 100);
+          logger.info(`📊 Token 计算: 已用=${used}, 剩余=${remaining}, 百分比=${percent}%`);
+          return percent;
         }
       } catch (e) {
-        // 解析失败，忽略
+        logger.warn(`📊 解析 Execution Info 失败: ${e.message}`);
       }
+    } else {
+      logger.info(`📊 未找到 Execution Info 标签`);
     }
     return null;
   }