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

Check for `\llm` in a loop to make sure alternate invokations are not…

Author: amjith

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`.

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()