@dyyz1993/pi-coding-agent 0.70.3 → 0.70.4

package/dist/core/agent-session.js

@@ -2800,15 +2800,10 @@ export class AgentSession {
         const contextWindow = model.contextWindow ?? 0;
         if (contextWindow <= 0)
             return undefined;
-        // After compaction, the last assistant usage reflects pre-compaction context size.
-        // We can only trust usage from an assistant that responded after the latest compaction.
-        // If no such assistant exists, context token count is unknown until the next LLM response.
         const branchEntries = this.sessionManager.getBranch();
         const latestCompaction = getLatestCompactionEntry(branchEntries);
         if (latestCompaction) {
-            // Check if there's a valid assistant usage after the compaction boundary
             const compactionIndex = branchEntries.lastIndexOf(latestCompaction);
-            let hasPostCompactionUsage = false;
             for (let i = branchEntries.length - 1; i > compactionIndex; i--) {
                 const entry = branchEntries[i];
                 if (entry.type === "message" && entry.message.role === "assistant") {
@@ -2816,18 +2811,16 @@ export class AgentSession {
                     if (assistant.stopReason !== "aborted" && assistant.stopReason !== "error") {
                         const contextTokens = calculateContextTokens(assistant.usage);
                         if (contextTokens > 0) {
-                            hasPostCompactionUsage = true;
+                            const percent = (contextTokens / contextWindow) * 100;
+                            return { tokens: contextTokens, contextWindow, percent };
                         }
-                        break;
                     }
+                    break;
                 }
             }
-            if (!hasPostCompactionUsage) {
-                return { tokens: null, contextWindow, percent: null };
-            }
         }
         const estimate = estimateContextTokens(this.messages);
-        const percent = (estimate.tokens / contextWindow) * 100;
+        const percent = contextWindow > 0 ? (estimate.tokens / contextWindow) * 100 : 0;
         return {
             tokens: estimate.tokens,
             contextWindow,