diff options
author | Heiner Lohaus <heiner@lohaus.eu> | 2023-10-10 09:49:29 +0200 |
---|---|---|
committer | Heiner Lohaus <heiner@lohaus.eu> | 2023-10-10 09:49:29 +0200 |
commit | 53192b86b129380660f7454170fa987faf2da3c5 (patch) | |
tree | 5797a34479840692f2ae5db9e63deacac9732bd7 /g4f/Provider/retry_provider.py | |
parent | ~ | g4f `v-0.1.5.7` (diff) | |
download | gpt4free-53192b86b129380660f7454170fa987faf2da3c5.tar gpt4free-53192b86b129380660f7454170fa987faf2da3c5.tar.gz gpt4free-53192b86b129380660f7454170fa987faf2da3c5.tar.bz2 gpt4free-53192b86b129380660f7454170fa987faf2da3c5.tar.lz gpt4free-53192b86b129380660f7454170fa987faf2da3c5.tar.xz gpt4free-53192b86b129380660f7454170fa987faf2da3c5.tar.zst gpt4free-53192b86b129380660f7454170fa987faf2da3c5.zip |
Diffstat (limited to '')
-rw-r--r-- | g4f/Provider/retry_provider.py | 13 |
1 files changed, 5 insertions, 8 deletions
diff --git a/g4f/Provider/retry_provider.py b/g4f/Provider/retry_provider.py index b49020b2..94b9b90a 100644 --- a/g4f/Provider/retry_provider.py +++ b/g4f/Provider/retry_provider.py @@ -2,7 +2,7 @@ from __future__ import annotations import random from typing import List, Type, Dict -from ..typing import CreateResult +from ..typing import CreateResult, Messages from .base_provider import BaseProvider, AsyncProvider from ..debug import logging @@ -10,10 +10,7 @@ from ..debug import logging class RetryProvider(AsyncProvider): __name__: str = "RetryProvider" working: bool = True - needs_auth: bool = False supports_stream: bool = True - supports_gpt_35_turbo: bool = False - supports_gpt_4: bool = False def __init__( self, @@ -27,7 +24,7 @@ class RetryProvider(AsyncProvider): def create_completion( self, model: str, - messages: List[Dict[str, str]], + messages: Messages, stream: bool = False, **kwargs ) -> CreateResult: @@ -54,17 +51,17 @@ class RetryProvider(AsyncProvider): if logging: print(f"{provider.__name__}: {e.__class__.__name__}: {e}") if started: - break + raise e self.raise_exceptions() async def create_async( self, model: str, - messages: List[Dict[str, str]], + messages: Messages, **kwargs ) -> str: - providers = [provider for provider in self.providers] + providers = self.providers if self.shuffle: random.shuffle(providers) |