From b929701e97e0bc638f61e244eb4c2abf92c36df4 Mon Sep 17 00:00:00 2001
From: Peter Steinberger <steipete@gmail.com>
Date: Wed, 29 Apr 2026 14:25:39 +0100
Subject: [PATCH] test(agents): capture codex payload mutations

---
 src/agents/pi-embedded-runner-extraparams.test.ts | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/src/agents/pi-embedded-runner-extraparams.test.ts b/src/agents/pi-embedded-runner-extraparams.test.ts
index c8dfc220b0f..768be43172e 100644
--- a/src/agents/pi-embedded-runner-extraparams.test.ts
+++ b/src/agents/pi-embedded-runner-extraparams.test.ts
@@ -416,7 +416,9 @@ function createTestOpenAIProviderWrapper(
   if (withDefaultTransport) {
     streamFn = createOpenAIDefaultTransportWrapper(streamFn);
   }
-  streamFn = createOpenAIAttributionHeadersWrapper(streamFn);
+  streamFn = createOpenAIAttributionHeadersWrapper(streamFn, {
+    codexNativeTransportStreamFn: params.context.streamFn,
+  });
 
   if (resolveOpenAIFastMode(params.context.extraParams)) {
     streamFn = createOpenAIFastModeWrapper(streamFn);