@tt-a1i/openpi 0.4.0 → 0.5.0

package/extensions/background-terminals/index.ts

@@ -32,6 +32,10 @@ import {
   OPENPI_TOOL_SURFACE,
   patchOwnedTools,
 } from "../shared/tool-surface.ts";
+import {
+  projectBackgroundTerminalCapability,
+  registerWebCapability,
+} from "../shared/web-observer-registry.ts";
 import type { TerminalSnapshot } from "./src/domain.ts";
 import {
   MAX_RUNNING,
@@ -95,6 +99,8 @@ interface WatchToolDetails {
 export default function (pi: ExtensionAPI) {
   let runtime: TerminalRuntime | undefined;
   let managerPromise: Promise<TerminalManagerShape> | undefined;
+  let managerForHooks: TerminalManagerShape | undefined;
+  let unregisterWebCapability: (() => void) | undefined;
   let sessionContext: ExtensionContext | undefined;
   let ui: ExtensionUIContext | undefined;
   let unsubStatus: (() => void) | undefined;
@@ -115,9 +121,21 @@ export default function (pi: ExtensionAPI) {
 
   /** Resolve the manager service once per runtime and wire the extension hooks. */
   const getManager = () => {
+    const scope = sessionContext?.sessionManager;
     managerPromise ??= getRuntime()
       .runPromise(TerminalManager)
       .then((manager) => {
+        managerForHooks = manager;
+        unregisterWebCapability?.();
+        unregisterWebCapability =
+          scope && sessionContext?.sessionManager === scope
+            ? registerWebCapability(scope, {
+                kind: "background-terminals",
+                snapshot: () =>
+                  projectBackgroundTerminalCapability(manager.view.list()),
+                subscribe: (listener) => manager.view.subscribe(listener),
+              })
+            : undefined;
         manager.view.setOnSettled(onSettled);
         unsubStatus?.();
         unsubStatus = manager.view.subscribe(() => updateWidget(manager));
@@ -214,6 +232,7 @@ export default function (pi: ExtensionAPI) {
         // costing a turn.
         resultDeliveryOptions(wake),
       );
+      for (const snap of snaps) managerForHooks?.view.releaseResult(snap.id);
       return true;
     } catch (error) {
       // Session may be shutting down, but retain the snapshot so any later
@@ -253,6 +272,7 @@ export default function (pi: ExtensionAPI) {
       stdout: { ...snap.stdout },
       stderr: { ...snap.stderr },
     });
+    managerForHooks?.view.retainResult(snap.id);
     // Pending settlements remain retractable while busy, so bg_status/bg_kill
     // can consume them before they are committed to context. bg_start applies
     // backpressure across running + pending + reserved work, keeping this map
@@ -310,7 +330,10 @@ export default function (pi: ExtensionAPI) {
     ui = undefined;
     const closing = runtime;
     runtime = undefined;
+    unregisterWebCapability?.();
+    unregisterWebCapability = undefined;
     managerPromise = undefined;
+    managerForHooks = undefined;
     await closing?.dispose();
   });
 
@@ -423,12 +446,16 @@ export default function (pi: ExtensionAPI) {
         );
       }
 
+      const text = buildStatusResult(snap);
       // This status is returning the settlement itself; a pending automatic
       // follow-up for the same settle would be a duplicate.
-      if (snap.status !== "running") resultDelivery.consume([snap.id]);
+      if (snap.status !== "running") {
+        resultDelivery.consume([snap.id]);
+        manager.view.releaseResult(snap.id);
+      }
 
       return {
-        content: [{ type: "text", text: buildStatusResult(snap) }],
+        content: [{ type: "text", text }],
         details: {
           id: snap.id,
           status: snap.status,
@@ -510,6 +537,7 @@ export default function (pi: ExtensionAPI) {
       // via the killInterest consumed flag). Remove any deferred automatic
       // delivery now that this tool returns the final state itself.
       resultDelivery.consume(ids);
+      for (const id of ids) manager.view.releaseResult(id);
 
       return {
         content: [{ type: "text", text: buildKillReport(report) }],

package/extensions/background-terminals/src/domain.ts

@@ -23,6 +23,8 @@ export type TerminalStatus =
 export interface OutputView {
   /** Decoded, possibly head-trimmed text (bounded by the in-memory cap). */
   readonly text: string;
+  /** Stateful, bounded projection safe for model-facing output. */
+  readonly modelSafeText: string;
   /** True total bytes ever received on this stream. */
   readonly totalBytes: number;
   /** Bytes dropped from the head of the in-memory view (0 = complete). */