From 4f1aa0333ec9cae6bf69c711735794c8c6677693 Mon Sep 17 00:00:00 2001 From: stellahsr Date: Tue, 12 Dec 2023 10:10:19 +0800 Subject: [PATCH] =?UTF-8?q?=E5=A2=9E=E5=8A=A0retry?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- metagpt/provider/openai_api.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/metagpt/provider/openai_api.py b/metagpt/provider/openai_api.py index 34e5693f8..d8d2e9a4f 100644 --- a/metagpt/provider/openai_api.py +++ b/metagpt/provider/openai_api.py @@ -15,6 +15,7 @@ from tenacity import ( retry, retry_if_exception_type, stop_after_attempt, + wait_random_exponential, wait_fixed, ) @@ -259,7 +260,8 @@ class OpenAIGPTAPI(BaseGPTAPI, RateLimiter): rsp = self.llm.ChatCompletion.create(**self._func_configs(messages, **kwargs)) self._update_costs(rsp.get("usage")) return rsp - + + @retry(wait=wait_random_exponential(min=1, max=60), stop=stop_after_attempt(6)) async def _achat_completion_function(self, messages: list[dict], **chat_configs) -> dict: rsp = await self.llm.ChatCompletion.acreate(**self._func_configs(messages, **chat_configs)) self._update_costs(rsp.get("usage"))