From 11099e17a053c2006caf2dc95020969e0f0c0e79 Mon Sep 17 00:00:00 2001 From: Your Name Date: Tue, 26 May 2026 17:16:42 +0000 Subject: [PATCH] address feedback --- packages/core/src/context/testing/contextTestUtils.ts | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/packages/core/src/context/testing/contextTestUtils.ts b/packages/core/src/context/testing/contextTestUtils.ts index 389e33df52..921edfe80a 100644 --- a/packages/core/src/context/testing/contextTestUtils.ts +++ b/packages/core/src/context/testing/contextTestUtils.ts @@ -181,8 +181,8 @@ export function createMockEnvironment( let env = new ContextEnvironmentImpl( () => llmClient as BaseLlmClient, - () => 'mock-session', - 'mock-model', + () => 'mock-model', + 'mock-session', 'mock-prompt-id', '/tmp/.gemini/trace', '/tmp/.gemini/tool-outputs', @@ -197,7 +197,7 @@ export function createMockEnvironment( if (overrides.llmClient) { env = new ContextEnvironmentImpl( () => overrides.llmClient!, - () => overrides.model!, + () => overrides.model ?? env.model, env.sessionId, env.promptId, env.traceDir,