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

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