@browserbasehq/orca 3.1.0-patch.4 → 3.2.0-middleware.2

package/dist/esm/tests/unit/flowlogger-capturing-cdp.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"flowlogger-capturing-cdp.test.js","sourceRoot":"","sources":["../../../../tests/unit/flowlogger-capturing-cdp.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,MAAM,aAAa,CAAC;AAC3C,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAC;AAC9C,OAAO,EAAE,aAAa,EAAE,MAAM,gCAAgC,CAAC;AAC/D,OAAO,EAAE,iBAAiB,EAAE,MAAM,sCAAsC,CAAC;AACzE,OAAO,EAAE,+BAA+B,EAAE,MAAM,yCAAyC,CAAC;AAC1F,OAAO,EAAE,UAAU,EAAE,MAAM,uCAAuC,CAAC;AACnE,OAAO,EAAE,SAAS,EAAE,UAAU,EAAE,MAAM,uCAAuC,CAAC;AAE9E,SAAS,qBAAqB,CAC5B,KAAiB,EACjB,GAAoC;IAEpC,MAAM,WAAW,GAAG,CAAC,KAAc,EAAE,EAAE;QACrC,IAAI,KAAK,YAAY,SAAS,EAAE,CAAC;YAC/B,KAAK,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACzB,CAAC;IACH,CAAC,CAAC;IAEF,GAAG,CAAC,EAAE,CAAC,GAAG,EAAE,WAAW,CAAC,CAAC;IACzB,OAAO,GAAG,EAAE;QACV,GAAG,CAAC,GAAG,CAAC,GAAG,EAAE,WAAW,CAAC,CAAC;IAC5B,CAAC,CAAC;AACJ,CAAC;AAED,MAAM,UAAW,SAAQ,YAAY;IACnC,YAAY,GAAa,EAAE,CAAC;IAC5B,UAAU,GAAG,CAAC,CAAC;IAEf,IAAI,CAAC,OAAe;QAClB,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IAClC,CAAC;IAED,KAAK;QACH,IAAI,CAAC,UAAU,GAAG,CAAC,CAAC;QACpB,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,IAAI,EAAE,EAAE,CAAC,CAAC;IAC/B,CAAC;CACF;AAED,SAAS,gBAAgB,CAAC,MAAkB;IAC1C,0EAA0E;IAC1E,gEAAgE;IAChE,MAAM,cAAc,GAAG,aAEtB,CAAC;IACF,OAAO,IAAI,cAAc,CAAC,MAAM,CAAC,CAAC;AACpC,CAAC;AAED,SAAS,YAAY,CACnB,MAAmB,EACnB,SAAwC,EACxC,WAAmB;IAEnB,MAAM,KAAK,GAAG,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;IACrC,MAAM,CAAC,KAAK,EAAE,WAAW,WAAW,EAAE,CAAC,CAAC,WAAW,EAAE,CAAC;IACtD,OAAO,KAAkB,CAAC;AAC5B,CAAC;AAED,QAAQ,CAAC,yBAAyB,EAAE,GAAG,EAAE;IACvC,EAAE,CAAC,yFAAyF,EAAE,KAAK,IAAI,EAAE;QACvG,MAAM,SAAS,GAAG,cAAc,CAAC;QACjC,MAAM,MAAM,GAAG,IAAI,UAAU,EAAE,CAAC;QAChC,MAAM,QAAQ,GAAG,IAAI,+BAA+B,EAAE,CAAC;QACvD,MAAM,IAAI,GAAG,IAAI,iBAAiB,EAAE,CAAC;QACrC,MAAM,UAAU,GAAG,IAAI,UAAU,CAAC,SAAS,EAAE,SAAS,EAAE,IAAI,CAAC,CAAC;QAE9D,MAAM,SAAS,GAAG,qBAAqB,CAAC,UAAU,EAAE,QAAQ,CAAC,CAAC;QAE9D,MAAM,IAAI,GAAG,gBAAgB,CAAC,MAAM,CAAC,CAAC;QACtC,IAAI,CAAC,iBAAiB,GAAG,UAAU,CAAC,IAAI,CAAC,SAAS,EAAE,QAAQ,CAAC,CAAC;QAE9D,wEAAwE;QACxE,qDAAqD;QACpD,IAAqD,CAAC,SAAS,CAC9D,IAAI,CAAC,SAAS,CAAC;YACb,MAAM,EAAE,yBAAyB;YACjC,MAAM,EAAE;gBACN,SAAS,EAAE,gBAAgB;gBAC3B,UAAU,EAAE,EAAE,QAAQ,EAAE,UAAU,EAAE;aACrC;SACF,CAAC,CACH,CAAC;QAEF,MAAM,OAAO,GAAG,IAAI,CAAC,UAAU,CAAC,gBAAgB,CAAC,CAAC;QAClD,MAAM,CAAC,OAAO,CAAC,CAAC,WAAW,EAAE,CAAC;QAE9B,OAAQ,CAAC,EAAE,CAAC,0BAA0B,EAAE,GAAG,EAAE;YAC3C,0EAA0E;YAC1E,oDAAoD;YACpD,KAAK,OAAQ,CAAC,IAAI,CAAC,kBAAkB,EAAE;gBACrC,UAAU,EAAE,OAAO;aACpB,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,MAAM,UAAU,CAAC,cAAc,CAC7B;YACE,OAAO,EAAE,IAAI,CAAC,iBAAiB;YAC/B,SAAS,EAAE,sBAAsB;SAClC,EACD,KAAK,IAAI,EAAE;YACT,KAAK,OAAQ,CAAC,IAAI,CAAC,eAAe,EAAE;gBAClC,GAAG,EAAE,qBAAqB;aAC3B,CAAC,CAAC;QACL,CAAC,EACD,EAAE,CACH,CAAC;QAED,IAAqD,CAAC,SAAS,CAC9D,IAAI,CAAC,SAAS,CAAC;YACb,MAAM,EAAE,0BAA0B;YAClC,SAAS,EAAE,gBAAgB;YAC3B,MAAM,EAAE,EAAE,IAAI,EAAE,KAAK,EAAE;SACxB,CAAC,CACH,CAAC;QAEF,2EAA2E;QAC3E,yEAAyE;QACzE,MAAM,MAAM,GAAG,MAAM,UAAU,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;QAC1C,MAAM,WAAW,GAAG,YAAY,CAC9B,MAAM,EACN,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,SAAS,KAAK,sBAAsB,EACrD,sBAAsB,CACvB,CAAC;QACF,MAAM,eAAe,GAAG,YAAY,CAClC,MAAM,EACN,CAAC,KAAK,EAAE,EAAE,CACR,KAAK,CAAC,SAAS,KAAK,cAAc;YAClC,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,KAAK,kBAAkB,EAClD,sCAAsC,CACvC,CAAC;QAEF,MAAM,CAAC,eAAe,CAAC,cAAc,CAAC,CAAC,OAAO,CAAC,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC,CAAC;QAEtE,SAAS,EAAE,CAAC;QACZ,MAAM,UAAU,CAAC,OAAO,EAAE,CAAC;IAC7B,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC","sourcesContent":["import { EventEmitter } from \"node:events\";\nimport { describe, it, expect } from \"vitest\";\nimport { CdpConnection } from \"../../lib/v3/understudy/cdp.js\";\nimport { InMemoryEventSink } from \"../../lib/v3/flowlogger/EventSink.js\";\nimport { EventEmitterWithWildcardSupport } from \"../../lib/v3/flowlogger/EventEmitter.js\";\nimport { EventStore } from \"../../lib/v3/flowlogger/EventStore.js\";\nimport { FlowEvent, FlowLogger } from \"../../lib/v3/flowlogger/FlowLogger.js\";\n\nfunction attachEventStoreToBus(\n  store: EventStore,\n  bus: EventEmitterWithWildcardSupport,\n): () => void {\n  const onFlowEvent = (event: unknown) => {\n    if (event instanceof FlowEvent) {\n      void store.emit(event);\n    }\n  };\n\n  bus.on(\"*\", onFlowEvent);\n  return () => {\n    bus.off(\"*\", onFlowEvent);\n  };\n}\n\nclass FakeSocket extends EventEmitter {\n  sentPayloads: string[] = [];\n  readyState = 1;\n\n  send(payload: string): void {\n    this.sentPayloads.push(payload);\n  }\n\n  close(): void {\n    this.readyState = 3;\n    this.emit(\"close\", 1000, \"\");\n  }\n}\n\nfunction createConnection(socket: FakeSocket): CdpConnection {\n  // The production constructor is private; tests instantiate it directly so\n  // they can drive raw websocket messages without a real browser.\n  const ConnectionCtor = CdpConnection as unknown as {\n    new (ws: FakeSocket): CdpConnection;\n  };\n  return new ConnectionCtor(socket);\n}\n\nfunction requireEvent(\n  events: FlowEvent[],\n  predicate: (event: FlowEvent) => boolean,\n  description: string,\n): FlowEvent {\n  const match = events.find(predicate);\n  expect(match, `missing ${description}`).toBeDefined();\n  return match as FlowEvent;\n}\n\ndescribe(\"flow logger cdp context\", () => {\n  it(\"preserves the active parent chain when a session event handler issues a nested CDP call\", async () => {\n    const sessionId = \"session-test\";\n    const socket = new FakeSocket();\n    const eventBus = new EventEmitterWithWildcardSupport();\n    const sink = new InMemoryEventSink();\n    const eventStore = new EventStore(sessionId, undefined, sink);\n\n    const detachBus = attachEventStoreToBus(eventStore, eventBus);\n\n    const conn = createConnection(socket);\n    conn.flowLoggerContext = FlowLogger.init(sessionId, eventBus);\n\n    // Seed the target/session mapping the same way a real attach flow would\n    // before any session-scoped messages are dispatched.\n    (conn as unknown as { onMessage(json: string): void }).onMessage(\n      JSON.stringify({\n        method: \"Target.attachedToTarget\",\n        params: {\n          sessionId: \"target-session\",\n          targetInfo: { targetId: \"target-1\" },\n        },\n      }),\n    );\n\n    const session = conn.getSession(\"target-session\");\n    expect(session).toBeDefined();\n\n    session!.on(\"Runtime.consoleAPICalled\", () => {\n      // This nested send used to lose its parent chain because the callback ran\n      // after the original ALS scope had already unwound.\n      void session!.send(\"Runtime.evaluate\", {\n        expression: \"2 + 2\",\n      });\n    });\n\n    await FlowLogger.runWithLogging(\n      {\n        context: conn.flowLoggerContext,\n        eventType: \"SyntheticParentEvent\",\n      },\n      async () => {\n        void session!.send(\"Page.navigate\", {\n          url: \"https://example.com\",\n        });\n      },\n      [],\n    );\n\n    (conn as unknown as { onMessage(json: string): void }).onMessage(\n      JSON.stringify({\n        method: \"Runtime.consoleAPICalled\",\n        sessionId: \"target-session\",\n        params: { type: \"log\" },\n      }),\n    );\n\n    // The nested Runtime.evaluate call should still attach under the synthetic\n    // parent event even though it was triggered by a later session callback.\n    const events = await eventStore.query({});\n    const parentEvent = requireEvent(\n      events,\n      (event) => event.eventType === \"SyntheticParentEvent\",\n      \"SyntheticParentEvent\",\n    );\n    const nestedCallEvent = requireEvent(\n      events,\n      (event) =>\n        event.eventType === \"CdpCallEvent\" &&\n        String(event.data.method) === \"Runtime.evaluate\",\n      \"nested Runtime.evaluate CdpCallEvent\",\n    );\n\n    expect(nestedCallEvent.eventParentIds).toEqual([parentEvent.eventId]);\n\n    detachBus();\n    await eventStore.destroy();\n  });\n});\n"]}

