fix: Throw warning when using transparent session resumption in ADK Live for Gemini API key

transparent session resumption is only supported in Vertex AI APIs

Co-authored-by: Hangfei Lin <hangfei@google.com>
PiperOrigin-RevId: 836715170
This commit is contained in:
Hangfei Lin
2025-11-25 10:04:57 -08:00
committed by Copybara-Service
parent d29261a3dc
commit 5cad8a7f58
2 changed files with 18 additions and 1 deletions
@@ -244,7 +244,7 @@ class GeminiLlmConnection(BaseLlmConnection):
]
yield LlmResponse(content=types.Content(role='model', parts=parts))
if message.session_resumption_update:
logger.info('Received session resumption message: %s', message)
logger.debug('Received session resumption message: %s', message)
yield (
LlmResponse(
live_session_resumption_update=message.session_resumption_update
+17
View File
@@ -325,6 +325,23 @@ class Gemini(BaseLlm):
types.Part.from_text(text=llm_request.config.system_instruction)
],
)
if (
llm_request.live_connect_config.session_resumption
and llm_request.live_connect_config.session_resumption.transparent
):
logger.debug(
'session resumption config: %s',
llm_request.live_connect_config.session_resumption,
)
logger.debug(
'self._api_backend: %s',
self._api_backend,
)
if self._api_backend == GoogleLLMVariant.GEMINI_API:
raise ValueError(
'Transparent session resumption is only supported for Vertex AI'
' backend. Please use Vertex AI backend.'
)
llm_request.live_connect_config.tools = llm_request.config.tools
logger.info('Connecting to live for model: %s', llm_request.model)
logger.debug('Connecting to live with llm_request:%s', llm_request)