Update models/gpt4_model.py

models/gpt4_model.py

@@ -21,16 +21,26 @@ class GPT4Model(BaseModel):
             api_key=os.environ["OPENAI_API_KEY"],
             base_url=os.environ["OPENAI_API_BASE"]
         )
-        print("start api call")
-        output = await client.chat.completions.create(
-            messages=messages,
-            model=self.generation_model,
-            temperature=self.temperature,
-            max_tokens=1000,
-        )
-        print("end api call")
-        response = output.choices[0].message.content
-        # content = response.choices[0]['message']['content']
+        try:
+            output = await client.chat.completions.create(
+                messages=messages,
+                model=self.generation_model,
+                temperature=self.temperature,
+                max_tokens=1000,
+            )
+            response = output.choices[0].message.content
+        except:
+            try:
+                output = await client.chat.completions.create(
+                    messages=messages,
+                    model=self.generation_model,
+                    temperature=self.temperature,
+                    max_tokens=1000,
+                )
+                response = output.choices[0].message.content
+            except:
+                response = "No answer provided."
+        return response
         
         return response