subashpoudel commited on
Commit
7fb95cb
·
1 Parent(s): 6178c40

Updated nodes.py

Browse files
Files changed (1) hide show
  1. my_agent/utils/nodes.py +1 -1
my_agent/utils/nodes.py CHANGED
@@ -100,7 +100,7 @@ def generate_story(state:State)-> State:
100
  messages = [SystemMessage(content=template)]
101
 
102
  response = react_agent.invoke({'messages':messages})
103
- response = response['messages'][-1]
104
  state.stories.append(response)
105
  # return State(messages="Story generated", topic=state.topic,stories=state.stories)
106
  return state
 
100
  messages = [SystemMessage(content=template)]
101
 
102
  response = react_agent.invoke({'messages':messages})
103
+ response = response['messages'][-1].content
104
  state.stories.append(response)
105
  # return State(messages="Story generated", topic=state.topic,stories=state.stories)
106
  return state