Spaces:
Sleeping
Sleeping
Commit
·
7fb95cb
1
Parent(s):
6178c40
Updated nodes.py
Browse files- 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
|