@h-rig/cli 0.0.6-alpha.23 → 0.0.6-alpha.24

package/dist/src/commands/run.js

@@ -600,6 +600,15 @@ function appendTranscript(state, label, text) {
     state.transcript.splice(0, state.transcript.length - MAX_TRANSCRIPT_LINES);
   }
 }
+function nativePiUi(ctx) {
+  const ui = ctx.ui;
+  return typeof ui.emitSessionEvent === "function" && typeof ui.appendSessionMessages === "function" ? ui : null;
+}
+function syncNativeDisplayCwd(ctx, state) {
+  const ui = nativePiUi(ctx);
+  if (ui?.setDisplayCwd && state.cwd)
+    ui.setDisplayCwd(state.cwd);
+}
 function parseExtensionUiRequest(value) {
   const request = recordOf(value) ?? {};
   const requestId = String(request.requestId ?? request.id ?? `ui-${Date.now()}`);
@@ -643,8 +652,13 @@ function renderBridgeWidget(state) {
 function updatePiUi(ctx, state) {
   ctx.ui.setTitle("Pi \xB7 Rig worker daemon");
   ctx.ui.setStatus("rig-worker-pi", state.wsConnected ? "worker Pi WS live" : state.status);
+  syncNativeDisplayCwd(ctx, state);
+  if (state.nativeStream && nativePiUi(ctx)) {
+    ctx.ui.setWidget("rig-worker-pi-transcript", undefined);
+    return;
+  }
   ctx.ui.setWorkingVisible(false);
-  ctx.ui.setWidget("rig-worker-pi-transcript", renderBridgeWidget(state), { placement: "aboveEditor" });
+  ctx.ui.setWidget("rig-worker-pi-transcript", [`Worker Pi daemon bridge \xB7 degraded widget transcript`, ...renderBridgeWidget(state)], { placement: "aboveEditor" });
 }
 function applyStatus(state, payload) {
   const status = recordOf(payload.status) ?? payload;
@@ -662,7 +676,7 @@ function applyMessage(state, message) {
   const label = role === "assistant" ? "Pi" : role === "user" ? "You" : role === "tool" || role === "toolResult" ? "Tool" : "System";
   appendTranscript(state, label, textFromContent(record.content ?? record.message ?? record.text ?? ""));
 }
-function applyPiEvent(state, eventValue) {
+function applyPiEvent(ctx, state, eventValue) {
   const event = recordOf(eventValue);
   if (!event)
     return;
@@ -670,11 +684,20 @@ function applyPiEvent(state, eventValue) {
   if (type === "agent_start") {
     state.streaming = true;
     state.status = "streaming";
+  } else if (type === "agent_end") {
+    state.streaming = false;
+    state.status = "idle";
+  } else if (type === "queue_update") {
+    const steering = Array.isArray(event.steering) ? event.steering.length : 0;
+    const followUp = Array.isArray(event.followUp) ? event.followUp.length : 0;
+    state.status = `queued \xB7 steer ${steering} \xB7 follow-up ${followUp}`;
+  }
+  const native = nativePiUi(ctx);
+  if (state.nativeStream && native?.emitSessionEvent) {
+    native.emitSessionEvent(eventValue);
     return;
   }
   if (type === "agent_end") {
-    state.streaming = false;
-    state.status = "idle";
     appendTranscript(state, "System", "Agent turn complete.");
     return;
   }
@@ -699,29 +722,60 @@ function applyPiEvent(state, eventValue) {
   }
   if (type === "tool_execution_end") {
     appendTranscript(state, event.isError === true ? "Error" : "Tool", asText(event.result ?? `${String(event.toolName ?? "tool")} complete`));
-    return;
-  }
-  if (type === "queue_update") {
-    const steering = Array.isArray(event.steering) ? event.steering.length : 0;
-    const followUp = Array.isArray(event.followUp) ? event.followUp.length : 0;
-    state.status = `queued \xB7 steer ${steering} \xB7 follow-up ${followUp}`;
   }
 }
+function firstPendingShell(state) {
+  return state.pendingShells[0];
+}
+function finishPendingShell(state, shell, result) {
+  const index = state.pendingShells.indexOf(shell);
+  if (index !== -1)
+    state.pendingShells.splice(index, 1);
+  shell.resolve(result);
+}
+function failPendingShell(state, shell, error) {
+  const index = state.pendingShells.indexOf(shell);
+  if (index !== -1)
+    state.pendingShells.splice(index, 1);
+  shell.reject(error);
+}
 function applyUiEvent(state, value) {
   const event = recordOf(value);
   if (!event)
     return;
   const type = String(event.type ?? "ui");
-  if (type === "shell.start")
-    appendTranscript(state, "Tool", `$ ${asText(event.command)}`);
-  else if (type === "shell.chunk")
-    appendTranscript(state, "Tool", asText(event.chunk));
-  else if (type === "shell.end")
-    appendTranscript(state, event.isError === true ? "Error" : "Tool", asText(event.output ?? `exit ${String(event.exitCode ?? "")}`));
-  else
-    appendTranscript(state, "System", `${type}: ${asText(event)}`);
-}
-function applyEnvelope(state, envelopeValue) {
+  if (type === "shell.chunk") {
+    const pending = firstPendingShell(state);
+    const chunk = asText(event.chunk);
+    if (pending) {
+      pending.sawChunk = true;
+      pending.onData(Buffer.from(chunk));
+    } else {
+      appendTranscript(state, "Tool", chunk);
+    }
+    return;
+  }
+  if (type === "shell.end") {
+    const pending = firstPendingShell(state);
+    const output = asText(event.output ?? "");
+    const exitCode = typeof event.exitCode === "number" ? event.exitCode : event.isError === true ? 1 : 0;
+    if (pending) {
+      if (output && !pending.sawChunk)
+        pending.onData(Buffer.from(output));
+      finishPendingShell(state, pending, { exitCode });
+    } else {
+      appendTranscript(state, event.isError === true ? "Error" : "Tool", output || `exit ${String(exitCode)}`);
+    }
+    return;
+  }
+  if (type === "shell.start") {
+    if (!firstPendingShell(state))
+      appendTranscript(state, "Tool", `$ ${asText(event.command)}`);
+    return;
+  }
+  appendTranscript(state, "System", `${type}: ${asText(event)}`);
+}
+function applyEnvelope(ctx, state, envelopeValue) {
   const envelope = recordOf(envelopeValue);
   if (!envelope)
     return;
@@ -730,7 +784,8 @@ function applyEnvelope(state, envelopeValue) {
     const metadata = recordOf(envelope.metadata);
     state.cwd = typeof metadata?.cwd === "string" ? metadata.cwd : state.cwd;
     state.status = "worker Pi daemon ready";
-    appendTranscript(state, "System", "Connected to worker Pi daemon.");
+    if (!state.nativeStream)
+      appendTranscript(state, "System", "Connected to worker Pi daemon.");
   } else if (type === "status.update") {
     applyStatus(state, envelope);
   } else if (type === "activity.update") {
@@ -742,10 +797,11 @@ function applyEnvelope(state, envelopeValue) {
   } else if (type === "pi.ui_event") {
     applyUiEvent(state, envelope.event);
   } else if (type === "pi.event") {
-    applyPiEvent(state, envelope.event);
+    applyPiEvent(ctx, state, envelope.event);
   } else if (type === "error") {
     appendTranscript(state, "Error", asText(envelope.message ?? envelope.detail ?? "unknown error"));
   }
+  syncNativeDisplayCwd(ctx, state);
 }
 async function waitForWorkerReady(options, ctx, state) {
   while (true) {
@@ -766,7 +822,7 @@ async function waitForWorkerReady(options, ctx, state) {
       continue;
     }
     const sessionRecord = recordOf(session) ?? {};
-    applyEnvelope(state, { type: "ready", metadata: sessionRecord.metadata ?? sessionRecord });
+    applyEnvelope(ctx, state, { type: "ready", metadata: sessionRecord.metadata ?? sessionRecord });
     updatePiUi(ctx, state);
     return true;
   }
@@ -796,7 +852,7 @@ async function connectWorkerStream(options, ctx, state) {
         if (!catchupDone)
           buffered.push(payload);
         else {
-          applyEnvelope(state, payload);
+          applyEnvelope(ctx, state, payload);
           updatePiUi(ctx, state);
         }
       } catch (error) {
@@ -819,8 +875,12 @@ async function connectWorkerStream(options, ctx, state) {
       getRunPiCommandsViaServer(options.context, options.runId)
     ]);
     const messages = Array.isArray(messagesPayload.messages) ? messagesPayload.messages : [];
-    for (const message of messages)
-      applyMessage(state, message);
+    const native = nativePiUi(ctx);
+    if (state.nativeStream && native?.appendSessionMessages)
+      native.appendSessionMessages(messages);
+    else
+      for (const message of messages)
+        applyMessage(state, message);
     applyStatus(state, statusPayload);
     const commands = Array.isArray(commandsPayload.commands) ? commandsPayload.commands : [];
     state.commands = commands.flatMap((command) => {
@@ -829,7 +889,7 @@ async function connectWorkerStream(options, ctx, state) {
     });
     catchupDone = true;
     for (const payload of buffered.splice(0))
-      applyEnvelope(state, payload);
+      applyEnvelope(ctx, state, payload);
     updatePiUi(ctx, state);
   } catch (error) {
     appendTranscript(state, "Error", `Worker Pi catch-up failed: ${error instanceof Error ? error.message : String(error)}`);
@@ -838,6 +898,51 @@ async function connectWorkerStream(options, ctx, state) {
   }
   await closePromise;
 }
+function createRemoteBashOperations(options, state, excludeFromContext) {
+  return {
+    exec(command, _cwd, execOptions) {
+      return new Promise((resolve3, reject) => {
+        const pending = {
+          command,
+          onData: execOptions.onData,
+          resolve: resolve3,
+          reject,
+          sawChunk: false
+        };
+        const cleanup = () => {
+          execOptions.signal?.removeEventListener("abort", onAbort);
+          if (timer)
+            clearTimeout(timer);
+        };
+        const onAbort = () => {
+          cleanup();
+          failPendingShell(state, pending, new Error("Remote worker shell command aborted locally."));
+        };
+        const timeoutMs = typeof execOptions.timeout === "number" && execOptions.timeout > 0 ? execOptions.timeout : 0;
+        const timer = timeoutMs > 0 ? setTimeout(() => {
+          cleanup();
+          failPendingShell(state, pending, new Error(`Remote worker shell command timed out after ${timeoutMs}ms.`));
+        }, timeoutMs) : null;
+        const wrappedResolve = pending.resolve;
+        const wrappedReject = pending.reject;
+        pending.resolve = (result) => {
+          cleanup();
+          wrappedResolve(result);
+        };
+        pending.reject = (error) => {
+          cleanup();
+          wrappedReject(error);
+        };
+        execOptions.signal?.addEventListener("abort", onAbort, { once: true });
+        state.pendingShells.push(pending);
+        sendRunPiShellViaServer(options.context, options.runId, `${excludeFromContext ? "!!" : "!"}${command}`).catch((error) => {
+          cleanup();
+          failPendingShell(state, pending, error instanceof Error ? error : new Error(String(error)));
+        });
+      });
+    }
+  };
+}
 async function answerPendingUi(options, state, line) {
   const pending = state.pendingUi;
   if (!pending)
@@ -903,13 +1008,22 @@ function createRigWorkerPiBridgeExtension(options) {
       commands: [],
       streaming: false,
       pendingUi: null,
-      wsConnected: false
+      pendingShells: [],
+      wsConnected: false,
+      nativeStream: false
     };
     if (options.initialMessageSent)
       appendTranscript(state, "System", "Initial message sent to worker Pi daemon.");
+    let nativePiUiContextAvailable = false;
+    pi.on("user_bash", (event) => {
+      state.nativeStream = Boolean(state.nativeStream || nativePiUiContextAvailable);
+      return { operations: createRemoteBashOperations(options, state, event.excludeFromContext === true) };
+    });
     pi.on("session_start", async (_event, ctx) => {
+      nativePiUiContextAvailable = Boolean(nativePiUi(ctx));
+      state.nativeStream = nativePiUiContextAvailable;
       updatePiUi(ctx, state);
-      ctx.ui.notify("Rig worker Pi bridge extension loaded", "info");
+      ctx.ui.notify(nativePiUiContextAvailable ? "Rig worker Pi native stream bridge loaded" : "Rig worker Pi bridge extension loaded (degraded widget fallback)", "info");
       ctx.ui.onTerminalInput((data) => {
         if (data.includes("\x04")) {
           ctx.shutdown();
@@ -923,6 +1037,8 @@ function createRigWorkerPiBridgeExtension(options) {
         const text = [editorText, inlineText].filter(Boolean).join(" ").trim();
         if (!text)
           return;
+        if (text.startsWith("!"))
+          return;
         ctx.ui.setEditorText("");
         routeInput(options, ctx, state, text).catch((error) => {
           appendTranscript(state, "Error", error instanceof Error ? error.message : String(error));

package/dist/src/commands/task.js

@@ -1003,6 +1003,15 @@ function appendTranscript(state, label, text) {
     state.transcript.splice(0, state.transcript.length - MAX_TRANSCRIPT_LINES);
   }
 }
+function nativePiUi(ctx) {
+  const ui = ctx.ui;
+  return typeof ui.emitSessionEvent === "function" && typeof ui.appendSessionMessages === "function" ? ui : null;
+}
+function syncNativeDisplayCwd(ctx, state) {
+  const ui = nativePiUi(ctx);
+  if (ui?.setDisplayCwd && state.cwd)
+    ui.setDisplayCwd(state.cwd);
+}
 function parseExtensionUiRequest(value) {
   const request = recordOf(value) ?? {};
   const requestId = String(request.requestId ?? request.id ?? `ui-${Date.now()}`);
@@ -1046,8 +1055,13 @@ function renderBridgeWidget(state) {
 function updatePiUi(ctx, state) {
   ctx.ui.setTitle("Pi \xB7 Rig worker daemon");
   ctx.ui.setStatus("rig-worker-pi", state.wsConnected ? "worker Pi WS live" : state.status);
+  syncNativeDisplayCwd(ctx, state);
+  if (state.nativeStream && nativePiUi(ctx)) {
+    ctx.ui.setWidget("rig-worker-pi-transcript", undefined);
+    return;
+  }
   ctx.ui.setWorkingVisible(false);
-  ctx.ui.setWidget("rig-worker-pi-transcript", renderBridgeWidget(state), { placement: "aboveEditor" });
+  ctx.ui.setWidget("rig-worker-pi-transcript", [`Worker Pi daemon bridge \xB7 degraded widget transcript`, ...renderBridgeWidget(state)], { placement: "aboveEditor" });
 }
 function applyStatus(state, payload) {
   const status = recordOf(payload.status) ?? payload;
@@ -1065,7 +1079,7 @@ function applyMessage(state, message2) {
   const label = role === "assistant" ? "Pi" : role === "user" ? "You" : role === "tool" || role === "toolResult" ? "Tool" : "System";
   appendTranscript(state, label, textFromContent(record.content ?? record.message ?? record.text ?? ""));
 }
-function applyPiEvent(state, eventValue) {
+function applyPiEvent(ctx, state, eventValue) {
   const event = recordOf(eventValue);
   if (!event)
     return;
@@ -1073,11 +1087,20 @@ function applyPiEvent(state, eventValue) {
   if (type === "agent_start") {
     state.streaming = true;
     state.status = "streaming";
+  } else if (type === "agent_end") {
+    state.streaming = false;
+    state.status = "idle";
+  } else if (type === "queue_update") {
+    const steering = Array.isArray(event.steering) ? event.steering.length : 0;
+    const followUp = Array.isArray(event.followUp) ? event.followUp.length : 0;
+    state.status = `queued \xB7 steer ${steering} \xB7 follow-up ${followUp}`;
+  }
+  const native = nativePiUi(ctx);
+  if (state.nativeStream && native?.emitSessionEvent) {
+    native.emitSessionEvent(eventValue);
     return;
   }
   if (type === "agent_end") {
-    state.streaming = false;
-    state.status = "idle";
     appendTranscript(state, "System", "Agent turn complete.");
     return;
   }
@@ -1102,29 +1125,60 @@ function applyPiEvent(state, eventValue) {
   }
   if (type === "tool_execution_end") {
     appendTranscript(state, event.isError === true ? "Error" : "Tool", asText(event.result ?? `${String(event.toolName ?? "tool")} complete`));
-    return;
-  }
-  if (type === "queue_update") {
-    const steering = Array.isArray(event.steering) ? event.steering.length : 0;
-    const followUp = Array.isArray(event.followUp) ? event.followUp.length : 0;
-    state.status = `queued \xB7 steer ${steering} \xB7 follow-up ${followUp}`;
   }
 }
+function firstPendingShell(state) {
+  return state.pendingShells[0];
+}
+function finishPendingShell(state, shell, result) {
+  const index = state.pendingShells.indexOf(shell);
+  if (index !== -1)
+    state.pendingShells.splice(index, 1);
+  shell.resolve(result);
+}
+function failPendingShell(state, shell, error) {
+  const index = state.pendingShells.indexOf(shell);
+  if (index !== -1)
+    state.pendingShells.splice(index, 1);
+  shell.reject(error);
+}
 function applyUiEvent(state, value) {
   const event = recordOf(value);
   if (!event)
     return;
   const type = String(event.type ?? "ui");
-  if (type === "shell.start")
-    appendTranscript(state, "Tool", `$ ${asText(event.command)}`);
-  else if (type === "shell.chunk")
-    appendTranscript(state, "Tool", asText(event.chunk));
-  else if (type === "shell.end")
-    appendTranscript(state, event.isError === true ? "Error" : "Tool", asText(event.output ?? `exit ${String(event.exitCode ?? "")}`));
-  else
-    appendTranscript(state, "System", `${type}: ${asText(event)}`);
-}
-function applyEnvelope(state, envelopeValue) {
+  if (type === "shell.chunk") {
+    const pending = firstPendingShell(state);
+    const chunk = asText(event.chunk);
+    if (pending) {
+      pending.sawChunk = true;
+      pending.onData(Buffer.from(chunk));
+    } else {
+      appendTranscript(state, "Tool", chunk);
+    }
+    return;
+  }
+  if (type === "shell.end") {
+    const pending = firstPendingShell(state);
+    const output = asText(event.output ?? "");
+    const exitCode = typeof event.exitCode === "number" ? event.exitCode : event.isError === true ? 1 : 0;
+    if (pending) {
+      if (output && !pending.sawChunk)
+        pending.onData(Buffer.from(output));
+      finishPendingShell(state, pending, { exitCode });
+    } else {
+      appendTranscript(state, event.isError === true ? "Error" : "Tool", output || `exit ${String(exitCode)}`);
+    }
+    return;
+  }
+  if (type === "shell.start") {
+    if (!firstPendingShell(state))
+      appendTranscript(state, "Tool", `$ ${asText(event.command)}`);
+    return;
+  }
+  appendTranscript(state, "System", `${type}: ${asText(event)}`);
+}
+function applyEnvelope(ctx, state, envelopeValue) {
   const envelope = recordOf(envelopeValue);
   if (!envelope)
     return;
@@ -1133,7 +1187,8 @@ function applyEnvelope(state, envelopeValue) {
     const metadata = recordOf(envelope.metadata);
     state.cwd = typeof metadata?.cwd === "string" ? metadata.cwd : state.cwd;
     state.status = "worker Pi daemon ready";
-    appendTranscript(state, "System", "Connected to worker Pi daemon.");
+    if (!state.nativeStream)
+      appendTranscript(state, "System", "Connected to worker Pi daemon.");
   } else if (type === "status.update") {
     applyStatus(state, envelope);
   } else if (type === "activity.update") {
@@ -1145,10 +1200,11 @@ function applyEnvelope(state, envelopeValue) {
   } else if (type === "pi.ui_event") {
     applyUiEvent(state, envelope.event);
   } else if (type === "pi.event") {
-    applyPiEvent(state, envelope.event);
+    applyPiEvent(ctx, state, envelope.event);
   } else if (type === "error") {
     appendTranscript(state, "Error", asText(envelope.message ?? envelope.detail ?? "unknown error"));
   }
+  syncNativeDisplayCwd(ctx, state);
 }
 async function waitForWorkerReady(options, ctx, state) {
   while (true) {
@@ -1169,7 +1225,7 @@ async function waitForWorkerReady(options, ctx, state) {
       continue;
     }
     const sessionRecord = recordOf(session) ?? {};
-    applyEnvelope(state, { type: "ready", metadata: sessionRecord.metadata ?? sessionRecord });
+    applyEnvelope(ctx, state, { type: "ready", metadata: sessionRecord.metadata ?? sessionRecord });
     updatePiUi(ctx, state);
     return true;
   }
@@ -1199,7 +1255,7 @@ async function connectWorkerStream(options, ctx, state) {
         if (!catchupDone)
           buffered.push(payload);
         else {
-          applyEnvelope(state, payload);
+          applyEnvelope(ctx, state, payload);
           updatePiUi(ctx, state);
         }
       } catch (error) {
@@ -1222,8 +1278,12 @@ async function connectWorkerStream(options, ctx, state) {
       getRunPiCommandsViaServer(options.context, options.runId)
     ]);
     const messages = Array.isArray(messagesPayload.messages) ? messagesPayload.messages : [];
-    for (const message2 of messages)
-      applyMessage(state, message2);
+    const native = nativePiUi(ctx);
+    if (state.nativeStream && native?.appendSessionMessages)
+      native.appendSessionMessages(messages);
+    else
+      for (const message2 of messages)
+        applyMessage(state, message2);
     applyStatus(state, statusPayload);
     const commands = Array.isArray(commandsPayload.commands) ? commandsPayload.commands : [];
     state.commands = commands.flatMap((command) => {
@@ -1232,7 +1292,7 @@ async function connectWorkerStream(options, ctx, state) {
     });
     catchupDone = true;
     for (const payload of buffered.splice(0))
-      applyEnvelope(state, payload);
+      applyEnvelope(ctx, state, payload);
     updatePiUi(ctx, state);
   } catch (error) {
     appendTranscript(state, "Error", `Worker Pi catch-up failed: ${error instanceof Error ? error.message : String(error)}`);
@@ -1241,6 +1301,51 @@ async function connectWorkerStream(options, ctx, state) {
   }
   await closePromise;
 }
+function createRemoteBashOperations(options, state, excludeFromContext) {
+  return {
+    exec(command, _cwd, execOptions) {
+      return new Promise((resolve3, reject) => {
+        const pending = {
+          command,
+          onData: execOptions.onData,
+          resolve: resolve3,
+          reject,
+          sawChunk: false
+        };
+        const cleanup = () => {
+          execOptions.signal?.removeEventListener("abort", onAbort);
+          if (timer)
+            clearTimeout(timer);
+        };
+        const onAbort = () => {
+          cleanup();
+          failPendingShell(state, pending, new Error("Remote worker shell command aborted locally."));
+        };
+        const timeoutMs = typeof execOptions.timeout === "number" && execOptions.timeout > 0 ? execOptions.timeout : 0;
+        const timer = timeoutMs > 0 ? setTimeout(() => {
+          cleanup();
+          failPendingShell(state, pending, new Error(`Remote worker shell command timed out after ${timeoutMs}ms.`));
+        }, timeoutMs) : null;
+        const wrappedResolve = pending.resolve;
+        const wrappedReject = pending.reject;
+        pending.resolve = (result) => {
+          cleanup();
+          wrappedResolve(result);
+        };
+        pending.reject = (error) => {
+          cleanup();
+          wrappedReject(error);
+        };
+        execOptions.signal?.addEventListener("abort", onAbort, { once: true });
+        state.pendingShells.push(pending);
+        sendRunPiShellViaServer(options.context, options.runId, `${excludeFromContext ? "!!" : "!"}${command}`).catch((error) => {
+          cleanup();
+          failPendingShell(state, pending, error instanceof Error ? error : new Error(String(error)));
+        });
+      });
+    }
+  };
+}
 async function answerPendingUi(options, state, line) {
   const pending = state.pendingUi;
   if (!pending)
@@ -1306,13 +1411,22 @@ function createRigWorkerPiBridgeExtension(options) {
       commands: [],
       streaming: false,
       pendingUi: null,
-      wsConnected: false
+      pendingShells: [],
+      wsConnected: false,
+      nativeStream: false
     };
     if (options.initialMessageSent)
       appendTranscript(state, "System", "Initial message sent to worker Pi daemon.");
+    let nativePiUiContextAvailable = false;
+    pi.on("user_bash", (event) => {
+      state.nativeStream = Boolean(state.nativeStream || nativePiUiContextAvailable);
+      return { operations: createRemoteBashOperations(options, state, event.excludeFromContext === true) };
+    });
     pi.on("session_start", async (_event, ctx) => {
+      nativePiUiContextAvailable = Boolean(nativePiUi(ctx));
+      state.nativeStream = nativePiUiContextAvailable;
       updatePiUi(ctx, state);
-      ctx.ui.notify("Rig worker Pi bridge extension loaded", "info");
+      ctx.ui.notify(nativePiUiContextAvailable ? "Rig worker Pi native stream bridge loaded" : "Rig worker Pi bridge extension loaded (degraded widget fallback)", "info");
       ctx.ui.onTerminalInput((data) => {
         if (data.includes("\x04")) {
           ctx.shutdown();
@@ -1326,6 +1440,8 @@ function createRigWorkerPiBridgeExtension(options) {
         const text = [editorText, inlineText].filter(Boolean).join(" ").trim();
         if (!text)
           return;
+        if (text.startsWith("!"))
+          return;
         ctx.ui.setEditorText("");
         routeInput(options, ctx, state, text).catch((error) => {
           appendTranscript(state, "Error", error instanceof Error ? error.message : String(error));