tooluniverse.memory_manager module

Memory Manager for Multi-Agent Systems Manages session-based memory for multi-agent workflows

class tooluniverse.memory_manager.MemoryManager

Bases: object

Manages session-based memory for multi-agent systems.

__init__(max_sessions=1000, session_timeout=3600)

create_session(user_id=None, session_name=None)

Create a new session and return its ID.

get_session(session_id)

Retrieve session data by ID, updating last-accessed timestamp.

update_session_context(session_id, updates)

Update the context dictionary for an existing session.

add_agent_result(session_id, agent_name, result, phase=None)

Record an agent execution result in the session history.

get_context_for_agent(session_id, agent_name)

Build a JSON context string suitable for passing to an agent.

set_current_phase(session_id, phase)

Set the current execution phase for a session.

get_session_summary(session_id)

Return a summary dictionary for the given session.

list_user_sessions(user_id)

List all sessions for a given user, sorted by last access time.

close_session(session_id)

Mark a session as closed.