Update display_result.py

src/langgraphagenticai/ui/streamlitui/display_result.py

@@ -1,5 +1,5 @@
 import streamlit as st
-from langchain_core.messages import HumanMessage,AIMessage
+from langchain_core.messages import HumanMessage,AIMessage,ToolMessage
 import json
 
 
@@ -38,4 +38,5 @@ class DisplayResultStreamlit:
                         st.write("Tool Call End")
                 elif type(message)==AIMessage and message.content:
                     with st.chat_message("assistant"):
-                        st.write(message.content)
+                        st.write(message.content)
+