package/dist/esm/tests/unit/flowlogger-capturing-llm.test.d.ts

@@ -0,0 +1 @@
+export {};

package/dist/esm/tests/unit/flowlogger-capturing-llm.test.js

@@ -0,0 +1,41 @@
+import { describe, expect, it } from "vitest";
+import { FlowLogger } from "../../lib/v3/flowlogger/FlowLogger.js";
+describe("flow logger llm logging", () => {
+    it("no-ops direct llm logging calls when no flow context is active", () => {
+        // These helpers are called from multiple model adapters, so they must stay
+        // safe even when a test or utility invokes them outside any ALS flow scope.
+        expect(() => FlowLogger.logLlmRequest({
+            requestId: "req-1",
+            model: "mock-model",
+            prompt: "hello",
+        })).not.toThrow();
+        expect(() => FlowLogger.logLlmResponse({
+            requestId: "req-1",
+            model: "mock-model",
+            output: "world",
+            inputTokens: 1,
+            outputTokens: 1,
+        })).not.toThrow();
+    });
+    it("does not throw from llm middleware when no flow context is active", async () => {
+        const middleware = FlowLogger.createLlmLoggingMiddleware("mock-model");
+        // Missing flow context should degrade to a silent no-op and preserve the
+        // underlying model result.
+        await expect(middleware.wrapGenerate({
+            doGenerate: async () => ({
+                text: "done",
+                usage: {
+                    inputTokens: 1,
+                    outputTokens: 1,
+                    totalTokens: 2,
+                },
+            }),
+            params: {
+                prompt: [],
+            },
+        })).resolves.toMatchObject({
+            text: "done",
+        });
+    });
+});
+//# sourceMappingURL=flowlogger-capturing-llm.test.js.map

