Merge pull request #213 from dbcli/litecli-llm-fix

Check for `\llm` in a loop to make sure alternate invokations are not…
dbcli · Feb 24, 2025 · 62e8de6 · 62e8de6
2 parents 4c1056f + 20f8b46
commit 62e8de6
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -5,6 +5,10 @@
 * Support `--help` in the `\llm`and `\llm+` command. ([#214](https://github.com/dbcli/litecli/pull/214))
 * Make the history file location configurable. ([#206](https://github.com/dbcli/litecli/issues/206))
 
+### Bug Fixes
+
+* Fix a bug where the `\llm` command on alternate invocations weren't detected correctly. (#211)
+
 ### Internal
 
 * Fix typo `pormpt`to `prompt` in `special/llm.py`.

diff --git a/litecli/main.py b/litecli/main.py
@@ -444,7 +444,7 @@ def one_iteration(text=None):
                     self.echo(str(e), err=True, fg="red")
                     return
 
-                if special.is_llm_command(text):
+                while special.is_llm_command(text):
                     try:
                         start = time()
                         cur = self.sqlexecute.conn and self.sqlexecute.conn.cursor()