@bunny-agent/runner-cli 0.9.50-release.1 → 0.9.51-beta.1

package/dist/bundle.mjs

@@ -5605,7 +5605,9 @@ function createPiRunner(options = {}) {
           ]
         });
         const eventQueue = [];
-        let isComplete = false;
+        let promptSettled = false;
+        let promptError;
+        let finalAgentEnd = void 0;
         let aborted = false;
         let wakeConsumer = null;
         const notify = () => {
@@ -5614,15 +5616,11 @@ function createPiRunner(options = {}) {
         };
         const unsubscribe = session.subscribe((e) => {
           eventQueue.push(e);
-          if (e.type === "agent_end") {
-            isComplete = true;
-          }
           notify();
         });
         const abortSignal = options.abortController?.signal;
         const abortHandler = () => {
           aborted = true;
-          isComplete = true;
           void session.abort();
           notify();
         };
@@ -5636,6 +5634,14 @@ function createPiRunner(options = {}) {
           traceRawMessage(cwd, null, true, options.env);
           const promptText = userInput;
           const promptPromise = session.prompt(promptText);
+          void promptPromise.then(() => {
+            promptSettled = true;
+            notify();
+          }, (error) => {
+            promptError = error;
+            promptSettled = true;
+            notify();
+          });
           const streamConverter = new PiAISDKStreamConverter({
             sessionId: session.sessionId,
             model,
@@ -5643,10 +5649,14 @@ function createPiRunner(options = {}) {
             getUsageFromAgentEndMessages,
             getErrorFromAgentEndMessages
           });
-          while (!isComplete || eventQueue.length > 0) {
+          while (!promptSettled && !aborted || eventQueue.length > 0) {
             while (eventQueue.length > 0) {
               const event = eventQueue.shift();
               traceRawMessage(cwd, event, false, options.env);
+              if (event.type === "agent_end") {
+                finalAgentEnd = event;
+                continue;
+              }
               const chunks = streamConverter.handleEvent(event, aborted);
               for (const chunk of chunks) {
                 yield chunk;
@@ -5658,30 +5668,35 @@ function createPiRunner(options = {}) {
               }
               break;
             }
-            if (!isComplete && eventQueue.length === 0) {
+            if (!promptSettled && !aborted && eventQueue.length === 0) {
               await new Promise((resolve4) => {
                 wakeConsumer = resolve4;
               });
             }
           }
-          if (streamConverter.finished) {
+          if (aborted) {
             return;
           }
-          try {
-            await promptPromise;
-          } catch (error) {
-            if (!streamConverter.finished) {
-              const message = error instanceof Error ? error.message : "Pi agent run failed.";
-              for (const chunk of streamConverter.forceError(message)) {
-                yield chunk;
-              }
+          if (promptError !== void 0) {
+            const message = promptError instanceof Error ? promptError.message : "Pi agent run failed.";
+            for (const chunk of streamConverter.forceError(message)) {
+              yield chunk;
             }
             return;
           }
-          if (!streamConverter.finished && session.agent.state.errorMessage) {
+          if (finalAgentEnd) {
+            const chunks = streamConverter.handleEvent(finalAgentEnd, false);
+            for (const chunk of chunks) {
+              yield chunk;
+            }
+          } else if (session.agent.state.errorMessage) {
             for (const chunk of streamConverter.forceError(session.agent.state.errorMessage)) {
               yield chunk;
             }
+          } else {
+            for (const chunk of streamConverter.forceError("Pi agent run completed without an agent_end event.")) {
+              yield chunk;
+            }
           }
         } finally {
           if (abortSignal) {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@bunny-agent/runner-cli",
-  "version": "0.9.50-release.1",
+  "version": "0.9.51-beta.1",
   "description": "BunnyAgent Runner CLI - Like gemini-cli or claude-code, runs in your local terminal with AI SDK UI streaming",
   "type": "module",
   "bin": {
@@ -53,8 +53,8 @@
     "esbuild": "^0.27.2",
     "typescript": "^5.3.0",
     "vitest": "^1.6.1",
-    "@bunny-agent/runner-harness": "0.1.1-beta.0",
     "@bunny-agent/runner-claude": "0.6.2",
+    "@bunny-agent/runner-harness": "0.1.1-beta.0",
     "@bunny-agent/runner-codex": "0.6.2",
     "@bunny-agent/runner-gemini": "0.6.2",
     "@bunny-agent/runner-opencode": "0.6.2",