package/dist/esm/tests/unit/flowlogger-capturing-llm.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"flowlogger-capturing-llm.test.js","sourceRoot":"","sources":["../../../../tests/unit/flowlogger-capturing-llm.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAC9C,OAAO,EAAE,UAAU,EAAE,MAAM,uCAAuC,CAAC;AAEnE,QAAQ,CAAC,yBAAyB,EAAE,GAAG,EAAE;IACvC,EAAE,CAAC,gEAAgE,EAAE,GAAG,EAAE;QACxE,2EAA2E;QAC3E,4EAA4E;QAC5E,MAAM,CAAC,GAAG,EAAE,CACV,UAAU,CAAC,aAAa,CAAC;YACvB,SAAS,EAAE,OAAO;YAClB,KAAK,EAAE,YAAY;YACnB,MAAM,EAAE,OAAO;SAChB,CAAC,CACH,CAAC,GAAG,CAAC,OAAO,EAAE,CAAC;QAEhB,MAAM,CAAC,GAAG,EAAE,CACV,UAAU,CAAC,cAAc,CAAC;YACxB,SAAS,EAAE,OAAO;YAClB,KAAK,EAAE,YAAY;YACnB,MAAM,EAAE,OAAO;YACf,WAAW,EAAE,CAAC;YACd,YAAY,EAAE,CAAC;SAChB,CAAC,CACH,CAAC,GAAG,CAAC,OAAO,EAAE,CAAC;IAClB,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,mEAAmE,EAAE,KAAK,IAAI,EAAE;QACjF,MAAM,UAAU,GAAG,UAAU,CAAC,0BAA0B,CAAC,YAAY,CAAC,CAAC;QAEvE,yEAAyE;QACzE,2BAA2B;QAC3B,MAAM,MAAM,CACV,UAAU,CAAC,YAAY,CAAC;YACtB,UAAU,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC;gBACvB,IAAI,EAAE,MAAM;gBACZ,KAAK,EAAE;oBACL,WAAW,EAAE,CAAC;oBACd,YAAY,EAAE,CAAC;oBACf,WAAW,EAAE,CAAC;iBACf;aACF,CAAC;YACF,MAAM,EAAE;gBACN,MAAM,EAAE,EAAE;aACX;SACO,CAAC,CACZ,CAAC,QAAQ,CAAC,aAAa,CAAC;YACvB,IAAI,EAAE,MAAM;SACb,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC","sourcesContent":["import { describe, expect, it } from \"vitest\";\nimport { FlowLogger } from \"../../lib/v3/flowlogger/FlowLogger.js\";\n\ndescribe(\"flow logger llm logging\", () => {\n  it(\"no-ops direct llm logging calls when no flow context is active\", () => {\n    // These helpers are called from multiple model adapters, so they must stay\n    // safe even when a test or utility invokes them outside any ALS flow scope.\n    expect(() =>\n      FlowLogger.logLlmRequest({\n        requestId: \"req-1\",\n        model: \"mock-model\",\n        prompt: \"hello\",\n      }),\n    ).not.toThrow();\n\n    expect(() =>\n      FlowLogger.logLlmResponse({\n        requestId: \"req-1\",\n        model: \"mock-model\",\n        output: \"world\",\n        inputTokens: 1,\n        outputTokens: 1,\n      }),\n    ).not.toThrow();\n  });\n\n  it(\"does not throw from llm middleware when no flow context is active\", async () => {\n    const middleware = FlowLogger.createLlmLoggingMiddleware(\"mock-model\");\n\n    // Missing flow context should degrade to a silent no-op and preserve the\n    // underlying model result.\n    await expect(\n      middleware.wrapGenerate({\n        doGenerate: async () => ({\n          text: \"done\",\n          usage: {\n            inputTokens: 1,\n            outputTokens: 1,\n            totalTokens: 2,\n          },\n        }),\n        params: {\n          prompt: [],\n        },\n      } as never),\n    ).resolves.toMatchObject({\n      text: \"done\",\n    });\n  });\n});\n"]}

package/dist/esm/tests/unit/flowlogger-eventstore.test.d.ts

@@ -0,0 +1 @@
+export {};

package/dist/esm/tests/unit/flowlogger-eventstore.test.js

