Update agent.py

agent.py

@@ -40,7 +40,7 @@ def simple_graph():
     ## Defining our nodes
     def assistant(state: MessagesState):
         """Assistant node"""
-        return {"messages": [sys_msg] + [chat_with_tools.invoke(state["messages"])]}
+        return {"messages": [state["messages"]] + [chat_with_tools.invoke(state["messages"])]}
     
     # Build graph / nodes
     builder = StateGraph(MessagesState)