Update agent.py

agent.py

@@ -48,9 +48,9 @@ def build_agent():
                             If you are asked for a comma separated list, apply the above rules depending of whether the element to be put in the list is a number or a string.
                             Your answer should only start with "FINAL ANSWER: ", then follows with the answer.
                         """
-        context = {"role": "system", "content": sys_msg}
-        
-        return context + state["messages"]
+        return {
+            "messages": [sys_msg] + state["messages"]
+        }
 
     ## The graph
     builder = StateGraph(AgentState)