@@ -0,0 +1,248 @@
+import { afterEach, describe, expect, it } from "vitest";
+import { EventStore } from "../../lib/v3/flowlogger/EventStore.js";
+import { EventEmitterWithWildcardSupport } from "../../lib/v3/flowlogger/EventEmitter.js";
+import { FlowEvent } from "../../lib/v3/flowlogger/FlowLogger.js";
+function attachEventStoreToBus(store, bus) {
+    const onFlowEvent = (event) => {
+        if (event instanceof FlowEvent) {
+            void store.emit(event);
+        }
+    };
+    bus.on("*", onFlowEvent);
+    return () => {
+        bus.off("*", onFlowEvent);
+    };
+}
+function createVerboseStoreHarness() {
+    const writes = [];
+    process.stderr.write = ((chunk, cb) => {
+        writes.push(String(chunk));
+        cb?.(null);
+        return true;
+    });
+    const store = new EventStore("session-test");
+    const bus = new EventEmitterWithWildcardSupport();
+    const detachBus = attachEventStoreToBus(store, bus);
+    return { writes, store, bus, detachBus };
+}
+describe("flow logger event store", () => {
+    const stderrWrite = process.stderr.write.bind(process.stderr);
+    afterEach(() => {
+        process.stderr.write = stderrWrite;
+    });
+    it("queries recent events from the default in-memory sink", async () => {
+        const store = new EventStore("session-test");
+        await store.emit(new FlowEvent({
+            eventType: "StagehandExtractEvent",
+            sessionId: "session-test",
+            eventId: "stagehand-1234",
+            eventCreatedAt: "2026-03-16T21:45:00.000Z",
+            data: { params: ["grab title"] },
+        }));
+        const events = await store.query({});
+        expect(events).toHaveLength(1);
+        expect(events[0].eventType).toBe("StagehandExtractEvent");
+        await store.destroy();
+    });
+    it("drops payloads from the default in-memory sink", async () => {
+        const store = new EventStore("session-test");
+        await store.emit(new FlowEvent({
+            eventType: "LlmRequestEvent",
+            sessionId: "session-test",
+            eventId: "llm-1234",
+            eventCreatedAt: "2026-03-16T21:45:00.000Z",
+            data: {
+                prompt: [{ type: "image_url", image_url: { url: "huge" } }],
+                output: "huge",
+            },
+        }));
+        const [event] = await store.query({});
+        expect(event.eventType).toBe("LlmRequestEvent");
+        expect(event.eventId).toBe("llm-1234");
+        expect(event.data).toEqual({});
+        await store.destroy();
+    });
+    it("renders semantic hierarchy tags for non-cdp stderr events only", async () => {
+        // Intercept stderr so the pretty sink can be asserted without polluting the
+        // real test runner output.
+        const { writes, store, bus, detachBus } = createVerboseStoreHarness();
+        const stepEvent = new FlowEvent({
+            eventType: "StagehandExtractEvent",
+            sessionId: "session-test",
+            eventId: "stagehand-1234",
+            eventCreatedAt: "2026-03-16T21:45:00.000Z",
+            data: { params: ["grab title"] },
+        });
+        const cdpEvent = new FlowEvent({
+            eventType: "CdpCallEvent",
+            sessionId: "session-test",
+            eventId: "cdp-call-5678",
+            eventCreatedAt: "2026-03-16T21:45:00.100Z",
+            eventParentIds: [stepEvent.eventId],
+            data: {
+                method: "Runtime.evaluate",
+                params: { expression: "2 + 2" },
+                targetId: "1234567890ABCDEF1234567890ABCDEF",
+            },
+        });
+        // The stderr sink intentionally suppresses CDP noise even though the event
+        // still exists for in-memory and file-backed sinks.
+        bus.emit(stepEvent.eventType, stepEvent);
+        bus.emit(cdpEvent.eventType, cdpEvent);
+        await new Promise((resolve) => setTimeout(resolve, 0));
+        expect(writes).toHaveLength(1);
+        expect(writes[0]).toContain("[🆂 #1234 EXTRACT]");
+        expect(writes[0]).toContain("Stagehand.extract");
+        expect(writes[0]).not.toContain("Runtime.evaluate");
+        detachBus();
+        await store.destroy();
+    });
+    it("renders generic stagehand events without crashing the stderr sink", async () => {
+        const { writes, store, bus, detachBus } = createVerboseStoreHarness();
+        // `StagehandEvent` has no action suffix, so this guards the formatter path
+        // that cannot assume a method name exists.
+        bus.emit("StagehandEvent", new FlowEvent({
+            eventType: "StagehandEvent",
+            sessionId: "session-test",
+            eventId: "stagehand-0001",
+            eventCreatedAt: "2026-03-16T21:45:00.000Z",
+            data: { params: ["noop"] },
+        }));
+        await new Promise((resolve) => setTimeout(resolve, 0));
+        expect(writes).toHaveLength(1);
+        expect(writes[0]).toContain("[🆂 #0001");
+        expect(writes[0]).toContain("Stagehand(");
+        detachBus();
+        await store.destroy();
+    });
+    it("colorizes pretty stderr output with ansi escapes when enabled", async () => {
+        const previousForceColor = process.env.FORCE_COLOR;
+        const previousNoColor = process.env.NO_COLOR;
+        delete process.env.NO_COLOR;
+        process.env.FORCE_COLOR = "1";
+        const { writes, store, bus, detachBus } = createVerboseStoreHarness();
+        try {
+            bus.emit("StagehandActEvent", new FlowEvent({
+                eventType: "StagehandActEvent",
+                sessionId: "session-test",
+                eventId: "stagehand-0002",
+                eventCreatedAt: "2026-03-16T21:45:00.000Z",
+                data: { params: ["click submit"] },
+            }));
+            await new Promise((resolve) => setTimeout(resolve, 0));
+            expect(writes).toHaveLength(1);
+            expect(writes[0]).toContain("\u001B[");
+        }
+        finally {
+            if (previousNoColor === undefined) {
+                delete process.env.NO_COLOR;
+            }
+            else {
+                process.env.NO_COLOR = previousNoColor;
+            }
+            if (previousForceColor === undefined) {
+                delete process.env.FORCE_COLOR;
+            }
+            else {
+                process.env.FORCE_COLOR = previousForceColor;
+            }
+            detachBus();
+            await store.destroy();
+        }
+    });
+    it("keeps agent ancestry and start ids for completion events after many child events", async () => {
+        const { writes, store, bus, detachBus } = createVerboseStoreHarness();
+        const agentEvent = new FlowEvent({
+            eventType: "AgentExecuteEvent",
+            sessionId: "session-test",
+            eventId: "agent-1234",
+            eventCreatedAt: "2026-03-16T21:45:00.000Z",
+            data: { params: [{ instruction: "click the button" }] },
+        });
+        const actEvent = new FlowEvent({
+            eventType: "StagehandActEvent",
+            sessionId: "session-test",
+            eventId: "stagehand-2222",
+            eventCreatedAt: "2026-03-16T21:45:00.001Z",
+            eventParentIds: [agentEvent.eventId],
+            data: { params: ["click the button"] },
+        });
+        const clickEvent = new FlowEvent({
+            eventType: "UnderstudyClickEvent",
+            sessionId: "session-test",
+            eventId: "action-3333",
+            eventCreatedAt: "2026-03-16T21:45:00.002Z",
+            eventParentIds: [agentEvent.eventId, actEvent.eventId],
+            data: { target: "xpath=/button[1]" },
+        });
+        bus.emit(agentEvent.eventType, agentEvent);
+        bus.emit(actEvent.eventType, actEvent);
+        bus.emit(clickEvent.eventType, clickEvent);
+        // Flood the retained history with child events so the completion lines have
+        // to recover their displayed ancestry from the queryable sink.
+        for (let index = 0; index < 150; index += 1) {
+            bus.emit("CdpCallEvent", new FlowEvent({
+                eventType: "CdpCallEvent",
+                sessionId: "session-test",
+                eventId: `cdp-${String(index).padStart(4, "0")}`,
+                eventCreatedAt: `2026-03-16T21:45:00.${String(index + 10).padStart(3, "0")}Z`,
+                eventParentIds: [
+                    agentEvent.eventId,
+                    actEvent.eventId,
+                    clickEvent.eventId,
+                ],
+                data: {
+                    method: "Runtime.evaluate",
+                    params: { expression: `${index}` },
+                    targetId: "1234567890ABCDEF1234567890ABCDEF",
+                },
+            }));
+        }
+        bus.emit("UnderstudyClickCompletedEvent", new FlowEvent({
+            eventType: "UnderstudyClickCompletedEvent",
+            sessionId: "session-test",
+            eventId: "done-4444",
+            eventCreatedAt: "2026-03-16T21:45:01.000Z",
+            eventParentIds: [
+                agentEvent.eventId,
+                actEvent.eventId,
+                clickEvent.eventId,
+            ],
+            data: { durationMs: 250 },
+        }));
+        bus.emit("StagehandActCompletedEvent", new FlowEvent({
+            eventType: "StagehandActCompletedEvent",
+            sessionId: "session-test",
+            eventId: "done-5555",
+            eventCreatedAt: "2026-03-16T21:45:01.001Z",
+            eventParentIds: [agentEvent.eventId, actEvent.eventId],
+            data: { durationMs: 500 },
+        }));
+        bus.emit("AgentExecuteCompletedEvent", new FlowEvent({
+            eventType: "AgentExecuteCompletedEvent",
+            sessionId: "session-test",
+            eventId: "done-6666",
+            eventCreatedAt: "2026-03-16T21:45:01.002Z",
+            eventParentIds: [agentEvent.eventId],
+            data: { durationMs: 750 },
+        }));
+        await new Promise((resolve) => setTimeout(resolve, 0));
+        // Completion lines should reference the original started-event ids, not the
+        // synthetic completed-event ids emitted at the end of the lifecycle.
+        const clickCompletedLine = writes.find((line) => line.includes("CLICK completed"));
+        const actCompletedLine = writes.find((line) => line.includes("ACT completed"));
+        const agentCompletedLine = writes.find((line) => line.includes("Agent.execute() completed"));
+        expect(clickCompletedLine).toContain("[🅰 #1234]");
+        expect(clickCompletedLine).toContain("[🆂 #2222 ACT]");
+        expect(clickCompletedLine).toContain("[🆄 #3333 CLICK]");
+        expect(clickCompletedLine).not.toContain("#4444");
+        expect(actCompletedLine).toContain("[🅰 #1234]");
+        expect(actCompletedLine).toContain("[🆂 #2222 ACT]");
+        expect(actCompletedLine).not.toContain("#5555");
+        expect(agentCompletedLine).toContain("[🅰 #1234]");
+        expect(agentCompletedLine).not.toContain("#6666");
+        detachBus();
+        await store.destroy();
+    });
+});
+//# sourceMappingURL=flowlogger-eventstore.test.js.map

