Automatically saving memory to memory bank at the end of a session

[zyn-google]

Is your feature request related to a problem? Please describe.
When using memory bank vertex ai service. The memory is lost when a session ends. We have to explicitly call add_session_to_memory to save it.

Describe the solution you'd like
Can we have the ADK to automatically save the memory to memory bank when a session ends?

[DeanChensj]

I think that's something we should consider, how do we define end of a session? If the session is end, can the user continue that session?

[ac-machache]

@DeanChensj

I think another potential solution would be to add support for creating memory that is directly controlled by the agent — not just through the generate_memories method, which ingests entire sessions.

According to the documentation, the memory bank does support this kind of fine-grained control. However, after reviewing the ADK memory module, I couldn’t find any implementation that allows agents to create or manage individual memories using the create_memory endpoint.

It would be helpful to expose or document how agents can use create_memory directly, rather than relying solely on generate_memories.

Would it make sense to extend or clarify the current memory API to support this?

[guillaumeblaquiere]

The end of session is hard (impossible) to define by the framework, it's known from the external environment.

The solution is to invoke the add_session_to_memory on the memory service. However, when we are using the api_server runner, it's impossible to access to this feature.

I recommend to add an endpoint to trigger externally to add the session to the memory, something like this:

 @app.post("/apps/{app_name}/users/{user_id}/sessions/{session_id}/add_to_memory")
    async def add_session_to_memory( app_name: str, user_id: str, session_id: str):
        session = await self.session_service.get_session(
            app_name=app_name,
            user_id=user_id,
            session_id=session_id
        )
        await self.memory_service.add_session_to_memory(session)

I will propose a pull request with this if you are OK