update callback

app.py

@@ -17,6 +17,8 @@ from langchain.agents import Tool
 from langchain.tools import BaseTool
 from langchain.memory import ConversationBufferWindowMemory
 from langchain.prompts import MessagesPlaceholder
+from langchain.agents import ConversationalChatAgent, AgentExecutor
+from langchain.callbacks import StreamlitCallbackHandler
 
 
 global CurrentAgent
@@ -183,6 +185,7 @@ agent_STRUCTURED_ZEROSHOT_REACT = initialize_agent(tools_remote, GPTfake,
 agent = agent_STRUCTURED_ZEROSHOT_REACT
 
 
+
 st.title("STLA-BABY")
 
 msgs = StreamlitChatMessageHistory()
@@ -203,6 +206,7 @@ if prompt := st.chat_input(placeholder="Input Your Request"):
 
     with st.chat_message("assistant"):
         # response = GPTfake(prompt)
-        response = agent.run(prompt)
+        st_cb = StreamlitCallbackHandler(st.container(), expand_new_thoughts=False)
+        response = agent(prompt, callbacks=[st_cb])
         st.write(response)
         print(msgs.messages)