Handle exceptions from running llm command and surface them appropr…

…iately.
dbcli · Jan 26, 2025 · 8672ad8 · 8672ad8
1 parent b8b65c8
commit 8672ad8
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/litecli/packages/special/llm.py b/litecli/packages/special/llm.py
@@ -48,7 +48,10 @@ def run_external_cmd(cmd, *args, capture_output=False, restart_cli=False, raise_
             except SystemExit as e:
                 code = e.code
                 if code != 0 and raise_exception:
-                    raise
+                    if capture_output:
+                        raise RuntimeError(buffer.getvalue())
+                    else:
+                        raise RuntimeError(f"Command {cmd} failed with exit code {code}.")
 
         if restart_cli and code == 0:
             os.execv(original_exe, [original_exe] + original_args)