change retry deco place

This commit is contained in:
geekan 2023-07-06 14:43:08 +08:00
parent 1de45add2e
commit 9f15569f04

View file

@ -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: