@koda-sl/baker-bridge 0.159.0-dev.aefecbe62 → 0.159.0-dev.c31d1bca3

package/dist/hono/bridge-job.test.js

@@ -70,6 +70,7 @@ function recordingRelay() {
     const reviewDigests = [];
     const retractedEvents = [];
     const toolOutcomeBatches = [];
+    const previewedTools = [];
     const contextUsageUpdates = [];
     return {
         assistantEvents,
@@ -80,9 +81,12 @@ function recordingRelay() {
         reviewDigests,
         retractedEvents,
         toolOutcomeBatches,
+        previewedTools,
         startPreview: () => undefined,
         previewText: () => undefined,
-        previewTools: () => undefined,
+        previewTools: (_target, tools) => {
+            previewedTools.push(tools);
+        },
         tagInputRequest: () => Promise.resolve(true),
         tagInputResolved: () => Promise.resolve(),
         flowInputRequest: () => Promise.resolve(true),
@@ -485,6 +489,44 @@ describe("BridgeJobExecutor", () => {
             },
         ]);
     });
+    it("previews a step with the input it streamed, not the empty one it opened with", async () => {
+        const relay = recordingRelay();
+        const executor = createExecutor({
+            relay,
+            queryResults: [
+                scriptedMessages({ type: "stream_event", event: { type: "message_start" } }, {
+                    type: "stream_event",
+                    event: {
+                        type: "content_block_start",
+                        index: 0,
+                        content_block: { type: "tool_use", id: "tool_1", name: "Bash", input: {} },
+                    },
+                }, {
+                    type: "stream_event",
+                    event: {
+                        type: "content_block_delta",
+                        index: 0,
+                        delta: { type: "input_json_delta", partial_json: '{"command":"ls","descr' },
+                    },
+                }, {
+                    type: "stream_event",
+                    event: {
+                        type: "content_block_delta",
+                        index: 0,
+                        delta: { type: "input_json_delta", partial_json: 'iption":"Checking your pages"}' },
+                    },
+                }, { type: "stream_event", event: { type: "content_block_stop", index: 0 } }, resultMessage(0.01, "session_next")),
+            ],
+        });
+        await executor.execute({ threadId: "thread_1", turnId: "turn_1", prompt: "Look", history: history([]) }, activeControls());
+        expect(relay.previewedTools.at(-1)).toEqual([
+            expect.objectContaining({
+                toolUseId: "tool_1",
+                toolName: "Bash",
+                input: { command: "ls", description: "Checking your pages" },
+            }),
+        ]);
+    });
     it("ends the turn with a legible error and clears the session once the control channel is closed", async () => {
         const clearCalls = [];
         const executor = createExecutor({