diff options
author | Tekky <98614666+xtekky@users.noreply.github.com> | 2023-10-10 16:30:05 +0200 |
---|---|---|
committer | GitHub <noreply@github.com> | 2023-10-10 16:30:05 +0200 |
commit | e06012ae067abf513633134f6bae41bfab2a6497 (patch) | |
tree | 7b2c4ea4cb468fc046fb4567292d3dfd5d0e7a19 /g4f/Provider/retry_provider.py | |
parent | ~ | Merge pull request #1033 from Commenter123321/main (diff) | |
parent | Some small fixes (diff) | |
download | gpt4free-e06012ae067abf513633134f6bae41bfab2a6497.tar gpt4free-e06012ae067abf513633134f6bae41bfab2a6497.tar.gz gpt4free-e06012ae067abf513633134f6bae41bfab2a6497.tar.bz2 gpt4free-e06012ae067abf513633134f6bae41bfab2a6497.tar.lz gpt4free-e06012ae067abf513633134f6bae41bfab2a6497.tar.xz gpt4free-e06012ae067abf513633134f6bae41bfab2a6497.tar.zst gpt4free-e06012ae067abf513633134f6bae41bfab2a6497.zip |
Diffstat (limited to 'g4f/Provider/retry_provider.py')
-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) |