feat: harden workflow gates and expose token usage

backend/test/pi-process-manager.test.ts

@@ -207,6 +207,43 @@ test("createFor does not prompt until start is called", async () => {
   mgr.teardown("sid-deferred");
 });
 
+test("configure gives the bridge the context window returned by set_model", async () => {
+  const child = recordingChild();
+  child.stdin.write = (data: unknown) => {
+    const request = JSON.parse(String(data));
+    child._writes.push(String(data));
+    if (request.type === "set_model") {
+      queueMicrotask(() => child.stdout.emit("data", `${JSON.stringify({
+        type: "response",
+        id: request.id,
+        success: true,
+        data: { contextWindow: 200_000 },
+      })}\n`));
+    }
+    return true;
+  };
+  const mgr = new PiProcessManager(loadConfig({}), { spawnFn: () => child as any });
+  const rt = mgr.createFor("context-window", {});
+  const seen: any[] = [];
+  rt.bridge.onClientEvent((event) => seen.push(event));
+
+  await mgr.configure(rt, { provider: "zai", model: "glm-5.2" });
+  child.stdout.emit("data", `${JSON.stringify({
+    type: "message_end",
+    message: {
+      role: "assistant",
+      stopReason: "stop",
+      usage: { input: 10, cacheRead: 20, output: 5, totalTokens: 35 },
+    },
+  })}\n`);
+
+  expect(seen).toContainEqual({
+    type: "usage",
+    usage: { input: 10, cacheRead: 20, output: 5, totalTokens: 35, contextWindow: 200_000 },
+  });
+  mgr.teardown("context-window");
+});
+
 test("a created runtime is active during configure/bootstrap", () => {
   const child = recordingChild();
   const mgr = new PiProcessManager(loadConfig({}), { spawnFn: () => child as any });