From 9f15569f047de001f411f6a44aae720feda0d63b Mon Sep 17 00:00:00 2001 From: geekan Date: Thu, 6 Jul 2023 14:43:08 +0800 Subject: [PATCH] change retry deco place --- metagpt/provider/openai_api.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/metagpt/provider/openai_api.py b/metagpt/provider/openai_api.py index c8a6c4489..c4b82d149 100644 --- a/metagpt/provider/openai_api.py +++ b/metagpt/provider/openai_api.py @@ -200,12 +200,12 @@ class OpenAIGPTAPI(BaseGPTAPI, RateLimiter): # messages = self.messages_to_dict(messages) return self._chat_completion(messages) - @retry(max_retries=6) async def acompletion(self, messages: list[dict]) -> dict: # if isinstance(messages[0], Message): # messages = self.messages_to_dict(messages) return await self._achat_completion(messages) + @retry(max_retries=6) async def acompletion_text(self, messages: list[dict], stream=False) -> str: """when streaming, print each token in place.""" if stream: