Update apps/agents/agents.py

apps/agents/agents.py

@@ -24,6 +24,7 @@ from typing import Any, Dict, List, Optional, Tuple, Union
 
 import gradio as gr
 import openai
+from openai import RateLimitError, APIError
 
 import tenacity
 
@@ -196,8 +197,8 @@ def role_playing_start(
             extend_task_specify_meta_dict=meta_dict,
             output_language=language,
         )
-    except (openai.error.RateLimitError, tenacity.RetryError,
-            RuntimeError) as ex:
+    except (RateLimitError, tenacity.RetryError,
+            RuntimeError, APIError) as ex:
         print("OpenAI API exception 0 " + str(ex))
         return (state, str(ex), "", [], gr.update())
 
@@ -247,8 +248,8 @@ def role_playing_chat_init(state) -> \
     try:
         init_assistant_msg: BaseMessage
         init_assistant_msg = session.init_chat()
-    except (openai.RateLimitError, tenacity.RetryError,
-            RuntimeError) as ex:
+    except (RateLimitError, tenacity.RetryError,
+            RuntimeError, APIError) as ex:
         print("OpenAI API exception 1 " + str(ex))
         state.session = None
         return state, state.chat, gr.update()
@@ -289,8 +290,8 @@ def role_playing_chat_cont(state) -> \
     try:
         assistant_response, user_response = session.step(
             state.saved_assistant_msg)
-    except (openai.RateLimitError, tenacity.RetryError,
-            RuntimeError) as ex:
+    except (RateLimitError, tenacity.RetryError,
+            RuntimeError, APIError) as ex:
         print("OpenAI API exception 2 " + str(ex))
         state.session = None
         return state, state.chat, gr.update(), gr.update()