use previewFeatures to determine which pro model to use for A2A (#15131)

This commit is contained in:
Sehoon Shon 2025-12-15 22:51:23 -05:00 committed by GitHub
parent 0c3eb82671
commit 2995af6a21
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -22,6 +22,7 @@ import {
DEFAULT_GEMINI_MODEL,
type ExtensionLoader,
startupProfiler,
PREVIEW_GEMINI_MODEL,
} from '@google/gemini-cli-core';
import { logger } from '../utils/logger.js';
@ -38,7 +39,9 @@ export async function loadConfig(
const configParams: ConfigParameters = {
sessionId: taskId,
model: DEFAULT_GEMINI_MODEL,
model: settings.general?.previewFeatures
? PREVIEW_GEMINI_MODEL
: DEFAULT_GEMINI_MODEL,
embeddingModel: DEFAULT_GEMINI_EMBEDDING_MODEL,
sandbox: undefined, // Sandbox might not be relevant for a server-side agent
targetDir: workspaceDir, // Or a specific directory the agent operates on