fix(diagnostics): track model stream progress

clawsweeper[bot] · clawsweeper[bot] · commit 44d84b498f54 · 2026-05-26T03:02:59.000Z
diff --git a/src/agents/pi-embedded-runner/run/attempt.model-diagnostic-events.test.ts b/src/agents/pi-embedded-runner/run/attempt.model-diagnostic-events.test.ts
@@ -265,6 +265,12 @@ describe("wrapStreamFnWithDiagnosticModelCallEvents", () => {
 
   it("does not retain stream progress activity when diagnostics are disabled", async () => {
     setDiagnosticsEnabledForProcess(false);
+    const runProgressEvents: DiagnosticEventPayload[] = [];
+    const stop = onInternalDiagnosticEvent((event) => {
+      if (event.type === "run.progress") {
+        runProgressEvents.push(event);
+      }
+    });
     async function* stream() {
       yield { type: "text_delta", delta: "first" };
       yield { type: "text_delta", delta: "second" };
@@ -282,15 +288,20 @@ describe("wrapStreamFnWithDiagnosticModelCallEvents", () => {
       },
     );
 
-    await drain(wrapped({} as never, {} as never, {} as never) as AsyncIterable<unknown>);
-    await waitForDiagnosticEventsDrained();
+    try {
+      await drain(wrapped({} as never, {} as never, {} as never) as AsyncIterable<unknown>);
+      await waitForDiagnosticEventsDrained();
+    } finally {
+      stop();
+    }
 
     expect(
       getDiagnosticSessionActivitySnapshot({
         sessionKey: "session-key",
         sessionId: "session-id",
       }),
     ).toEqual({});
+    expect(runProgressEvents).toEqual([]);
   });
 
   it("counts async onPayload replacements instead of raw payload content", async () => {
