Limit Log Message Length

st_app.py

@@ -22,6 +22,10 @@ def show_example_questions():
             return True
     return False
 
+def format_log_msg(log_msg: str):
+    max_log_msg_size = 500
+    return log_msg if len(log_msg) <= max_log_msg_size else log_msg[:max_log_msg_size]+'...'
+
 def agent_progress_callback(status_type: AgentStatusType, msg: str):
     output = f'<span style="color:blue;">{status_type.value}</span>: {msg}'
     st.session_state.log_messages.append(output)
@@ -40,12 +44,12 @@ def agent_progress_callback(status_type: AgentStatusType, msg: str):
         max_log_msg_size = 200
         with st.session_state.status:
             for log_msg in st.session_state.log_messages:
-                st.markdown(log_msg[:max_log_msg_size]+'...', unsafe_allow_html=True)
+                st.markdown(format_log_msg(log_msg), unsafe_allow_html=True)
 
 @st.dialog(title="Agent logs", width='large')
 def show_modal():
     for log_msg in st.session_state.log_messages:
-        st.write(log_msg, unsafe_allow_html=True)
+        st.markdown(format_log_msg(log_msg), unsafe_allow_html=True)
 
 async def launch_bot():
     def reset():