From 4f6d33920a2d4b8fdd51ca17aa4df56da8e3c418 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Florian=20Ho=CC=88nicke?= <hoenicke.florian@gmail.com>
Date: Mon, 10 Apr 2023 21:37:48 +0200
Subject: [PATCH] fix: rate limiting error

---
 src/gpt.py | 19 ++++++++++++-------
 1 file changed, 12 insertions(+), 7 deletions(-)

diff --git a/src/gpt.py b/src/gpt.py
index 02631e4..f5cf175 100644
--- a/src/gpt.py
+++ b/src/gpt.py
@@ -32,13 +32,18 @@ class GPTSession:
 
     def is_gpt4_available(self):
         try:
-            openai.ChatCompletion.create(
-                model="gpt-4",
-                messages=[{
-                    "role": 'system',
-                    "content": 'test'
-                }]
-            )
+            for i in range(5):
+                try:
+                    openai.ChatCompletion.create(
+                        model="gpt-4",
+                        messages=[{
+                            "role": 'system',
+                            "content": 'test'
+                        }]
+                    )
+                except RateLimitError:
+                    sleep(1)
+                    continue
             return True
         except openai.error.InvalidRequestError:
             return False