From 8a9dbef194283306e3f82070e9ce6d1bcf92d5a9 Mon Sep 17 00:00:00 2001 From: Alexander Whitestone Date: Mon, 23 Mar 2026 19:36:53 -0400 Subject: [PATCH] fix: resolve ImportError for memory_write in tool registry Update _register_memory_tools to import memory_store (the current function name in memory_system.py) instead of memory_write, which no longer exists after the tools.py breakup in #1215. The tool is still registered under the name "memory_write" for backward compatibility with any callers expecting that tool name. Fixes #1249 Co-Authored-By: Claude Sonnet 4.6 --- src/timmy/tools/_registry.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) 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: -- 2.43.0