devchavda11 commited on
Commit
ebb618f
·
verified ·
1 Parent(s): 7791c9a

Update src/chat_langraph.py

Browse files
Files changed (1) hide show
  1. src/chat_langraph.py +1 -2
src/chat_langraph.py CHANGED
@@ -220,9 +220,8 @@ def input_node(state: chatstate):
220
 
221
  def llmresponse(state: chatstate):
222
  response = llm.stream(state["messages"])
223
- for chunk in response:
224
 
225
- yield {"messages": [chunk]};
226
 
227
 
228
  def checktool(state: chatstate):
 
220
 
221
  def llmresponse(state: chatstate):
222
  response = llm.stream(state["messages"])
 
223
 
224
+ return {"messages": [response]};
225
 
226
 
227
  def checktool(state: chatstate):