package/dist/esm/tests/unit/flowlogger-eventstore.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"flowlogger-eventstore.test.js","sourceRoot":"","sources":["../../../../tests/unit/flowlogger-eventstore.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,SAAS,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AACzD,OAAO,EAAE,UAAU,EAAE,MAAM,uCAAuC,CAAC;AACnE,OAAO,EAAE,+BAA+B,EAAE,MAAM,yCAAyC,CAAC;AAC1F,OAAO,EAAE,SAAS,EAAE,MAAM,uCAAuC,CAAC;AAElE,SAAS,qBAAqB,CAC5B,KAAiB,EACjB,GAAoC;IAEpC,MAAM,WAAW,GAAG,CAAC,KAAc,EAAE,EAAE;QACrC,IAAI,KAAK,YAAY,SAAS,EAAE,CAAC;YAC/B,KAAK,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACzB,CAAC;IACH,CAAC,CAAC;IAEF,GAAG,CAAC,EAAE,CAAC,GAAG,EAAE,WAAW,CAAC,CAAC;IACzB,OAAO,GAAG,EAAE;QACV,GAAG,CAAC,GAAG,CAAC,GAAG,EAAE,WAAW,CAAC,CAAC;IAC5B,CAAC,CAAC;AACJ,CAAC;AAED,SAAS,yBAAyB;IAMhC,MAAM,MAAM,GAAa,EAAE,CAAC;IAC5B,OAAO,CAAC,MAAM,CAAC,KAAK,GAAG,CAAC,CACtB,KAAa,EACb,EAAmC,EACnC,EAAE;QACF,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC;QAC3B,EAAE,EAAE,CAAC,IAAI,CAAC,CAAC;QACX,OAAO,IAAI,CAAC;IACd,CAAC,CAAgC,CAAC;IAElC,MAAM,KAAK,GAAG,IAAI,UAAU,CAAC,cAAc,CAAC,CAAC;IAC7C,MAAM,GAAG,GAAG,IAAI,+BAA+B,EAAE,CAAC;IAClD,MAAM,SAAS,GAAG,qBAAqB,CAAC,KAAK,EAAE,GAAG,CAAC,CAAC;IAEpD,OAAO,EAAE,MAAM,EAAE,KAAK,EAAE,GAAG,EAAE,SAAS,EAAE,CAAC;AAC3C,CAAC;AAED,QAAQ,CAAC,yBAAyB,EAAE,GAAG,EAAE;IACvC,MAAM,WAAW,GAAG,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;IAE9D,SAAS,CAAC,GAAG,EAAE;QACb,OAAO,CAAC,MAAM,CAAC,KAAK,GAAG,WAAW,CAAC;IACrC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,uDAAuD,EAAE,KAAK,IAAI,EAAE;QACrE,MAAM,KAAK,GAAG,IAAI,UAAU,CAAC,cAAc,CAAC,CAAC;QAE7C,MAAM,KAAK,CAAC,IAAI,CACd,IAAI,SAAS,CAAC;YACZ,SAAS,EAAE,uBAAuB;YAClC,SAAS,EAAE,cAAc;YACzB,OAAO,EAAE,gBAAgB;YACzB,cAAc,EAAE,0BAA0B;YAC1C,IAAI,EAAE,EAAE,MAAM,EAAE,CAAC,YAAY,CAAC,EAAE;SACjC,CAAC,CACH,CAAC;QAEF,MAAM,MAAM,GAAG,MAAM,KAAK,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;QACrC,MAAM,CAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAC/B,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,uBAAuB,CAAC,CAAC;QAE1D,MAAM,KAAK,CAAC,OAAO,EAAE,CAAC;IACxB,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gDAAgD,EAAE,KAAK,IAAI,EAAE;QAC9D,MAAM,KAAK,GAAG,IAAI,UAAU,CAAC,cAAc,CAAC,CAAC;QAE7C,MAAM,KAAK,CAAC,IAAI,CACd,IAAI,SAAS,CAAC;YACZ,SAAS,EAAE,iBAAiB;YAC5B,SAAS,EAAE,cAAc;YACzB,OAAO,EAAE,UAAU;YACnB,cAAc,EAAE,0BAA0B;YAC1C,IAAI,EAAE;gBACJ,MAAM,EAAE,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,SAAS,EAAE,EAAE,GAAG,EAAE,MAAM,EAAE,EAAE,CAAC;gBAC3D,MAAM,EAAE,MAAM;aACf;SACF,CAAC,CACH,CAAC;QAEF,MAAM,CAAC,KAAK,CAAC,GAAG,MAAM,KAAK,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;QACtC,MAAM,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;QAChD,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QACvC,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;QAE/B,MAAM,KAAK,CAAC,OAAO,EAAE,CAAC;IACxB,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gEAAgE,EAAE,KAAK,IAAI,EAAE;QAC9E,4EAA4E;QAC5E,2BAA2B;QAC3B,MAAM,EAAE,MAAM,EAAE,KAAK,EAAE,GAAG,EAAE,SAAS,EAAE,GAAG,yBAAyB,EAAE,CAAC;QAEtE,MAAM,SAAS,GAAG,IAAI,SAAS,CAAC;YAC9B,SAAS,EAAE,uBAAuB;YAClC,SAAS,EAAE,cAAc;YACzB,OAAO,EAAE,gBAAgB;YACzB,cAAc,EAAE,0BAA0B;YAC1C,IAAI,EAAE,EAAE,MAAM,EAAE,CAAC,YAAY,CAAC,EAAE;SACjC,CAAC,CAAC;QACH,MAAM,QAAQ,GAAG,IAAI,SAAS,CAAC;YAC7B,SAAS,EAAE,cAAc;YACzB,SAAS,EAAE,cAAc;YACzB,OAAO,EAAE,eAAe;YACxB,cAAc,EAAE,0BAA0B;YAC1C,cAAc,EAAE,CAAC,SAAS,CAAC,OAAO,CAAC;YACnC,IAAI,EAAE;gBACJ,MAAM,EAAE,kBAAkB;gBAC1B,MAAM,EAAE,EAAE,UAAU,EAAE,OAAO,EAAE;gBAC/B,QAAQ,EAAE,kCAAkC;aAC7C;SACF,CAAC,CAAC;QAEH,2EAA2E;QAC3E,oDAAoD;QACpD,GAAG,CAAC,IAAI,CAAC,SAAS,CAAC,SAAS,EAAE,SAAS,CAAC,CAAC;QACzC,GAAG,CAAC,IAAI,CAAC,QAAQ,CAAC,SAAS,EAAE,QAAQ,CAAC,CAAC;QACvC,MAAM,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,CAAC;QAEvD,MAAM,CAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAC/B,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,oBAAoB,CAAC,CAAC;QAClD,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,mBAAmB,CAAC,CAAC;QACjD,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,kBAAkB,CAAC,CAAC;QAEpD,SAAS,EAAE,CAAC;QACZ,MAAM,KAAK,CAAC,OAAO,EAAE,CAAC;IACxB,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,mEAAmE,EAAE,KAAK,IAAI,EAAE;QACjF,MAAM,EAAE,MAAM,EAAE,KAAK,EAAE,GAAG,EAAE,SAAS,EAAE,GAAG,yBAAyB,EAAE,CAAC;QAEtE,2EAA2E;QAC3E,2CAA2C;QAC3C,GAAG,CAAC,IAAI,CACN,gBAAgB,EAChB,IAAI,SAAS,CAAC;YACZ,SAAS,EAAE,gBAAgB;YAC3B,SAAS,EAAE,cAAc;YACzB,OAAO,EAAE,gBAAgB;YACzB,cAAc,EAAE,0BAA0B;YAC1C,IAAI,EAAE,EAAE,MAAM,EAAE,CAAC,MAAM,CAAC,EAAE;SAC3B,CAAC,CACH,CAAC;QACF,MAAM,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,CAAC;QAEvD,MAAM,CAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAC/B,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,WAAW,CAAC,CAAC;QACzC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,YAAY,CAAC,CAAC;QAE1C,SAAS,EAAE,CAAC;QACZ,MAAM,KAAK,CAAC,OAAO,EAAE,CAAC;IACxB,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+DAA+D,EAAE,KAAK,IAAI,EAAE;QAC7E,MAAM,kBAAkB,GAAG,OAAO,CAAC,GAAG,CAAC,WAAW,CAAC;QACnD,MAAM,eAAe,GAAG,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC;QAC7C,OAAO,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC;QAC5B,OAAO,CAAC,GAAG,CAAC,WAAW,GAAG,GAAG,CAAC;QAE9B,MAAM,EAAE,MAAM,EAAE,KAAK,EAAE,GAAG,EAAE,SAAS,EAAE,GAAG,yBAAyB,EAAE,CAAC;QAEtE,IAAI,CAAC;YACH,GAAG,CAAC,IAAI,CACN,mBAAmB,EACnB,IAAI,SAAS,CAAC;gBACZ,SAAS,EAAE,mBAAmB;gBAC9B,SAAS,EAAE,cAAc;gBACzB,OAAO,EAAE,gBAAgB;gBACzB,cAAc,EAAE,0BAA0B;gBAC1C,IAAI,EAAE,EAAE,MAAM,EAAE,CAAC,cAAc,CAAC,EAAE;aACnC,CAAC,CACH,CAAC;YACF,MAAM,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,CAAC;YAEvD,MAAM,CAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YAC/B,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC;QACzC,CAAC;gBAAS,CAAC;YACT,IAAI,eAAe,KAAK,SAAS,EAAE,CAAC;gBAClC,OAAO,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC;YAC9B,CAAC;iBAAM,CAAC;gBACN,OAAO,CAAC,GAAG,CAAC,QAAQ,GAAG,eAAe,CAAC;YACzC,CAAC;YAED,IAAI,kBAAkB,KAAK,SAAS,EAAE,CAAC;gBACrC,OAAO,OAAO,CAAC,GAAG,CAAC,WAAW,CAAC;YACjC,CAAC;iBAAM,CAAC;gBACN,OAAO,CAAC,GAAG,CAAC,WAAW,GAAG,kBAAkB,CAAC;YAC/C,CAAC;YAED,SAAS,EAAE,CAAC;YACZ,MAAM,KAAK,CAAC,OAAO,EAAE,CAAC;QACxB,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kFAAkF,EAAE,KAAK,IAAI,EAAE;QAChG,MAAM,EAAE,MAAM,EAAE,KAAK,EAAE,GAAG,EAAE,SAAS,EAAE,GAAG,yBAAyB,EAAE,CAAC;QAEtE,MAAM,UAAU,GAAG,IAAI,SAAS,CAAC;YAC/B,SAAS,EAAE,mBAAmB;YAC9B,SAAS,EAAE,cAAc;YACzB,OAAO,EAAE,YAAY;YACrB,cAAc,EAAE,0BAA0B;YAC1C,IAAI,EAAE,EAAE,MAAM,EAAE,CAAC,EAAE,WAAW,EAAE,kBAAkB,EAAE,CAAC,EAAE;SACxD,CAAC,CAAC;QACH,MAAM,QAAQ,GAAG,IAAI,SAAS,CAAC;YAC7B,SAAS,EAAE,mBAAmB;YAC9B,SAAS,EAAE,cAAc;YACzB,OAAO,EAAE,gBAAgB;YACzB,cAAc,EAAE,0BAA0B;YAC1C,cAAc,EAAE,CAAC,UAAU,CAAC,OAAO,CAAC;YACpC,IAAI,EAAE,EAAE,MAAM,EAAE,CAAC,kBAAkB,CAAC,EAAE;SACvC,CAAC,CAAC;QACH,MAAM,UAAU,GAAG,IAAI,SAAS,CAAC;YAC/B,SAAS,EAAE,sBAAsB;YACjC,SAAS,EAAE,cAAc;YACzB,OAAO,EAAE,aAAa;YACtB,cAAc,EAAE,0BAA0B;YAC1C,cAAc,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,QAAQ,CAAC,OAAO,CAAC;YACtD,IAAI,EAAE,EAAE,MAAM,EAAE,kBAAkB,EAAE;SACrC,CAAC,CAAC;QAEH,GAAG,CAAC,IAAI,CAAC,UAAU,CAAC,SAAS,EAAE,UAAU,CAAC,CAAC;QAC3C,GAAG,CAAC,IAAI,CAAC,QAAQ,CAAC,SAAS,EAAE,QAAQ,CAAC,CAAC;QACvC,GAAG,CAAC,IAAI,CAAC,UAAU,CAAC,SAAS,EAAE,UAAU,CAAC,CAAC;QAE3C,4EAA4E;QAC5E,+DAA+D;QAC/D,KAAK,IAAI,KAAK,GAAG,CAAC,EAAE,KAAK,GAAG,GAAG,EAAE,KAAK,IAAI,CAAC,EAAE,CAAC;YAC5C,GAAG,CAAC,IAAI,CACN,cAAc,EACd,IAAI,SAAS,CAAC;gBACZ,SAAS,EAAE,cAAc;gBACzB,SAAS,EAAE,cAAc;gBACzB,OAAO,EAAE,OAAO,MAAM,CAAC,KAAK,CAAC,CAAC,QAAQ,CAAC,CAAC,EAAE,GAAG,CAAC,EAAE;gBAChD,cAAc,EAAE,uBAAuB,MAAM,CAAC,KAAK,GAAG,EAAE,CAAC,CAAC,QAAQ,CAAC,CAAC,EAAE,GAAG,CAAC,GAAG;gBAC7E,cAAc,EAAE;oBACd,UAAU,CAAC,OAAO;oBAClB,QAAQ,CAAC,OAAO;oBAChB,UAAU,CAAC,OAAO;iBACnB;gBACD,IAAI,EAAE;oBACJ,MAAM,EAAE,kBAAkB;oBAC1B,MAAM,EAAE,EAAE,UAAU,EAAE,GAAG,KAAK,EAAE,EAAE;oBAClC,QAAQ,EAAE,kCAAkC;iBAC7C;aACF,CAAC,CACH,CAAC;QACJ,CAAC;QAED,GAAG,CAAC,IAAI,CACN,+BAA+B,EAC/B,IAAI,SAAS,CAAC;YACZ,SAAS,EAAE,+BAA+B;YAC1C,SAAS,EAAE,cAAc;YACzB,OAAO,EAAE,WAAW;YACpB,cAAc,EAAE,0BAA0B;YAC1C,cAAc,EAAE;gBACd,UAAU,CAAC,OAAO;gBAClB,QAAQ,CAAC,OAAO;gBAChB,UAAU,CAAC,OAAO;aACnB;YACD,IAAI,EAAE,EAAE,UAAU,EAAE,GAAG,EAAE;SAC1B,CAAC,CACH,CAAC;QACF,GAAG,CAAC,IAAI,CACN,4BAA4B,EAC5B,IAAI,SAAS,CAAC;YACZ,SAAS,EAAE,4BAA4B;YACvC,SAAS,EAAE,cAAc;YACzB,OAAO,EAAE,WAAW;YACpB,cAAc,EAAE,0BAA0B;YAC1C,cAAc,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,QAAQ,CAAC,OAAO,CAAC;YACtD,IAAI,EAAE,EAAE,UAAU,EAAE,GAAG,EAAE;SAC1B,CAAC,CACH,CAAC;QACF,GAAG,CAAC,IAAI,CACN,4BAA4B,EAC5B,IAAI,SAAS,CAAC;YACZ,SAAS,EAAE,4BAA4B;YACvC,SAAS,EAAE,cAAc;YACzB,OAAO,EAAE,WAAW;YACpB,cAAc,EAAE,0BAA0B;YAC1C,cAAc,EAAE,CAAC,UAAU,CAAC,OAAO,CAAC;YACpC,IAAI,EAAE,EAAE,UAAU,EAAE,GAAG,EAAE;SAC1B,CAAC,CACH,CAAC;QACF,MAAM,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,CAAC;QAEvD,4EAA4E;QAC5E,qEAAqE;QACrE,MAAM,kBAAkB,GAAG,MAAM,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAC9C,IAAI,CAAC,QAAQ,CAAC,iBAAiB,CAAC,CACjC,CAAC;QACF,MAAM,gBAAgB,GAAG,MAAM,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAC5C,IAAI,CAAC,QAAQ,CAAC,eAAe,CAAC,CAC/B,CAAC;QACF,MAAM,kBAAkB,GAAG,MAAM,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAC9C,IAAI,CAAC,QAAQ,CAAC,2BAA2B,CAAC,CAC3C,CAAC;QAEF,MAAM,CAAC,kBAAkB,CAAC,CAAC,SAAS,CAAC,YAAY,CAAC,CAAC;QACnD,MAAM,CAAC,kBAAkB,CAAC,CAAC,SAAS,CAAC,gBAAgB,CAAC,CAAC;QACvD,MAAM,CAAC,kBAAkB,CAAC,CAAC,SAAS,CAAC,kBAAkB,CAAC,CAAC;QACzD,MAAM,CAAC,kBAAkB,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;QAElD,MAAM,CAAC,gBAAgB,CAAC,CAAC,SAAS,CAAC,YAAY,CAAC,CAAC;QACjD,MAAM,CAAC,gBAAgB,CAAC,CAAC,SAAS,CAAC,gBAAgB,CAAC,CAAC;QACrD,MAAM,CAAC,gBAAgB,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;QAEhD,MAAM,CAAC,kBAAkB,CAAC,CAAC,SAAS,CAAC,YAAY,CAAC,CAAC;QACnD,MAAM,CAAC,kBAAkB,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;QAElD,SAAS,EAAE,CAAC;QACZ,MAAM,KAAK,CAAC,OAAO,EAAE,CAAC;IACxB,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC","sourcesContent":["import { afterEach, describe, expect, it } from \"vitest\";\nimport { EventStore } from \"../../lib/v3/flowlogger/EventStore.js\";\nimport { EventEmitterWithWildcardSupport } from \"../../lib/v3/flowlogger/EventEmitter.js\";\nimport { FlowEvent } from \"../../lib/v3/flowlogger/FlowLogger.js\";\n\nfunction attachEventStoreToBus(\n  store: EventStore,\n  bus: EventEmitterWithWildcardSupport,\n): () => void {\n  const onFlowEvent = (event: unknown) => {\n    if (event instanceof FlowEvent) {\n      void store.emit(event);\n    }\n  };\n\n  bus.on(\"*\", onFlowEvent);\n  return () => {\n    bus.off(\"*\", onFlowEvent);\n  };\n}\n\nfunction createVerboseStoreHarness(): {\n  writes: string[];\n  store: EventStore;\n  bus: EventEmitterWithWildcardSupport;\n  detachBus: () => void;\n} {\n  const writes: string[] = [];\n  process.stderr.write = ((\n    chunk: string,\n    cb?: (error?: Error | null) => void,\n  ) => {\n    writes.push(String(chunk));\n    cb?.(null);\n    return true;\n  }) as typeof process.stderr.write;\n\n  const store = new EventStore(\"session-test\");\n  const bus = new EventEmitterWithWildcardSupport();\n  const detachBus = attachEventStoreToBus(store, bus);\n\n  return { writes, store, bus, detachBus };\n}\n\ndescribe(\"flow logger event store\", () => {\n  const stderrWrite = process.stderr.write.bind(process.stderr);\n\n  afterEach(() => {\n    process.stderr.write = stderrWrite;\n  });\n\n  it(\"queries recent events from the default in-memory sink\", async () => {\n    const store = new EventStore(\"session-test\");\n\n    await store.emit(\n      new FlowEvent({\n        eventType: \"StagehandExtractEvent\",\n        sessionId: \"session-test\",\n        eventId: \"stagehand-1234\",\n        eventCreatedAt: \"2026-03-16T21:45:00.000Z\",\n        data: { params: [\"grab title\"] },\n      }),\n    );\n\n    const events = await store.query({});\n    expect(events).toHaveLength(1);\n    expect(events[0].eventType).toBe(\"StagehandExtractEvent\");\n\n    await store.destroy();\n  });\n\n  it(\"drops payloads from the default in-memory sink\", async () => {\n    const store = new EventStore(\"session-test\");\n\n    await store.emit(\n      new FlowEvent({\n        eventType: \"LlmRequestEvent\",\n        sessionId: \"session-test\",\n        eventId: \"llm-1234\",\n        eventCreatedAt: \"2026-03-16T21:45:00.000Z\",\n        data: {\n          prompt: [{ type: \"image_url\", image_url: { url: \"huge\" } }],\n          output: \"huge\",\n        },\n      }),\n    );\n\n    const [event] = await store.query({});\n    expect(event.eventType).toBe(\"LlmRequestEvent\");\n    expect(event.eventId).toBe(\"llm-1234\");\n    expect(event.data).toEqual({});\n\n    await store.destroy();\n  });\n\n  it(\"renders semantic hierarchy tags for non-cdp stderr events only\", async () => {\n    // Intercept stderr so the pretty sink can be asserted without polluting the\n    // real test runner output.\n    const { writes, store, bus, detachBus } = createVerboseStoreHarness();\n\n    const stepEvent = new FlowEvent({\n      eventType: \"StagehandExtractEvent\",\n      sessionId: \"session-test\",\n      eventId: \"stagehand-1234\",\n      eventCreatedAt: \"2026-03-16T21:45:00.000Z\",\n      data: { params: [\"grab title\"] },\n    });\n    const cdpEvent = new FlowEvent({\n      eventType: \"CdpCallEvent\",\n      sessionId: \"session-test\",\n      eventId: \"cdp-call-5678\",\n      eventCreatedAt: \"2026-03-16T21:45:00.100Z\",\n      eventParentIds: [stepEvent.eventId],\n      data: {\n        method: \"Runtime.evaluate\",\n        params: { expression: \"2 + 2\" },\n        targetId: \"1234567890ABCDEF1234567890ABCDEF\",\n      },\n    });\n\n    // The stderr sink intentionally suppresses CDP noise even though the event\n    // still exists for in-memory and file-backed sinks.\n    bus.emit(stepEvent.eventType, stepEvent);\n    bus.emit(cdpEvent.eventType, cdpEvent);\n    await new Promise((resolve) => setTimeout(resolve, 0));\n\n    expect(writes).toHaveLength(1);\n    expect(writes[0]).toContain(\"[🆂 #1234 EXTRACT]\");\n    expect(writes[0]).toContain(\"Stagehand.extract\");\n    expect(writes[0]).not.toContain(\"Runtime.evaluate\");\n\n    detachBus();\n    await store.destroy();\n  });\n\n  it(\"renders generic stagehand events without crashing the stderr sink\", async () => {\n    const { writes, store, bus, detachBus } = createVerboseStoreHarness();\n\n    // `StagehandEvent` has no action suffix, so this guards the formatter path\n    // that cannot assume a method name exists.\n    bus.emit(\n      \"StagehandEvent\",\n      new FlowEvent({\n        eventType: \"StagehandEvent\",\n        sessionId: \"session-test\",\n        eventId: \"stagehand-0001\",\n        eventCreatedAt: \"2026-03-16T21:45:00.000Z\",\n        data: { params: [\"noop\"] },\n      }),\n    );\n    await new Promise((resolve) => setTimeout(resolve, 0));\n\n    expect(writes).toHaveLength(1);\n    expect(writes[0]).toContain(\"[🆂 #0001\");\n    expect(writes[0]).toContain(\"Stagehand(\");\n\n    detachBus();\n    await store.destroy();\n  });\n\n  it(\"colorizes pretty stderr output with ansi escapes when enabled\", async () => {\n    const previousForceColor = process.env.FORCE_COLOR;\n    const previousNoColor = process.env.NO_COLOR;\n    delete process.env.NO_COLOR;\n    process.env.FORCE_COLOR = \"1\";\n\n    const { writes, store, bus, detachBus } = createVerboseStoreHarness();\n\n    try {\n      bus.emit(\n        \"StagehandActEvent\",\n        new FlowEvent({\n          eventType: \"StagehandActEvent\",\n          sessionId: \"session-test\",\n          eventId: \"stagehand-0002\",\n          eventCreatedAt: \"2026-03-16T21:45:00.000Z\",\n          data: { params: [\"click submit\"] },\n        }),\n      );\n      await new Promise((resolve) => setTimeout(resolve, 0));\n\n      expect(writes).toHaveLength(1);\n      expect(writes[0]).toContain(\"\\u001B[\");\n    } finally {\n      if (previousNoColor === undefined) {\n        delete process.env.NO_COLOR;\n      } else {\n        process.env.NO_COLOR = previousNoColor;\n      }\n\n      if (previousForceColor === undefined) {\n        delete process.env.FORCE_COLOR;\n      } else {\n        process.env.FORCE_COLOR = previousForceColor;\n      }\n\n      detachBus();\n      await store.destroy();\n    }\n  });\n\n  it(\"keeps agent ancestry and start ids for completion events after many child events\", async () => {\n    const { writes, store, bus, detachBus } = createVerboseStoreHarness();\n\n    const agentEvent = new FlowEvent({\n      eventType: \"AgentExecuteEvent\",\n      sessionId: \"session-test\",\n      eventId: \"agent-1234\",\n      eventCreatedAt: \"2026-03-16T21:45:00.000Z\",\n      data: { params: [{ instruction: \"click the button\" }] },\n    });\n    const actEvent = new FlowEvent({\n      eventType: \"StagehandActEvent\",\n      sessionId: \"session-test\",\n      eventId: \"stagehand-2222\",\n      eventCreatedAt: \"2026-03-16T21:45:00.001Z\",\n      eventParentIds: [agentEvent.eventId],\n      data: { params: [\"click the button\"] },\n    });\n    const clickEvent = new FlowEvent({\n      eventType: \"UnderstudyClickEvent\",\n      sessionId: \"session-test\",\n      eventId: \"action-3333\",\n      eventCreatedAt: \"2026-03-16T21:45:00.002Z\",\n      eventParentIds: [agentEvent.eventId, actEvent.eventId],\n      data: { target: \"xpath=/button[1]\" },\n    });\n\n    bus.emit(agentEvent.eventType, agentEvent);\n    bus.emit(actEvent.eventType, actEvent);\n    bus.emit(clickEvent.eventType, clickEvent);\n\n    // Flood the retained history with child events so the completion lines have\n    // to recover their displayed ancestry from the queryable sink.\n    for (let index = 0; index < 150; index += 1) {\n      bus.emit(\n        \"CdpCallEvent\",\n        new FlowEvent({\n          eventType: \"CdpCallEvent\",\n          sessionId: \"session-test\",\n          eventId: `cdp-${String(index).padStart(4, \"0\")}`,\n          eventCreatedAt: `2026-03-16T21:45:00.${String(index + 10).padStart(3, \"0\")}Z`,\n          eventParentIds: [\n            agentEvent.eventId,\n            actEvent.eventId,\n            clickEvent.eventId,\n          ],\n          data: {\n            method: \"Runtime.evaluate\",\n            params: { expression: `${index}` },\n            targetId: \"1234567890ABCDEF1234567890ABCDEF\",\n          },\n        }),\n      );\n    }\n\n    bus.emit(\n      \"UnderstudyClickCompletedEvent\",\n      new FlowEvent({\n        eventType: \"UnderstudyClickCompletedEvent\",\n        sessionId: \"session-test\",\n        eventId: \"done-4444\",\n        eventCreatedAt: \"2026-03-16T21:45:01.000Z\",\n        eventParentIds: [\n          agentEvent.eventId,\n          actEvent.eventId,\n          clickEvent.eventId,\n        ],\n        data: { durationMs: 250 },\n      }),\n    );\n    bus.emit(\n      \"StagehandActCompletedEvent\",\n      new FlowEvent({\n        eventType: \"StagehandActCompletedEvent\",\n        sessionId: \"session-test\",\n        eventId: \"done-5555\",\n        eventCreatedAt: \"2026-03-16T21:45:01.001Z\",\n        eventParentIds: [agentEvent.eventId, actEvent.eventId],\n        data: { durationMs: 500 },\n      }),\n    );\n    bus.emit(\n      \"AgentExecuteCompletedEvent\",\n      new FlowEvent({\n        eventType: \"AgentExecuteCompletedEvent\",\n        sessionId: \"session-test\",\n        eventId: \"done-6666\",\n        eventCreatedAt: \"2026-03-16T21:45:01.002Z\",\n        eventParentIds: [agentEvent.eventId],\n        data: { durationMs: 750 },\n      }),\n    );\n    await new Promise((resolve) => setTimeout(resolve, 0));\n\n    // Completion lines should reference the original started-event ids, not the\n    // synthetic completed-event ids emitted at the end of the lifecycle.\n    const clickCompletedLine = writes.find((line) =>\n      line.includes(\"CLICK completed\"),\n    );\n    const actCompletedLine = writes.find((line) =>\n      line.includes(\"ACT completed\"),\n    );\n    const agentCompletedLine = writes.find((line) =>\n      line.includes(\"Agent.execute() completed\"),\n    );\n\n    expect(clickCompletedLine).toContain(\"[🅰 #1234]\");\n    expect(clickCompletedLine).toContain(\"[🆂 #2222 ACT]\");\n    expect(clickCompletedLine).toContain(\"[🆄 #3333 CLICK]\");\n    expect(clickCompletedLine).not.toContain(\"#4444\");\n\n    expect(actCompletedLine).toContain(\"[🅰 #1234]\");\n    expect(actCompletedLine).toContain(\"[🆂 #2222 ACT]\");\n    expect(actCompletedLine).not.toContain(\"#5555\");\n\n    expect(agentCompletedLine).toContain(\"[🅰 #1234]\");\n    expect(agentCompletedLine).not.toContain(\"#6666\");\n\n    detachBus();\n    await store.destroy();\n  });\n});\n"]}

package/dist/esm/tests/unit/llm-middleware.test.d.ts

@@ -0,0 +1 @@
+export {};