ADK changes

PiperOrigin-RevId: 754131080
This commit is contained in:
Shangjie Chen
2025-05-02 14:18:38 -07:00
committed by Copybara-Service
parent 879064343c
commit bcf1deb582
7 changed files with 15 additions and 13 deletions

View File

@@ -297,14 +297,14 @@ class Runner:
self.session_service.append_event(session=session, event=event)
yield event
def close_session(self, session: Session):
async def close_session(self, session: Session):
"""Closes a session and adds it to the memory service (experimental feature).
Args:
session: The session to close.
"""
if self.memory_service:
self.memory_service.add_session_to_memory(session)
await self.memory_service.add_session_to_memory(session)
self.session_service.close_session(session=session)
def _find_agent_to_run(