NeoAnd/adk-python
1
0
Fork 0
mirror of https://github.com/EvolutionAPI/adk-python.git synced 2025-07-16 04:02:55 -06:00
Code Issues Actions Packages Projects Releases Wiki Activity

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

Browse Source 
PiperOrigin-RevId: 765448305
This commit is contained in:
Xiang (Sean) Zhou 2025-05-30 20:59:50 -07:00 committed by Copybara-Service
parent 3ec496dd81
commit 5d297169d0
1 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
src/google/adk/models/google_llm.py
View File

@ -83,7 +83,7 @@ class Gemini(BaseLlm):
Yields: Yields:
LlmResponse: The model response. LlmResponse: The model response.
""" """
self._preprocess_request(llm_request)
self._maybe_append_user_content(llm_request) self._maybe_append_user_content(llm_request)
logger.info( logger.info(
'Sending out request, model: %s, backend: %s, stream: %s', 'Sending out request, model: %s, backend: %s, stream: %s',
@ -237,6 +237,12 @@ class Gemini(BaseLlm):
) as live_session: ) as live_session:
yield GeminiLlmConnection(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( def _build_function_declaration_log(
func_decl: types.FunctionDeclaration, func_decl: types.FunctionDeclaration,