@bunny-agent/runner-cli 0.10.0-beta.1 → 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;
@@ -40245,7 +40362,7 @@ var SourceRegistry = class {
     }
   }
   register(source, toolCallIds = [], save = true, restoredReferences) {
-    if (toolCallIds.some((id) => typeof id !== "string" || id.trim().length === 0 || id.length > 1024)) {
+    if (toolCallIds.some((id) => typeof id !== "string" || id.trim().length === 0)) {
       this.diagnose("invalid_version", "Rejected malformed source provenance.");
       return void 0;
     }
@@ -40339,7 +40456,7 @@ function isRegisteredSource(value) {
   if (!isWebSearchSource(value))
     return false;
   const entry = value;
-  return deriveSourceId(entry.url) === entry.source_id && (entry.references === void 0 || typeof entry.references === "object" && entry.references !== null && Object.keys(entry.references).length <= 50 && Object.entries(entry.references).every(([ref, call]) => ref.length <= 64 && /^turn\d+url\d+$/.test(ref) && Number.isSafeInteger(Number(/^turn(\d+)/.exec(ref)?.[1])) && Number.isSafeInteger(Number(/url(\d+)$/.exec(ref)?.[1])) && typeof call === "string" && Array.isArray(entry.tool_call_ids) && entry.tool_call_ids.includes(call))) && Array.isArray(entry.tool_call_ids) && entry.tool_call_ids.length <= 50 && entry.tool_call_ids.every((id) => typeof id === "string" && id.trim().length > 0 && id.length <= 1024);
+  return deriveSourceId(entry.url) === entry.source_id && (entry.references === void 0 || typeof entry.references === "object" && entry.references !== null && Object.keys(entry.references).length <= 50 && Object.entries(entry.references).every(([ref, call]) => ref.length <= 64 && /^turn\d+url\d+$/.test(ref) && Number.isSafeInteger(Number(/^turn(\d+)/.exec(ref)?.[1])) && Number.isSafeInteger(Number(/url(\d+)$/.exec(ref)?.[1])) && typeof call === "string" && Array.isArray(entry.tool_call_ids) && entry.tool_call_ids.includes(call))) && Array.isArray(entry.tool_call_ids) && entry.tool_call_ids.length <= 50 && entry.tool_call_ids.every((id) => typeof id === "string" && id.trim().length > 0);
 }
 
 // ../../packages/runner-pi/dist/citation-session-cache.js
@@ -40357,7 +40474,7 @@ function sourcesFromEntry(value) {
   const record3 = entry.message?.details?.web_search;
   if (entry.type === "message" && entry.message?.role === "toolResult" && !entry.message.isError && isValidWebSearchCallRecord(record3) && record3.status === "completed") {
     const callId = entry.message.toolCallId;
-    if (typeof callId !== "string" || callId.trim().length === 0 || callId.length > 1024)
+    if (typeof callId !== "string" || callId.trim().length === 0)
       return [];
     return record3.action.sources.map((source) => ({
       ...source,
@@ -40469,7 +40586,7 @@ function createCitationSessionRegistry(store, enabled) {
     const entry = value;
     if (entry.type === "custom" && entry.customType === CITATION_CACHE_TYPE && entry.data?.schema_version === 1 && Array.isArray(entry.data.calls)) {
       for (const call of entry.data.calls.slice(0, 50)) {
-        if (call && typeof call.id === "string" && call.id.trim().length > 0 && call.id.length <= 1024 && Number.isSafeInteger(call.turn) && call.turn >= 0)
+        if (call && typeof call.id === "string" && call.id.trim().length > 0 && Number.isSafeInteger(call.turn) && call.turn >= 0)
           registry3.reserveSearch(call.id, call.turn);
       }
     }
@@ -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.1",
+  "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": {
@@ -59,16 +59,16 @@
     "esbuild": "^0.27.2",
     "typescript": "^5.3.0",
     "vitest": "^1.6.1",
+    "@bunny-agent/apply-patch": "0.1.0",
     "@bunny-agent/runner-claude": "0.6.2",
-    "@bunny-agent/runner-codex": "0.6.2",
     "@bunny-agent/runner-copilot": "0.1.0",
-    "@bunny-agent/manager": "0.10.0-beta.1",
     "@bunny-agent/runner-gemini": "0.6.2",
-    "@bunny-agent/apply-patch": "0.1.0",
     "@bunny-agent/runner-harness": "0.1.1-beta.0",
+    "@bunny-agent/runner-opencode": "0.6.2",
     "@bunny-agent/runner-pi": "0.6.4-beta.0",
     "@bunny-agent/server-acp": "0.1.0",
-    "@bunny-agent/runner-opencode": "0.6.2"
+    "@bunny-agent/manager": "0.10.0-beta.4",
+    "@bunny-agent/runner-codex": "0.6.2"
   },
   "scripts": {
     "build": "tsc && pnpm bundle",