diff --git a/packages/core/src/core/contentGenerator.ts b/packages/core/src/core/contentGenerator.ts index 740bede47c9..9d1e24c4fca 100644 --- a/packages/core/src/core/contentGenerator.ts +++ b/packages/core/src/core/contentGenerator.ts @@ -130,6 +130,7 @@ export async function createContentGenerator( const customHeadersMap = parseCustomHeaders(customHeadersEnv); const apiKeyAuthMechanism = process.env['GEMINI_API_KEY_AUTH_MECHANISM'] || 'x-goog-api-key'; + const apiVersionEnv = process.env['GOOGLE_GENAI_API_VERSION']?.trim(); const baseHeaders: Record = { ...customHeadersMap, @@ -179,6 +180,7 @@ export async function createContentGenerator( apiKey: config.apiKey === '' ? undefined : config.apiKey, vertexai: config.vertexai, httpOptions, + ...(apiVersionEnv && { apiVersion: apiVersionEnv }), }); return new LoggingContentGenerator(googleGenAI.models, gcConfig); }