From 9af2f481b22fc40a6fa11a7f47b2467d8a2e6f7b Mon Sep 17 00:00:00 2001
From: phernandez <phernandez@basicmachines.co>
Date: Sat, 18 Jan 2025 21:21:47 -0600
Subject: [PATCH] improve tool imports

---
 src/basic_memory/mcp/main.py | 4 +---
 1 file changed, 1 insertion(+), 3 deletions(-)

diff --git a/src/basic_memory/mcp/main.py b/src/basic_memory/mcp/main.py
index b4623206..1afe1621 100644
--- a/src/basic_memory/mcp/main.py
+++ b/src/basic_memory/mcp/main.py
@@ -11,9 +11,7 @@ from basic_memory.config import config
 from basic_memory.mcp.server import mcp
 
 # Import tools to register them
-from basic_memory.mcp.tools import knowledge, search, discussion, activity, ai_edit
-
-__all__ = ["knowledge", "search", "discussion", "activity", "ai_edit"]
+import basic_memory.mcp.tools # noqa: F401
 
 
 if __name__ == "__main__":