diff --git a/packages/cli/src/gemini.tsx b/packages/cli/src/gemini.tsx index 565625d8..64bb5c84 100644 --- a/packages/cli/src/gemini.tsx +++ b/packages/cli/src/gemini.tsx @@ -24,6 +24,7 @@ import { import { themeManager } from './ui/themes/theme-manager.js'; import { getStartupWarnings } from './utils/startupWarnings.js'; import { getUserStartupWarnings } from './utils/userStartupWarnings.js'; +import { ConsolePatcher } from './ui/utils/ConsolePatcher.js'; import { runNonInteractive } from './nonInteractiveCli.js'; import { loadExtensions } from './config/extension.js'; import { cleanupCheckpoints, registerCleanup } from './utils/cleanup.js'; @@ -158,6 +159,13 @@ export async function main() { argv, ); + const consolePatcher = new ConsolePatcher({ + stderr: true, + debugMode: config.getDebugMode(), + }); + consolePatcher.patch(); + registerCleanup(consolePatcher.cleanup); + dns.setDefaultResultOrder( validateDnsResolutionOrder(settings.merged.dnsResolutionOrder), );