Update src/streamlit_app.py

src/streamlit_app.py

@@ -3,7 +3,7 @@ import streamlit as st
 from typing import Literal
 from tavily import TavilyClient
 from deepagents import create_deep_agent
-#from langchain.callbacks.streamlit import StreamlitCallbackHandler
+from langchain.callbacks.streamlit import StreamlitCallbackHandler
 from langgraph.store.memory import InMemoryStore
 from langchain.chat_models import init_chat_model
 from langchain_huggingface import ChatHuggingFace, HuggingFacePipeline
@@ -119,7 +119,7 @@ st.caption(f"Running on: **{provider}** backend")
 if prompt := st.chat_input("Ask your research question..."):
     st.chat_message("user").write(prompt)
     with st.chat_message("assistant"):
-        #st_callback = StreamlitCallbackHandler(st.container())
+        st_callback = StreamlitCallbackHandler(st.container())
         response = agent.invoke({"input": prompt}, {"callbacks": [st_callback]})
         output = response.get("output", str(response))
         st.write(output)