From eb793f3a6d147f6d83584663ca30821d3b2ddcf8 Mon Sep 17 00:00:00 2001
From: Dillon Laird <dillonalaird@gmail.com>
Date: Mon, 22 Apr 2024 19:12:11 -0700
Subject: [PATCH] register tool returns tool'

---
 vision_agent/tools/tools.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/vision_agent/tools/tools.py b/vision_agent/tools/tools.py
index 711fc894..67dc2292 100644
--- a/vision_agent/tools/tools.py
+++ b/vision_agent/tools/tools.py
@@ -52,6 +52,9 @@ class Tool(ABC):
     description: str
     usage: Dict
 
+    def __call__(self, *args: Any, **kwargs: Any) -> Any:
+        raise NotImplementedError
+
 
 class NoOp(Tool):
     name = "noop_"
@@ -765,7 +768,7 @@ def __call__(self, equation: str) -> float:
 }
 
 
-def register_tool(tool: Type[Tool]) -> None:
+def register_tool(tool: Type[Tool]) -> Type[Tool]:
     r"""Add a tool to the list of available tools.
 
     Parameters:
@@ -787,6 +790,7 @@ def register_tool(tool: Type[Tool]) -> None:
         "usage": tool.usage,
         "class": tool,
     }
+    return tool
 
 
 def _send_inference_request(