Update agent.py

agent.py

@@ -49,8 +49,9 @@ def build_agent():
                             Your answer should only start with "FINAL ANSWER: ", then follows with the answer.
                         """
         return {
-            "messages": [sys_msg] + state["messages"]
-        }
+                "messages": [{"role": "system", "content": sys_msg}] + state["messages"],
+                "wiki_table": state.get("wiki_table")
+            }
 
     ## The graph
     builder = StateGraph(AgentState)