Spaces:

Suraj-Yadav
/

Basic-Chatbot-LangGraph

Sleeping

App Files Files Community

Suraj Yadav commited on Mar 30

Commit

7d93036

1 Parent(s): 3de25ad

Update the Chat UI: Putting tool call msg in st.expander of streamlit.

Browse files

Files changed (2) hide show

src/basicchatbot/ui/streamlitui/__pycache__/display_result.cpython-311.pyc +0 -0
src/basicchatbot/ui/streamlitui/display_result.py +5 -3

src/basicchatbot/ui/streamlitui/__pycache__/display_result.cpython-311.pyc CHANGED Viewed

Binary files a/src/basicchatbot/ui/streamlitui/__pycache__/display_result.cpython-311.pyc and b/src/basicchatbot/ui/streamlitui/__pycache__/display_result.cpython-311.pyc differ

src/basicchatbot/ui/streamlitui/display_result.py CHANGED Viewed

@@ -83,9 +83,11 @@ class DisplayResultStreamlit:
                     st.session_state.message_history.append({"role": "assistant", "message": tool_message})
                     with st.chat_message("assistant"):
-                        st.markdown("🔧 **Tool Call Start**")
-                        st.markdown(message.content)
-                        st.markdown("🔧 **Tool Call End**")
         except Exception as e:
             st.error(f"Error processing response: {str(e)}")

                     st.session_state.message_history.append({"role": "assistant", "message": tool_message})
                     with st.chat_message("assistant"):
+                        # st.markdown("🔧 **Tool Call Start**")
+                        # st.markdown(message.content)
+                        with st.expander(label="Tool Call",expanded=False):
+                            st.markdown(tool_message)
+                        # st.markdown("🔧 **Tool Call End**")
         except Exception as e:
             st.error(f"Error processing response: {str(e)}")