Update app.py

app.py

@@ -7,7 +7,11 @@ from langchain_core.vectorstores import InMemoryVectorStore
 import os
 from langchain_core.chat_history import InMemoryChatMessageHistory, BaseChatMessageHistory
 import time
+
+from langgraph.errors import GraphRecursionError
+
 from graph import get_graph
+from langchain_core.runnables import RunnableConfig
 
 if 'read_file' not in st.session_state:
     st.session_state.read_file = False
@@ -77,7 +81,8 @@ if st.session_state.read_file:
                 'next_node': None,
                 'history': get_session_by_id('chat42').messages,
             },
-            stream_mode='messages'
+            stream_mode='messages',
+            config = RunnableConfig(recursion_limit=4)
         )
 
         get_session_by_id('chat42').add_message(HumanMessage(content=prompt))
@@ -93,20 +98,29 @@ if st.session_state.read_file:
             placeholders = {}
             prompt_message_placeholder = st.empty()
 
-            for msg in get_message():
-                full_response += msg
-                if '<tool>' in full_response:
-                    with tool_placeholder.status('Reading document...', expanded=True):
-                        if 'tool_message_placeholder' not in placeholders:
-                            placeholders['tool_message_placeholder'] = st.empty()
-                        placeholders['tool_message_placeholder'].write(full_response
-                                                                       .replace('<tool>', '')
-                                                                       .replace('</tool>', '')
-                                                                       .replace('retriever', 'Retrieving document'))
-                        prompt_message_placeholder.empty()
-                if '</tool>' in full_response:
-                    full_response = ''
-                    continue
-                else:
+            try:
+                for msg in get_message():
+                    full_response += msg
+                    if '<tool>' in full_response:
+                        with tool_placeholder.status('Reading document...', expanded=True):
+                            if 'tool_message_placeholder' not in placeholders:
+                                placeholders['tool_message_placeholder'] = st.empty()
+                            placeholders['tool_message_placeholder'].write(full_response
+                                                                           .replace('<tool>', '')
+                                                                           .replace('</tool>', '')
+                                                                           .replace('retriever', 'Retrieving document'))
+                            prompt_message_placeholder.empty()
+                    if '</tool>' in full_response:
+                        full_response = ''
+                        continue
+                    else:
+                        prompt_message_placeholder.write(full_response)
+            except GraphRecursionError:
+                message = 'Não consegui responder a sua pergunta. 😥 Poderia me perguntar outra coisa?'
+                full_response = ''
+                for letter in message:
+                    full_response += letter
+                    time.sleep(0.015)
                     prompt_message_placeholder.write(full_response)
+
         get_session_by_id('chat42').add_message(AIMessage(content=full_response))