diff --git a/src/timmy/tools/_registry.py b/src/timmy/tools/_registry.py index ed5b12a6..567b4fa6 100644 --- a/src/timmy/tools/_registry.py +++ b/src/timmy/tools/_registry.py @@ -90,10 +90,10 @@ def _register_grok_tool(toolkit: Toolkit) -> None: def _register_memory_tools(toolkit: Toolkit) -> None: """Register memory search, write, and forget tools.""" try: - from timmy.memory_system import memory_forget, memory_read, memory_search, memory_write + from timmy.memory_system import memory_forget, memory_read, memory_search, memory_store toolkit.register(memory_search, name="memory_search") - toolkit.register(memory_write, name="memory_write") + toolkit.register(memory_store, name="memory_write") toolkit.register(memory_read, name="memory_read") toolkit.register(memory_forget, name="memory_forget") except (ImportError, AttributeError) as exc: