fix: Create a new session resumption config if it's None

Co-authored-by: Xiang (Sean) Zhou <seanzhougoogle@google.com>
PiperOrigin-RevId: 853933839
This commit is contained in:
Xiang (Sean) Zhou
2026-01-08 16:49:52 -08:00
committed by Copybara-Service
parent 23d330eef1
commit d6c964e05e
@@ -116,6 +116,10 @@ class BaseLlmFlow(ABC):
attempt += 1
if not llm_request.live_connect_config:
llm_request.live_connect_config = types.LiveConnectConfig()
if not llm_request.live_connect_config.session_resumption:
llm_request.live_connect_config.session_resumption = (
types.SessionResumptionConfig()
)
llm_request.live_connect_config.session_resumption.handle = (
invocation_context.live_session_resumption_handle
)