@bunny-agent/runner-cli 0.10.0-beta.3 → 0.10.0-beta.4

package/dist/bundle.mjs

@@ -39586,6 +39586,123 @@ function createOpenCodeRunner(options = {}) {
   });
 }
 
+// ../../packages/runner-pi/dist/live-control.js
+import { AsyncLocalStorage } from "node:async_hooks";
+import { randomBytes as randomBytes2 } from "node:crypto";
+function createPiLiveController(session, onFirstStreamInvocation) {
+  const originalSteeringMode = session.agent.steeringMode;
+  session.agent.steeringMode = "one-at-a-time";
+  const controllerNonce = randomBytes2(16).toString("hex");
+  const dispatchContext = new AsyncLocalStorage();
+  const dispatches = /* @__PURE__ */ new Map();
+  let consumedCallback;
+  let disposed = false;
+  let firstStreamInvocationReported = false;
+  const reportIfConfirmed = (dispatch) => {
+    if (!disposed && dispatch.outcome === "queued" && dispatch.captured.length === 1 && dispatch.observed && !dispatch.reported) {
+      dispatch.reported = true;
+      consumedCallback?.(dispatch.id);
+    }
+  };
+  const originalAgentSteer = session.agent.steer;
+  const installedAgentSteer = (message) => {
+    const dispatch = dispatchContext.getStore();
+    if (dispatch && message.role === "user") {
+      dispatch.captured.push(message);
+      message.__bunnyLiveControl = {
+        controllerNonce,
+        stableId: dispatch.id
+      };
+    }
+    originalAgentSteer.call(session.agent, message);
+  };
+  session.agent.steer = installedAgentSteer;
+  const originalStreamFunction = session.agent.streamFunction;
+  const wrappedStreamFunction = (model, context, streamOptions) => {
+    const observed = /* @__PURE__ */ new Set();
+    const sanitizedMessages = context.messages.map((candidate) => {
+      if (!Object.getOwnPropertyDescriptor(candidate, "__bunnyLiveControl"))
+        return candidate;
+      const { __bunnyLiveControl: marker, ...rest } = candidate;
+      if (marker && marker.controllerNonce === controllerNonce && candidate.role === "user") {
+        const dispatch = dispatches.get(marker.stableId);
+        if (dispatch)
+          observed.add(dispatch);
+      }
+      return rest;
+    });
+    const providerContext = {
+      ...context,
+      messages: sanitizedMessages
+    };
+    const result = originalStreamFunction.call(session.agent, model, providerContext, streamOptions);
+    return Promise.resolve(result).then((stream) => {
+      if (!disposed && !firstStreamInvocationReported) {
+        firstStreamInvocationReported = true;
+        onFirstStreamInvocation?.();
+      }
+      for (const dispatch of observed) {
+        dispatch.observed = true;
+        reportIfConfirmed(dispatch);
+      }
+      return stream;
+    });
+  };
+  session.agent.streamFunction = wrappedStreamFunction;
+  return {
+    async steer(message) {
+      if (disposed)
+        return { disposition: "rejected", reason: "controller disposed" };
+      if (dispatches.has(message.id)) {
+        return {
+          disposition: "rejected",
+          reason: "message id already dispatched"
+        };
+      }
+      const dispatch = {
+        id: message.id,
+        captured: [],
+        observed: false,
+        reported: false
+      };
+      dispatches.set(message.id, dispatch);
+      const images = message.images?.map((image) => ({
+        type: "image",
+        data: image.data,
+        mimeType: image.mimeType
+      }));
+      try {
+        dispatch.outcome = await dispatchContext.run(dispatch, () => session.steer(message.text, images?.length ? images : void 0));
+      } catch (error109) {
+        if (dispatch.captured.length > 0)
+          throw error109;
+        return {
+          disposition: "rejected",
+          reason: error109 instanceof Error ? error109.message : String(error109)
+        };
+      }
+      if (dispatch.outcome === "handled")
+        return { disposition: "handled" };
+      reportIfConfirmed(dispatch);
+      return { disposition: "queued" };
+    },
+    onConsumed(callback) {
+      consumedCallback = callback;
+    },
+    dispose() {
+      if (disposed)
+        return;
+      disposed = true;
+      if (session.agent.steer === installedAgentSteer)
+        session.agent.steer = originalAgentSteer;
+      if (session.agent.streamFunction === wrappedStreamFunction)
+        session.agent.streamFunction = originalStreamFunction;
+      session.agent.steeringMode = originalSteeringMode;
+      dispatches.clear();
+    }
+  };
+}
+
 // ../../packages/runner-pi/dist/mcp-config.js
 import { createJiti } from "jiti";
 var MAX_SERVERS = 50;
@@ -45818,6 +45935,14 @@ function createPiRunner(options = {}) {
           wakeConsumer?.();
           wakeConsumer = null;
         };
+        let liveControlRegistered = false;
+        const registerLiveControl = options.liveControl;
+        const liveController = registerLiveControl ? createPiLiveController(session, () => {
+          if (liveControlRegistered)
+            return;
+          liveControlRegistered = true;
+          registerLiveControl(liveController);
+        }) : void 0;
         const unsubscribe = session.subscribe((e) => {
           eventQueue.push(e);
           notify();
@@ -45915,6 +46040,7 @@ function createPiRunner(options = {}) {
           if (abortSignal) {
             abortSignal.removeEventListener("abort", abortHandler);
           }
+          liveController?.dispose();
           unsubscribe();
           await disposePiSession(session);
         }
@@ -46119,7 +46245,8 @@ function dispatchRunner(runner, base, cwd, options, input2) {
         mcpConfig: options.mcpConfig,
         effort: options.effort,
         systemEnv: options.systemEnv,
-        searchContractVersion: options.searchContractVersion
+        searchContractVersion: options.searchContractVersion,
+        liveControl: options.liveControl
       }).run(hasStructuredInput ? input2 : compiled.text);
     }
     case "opencode":

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@bunny-agent/runner-cli",
-  "version": "0.10.0-beta.3",
+  "version": "0.10.0-beta.4",
   "description": "BunnyAgent Runner CLI - Like gemini-cli or claude-code, runs in your local terminal with AI SDK UI streaming",
   "type": "module",
   "bin": {
@@ -60,15 +60,15 @@
     "typescript": "^5.3.0",
     "vitest": "^1.6.1",
     "@bunny-agent/apply-patch": "0.1.0",
-    "@bunny-agent/manager": "0.10.0-beta.3",
     "@bunny-agent/runner-claude": "0.6.2",
     "@bunny-agent/runner-copilot": "0.1.0",
     "@bunny-agent/runner-gemini": "0.6.2",
     "@bunny-agent/runner-harness": "0.1.1-beta.0",
     "@bunny-agent/runner-opencode": "0.6.2",
-    "@bunny-agent/runner-codex": "0.6.2",
     "@bunny-agent/runner-pi": "0.6.4-beta.0",
-    "@bunny-agent/server-acp": "0.1.0"
+    "@bunny-agent/server-acp": "0.1.0",
+    "@bunny-agent/manager": "0.10.0-beta.4",
+    "@bunny-agent/runner-codex": "0.6.2"
   },
   "scripts": {
     "build": "tsc && pnpm bundle",