Spaces:
Sleeping
Sleeping
Update src/streamlit_app.py
Browse files- 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]
|