Skip to content

Commit cf8778a

Browse files
sehoon38gemini-cli-robot
authored andcommitted
use previewFeatures to determine which pro model to use for A2A (#15131)
1 parent 5977ce8 commit cf8778a

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

packages/a2a-server/src/config/config.ts

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -22,6 +22,7 @@ import {
2222
DEFAULT_GEMINI_MODEL,
2323
type ExtensionLoader,
2424
startupProfiler,
25+
PREVIEW_GEMINI_MODEL,
2526
} from '@google/gemini-cli-core';
2627

2728
import { logger } from '../utils/logger.js';
@@ -38,7 +39,9 @@ export async function loadConfig(
3839

3940
const configParams: ConfigParameters = {
4041
sessionId: taskId,
41-
model: DEFAULT_GEMINI_MODEL,
42+
model: settings.general?.previewFeatures
43+
? PREVIEW_GEMINI_MODEL
44+
: DEFAULT_GEMINI_MODEL,
4245
embeddingModel: DEFAULT_GEMINI_EMBEDDING_MODEL,
4346
sandbox: undefined, // Sandbox might not be relevant for a server-side agent
4447
targetDir: workspaceDir, // Or a specific directory the agent operates on

0 commit comments

Comments
 (0)