retry use wait_random_exponential

This commit is contained in:
better629 2023-12-15 17:30:25 +08:00
parent 4127ef8570
commit 70cbfb1e48

View file

@ -7,7 +7,7 @@ from tenacity import (
retry,
retry_if_exception_type,
stop_after_attempt,
wait_fixed,
wait_random_exponential,
)
import google.generativeai as genai
from google.ai import generativelanguage as glm
@ -139,7 +139,7 @@ class GeminiGPTAPI(BaseGPTAPI):
@retry(
stop=stop_after_attempt(3),
wait=wait_fixed(1),
wait=wait_random_exponential(min=1, max=60),
after=after_log(logger, logger.level("WARNING").name),
retry=retry_if_exception_type(ConnectionError),
retry_error_callback=log_and_reraise