@caupulican/pi-adaptative 0.75.6 → 0.75.7

package/dist/core/agent-session.js

@@ -132,6 +132,7 @@ export class AgentSession {
         this._unsubscribeAgent = this.agent.subscribe(this._handleAgentEvent);
         this._installAgentToolHooks();
         this._installAgentContextTransform();
+        this._installAgentTurnRefresh();
         this._buildRuntime({
             activeToolNames: this._initialActiveToolNames,
             includeAllExtensionTools: true,
@@ -202,6 +203,30 @@ export class AgentSession {
             }
         };
     }
+    _installAgentTurnRefresh() {
+        const previousPrepareNextTurn = this.agent.prepareNextTurn?.bind(this.agent);
+        this.agent.prepareNextTurn = async (signal) => {
+            const previous = previousPrepareNextTurn ? await previousPrepareNextTurn(signal) : undefined;
+            const snapshot = this._createAgentContextSnapshot();
+            return {
+                ...previous,
+                context: {
+                    ...(previous?.context ?? snapshot),
+                    systemPrompt: snapshot.systemPrompt,
+                    tools: snapshot.tools,
+                },
+                model: previous?.model ?? this.agent.state.model,
+                thinkingLevel: previous?.thinkingLevel ?? this.agent.state.thinkingLevel,
+            };
+        };
+    }
+    _createAgentContextSnapshot() {
+        return {
+            systemPrompt: this.agent.state.systemPrompt,
+            messages: this.agent.state.messages.slice(),
+            tools: this.agent.state.tools.slice(),
+        };
+    }
     _estimateCurrentContextTokens(messages) {
         const estimate = estimateContextTokens(messages);
         const compactionEntry = getLatestCompactionEntry(this.sessionManager.getBranch());
@@ -1877,12 +1902,9 @@ export class AgentSession {
                 })();
             },
             reload: () => {
-                if (this.isStreaming) {
-                    throw new Error("Cannot reload while the agent is streaming.");
-                }
                 const actions = this._extensionCommandContextActions;
-                if (!actions) {
-                    throw new Error("Reload is unavailable before extension command context is bound.");
+                if (this.isStreaming || !actions) {
+                    return this.reload();
                 }
                 return actions.reload();
             },