fix cli_demo

2025-12-16 11:50:35 +08:00 · 2024-01-20 23:27:10 +08:00
parent cf818a2598
commit c550987a72
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})