fix: re-raise tool use errors in tests

ErikBjare · Oct 12, 2024 · 0ada191 · 0ada191
1 parent 732c5b8
commit 0ada191
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/gptme/tools/base.py b/gptme/tools/base.py
@@ -98,6 +98,9 @@ def execute(self, ask: bool) -> Generator[Message, None, None]:
             try:
                 yield from tool.execute(self.content, ask, self.args)
             except Exception as e:
+                # if we are testing, raise the exception
+                if "pytest" in globals():
+                    raise e
                 yield Message("system", f"Error executing tool '{self.tool}': {e}")
         else:
             logger.warning(f"Tool '{self.tool}' is not available for execution.")