added more output

app.py

@@ -58,7 +58,8 @@ def create_agent(llm: ChatOpenAI, tools: list, task: str):
 # AIMessage will have all kinds of metadata, so treat it all as HumanMessage I suppose?
 def agent_node(state: AgentState, agent, name):
     result = agent.invoke(state)
-    return {"messages": [HumanMessage(content=result["output"], name=name)]}
+    # return {"messages": [HumanMessage(content=result["output"], name=name)]}
+    return {"messages": [result]}
 
 # I need to write the message to state here? or is that handled automatically?
 def chain_node(state: AgentState, chain, name):