fix: catch exceptions when executing tools

ErikBjare · Oct 10, 2024 · 9560660 · 9560660
1 parent 771734c
commit 9560660
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/gptme/tools/base.py b/gptme/tools/base.py
@@ -95,7 +95,10 @@ def execute(self, ask: bool) -> Generator[Message, None, None]:
 
         tool = get_tool(self.tool)
         if tool and tool.execute:
-            yield from tool.execute(self.content, ask, self.args)
+            try:
+                yield from tool.execute(self.content, ask, self.args)
+            except Exception as e:
+                yield Message("system", f"Error executing tool '{self.tool}': {e}")
         else:
             logger.warning(f"Tool '{self.tool}' is not available for execution.")