feat: disable colors if agent is detected (#9851)

sheremet-va · AriPerkkio · web-flow · commit 6f97b55ddc44 · 2026-03-26T10:41:26.000+01:00
Co-authored-by: Ari Perkkiö &lt;ari.perkkio@gmail.com&gt;
diff --git a/packages/vitest/src/node/cli/cac.ts b/packages/vitest/src/node/cli/cac.ts
@@ -5,8 +5,9 @@ import type { CLIOption, CLIOptions as CLIOptionsConfig } from './cli-config'
 import { toArray } from '@vitest/utils/helpers'
 import cac from 'cac'
 import { normalize } from 'pathe'
-import c from 'tinyrainbow'
+import c, { disableDefaultColors } from 'tinyrainbow'
 import { version } from '../../../package.json' with { type: 'json' }
+import { isAgent } from '../../utils/env'
 import { benchCliOptionsConfig, cliOptionsConfig, collectCliOptionsConfig } from './cli-config'
 import { setupTabCompletions } from './completions'
 
@@ -74,6 +75,10 @@ function addCliOptions(cli: CAC | Command, options: CLIOptionsConfig<any>) {
 }
 
 export function createCLI(options: CliParseOptions = {}): CAC {
+  if (isAgent) {
+    disableDefaultColors()
+  }
+
   const cli = cac('vitest')
 
   cli.version(version)
diff --git a/packages/vitest/src/node/cli/cli-config.ts b/packages/vitest/src/node/cli/cli-config.ts
@@ -1,7 +1,6 @@
 import type { ApiConfig } from '../types/config'
 import type { CliOptions } from './cli-api'
 import { defaultBrowserPort, defaultPort } from '../../constants'
-import { ReportersMap } from '../reporters'
 
 type NestedOption<T, V = Extract<T, Record<string, any>>> = V extends
   | never
@@ -141,7 +140,7 @@ export const cliOptionsConfig: VitestCLIOptions = {
   },
   reporters: {
     alias: 'reporter',
-    description: `Specify reporters (${Object.keys(ReportersMap).join(', ')})`,
+    description: `Specify reporters (default, agent, blob, verbose, dot, json, tap, tap-flat, junit, tree, hanging-process, github-actions)`,
     argument: '<name>',
     subcommands: null, // don't support custom objects
     array: true,
diff --git a/packages/vitest/src/node/config/serializeConfig.ts b/packages/vitest/src/node/config/serializeConfig.ts
@@ -1,6 +1,7 @@
 import type { TestProject } from '../project'
 import type { ApiConfig, SerializedConfig } from '../types/config'
 import { configDefaults } from '../../defaults'
+import { isAgent } from '../../utils/env'
 
 export function serializeConfig(project: TestProject): SerializedConfig {
   const { config, globalConfig } = project
@@ -146,5 +147,6 @@ export function serializeConfig(project: TestProject): SerializedConfig {
       config.slowTestThreshold
       ?? globalConfig.slowTestThreshold
       ?? configDefaults.slowTestThreshold,
+    isAgent,
   }
 }
diff --git a/packages/vitest/src/runtime/config.ts b/packages/vitest/src/runtime/config.ts
@@ -146,6 +146,7 @@ export interface SerializedConfig {
   tagsFilter: string[] | undefined
   strictTags: boolean
   slowTestThreshold: number | undefined
+  isAgent: boolean
 }
 
 export interface SerializedCoverageConfig {
diff --git a/packages/vitest/src/runtime/workers/init.ts b/packages/vitest/src/runtime/workers/init.ts
@@ -3,6 +3,7 @@ import type { WorkerRequest, WorkerResponse } from '../../node/pools/types'
 import type { WorkerSetupContext } from '../../types/worker'
 import type { VitestWorker } from './types'
 import { serializeError } from '@vitest/utils/error'
+import { disableDefaultColors } from 'tinyrainbow'
 import { Traces } from '../../utils/traces'
 import * as listeners from '../listeners'
 import { createRuntimeRpc } from '../rpc'
@@ -49,6 +50,10 @@ export function init(worker: Options): void {
         process.env.VITEST_WORKER_ID = String(message.workerId)
         reportMemory = message.options.reportMemory
 
+        if (message.context.config.isAgent) {
+          disableDefaultColors()
+        }
+
         traces ??= await new Traces({
           enabled: message.traces.enabled,
           sdkPath: message.traces.sdkPath,
diff --git a/pnpm-lock.yaml b/pnpm-lock.yaml
diff --git a/pnpm-workspace.yaml b/pnpm-workspace.yaml
@@ -85,7 +85,7 @@ catalog:
   tinyexec: ^1.0.2
   tinyglobby: ^0.2.15
   tinyhighlight: ^0.3.2
-  tinyrainbow: ^3.0.3
+  tinyrainbow: ^3.1.0
   tinyspy: ^4.0.4
   typescript: ^5.9.3
   unocss: ^66.6.6
diff --git a/test/config/test/console-color.test.ts b/test/config/test/console-color.test.ts
@@ -1,5 +1,5 @@
 import { expect, test } from 'vitest'
-import { runVitest } from '../../test-utils'
+import { runVitest, runVitestCli } from '../../test-utils'
 
 test('with color', async () => {
   const { stdout } = await runVitest({
@@ -30,6 +30,20 @@ test('without color', async () => {
   expect(stdout).not.toContain('\x1B[33mtrue\x1B[39m\n')
 })
 
+test('agent', async () => {
+  // Agent check is done on module import, so new process is needed
+  const { stdout } = await runVitestCli({
+    preserveAnsi: true,
+    nodeOptions: { env: { AI_AGENT: 'copilot' } },
+  }, '--root', 'fixtures/console-color', '--reporter', 'default')
+
+  expect.soft(stdout).toContain('true\n')
+  expect.soft(stdout).not.toContain('\x1B[33mtrue\x1B[39m\n')
+
+  expect.soft(stdout).toContain(' RUN')
+  expect.soft(stdout).not.toContain('\x1B[46m RUN')
+})
+
 test.skipIf(process.platform === 'win32')('without color, forks pool in non-TTY parent', async () => {
   const { stdout } = await runVitest({
     root: 'fixtures/console-color',

Original file line number	Diff line number	Diff line change
`@@ -1,6 +1,7 @@`
`1`	`1`	`import type { TestProject } from '../project'`
`2`	`2`	`import type { ApiConfig, SerializedConfig } from '../types/config'`
`3`	`3`	`import { configDefaults } from '../../defaults'`
	`4`	`+import { isAgent } from '../../utils/env'`
`4`	`5`
`5`	`6`	`export function serializeConfig(project: TestProject): SerializedConfig {`
`6`	`7`	`const { config, globalConfig } = project`
`@@ -146,5 +147,6 @@ export function serializeConfig(project: TestProject): SerializedConfig {`
`146`	`147`	`config.slowTestThreshold`
`147`	`148`	`?? globalConfig.slowTestThreshold`
`148`	`149`	`?? configDefaults.slowTestThreshold,`
	`150`	`+ isAgent,`
`149`	`151`	`}`
`150`	`152`	`}`
Original file line number	Diff line number	Diff line change
`@@ -146,6 +146,7 @@ export interface SerializedConfig {`
`146`	`146`	`tagsFilter: string[] \| undefined`
`147`	`147`	`strictTags: boolean`
`148`	`148`	`slowTestThreshold: number \| undefined`
	`149`	`+ isAgent: boolean`
`149`	`150`	`}`
`150`	`151`
`151`	`152`	`export interface SerializedCoverageConfig {`