Update agent.py

agent.py

@@ -177,7 +177,7 @@ if __name__ == "__main__":
         print(f"\n--- Testing Question {i+1}: {question}")
         try:
             # LangGraph returns the final state after execution completes or hits recursion limit
-            final_state = graph.invoke({"messages": [HumanMessage(content=question)]})
+            final_state = graph.invoke({"messages": [SystemMessage(content=system_prompt), HumanMessage(content=question)]})
             print("\n--- Final State Messages ---")
             for m in final_state["messages"]:
                  m.pretty_print()