From 43ba638ac4d641747a5f5e99938008ff3fcb703b Mon Sep 17 00:00:00 2001 From: Paul Gauthier Date: Tue, 20 Jun 2023 15:46:22 -0700 Subject: [PATCH] refac --- aider/coder.py | 2 +- aider/models.py | 6 +++--- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/aider/coder.py b/aider/coder.py index 9e92445f1..40b9cd90b 100755 --- a/aider/coder.py +++ b/aider/coder.py @@ -81,7 +81,7 @@ class Coder: self.console = Console(force_terminal=True, no_color=True) main_model = models.Model(main_model) - if not main_model.is_always_available(): + if not main_model.always_available: if not self.check_model_availability(main_model): if main_model != models.GPT4: self.io.tool_error( diff --git a/aider/models.py b/aider/models.py index f355c00a7..fe5786d59 100644 --- a/aider/models.py +++ b/aider/models.py @@ -2,6 +2,8 @@ import re class Model: + always_available = False + def __init__(self, name, tokens=None): self.name = name if tokens is None: @@ -19,6 +21,7 @@ class Model: if self.is_gpt35(): self.edit_format = "whole" + self.always_available = True return raise ValueError(f"Unsupported model: {name}") @@ -29,9 +32,6 @@ class Model: def is_gpt35(self): return self.name.startswith("gpt-3.5-turbo") - def is_always_available(self): - return self.is_gpt35() - GPT4 = Model("gpt-4", 8) GPT35 = Model("gpt-3.5-turbo")