diff --git a/integration-tests/test-helper.ts b/integration-tests/test-helper.ts index e928cbe45f..63362fc257 100644 --- a/integration-tests/test-helper.ts +++ b/integration-tests/test-helper.ts @@ -9,6 +9,7 @@ import { mkdirSync, writeFileSync, readFileSync } from 'node:fs'; import { join, dirname } from 'node:path'; import { fileURLToPath } from 'node:url'; import { env } from 'node:process'; +import { DEFAULT_GEMINI_MODEL } from '../packages/core/src/config/models.js'; import fs from 'node:fs'; import * as pty from '@lydell/node-pty'; @@ -152,6 +153,7 @@ export class TestRig { otlpEndpoint: '', outfile: telemetryPath, }, + model: DEFAULT_GEMINI_MODEL, sandbox: env.GEMINI_SANDBOX !== 'false' ? env.GEMINI_SANDBOX : false, ...options.settings, // Allow tests to override/add settings };