fix cli_demo

Former-commit-id: e8336b3653f43618cf7cd70f8da004208de970c0
2024-01-20 23:27:10 +08:00
parent 7543dc4a9d
commit aed00a97b6
4 changed files with 6 additions and 6 deletions
--- a/src/cli_demo.py
+++ b/src/cli_demo.py
@@ -34,6 +34,7 @@ def main():
            print("History has been removed.")
            continue

+        messages.append({"role": "user", "content": query})
        print("Assistant: ", end="", flush=True)

        response = ""
@@ -41,8 +42,6 @@ def main():
            print(new_text, end="", flush=True)
            response += new_text
        print()
-
-        messages.append({"role": "user", "content": query})
        messages.append({"role": "assistant", "content": response})