Update cli.py

Author: lunaticsm

duck_chat/cli.py

@@ -54,19 +54,21 @@ async def run(self) -> None:
 
                 user_input = self.get_user_input()
 
-                # user input is command
                 if user_input.startswith("/"):
                     await self.command_parsing(user_input.split(), chat)
                     continue
 
-                # empty user input
                 if not user_input:
                     print("Bad input")
                     continue
 
                 print(f"\033[1;4m>>> Response №{self.COUNT}:\033[0m", end="\n")
                 try:
-                    self.answer_print(await chat.ask_question(user_input))
+                    stream = "--stream" in user_input
+                    if stream:
+                        user_input = user_input.replace("--stream", "").strip()
+
+                    self.answer_print(await chat.ask_question(user_input, stream=stream))
                 except DuckChatException as e:
                     print(f"Error occurred: {str(e)}")
                 else:
@@ -133,7 +135,7 @@ async def command_parsing(self, args: list[str], chat: DuckChat) -> None:
                 print("Type \033[1;4m/help\033[0m to display the help")
 
     def answer_print(self, query: str) -> None:
-        if "`" in query:  # block of code
+        if "`" in query:
             self.console.print(Markdown(query))
         else:
             print(query)