From d2b9cb9e98cf1cc97de7bd09fe0240eb102ecada Mon Sep 17 00:00:00 2001
From: Paul Gauthier <aider@paulg.org>
Date: Fri, 23 Jun 2023 13:29:09 -0700
Subject: [PATCH] added --no-stream

---
 aider/coders/base_coder.py | 27 +++++++++++++++++++++++++--
 aider/main.py              |  9 +++++++++
 2 files changed, 34 insertions(+), 2 deletions(-)

diff --git a/aider/coders/base_coder.py b/aider/coders/base_coder.py
index 3a000b427..250f50e57 100755
--- a/aider/coders/base_coder.py
+++ b/aider/coders/base_coder.py
@@ -88,6 +88,7 @@ class Coder:
         map_tokens=1024,
         verbose=False,
         assistant_output_color="blue",
+        stream=True,
     ):
         self.verbose = verbose
         self.abs_fnames = set()
@@ -96,6 +97,7 @@ class Coder:
         self.num_control_c = 0
 
         self.io = io
+        self.stream = stream
 
         if not auto_commits:
             dirty_commits = False
@@ -475,7 +477,7 @@ class Coder:
             model=model,
             messages=messages,
             temperature=0,
-            stream=True,
+            stream=self.stream,
         )
         if functions is not None:
             kwargs["functions"] = self.functions
@@ -493,7 +495,10 @@ class Coder:
         interrupted = False
         try:
             completion = self.send_with_retries(model, messages, functions)
-            self.show_send_output(completion, silent)
+            if self.stream:
+                self.show_send_output_stream(completion, silent)
+            else:
+                self.show_send_output(completion, silent)
         except KeyboardInterrupt:
             interrupted = True
 
@@ -511,6 +516,24 @@ class Coder:
         return interrupted
 
     def show_send_output(self, completion, silent):
+        try:
+            self.partial_response_function_call = completion.choices[0].function_call
+        except AttributeError:
+            pass
+
+        try:
+            self.partial_response_content = completion.choices[0].message.content
+        except AttributeError:
+            pass
+
+        show_resp = self.render_incremental_response(True)
+        if self.pretty:
+            md = Markdown(show_resp, style=self.assistant_output_color, code_theme="default")
+            self.io.console.print(md)
+        else:
+            print(show_resp)
+
+    def show_send_output_stream(self, completion, silent):
         live = None
         if self.pretty and not silent:
             live = Live(vertical_overflow="scroll")
diff --git a/aider/main.py b/aider/main.py
index 8c669cecf..3b50c066f 100644
--- a/aider/main.py
+++ b/aider/main.py
@@ -111,6 +111,13 @@ def main(args=None, input=None, output=None):
         dest="pretty",
         help="Disable pretty, colorized output",
     )
+    parser.add_argument(
+        "--no-stream",
+        action="store_false",
+        dest="stream",
+        default=True,
+        help="Disable streaming responses",
+    )
     parser.add_argument(
         "--user-input-color",
         default="green",
@@ -257,6 +264,7 @@ def main(args=None, input=None, output=None):
         map_tokens=args.map_tokens,
         verbose=args.verbose,
         assistant_output_color=args.assistant_output_color,
+        stream=args.stream,
     )
 
     if args.dirty_commits:
@@ -270,6 +278,7 @@ def main(args=None, input=None, output=None):
 
     io.tool_output("Use /help to see in-chat commands.")
     if args.command:
+        io.tool_output()
         coder.run(with_message=args.command)
     else:
         coder.run()