From 1f2aa1a6c8b01aea0cb92a7bc16b400140abbff7 Mon Sep 17 00:00:00 2001 From: Paul Gauthier Date: Thu, 1 Jun 2023 08:39:27 -0700 Subject: [PATCH] Auto fallback to 3.5 --- aider/coder.py | 14 +++++++++----- 1 file changed, 9 insertions(+), 5 deletions(-) diff --git a/aider/coder.py b/aider/coder.py index e7763dc07..acb807813 100755 --- a/aider/coder.py +++ b/aider/coder.py @@ -31,9 +31,8 @@ class Coder: def check_model_availability(self, main_model): available_models = openai.Model.list() - model_ids = [model.id for model in available_models['data']] - if main_model not in model_ids: - raise ValueError(f"Model {main_model} is not available. Please choose from the available models: {model_ids}") + model_ids = [model.id for model in available_models["data"]] + return main_model in model_ids def __init__( self, @@ -70,11 +69,16 @@ class Coder: self.console = Console(force_terminal=True, no_color=True) self.commands = Commands(self.io, self) - self.check_model_availability(main_model) + if not self.check_model_availability(main_model): + self.io.tool_error( + f"Your openai key does not support {main_model}, using gpt-3.5-turbo." + ) + main_model = "gpt-3.5-turbo" + self.main_model = main_model if main_model == "gpt-3.5-turbo": self.io.tool_error( - f"aider doesn't work well with {main_model}, use gpt-4 for best results." + f"Aider doesn't work well with {main_model}, use gpt-4 for best results." ) self.set_repo(fnames)