add dall-e-2 retry_count limit

2026-03-03 00:40:45 +08:00 · 2024-03-30 01:07:52 +08:00
parent 9d394adf24
commit 22c7f8fe9e
1 changed files with 3 additions and 0 deletions
--- a/bot/chatgpt/chat_gpt_bot.py
+++ b/bot/chatgpt/chat_gpt_bot.py
@@ -187,8 +187,11 @@ class AzureChatGPTBot(ChatGPTBot):
                operation_location = submission.headers['operation-location']
                status = ""
                while (status != "succeeded"):
+                    if retry_count > 3:
+                        return False, "图片生成失败"
                    response = requests.get(operation_location, headers=headers)
                    status = response.json()['status']
+                    retry_count += 1
                image_url = response.json()['result']['data'][0]['url']
                return True, image_url
            except Exception as e: