From ff57a5b2670acda14d83a97c85276d68ab05e7df Mon Sep 17 00:00:00 2001
From: Dominik Jain <dominik.jain@oraios-ai.de>
Date: Sat, 5 Jul 2025 19:06:29 +0200
Subject: [PATCH] Limit tool registry to serena.tools

---
 src/serena/tools/tools_base.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/src/serena/tools/tools_base.py b/src/serena/tools/tools_base.py
index 1d052972..b0a3cd73 100644
--- a/src/serena/tools/tools_base.py
+++ b/src/serena/tools/tools_base.py
@@ -320,7 +320,9 @@ class ToolRegistry:
         """
         Iterate over Tool subclasses.
         """
-        yield from iter_subclasses(Tool)
+        for cls in iter_subclasses(Tool):
+            if cls.__module__.startswith("serena.tools"):
+                yield cls
 
     @classmethod
     def _get_tool_dict(cls) -> dict[str, type[Tool]]: