From ad3eeca80edd8c5642c2212ba13a44207d5b590f Mon Sep 17 00:00:00 2001 From: Paul Gauthier Date: Fri, 8 Sep 2023 08:28:12 -0700 Subject: [PATCH] Added __all__ to models/__init__ --- aider/models/__init__.py | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/aider/models/__init__.py b/aider/models/__init__.py index f299009c6..fa8aa3673 100644 --- a/aider/models/__init__.py +++ b/aider/models/__init__.py @@ -1,5 +1,15 @@ from .model import Model +from .openai import OpenAIModel +from .openrouter import OpenRouterModel GPT4 = Model.create("gpt-4") GPT35 = Model.create("gpt-3.5-turbo") GPT35_16k = Model.create("gpt-3.5-turbo-16k") + +__all__ = [ + OpenAIModel, + OpenRouterModel, + GPT4, + GPT35, + GPT35_16k, +]