Update src/streamlit_app.py

src/streamlit_app.py

@@ -149,17 +149,17 @@ if "current_chat_id" in st.session_state:
                 config=set_config()):
     # The output of a streaming node is the key of the node itself
                      if "llmresponse" in chunk:
-                     ai_message = chunk["llmresponse"]["messages"][-1]
-                     full_response += ai_message.content
-                     response_placeholder.markdown(full_response + "▌") # Add a cursor effect
+                         ai_message = chunk["llmresponse"]["messages"][-1]
+                         full_response += ai_message.content
+                         response_placeholder.markdown(full_response + "▌") # Add a cursor effect
     
     # Handle tool calls if they appear in the stream
                      elif "tool_node" in chunk:
         # Since we are streaming, render the tool message right away
         # And clear the response placeholder for the next text chunk
-                     response_placeholder.empty() 
-                     tool_messages = chunk["tool_node"]["messages"]
-                     for tool_message in tool_messages:
-                        render_tool_message(tool_message)
+                         response_placeholder.empty() 
+                         tool_messages = chunk["tool_node"]["messages"]
+                         for tool_message in tool_messages:
+                             render_tool_message(tool_message)
 
             response_placeholder.markdown(full_response) # Final update without cursor