Added __all__ to models/__init__

This commit is contained in:
Paul Gauthier 2023-09-08 08:28:12 -07:00
parent bfbd7b8416
commit ad3eeca80e

View file

@ -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,
]