From 5d297169d08a2d0ea1a07641da2ac39fa46b68a4 Mon Sep 17 00:00:00 2001 From: "Xiang (Sean) Zhou" Date: Fri, 30 May 2025 20:59:50 -0700 Subject: [PATCH] fix: remove labels from config when using API key from Google AI Studio to call model PiperOrigin-RevId: 765448305 --- src/google/adk/models/google_llm.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/src/google/adk/models/google_llm.py b/src/google/adk/models/google_llm.py index 7a3932a..c9d021f 100644 --- a/src/google/adk/models/google_llm.py +++ b/src/google/adk/models/google_llm.py @@ -83,7 +83,7 @@ class Gemini(BaseLlm): Yields: LlmResponse: The model response. """ - + self._preprocess_request(llm_request) self._maybe_append_user_content(llm_request) logger.info( 'Sending out request, model: %s, backend: %s, stream: %s', @@ -237,6 +237,12 @@ class Gemini(BaseLlm): ) as live_session: yield GeminiLlmConnection(live_session) + def _preprocess_request(self, llm_request: LlmRequest) -> None: + + if llm_request.config and self._api_backend == 'ml_dev': + # Using API key from Google AI Studio to call model doesn't support labels. + llm_request.config.labels = None + def _build_function_declaration_log( func_decl: types.FunctionDeclaration,