Update PlanFileEditAtomicFlow.py

PlanFileEditAtomicFlow.py

@@ -38,7 +38,7 @@ class PlanFileEditAtomicFlow(AtomicFlow):
             return False, str(e), file_location, 0
 
     def _check_input(self, input_data: Dict[str, Any]):
-        assert "plan" in input_data, "plan is not passed to PlanFileEditAtomicFlow"
+        assert any(item in input_data for item in ["plan", "new_plan"]) in input_data, "plan or new_plan is not passed to PlanFileEditAtomicFlow"
         assert "plan_file_location" in input_data, "plan_file_location not passed to PlanFileEditAtomicFlow"
         plan_file_loc = input_data["plan_file_location"]
         assert os.path.exists(plan_file_loc), f"{plan_file_loc} does not exist"