Merge pull request #201 from dbcli/redirect-stderr

Redirect both stdout and stderr while capturing output.
dbcli · Jan 26, 2025 · d0ad299 · d0ad299
2 parents d06888a + b3b6139
commit d0ad299
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 3 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -1,3 +1,9 @@
+## 1.14.1 - 2025-01-25
+
+### Bug Fixes
+
+* Capture stderr in addition to stdout when capturing output from `llm` cli.
+
 ## 1.14.0 - 2025-01-22
 
 ### Features

diff --git a/litecli/packages/special/llm.py b/litecli/packages/special/llm.py
@@ -36,9 +36,10 @@ def run_external_cmd(cmd, *args, capture_output=False, restart_cli=False, raise_
 
         if capture_output:
             buffer = io.StringIO()
-            redirect = contextlib.redirect_stdout(buffer)
+            redirect = contextlib.ExitStack()
+            redirect.enter_context(contextlib.redirect_stdout(buffer))
+            redirect.enter_context(contextlib.redirect_stderr(buffer))
         else:
-            # Use nullcontext to do nothing when not capturing output
             redirect = contextlib.nullcontext()
 
         with redirect:
@@ -172,7 +173,6 @@ def initialize_llm():
     if click.confirm("This feature requires additional libraries. Install LLM library?", default=True):
         click.echo("Installing LLM library. Please wait...")
         run_external_cmd("pip", "install", "--quiet", "llm", restart_cli=True)
-        ensure_litecli_template()
 
 
 def ensure_litecli_template(replace=False):