Update models/gpt4_model.py

models/gpt4_model.py

@@ -13,12 +13,28 @@ class GPT4Model(BaseModel):
         self.temperature = temperature
     
     def respond(self, messages: list) -> str:
-        response = openai.ChatCompletion.create(
-            messages=messages,
-            model=self.generation_model,
-            temperature=self.temperature,
-            max_tokens=1000,
-        ).choices[0]['message']['content']
+        try:
+            response = openai.ChatCompletion.create(
+                messages=messages,
+                model=self.generation_model,
+                temperature=self.temperature,
+                max_tokens=1000,
+            ).choices[0]['message']['content']
+        except:
+            try:
+                response = openai.ChatCompletion.create(
+                    messages=messages,
+                    model=self.generation_model,
+                    temperature=self.temperature,
+                    max_tokens=1000,
+                ).choices[0]['message']['content']
+            except:
+                response = openai.ChatCompletion.create(
+                    messages=messages,
+                    model=self.generation_model,
+                    temperature=self.temperature,
+                    max_tokens=1000,
+                ).choices[0]['message']['content']
         # content = response.choices[0]['message']['content']
         
         return response