summaryrefslogtreecommitdiffstats
path: root/g4f/Provider/retry_provider.py
diff options
context:
space:
mode:
authorTekky <98614666+xtekky@users.noreply.github.com>2023-10-10 16:30:05 +0200
committerGitHub <noreply@github.com>2023-10-10 16:30:05 +0200
commite06012ae067abf513633134f6bae41bfab2a6497 (patch)
tree7b2c4ea4cb468fc046fb4567292d3dfd5d0e7a19 /g4f/Provider/retry_provider.py
parent~ | Merge pull request #1033 from Commenter123321/main (diff)
parentSome small fixes (diff)
downloadgpt4free-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.py13
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)