From c3e913466f83089b418401f29a3b7607ac34aed8 Mon Sep 17 00:00:00 2001 From: eycjur Date: Sat, 20 Sep 2025 15:25:31 +0900 Subject: [PATCH] Fix API key passing for Gemini token counting endpoints --- litellm/llms/gemini/google_genai/transformation.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/litellm/llms/gemini/google_genai/transformation.py b/litellm/llms/gemini/google_genai/transformation.py index 28142f72739..f38c772e355 100644 --- a/litellm/llms/gemini/google_genai/transformation.py +++ b/litellm/llms/gemini/google_genai/transformation.py @@ -121,7 +121,8 @@ class GoogleGenAIConfig(BaseGoogleGenAIGenerateContentConfig, VertexLLM): default_headers = { "Content-Type": "application/json", } - gemini_api_key = self._get_google_ai_studio_api_key(dict(litellm_params or {})) + # Use the passed api_key first, then fall back to litellm_params and environment + gemini_api_key = api_key or self._get_google_ai_studio_api_key(dict(litellm_params or {})) if gemini_api_key is not None: default_headers[self.XGOOGLE_API_KEY] = gemini_api_key if headers is not None: