register tool returns tool'

landing-ai · Apr 23, 2024 · eb793f3 · eb793f3
1 parent 9effc1f
commit eb793f3
Showing 1 changed file with 5 additions and 1 deletion.
diff --git 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(