diff options
author | Heiner Lohaus <heiner@lohaus.eu> | 2023-10-05 05:13:37 +0200 |
---|---|---|
committer | Heiner Lohaus <heiner@lohaus.eu> | 2023-10-05 05:13:37 +0200 |
commit | 88d2cbff099df00944ed6dfb6c73b1b5e8dfc7f9 (patch) | |
tree | 7e6e3b6c179cf1a695f9d80026d544a4fd3c5203 /g4f/Provider/retry_provider.py | |
parent | ~ | g4f v-0.1.4.8 - Fixed `g4f.Provider.Bing` (diff) | |
download | gpt4free-88d2cbff099df00944ed6dfb6c73b1b5e8dfc7f9.tar gpt4free-88d2cbff099df00944ed6dfb6c73b1b5e8dfc7f9.tar.gz gpt4free-88d2cbff099df00944ed6dfb6c73b1b5e8dfc7f9.tar.bz2 gpt4free-88d2cbff099df00944ed6dfb6c73b1b5e8dfc7f9.tar.lz gpt4free-88d2cbff099df00944ed6dfb6c73b1b5e8dfc7f9.tar.xz gpt4free-88d2cbff099df00944ed6dfb6c73b1b5e8dfc7f9.tar.zst gpt4free-88d2cbff099df00944ed6dfb6c73b1b5e8dfc7f9.zip |
Diffstat (limited to 'g4f/Provider/retry_provider.py')
-rw-r--r-- | g4f/Provider/retry_provider.py | 9 |
1 files changed, 8 insertions, 1 deletions
diff --git a/g4f/Provider/retry_provider.py b/g4f/Provider/retry_provider.py index e1a9cd1f..c1672aba 100644 --- a/g4f/Provider/retry_provider.py +++ b/g4f/Provider/retry_provider.py @@ -4,6 +4,7 @@ import random from ..typing import CreateResult from .base_provider import BaseProvider, AsyncProvider +from ..debug import logging class RetryProvider(AsyncProvider): @@ -41,6 +42,8 @@ class RetryProvider(AsyncProvider): started = False for provider in providers: try: + if logging: + print(f"Using {provider.__name__} provider") for token in provider.create_completion(model, messages, stream, **kwargs): yield token started = True @@ -48,6 +51,8 @@ class RetryProvider(AsyncProvider): return except Exception as e: self.exceptions[provider.__name__] = e + if logging: + print(f"{provider.__name__}: {e.__class__.__name__}: {e}") if started: break @@ -59,7 +64,7 @@ class RetryProvider(AsyncProvider): messages: list[dict[str, str]], **kwargs ) -> str: - providers = [provider for provider in self.providers if issubclass(provider, AsyncProvider)] + providers = [provider for provider in self.providers] if self.shuffle: random.shuffle(providers) @@ -69,6 +74,8 @@ class RetryProvider(AsyncProvider): return await provider.create_async(model, messages, **kwargs) except Exception as e: self.exceptions[provider.__name__] = e + if logging: + print(f"{provider.__name__}: {e.__class__.__name__}: {e}") self.raise_exceptions() |