fix: catch exceptions in shell tool

ErikBjare · Sep 28, 2024 · f42be8f · f42be8f
1 parent 46340ca
commit f42be8f
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/gptme/tools/shell.py b/gptme/tools/shell.py
@@ -262,7 +262,11 @@ def execute_shell(
         print()
 
     if not ask or confirm:
-        returncode, stdout, stderr = shell.run(cmd)
+        try:
+            returncode, stdout, stderr = shell.run(cmd)
+        except Exception as e:
+            yield Message("system", f"Error: {e}")
+            return
         stdout = _shorten_stdout(stdout.strip(), pre_tokens=2000, post_tokens=8000)
         stderr = _shorten_stdout(stderr.strip(), pre_tokens=2000, post_tokens=2000)