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