Update app.py

app.py

@@ -528,16 +528,20 @@ Write a detailed and complete response that answers the following user question:
             response = ""
             for i in range(num_calls):
                 logging.info(f"API call {i+1}/{num_calls}")
-                for message in client.chat_completion(
-                    messages=[{"role": "user", "content": prompt}],
-                    max_tokens=10000,
-                    temperature=temperature,
-                    stream=True,
-                ):
-                    if message.choices and message.choices[0].delta and message.choices[0].delta.content:
-                        chunk = message.choices[0].delta.content
-                        response += chunk
-                        yield response  # Yield partial response
+                try:
+                    for message in client.chat_completion(
+                        messages=[{"role": "user", "content": prompt}],
+                        max_tokens=10000,
+                        temperature=temperature,
+                        stream=True,
+                    ):
+                        if message.choices and message.choices[0].delta and message.choices[0].delta.content:
+                            chunk = message.choices[0].delta.content
+                            response += chunk
+                            yield response  # Yield partial response
+                except Exception as e:
+                    logging.error(f"Error in API call {i+1}: {str(e)}")
+                    yield f"Error in API call {i+1}: {str(e)}. Attempting next call..."
             
             logging.info("Finished generating response")