Skip to content

Commit 0bc767e

Browse files
google-genai-botcopybara-github
authored andcommitted
feat: add BufferableSessionService
PiperOrigin-RevId: 912057968
1 parent 83ae405 commit 0bc767e

2 files changed

Lines changed: 11 additions & 0 deletions

File tree

src/google/adk/runners.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1616,6 +1616,10 @@ async def close(self):
16161616
if self.plugin_manager:
16171617
await self.plugin_manager.close()
16181618

1619+
# Close Session Service
1620+
if self.session_service:
1621+
await self.session_service.flush()
1622+
16191623
logger.info('Runner closed.')
16201624

16211625
if sys.version_info < (3, 11):

src/google/adk/sessions/base_session_service.py

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -124,6 +124,13 @@ async def append_event(self, session: Session, event: Event) -> Event:
124124
session.events.append(event)
125125
return event
126126

127+
async def flush(self):
128+
"""Flushes any buffered events.
129+
130+
For non-buffering implementations, this can be a no-op.
131+
"""
132+
pass
133+
127134
def _apply_temp_state(self, session: Session, event: Event) -> None:
128135
"""Applies temp-scoped state delta to the in-memory session state.
129136

0 commit comments

Comments
 (0)