Memory Backends

Use MemoryBackend when you want to plug custom memory into the high-level Agent wrapper.

What it provides

MemoryBackend is the Python-side contract for agent memory. You implement the storage and retrieval behavior, then register it with Agent through as_memory_module().

Each backend method may be implemented either synchronously or as an async def coroutine method. The wrapper supports both forms.

The class defines:

• record(session_id, user_msg, assistant_msg)
• recall(session_id, query, max_entries)
• flush(session_id)
• consolidate(session_id): optional
• as_memory_module()

Return expectations:

• record() may return None or an awaitable that resolves to None
• recall() may return list[MemoryEntry] or an awaitable that resolves to that list
• flush() may return None or an awaitable that resolves to None
• consolidate() may return None or an awaitable that resolves to None

Minimal shape

from enki_py import MemoryBackend, MemoryEntry


class MyMemory(MemoryBackend):
    name = "my_memory"

    async def record(self, session_id: str, user_msg: str, assistant_msg: str) -> None:
        ...

    async def recall(
        self,
        session_id: str,
        query: str,
        max_entries: int,
    ) -> list[MemoryEntry]:
        ...

    async def flush(self, session_id: str) -> None:
        ...

Registering with an agent

Create the backend instance, convert it to a MemoryModule, and pass it to Agent.

from enki_py import Agent

memory = MyMemory()

agent = Agent(
    "ollama::qwen3.5:latest",
    instructions="Answer clearly and keep responses short.",
    memories=[memory.as_memory_module()],
)

When to use consolidate()

Override consolidate() if your backend needs a compaction or summarization hook. If you do not need it, inherit the default no-op implementation.

async def consolidate(self, session_id: str) -> None:
    ...

If you prefer, the same methods can also be plain synchronous def functions.

Related types

• MemoryEntry: the unit returned by recall()
• MemoryKind: categorizes entries such as recent messages and summaries
• MemoryModule: the registration object built by as_memory_module()

See Memory Examples for a full implementation.