summaryrefslogtreecommitdiffstats
path: root/g4f/__init__.py
diff options
context:
space:
mode:
authorHeiner Lohaus <hlohaus@users.noreply.github.com>2024-03-14 15:22:09 +0100
committerHeiner Lohaus <hlohaus@users.noreply.github.com>2024-03-14 15:22:09 +0100
commit012f40078a59e6f725f57cdb013451fe0b268206 (patch)
treea3f1da46a1efb97854dd50913fb8d46e8d27e70e /g4f/__init__.py
parentAdd webview support to OpenaiChat (diff)
downloadgpt4free-012f40078a59e6f725f57cdb013451fe0b268206.tar
gpt4free-012f40078a59e6f725f57cdb013451fe0b268206.tar.gz
gpt4free-012f40078a59e6f725f57cdb013451fe0b268206.tar.bz2
gpt4free-012f40078a59e6f725f57cdb013451fe0b268206.tar.lz
gpt4free-012f40078a59e6f725f57cdb013451fe0b268206.tar.xz
gpt4free-012f40078a59e6f725f57cdb013451fe0b268206.tar.zst
gpt4free-012f40078a59e6f725f57cdb013451fe0b268206.zip
Diffstat (limited to 'g4f/__init__.py')
-rw-r--r--g4f/__init__.py9
1 files changed, 6 insertions, 3 deletions
diff --git a/g4f/__init__.py b/g4f/__init__.py
index 441225f1..dce116c4 100644
--- a/g4f/__init__.py
+++ b/g4f/__init__.py
@@ -69,11 +69,14 @@ def get_model_and_provider(model : Union[Model, str],
if isinstance(model, Model):
model = model.name
- if ignored and isinstance(provider, BaseRetryProvider):
- provider.providers = [p for p in provider.providers if p.__name__ not in ignored]
-
if not ignore_working and not provider.working:
raise ProviderNotWorkingError(f'{provider.__name__} is not working')
+
+ if not ignore_working and isinstance(provider, BaseRetryProvider):
+ provider.providers = [p for p in provider.providers if p.working]
+
+ if ignored and isinstance(provider, BaseRetryProvider):
+ provider.providers = [p for p in provider.providers if p.__name__ not in ignored]
if not ignore_stream and not provider.supports_stream and stream:
raise StreamNotSupportedError(f'{provider.__name__} does not support "stream" argument')