From 8c1e1c72672675ca52ccc2f975e636b40747ea82 Mon Sep 17 00:00:00 2001 From: Paul Gauthier Date: Wed, 21 Jun 2023 21:08:22 -0700 Subject: [PATCH] handle not live case --- aider/coders/base_coder.py | 2 +- aider/coders/wholefile_coder.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/aider/coders/base_coder.py b/aider/coders/base_coder.py index 59e6b6a48..d7315fb04 100755 --- a/aider/coders/base_coder.py +++ b/aider/coders/base_coder.py @@ -532,8 +532,8 @@ class Coder: sys.stdout.write(text) sys.stdout.flush() finally: - self.live_incremental_response(live, True) if live: + self.live_incremental_response(live, True) live.stop() def live_incremental_response(self, live, final): diff --git a/aider/coders/wholefile_coder.py b/aider/coders/wholefile_coder.py index 1d50c5696..da870f8ca 100644 --- a/aider/coders/wholefile_coder.py +++ b/aider/coders/wholefile_coder.py @@ -20,7 +20,7 @@ class WholeFileCoder(Coder): else: self.cur_messages += [dict(role="assistant", content=content)] - def modify_incremental_response(self): + def modify_incremental_response(self, final): resp = self.partial_response_content return self.update_files(resp, mode="diff")