Fixed Tool.run after tests

antoninoLorenzo · Aug 11, 2024 · ed9e083 · ed9e083
1 parent bf6c119
commit ed9e083
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/agent/tools/base.py b/src/agent/tools/base.py
@@ -41,7 +41,9 @@ def load_tool(path: str):
     def run(*args):
         """Execute a tool"""
         if not isinstance(args[0], str):
-            raise ValueError(f'Argument must be a string found {type(args[0])}')
+            raise TypeError(f'Argument must be a string found {type(args[0])}.')
+        elif len(args[0]) == 0:
+            raise ValueError(f'String is empty.')
 
         command = args[0].encode('utf-8').decode('utf-8')
         arguments = command.split()