Update models/gpt4_model.py

models/gpt4_model.py

@@ -22,16 +22,27 @@ class GPT4Model(BaseModel):
             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,
+            )
+            print("end api call")
+            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,
+                )
+                print("end api call")
+                response = output.choices[0].message.content
+            except:
+                response = "No answer provided."
         return response
     
     def respond(self, messages: list[dict]) -> str:
@@ -41,12 +52,23 @@ class GPT4Model(BaseModel):
         )
         # OpenAI.api_key=os.environ["OPENAI_API_KEY"]
         # OpenAI.api_base=os.environ["OPENAI_API_BASE"]
-        response = client.chat.completions.create(
-            messages=messages,
-            model=self.generation_model,
-            temperature=self.temperature,
-            max_tokens=1000,
-        ).choices[0].message.content
+        try:
+            response = client.chat.completions.create(
+                messages=messages,
+                model=self.generation_model,
+                temperature=self.temperature,
+                max_tokens=1000,
+            ).choices[0].message.content
+        except:
+            try:
+                response = client.chat.completions.create(
+                    messages=messages,
+                    model=self.generation_model,
+                    temperature=self.temperature,
+                    max_tokens=1000,
+                ).choices[0].message.content
+            except:
+                response = "No answer provided."
         return response
     
     def embedding(self, texts: list[str]) -> list[float]: