Upload 3 files

PlanFileEditAtomicFlow.py

@@ -14,10 +14,10 @@ class PlanFileEditAtomicFlow(AtomicFlow):
                 "# Please do provide your feedback in the thoughts section so that JARVIS knows what you "
                 "are doing.\n"
                 "###########\n"
-                "# Plan:\n" +
+                "Plan:\n" +
                 plan_str +
                 "\n############\n"
-                "# Thoughts:"
+                "Thoughts:"
         )
         return content
 
@@ -39,15 +39,14 @@ class PlanFileEditAtomicFlow(AtomicFlow):
 
     def _check_input(self, input_data: Dict[str, Any]):
         assert "plan" in input_data, "plan is not passed to PlanFileEditAtomicFlow"
-        assert "memory_files" in input_data, "memory_files is not passed to PlanFileEditAtomicFlow"
-        assert "plan" in input_data["memory_files"], "plan not in memory files"
-        plan_file_loc = input_data["memory_files"]["plan"]
+        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"
         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']
-        plan_file_location = input_data["memory_files"]["plan"]
+        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)
         return content_to_write, file_location_to_write

PlanFileEditAtomicFlow.yaml

@@ -3,7 +3,7 @@ description: "A flow that writes plan to a temp file"
 
 input_interface:
   - "plan"
-  - "memory_files" # e.g. in Planner of ExtLib, it takes the memory_files of ExtLib flow
+  - "plan_file_location"
 
 output_interface:
   - "plan_editor_output"