From 51d423fb00989afdcdfb2e7e8cc58085e3951f10 Mon Sep 17 00:00:00 2001 From: Danny Avila Date: Sun, 26 Jul 2026 17:27:39 -0400 Subject: [PATCH] fix: preserve agent context in mock stream responses --- e2e/setup/fake-model.js | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/e2e/setup/fake-model.js b/e2e/setup/fake-model.js index f63ac9c8ad..e284caabc8 100644 --- a/e2e/setup/fake-model.js +++ b/e2e/setup/fake-model.js @@ -1173,7 +1173,9 @@ module.exports = function fakeModelHook(run, context) { sleep, toolCalls, thrownError, - resolveOnStream: (streamMessages) => - approvalOutcomeResponses(streamMessages) ?? resolveOnStream?.(streamMessages) ?? null, + resolveOnStream: (streamMessages, streamOptions, runManager) => + approvalOutcomeResponses(streamMessages) ?? + resolveOnStream?.(streamMessages, streamOptions, runManager) ?? + null, }); };