diff --git a/packages/ai/src/gemini/index.ts b/packages/ai/src/gemini/index.ts index 4b170012b3..1eec421b64 100644 --- a/packages/ai/src/gemini/index.ts +++ b/packages/ai/src/gemini/index.ts @@ -4,6 +4,7 @@ import { GenerateContentParameters, Part, GenerateContentResponseUsageMetadata, + GoogleGenAIOptions, } from '@google/genai' import { PostHog } from 'posthog-node' import { @@ -18,7 +19,7 @@ import { sanitizeGemini } from '../sanitization' import type { TokenUsage, FormattedContent, FormattedContentItem, FormattedMessage } from '../types' import { isString } from '../typeGuards' -interface MonitoringGeminiConfig { +interface MonitoringGeminiConfig extends GoogleGenAIOptions { apiKey?: string vertexai?: boolean project?: string