diff options
author | xtekky <98614666+xtekky@users.noreply.github.com> | 2023-08-07 13:20:27 +0200 |
---|---|---|
committer | GitHub <noreply@github.com> | 2023-08-07 13:20:27 +0200 |
commit | 8d024192df662d8445265c9a4f5c7bea908fc0a6 (patch) | |
tree | 4749eb44df477c790442d912a40b86313e23d39d /g4f/__init__.py | |
parent | update image (diff) | |
parent | refactor: refactor models (diff) | |
download | gpt4free-8d024192df662d8445265c9a4f5c7bea908fc0a6.tar gpt4free-8d024192df662d8445265c9a4f5c7bea908fc0a6.tar.gz gpt4free-8d024192df662d8445265c9a4f5c7bea908fc0a6.tar.bz2 gpt4free-8d024192df662d8445265c9a4f5c7bea908fc0a6.tar.lz gpt4free-8d024192df662d8445265c9a4f5c7bea908fc0a6.tar.xz gpt4free-8d024192df662d8445265c9a4f5c7bea908fc0a6.tar.zst gpt4free-8d024192df662d8445265c9a4f5c7bea908fc0a6.zip |
Diffstat (limited to '')
-rw-r--r-- | g4f/__init__.py | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/g4f/__init__.py b/g4f/__init__.py index 09b24b55..e5d3d4bf 100644 --- a/g4f/__init__.py +++ b/g4f/__init__.py @@ -1,12 +1,12 @@ import sys from . import Provider -from g4f.models import Model, ModelUtils +from g4f import models logging = False class ChatCompletion: @staticmethod - def create(model: Model.model or str, messages: list, provider: Provider.Provider = None, stream: bool = False, auth: str = False, **kwargs): + def create(model: models.Model | str, messages: list, provider: Provider.Provider = None, stream: bool = False, auth: str = False, **kwargs): kwargs['auth'] = auth if provider and provider.working == False: return f'{provider.__name__} is not working' @@ -19,7 +19,7 @@ class ChatCompletion: try: if isinstance(model, str): try: - model = ModelUtils.convert[model] + model = models.ModelUtils.convert[model] except KeyError: raise Exception(f'The model: {model} does not exist') |