From 98097005c70b8a2c3304229bbcb355d12c3f8921 Mon Sep 17 00:00:00 2001 From: Joshua Vial Date: Wed, 6 Sep 2023 15:00:35 +1200 Subject: [PATCH] small fix for better error handling if the summarizer ever fails - fix #231 --- aider/coders/base_coder.py | 6 +++++- aider/history.py | 2 ++ 2 files changed, 7 insertions(+), 1 deletion(-) diff --git a/aider/coders/base_coder.py b/aider/coders/base_coder.py index d103396d9..c10118924 100755 --- a/aider/coders/base_coder.py +++ b/aider/coders/base_coder.py @@ -359,7 +359,11 @@ class Coder: self.summarizer_thread.start() def summarize_worker(self): - self.summarized_done_messages = self.summarizer.summarize(self.done_messages) + try: + self.summarized_done_messages = self.summarizer.summarize(self.done_messages) + except ValueError as err: + self.io.tool_error(err.args[0]) + if self.verbose: self.io.tool_output("Finished summarizing chat history.") diff --git a/aider/history.py b/aider/history.py index 36d148344..56e8a8423 100644 --- a/aider/history.py +++ b/aider/history.py @@ -87,6 +87,8 @@ class ChatSummary: ] summary = simple_send_with_retries(self.model.name, messages) + if summary is None: + raise ValueError(f"summarizer unexpectedly failed for {self.model.name}") summary = prompts.summary_prefix + summary return [dict(role="user", content=summary)]