fix: remove labels from config when using API key from Google AI Studio to call model

seanzhougoogle · copybara-github · commit 5d297169d08a · 2025-05-30T21:00:23.000-07:00
PiperOrigin-RevId: 765448305
diff --git a/src/google/adk/models/google_llm.py b/src/google/adk/models/google_llm.py
@@ -83,7 +83,7 @@ async def generate_content_async(
     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 @@ async def connect(self, llm_request: LlmRequest) -> BaseLlmConnection:
     ) 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,
