@fleetagent/pi-coding-agent 0.0.7 → 0.0.9

package/dist/core/agent-session.js

@@ -168,6 +168,7 @@ export class AgentSession {
         // (session persistence, extensions, auto-compaction, retry logic)
         this._unsubscribeAgent = this.agent.subscribe(this._handleAgentEvent);
         this._installAgentToolHooks();
+        this._installAgentTurnPreparation();
         this._buildRuntime({
             activeToolNames: this._initialActiveToolNames,
             includeAllExtensionTools: true,
@@ -256,6 +257,27 @@ export class AgentSession {
             };
         };
     }
+    _installAgentTurnPreparation() {
+        const previousPrepareNextTurn = this.agent.prepareNextTurn;
+        this.agent.prepareNextTurn = async (signal) => {
+            const previousUpdate = await previousPrepareNextTurn?.(signal);
+            const compacted = await this._drainExtensionCompactionQueue("between-turns");
+            if (!compacted) {
+                return previousUpdate;
+            }
+            return {
+                ...previousUpdate,
+                context: this._buildCurrentAgentContext(previousUpdate?.context),
+            };
+        };
+    }
+    _buildCurrentAgentContext(previousContext) {
+        return {
+            systemPrompt: previousContext?.systemPrompt ?? this.agent.state.systemPrompt,
+            messages: this.agent.state.messages.slice(),
+            tools: previousContext?.tools ?? this.agent.state.tools.slice(),
+        };
+    }
     // =========================================================================
     // Event Subscription
     // =========================================================================
@@ -1529,30 +1551,33 @@ export class AgentSession {
             void this._drainExtensionCompactionQueue();
         }, 0);
     }
-    async _drainExtensionCompactionQueue() {
+    async _drainExtensionCompactionQueue(mode = "idle") {
         if (this._extensionCompactionRunning) {
-            return;
+            return false;
         }
         this._extensionCompactionRunning = true;
+        let compacted = false;
         try {
-            while (!this.isStreaming) {
+            while (mode === "between-turns" || !this.isStreaming) {
                 const options = this._extensionCompactionQueue.shift();
                 if (!options) {
                     break;
                 }
-                await this._runRequestedExtensionCompaction(options);
+                compacted = true;
+                await this._runRequestedExtensionCompaction(options, mode === "idle");
             }
         }
         finally {
             this._extensionCompactionRunning = false;
         }
-        if (!this.isStreaming) {
+        if (mode === "idle" && !this.isStreaming) {
             this._scheduleExtensionCompactions();
         }
+        return compacted;
     }
-    async _runRequestedExtensionCompaction(options) {
+    async _runRequestedExtensionCompaction(options, abortActiveRun) {
         try {
-            const result = await this.compact(options.customInstructions);
+            const result = await this._compactSession(options.customInstructions, { abortActiveRun });
             options.onComplete?.(result);
         }
         catch (error) {
@@ -1566,8 +1591,15 @@ export class AgentSession {
      * @param customInstructions Optional instructions for the compaction summary
      */
     async compact(customInstructions) {
-        this._disconnectFromAgent();
-        await this.abort();
+        return this._compactSession(customInstructions, { abortActiveRun: true });
+    }
+    async _compactSession(customInstructions, options) {
+        let disconnected = false;
+        if (options.abortActiveRun) {
+            this._disconnectFromAgent();
+            disconnected = true;
+            await this.abort();
+        }
         this._compactionAbortController = new AbortController();
         this._emit({ type: "compaction_start", reason: "manual" });
         try {
@@ -1669,7 +1701,9 @@ export class AgentSession {
         }
         finally {
             this._compactionAbortController = undefined;
-            this._reconnectToAgent();
+            if (disconnected) {
+                this._reconnectToAgent();
+            }
         }
     }
     /**