devchavda11 commited on
Commit
ad95b8d
·
verified ·
1 Parent(s): 34c7eeb

Update src/streamlit_app.py

Browse files
Files changed (1) hide show
  1. src/streamlit_app.py +1 -1
src/streamlit_app.py CHANGED
@@ -134,7 +134,7 @@ if user_input := st.chat_input("Your message:"):
134
  input_for_graph = {"messages": [HumanMessage(content=user_input)]}
135
 
136
  # Use stream_mode='values' to get outputs from each node as they run
137
- for chunk in workflow.stream(input_for_graph, config=set_config()):
138
  # Check for output from the LLM node
139
  if "llmresponse" in chunk:
140
  ai_message_chunk = chunk["llmresponse"]["messages"][-1]
 
134
  input_for_graph = {"messages": [HumanMessage(content=user_input)]}
135
 
136
  # Use stream_mode='values' to get outputs from each node as they run
137
+ for chunk in workflow.stream(input_for_graph, config=set_config() , stream_mode = "messages"):
138
  # Check for output from the LLM node
139
  if "llmresponse" in chunk:
140
  ai_message_chunk = chunk["llmresponse"]["messages"][-1]