From defcd5a7544f59459ff7c5ee6d3a04b80f89f2cc Mon Sep 17 00:00:00 2001
From: Abhi <abhipatel@google.com>
Date: Mon, 23 Mar 2026 13:22:24 -0400
Subject: [PATCH] test-utils: update default integration test model to
 PREVIEW_GEMINI_FLASH_MODEL

---
 packages/test-utils/src/test-rig.ts | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/packages/test-utils/src/test-rig.ts b/packages/test-utils/src/test-rig.ts
index ee091bee92..e9339406f9 100644
--- a/packages/test-utils/src/test-rig.ts
+++ b/packages/test-utils/src/test-rig.ts
@@ -11,7 +11,10 @@ import { join, dirname } from 'node:path';
 import { fileURLToPath } from 'node:url';
 import { env } from 'node:process';
 import { setTimeout as sleep } from 'node:timers/promises';
-import { DEFAULT_GEMINI_MODEL, GEMINI_DIR } from '@google/gemini-cli-core';
+import {
+  GEMINI_DIR,
+  PREVIEW_GEMINI_FLASH_MODEL,
+} from '@google/gemini-cli-core';
 export { GEMINI_DIR };
 import * as pty from '@lydell/node-pty';
 import stripAnsi from 'strip-ansi';
@@ -456,7 +459,7 @@ export class TestRig {
         ...(env['GEMINI_TEST_TYPE'] === 'integration'
           ? {
               model: {
-                name: DEFAULT_GEMINI_MODEL,
+                name: PREVIEW_GEMINI_FLASH_MODEL,
               },
             }
           : {}),