summaryrefslogtreecommitdiffstats
path: root/g4f/__init__.py
diff options
context:
space:
mode:
authorHeiner Lohaus <hlohaus@users.noreply.github.com>2024-01-23 12:17:16 +0100
committerHeiner Lohaus <hlohaus@users.noreply.github.com>2024-01-23 12:17:16 +0100
commite4f743881c915ba250b859d01ef78c7b90a96bf5 (patch)
tree22e1897967212a209751880cd6f9e3bac82bb9d9 /g4f/__init__.py
parentCheck for comments in copilot (diff)
downloadgpt4free-e4f743881c915ba250b859d01ef78c7b90a96bf5.tar
gpt4free-e4f743881c915ba250b859d01ef78c7b90a96bf5.tar.gz
gpt4free-e4f743881c915ba250b859d01ef78c7b90a96bf5.tar.bz2
gpt4free-e4f743881c915ba250b859d01ef78c7b90a96bf5.tar.lz
gpt4free-e4f743881c915ba250b859d01ef78c7b90a96bf5.tar.xz
gpt4free-e4f743881c915ba250b859d01ef78c7b90a96bf5.tar.zst
gpt4free-e4f743881c915ba250b859d01ef78c7b90a96bf5.zip
Diffstat (limited to 'g4f/__init__.py')
-rw-r--r--g4f/__init__.py9
1 files changed, 5 insertions, 4 deletions
diff --git a/g4f/__init__.py b/g4f/__init__.py
index 2b0e5b46..996acf6c 100644
--- a/g4f/__init__.py
+++ b/g4f/__init__.py
@@ -45,12 +45,13 @@ def get_model_and_provider(model : Union[Model, str],
else:
raise ProviderNotFoundError(f'Provider not found: {provider}')
+ if isinstance(model, str):
+ if model in ModelUtils.convert:
+ model = ModelUtils.convert[model]
+
if not provider:
if isinstance(model, str):
- if model in ModelUtils.convert:
- model = ModelUtils.convert[model]
- else:
- raise ModelNotFoundError(f'Model not found: {model}')
+ raise ModelNotFoundError(f'Model not found: {model}')
provider = model.best_provider
if not provider: