@caupulican/pi-adaptative 0.81.17 → 0.81.19

package/dist/core/agent-session.js

@@ -1613,6 +1613,7 @@ export class AgentSession {
         // accumulate until the interactive Node process hits the V8 heap limit.
         if (event.type === "message_end") {
             compactToolResultDetailsForRetention(event.message);
+            let messagePersisted = false;
             // While a cheap routed turn is buffering, its messages are captured for later flush/discard
             // instead of persisted here (see ModelRouterController.captureSessionMessage).
             if (this._modelRouter.captureSessionMessage(event.message)) {
@@ -1622,18 +1623,23 @@ export class AgentSession {
             else if (event.message.role === "custom") {
                 // Persist as CustomMessageEntry
                 this.sessionManager.appendCustomMessageEntry(event.message.customType, event.message.content, event.message.display, event.message.details);
+                messagePersisted = true;
             }
             else if (event.message.role === "user" ||
                 event.message.role === "assistant" ||
                 event.message.role === "toolResult") {
                 // Regular LLM message - persist as SessionMessageEntry
                 this.sessionManager.appendMessage(event.message);
+                messagePersisted = true;
             }
             // Other message types (bashExecution, compactionSummary, branchSummary) are persisted elsewhere
             // Track assistant message for auto-compaction (checked on agent_end)
             if (event.message.role === "assistant") {
                 this._lastAssistantMessage = event.message;
                 const assistantMsg = event.message;
+                if (messagePersisted) {
+                    this._pipeline.observeProviderUsage(this.agent.state.messages, assistantMsg);
+                }
                 if (assistantMsg.stopReason !== "error") {
                     this._overflowRecoveryAttempted = false;
                 }
@@ -3033,6 +3039,18 @@ export class AgentSession {
         }
         return false;
     }
+    _measureLiveContextTokensForCompaction() {
+        const estimatedTokens = this._pipeline.estimateCurrentContextTokens(this.agent.state.messages);
+        const assistantMessage = findLastAssistantMessage(this.agent.state.messages);
+        if (!assistantMessage || assistantMessage.stopReason === "error" || assistantMessage.stopReason === "aborted") {
+            return estimatedTokens;
+        }
+        const compactionEntry = getLatestCompactionEntry(this.sessionManager.getBranch());
+        if (compactionEntry && assistantMessage.timestamp <= new Date(compactionEntry.timestamp).getTime()) {
+            return estimatedTokens;
+        }
+        return Math.max(calculateContextTokens(assistantMessage.usage), estimatedTokens);
+    }
     /**
      * Internal: Run auto-compaction with events.
      */
@@ -3062,7 +3080,7 @@ export class AgentSession {
             const margin = Math.max(0, Math.floor(0.01 * contextWindow));
             const outcome = await runCompactionLoop({
                 getBranch: () => this.sessionManager.getBranch(),
-                measureLiveTokens: () => this._pipeline.estimateCurrentContextTokens(this.agent.state.messages),
+                measureLiveTokens: () => this._measureLiveContextTokensForCompaction(),
                 shouldCompact: reason === "overflow"
                     ? () => true
                     : (tokens) => shouldCompact(tokens, contextWindow, settings, model.autoCompactionTriggerTokens),
@@ -3771,4 +3789,13 @@ export class AgentSession {
         return this._extensionRunner;
     }
 }
+function findLastAssistantMessage(messages) {
+    for (let i = messages.length - 1; i >= 0; i--) {
+        const message = messages[i];
+        if (message.role === "assistant") {
+            return message;
+        }
+    }
+    return undefined;
+}
 //# sourceMappingURL=agent-session.js.map