@greppleai/grip 0.0.1 → 0.1.4

package/dist/core/agent-session.js

@@ -696,7 +696,12 @@ export class AgentSession {
         if (this._sandboxToolOperations)
             throw new Error("A session sandbox backend is already active");
         const previousToolOperations = this._toolOperations;
-        const operations = new DeferredRemoteToolOperations(expectedCwd);
+        const operations = new DeferredRemoteToolOperations(expectedCwd, () => {
+            if (this._sandboxToolOperations !== operations || this._toolOperations !== operations)
+                return;
+            this.abortRetry();
+            this.agent.abort();
+        });
         try {
             await configure(operations);
         }
@@ -708,6 +713,8 @@ export class AgentSession {
         this._toolOperations = operations;
         this._sandboxToolOperations = operations;
         try {
+            if (operations.isDisconnected())
+                throw new Error("Sandbox connection lost during activation");
             await this.applyToolOperationsBackendChange(operations);
         }
         catch (error) {
@@ -2352,6 +2359,8 @@ export class AgentSession {
     // Prompting
     // =========================================================================
     async _continueAgentIfReady() {
+        if (this._sandboxToolOperations?.isDisconnected())
+            return false;
         const lastMessage = this.agent.state.messages.at(-1);
         if (lastMessage?.role === "assistant" && !this.agent.hasQueuedMessages()) {
             return false;
@@ -2360,10 +2369,13 @@ export class AgentSession {
         return true;
     }
     _runAgentPrompt(messages) {
+        if (this._sandboxToolOperations?.isDisconnected()) {
+            throw new Error("Sandbox connection lost; reconnect before prompting");
+        }
         return this._runActivity("agent run", async () => {
             try {
                 await this.agent.prompt(messages);
-                while (await this._handlePostAgentRun()) {
+                while (!this._sandboxToolOperations?.isDisconnected() && (await this._handlePostAgentRun())) {
                     if (!(await this._continueAgentIfReady()))
                         break;
                 }
@@ -2855,6 +2867,9 @@ export class AgentSession {
      * @throws Error if manual or extension-requested compaction is in progress
      */
     async prompt(text, options) {
+        if (this._sandboxToolOperations?.isDisconnected()) {
+            throw new Error("Sandbox connection lost; reconnect before prompting");
+        }
         const promptId = randomUUID();
         const state = {
             text,