Skip to content

Commit eb39140

Browse files
authored
Merge pull request #415 from LlmKira/dev
⬆️ fix(llmkira/openai): add wait_exponential to retry configu…
2 parents 63fc9b4 + 0da504e commit eb39140

File tree

1 file changed

+6
-2
lines changed

1 file changed

+6
-2
lines changed

llmkira/openai/request.py

+6-2
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@
66
from loguru import logger
77
from pydantic import ConfigDict, BaseModel, Field, field_validator, model_validator
88
from pydantic import SecretStr
9-
from tenacity import retry, stop_after_attempt
9+
from tenacity import retry, stop_after_attempt, wait_exponential
1010

1111
from llmkira.openai._excption import raise_error, NetworkError, UnexpectedFormatError
1212
from .cell import (
@@ -181,7 +181,11 @@ def check_vision(self):
181181
]
182182
return self
183183

184-
@retry(stop=stop_after_attempt(3), reraise=True)
184+
@retry(
185+
stop=stop_after_attempt(3),
186+
reraise=True,
187+
wait=wait_exponential(multiplier=1, min=4, max=10),
188+
)
185189
async def request(self, session: OpenAICredential) -> OpenAIResult:
186190
"""
187191
:param session: Credential

0 commit comments

Comments
 (0)