fix: console patcher test errors

packages/cli/src/nonInteractive/session.ts

@@ -40,6 +40,7 @@ import {
 } from './types.js';
 import type { LoadedSettings } from '../config/settings.js';
 import { runNonInteractive } from '../nonInteractiveCli.js';
+import { ConsolePatcher } from '../ui/utils/ConsolePatcher.js';
 
 const SESSION_STATE = {
   INITIALIZING: 'initializing',
@@ -695,6 +696,12 @@ export async function runNonInteractiveStreamJson(
   input: string,
   _promptId: string,
 ): Promise<void> {
+  const consolePatcher = new ConsolePatcher({
+    debugMode: config.getDebugMode(),
+  });
+  consolePatcher.patch();
+
+  try {
     // Create initial user message from prompt input if provided
     let initialPrompt: CLIUserMessage | undefined = undefined;
     if (input && input.trim().length > 0) {
@@ -712,4 +719,7 @@ export async function runNonInteractiveStreamJson(
 
     const manager = new SessionManager(config, settings, initialPrompt);
     await manager.run();
+  } finally {
+    consolePatcher.cleanup();
+  }
 }