From 95f994d7cad221087f8ee55144854ad0c060e332 Mon Sep 17 00:00:00 2001 From: Ankur Sharma Date: Thu, 15 May 2025 10:34:44 -0700 Subject: [PATCH] chore:add in memory session service to in memory runner. PiperOrigin-RevId: 759206471 --- src/google/adk/runners.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/src/google/adk/runners.py b/src/google/adk/runners.py index 3c384a8..d8cdd63 100644 --- a/src/google/adk/runners.py +++ b/src/google/adk/runners.py @@ -481,6 +481,8 @@ class InMemoryRunner(Runner): agent: The root agent to run. app_name: The application name of the runner. Defaults to 'InMemoryRunner'. + _in_memory_session_service: Deprecated. Please don't use. The in-memory + session service for the runner. """ def __init__(self, agent: LlmAgent, *, app_name: str = 'InMemoryRunner'): @@ -491,10 +493,11 @@ class InMemoryRunner(Runner): app_name: The application name of the runner. Defaults to 'InMemoryRunner'. """ + self._in_memory_session_service = InMemorySessionService() super().__init__( app_name=app_name, agent=agent, artifact_service=InMemoryArtifactService(), - session_service=InMemorySessionService(), + session_service=self._in_memory_session_service, memory_service=InMemoryMemoryService(), )