mirror of
https://github.com/encounter/adk-python.git
synced 2026-07-09 18:19:28 -07:00
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:
committed by
Copybara-Service
parent
d29261a3dc
commit
5cad8a7f58
@@ -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
|
||||
|
||||
@@ -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)
|
||||
|
||||
Reference in New Issue
Block a user