Update PlanFileEditAtomicFlow.py

PlanFileEditAtomicFlow.py

@@ -43,7 +43,23 @@ class PlanFileEditAtomicFlow(AtomicFlow):
         assert os.path.isfile(plan_file_loc), f"{plan_file_loc} is not a file"
 
     def _generate_input_to_writer(self, input_data: Dict[str, Any]):
-        plan_str = input_data['plan'] if "plan" in input_data else input_data['new_plan']
+        """
+        sometimes the plan generator will return an arrary of indexed plans, like
+        [
+            "1. Extend the code library with a function named 'import_libraries'. This function should import necessary libraries for the task...",
+            "2. Extend the code library with a function named 'fetch_stock_prices'. This function should take two inputs: 'company_code' and 'duration'...",
+            "3. Investigate the issue with importing the 'fetch_stock_prices' function from the library..."
+        ]
+        In this case we need to prase this format accorrdingly.
+        """
+        
+        plan = input_data['plan'] if "plan" in input_data else input_data['new_plan']
+        if isinstance(plan, str):
+            plan_str = plan
+        elif isinstance(plan, list):
+            plan_str = "\n".join(plan)
+        else:
+             raise TypeError("plan is neither a string nor a list")
         plan_file_location = input_data["plan_file_location"]
         content_to_write = self._generate_content(plan_file_location, plan_str)
         file_location_to_write = self._generate_temp_file_location(plan_file_location)