Update streamly.py

AdieLaine · Jul 24, 2024 · aeff73d · aeff73d
1 parent a64ed54
commit aeff73d
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/streamly.py b/streamly.py
@@ -212,6 +212,12 @@ def on_chat_submit(chat_input, api_key, latest_updates, use_langchain=False):
             st.session_state.history.append({"role": "user", "content": user_input})
             st.session_state.history.append({"role": "assistant", "content": assistant_reply})
 
+    except OpenAIError.APIConnectionError as e:
+        logging.error(f"Error occurred: {e}")
+        error_message = f"OpenAI Error: {str(e)}"
+        st.error(error_message)
+        #st.session_state.history.append({"role": "assistant", "content": error_message})
+
 def main():
     """
     Display Streamlit updates and handle the chat interface.