Upload 8 files

PlanWriterFlow.py

@@ -17,11 +17,11 @@ class PlanWriterFlow(ContentWriterFlow):
         command = output_payload["command"]
         if command == "finish":
             # ~~~ fetch temp file location, plan content, memory file (of upper level flow e.g. ExtLib) from flow state
-            keys_to_fetch_from_state = ["temp_plan_file_location", "plan", "plan_file_location"]
+            keys_to_fetch_from_state = ["temp_plan_file_location", "plan", "memory_files"]
             fetched_state = self._fetch_state_attributes_by_keys(keys=keys_to_fetch_from_state)
             temp_plan_file_location = fetched_state["temp_plan_file_location"]
             plan_content = fetched_state["plan"]
-            plan_file_location = fetched_state["plan_file_location"]
+            plan_file_location = fetched_state["memory_files"]["plan"]
 
             # ~~~ delete the temp plan file ~~~
             if os.path.exists(temp_plan_file_location):
@@ -35,7 +35,7 @@ class PlanWriterFlow(ContentWriterFlow):
             return {
                 "EARLY_EXIT": True,
                 "plan": plan_content,
-                "summary": "PlanWriter: " + output_payload["command_args"]["summary"],
+                "summary": "ExtendLibrary/PlanWriter: " + output_payload["command_args"]["summary"],
                 "status": "finished"
             }
         elif command == "manual_finish":
@@ -49,13 +49,13 @@ class PlanWriterFlow(ContentWriterFlow):
             return {
                 "EARLY_EXIT": True,
                 "plan": "no plan was generated",
-                "summary": "PlanWriter: PlanWriter was terminated explicitly by the user, process is unfinished",
+                "summary": "ExtendLibrary/PlanWriter: PlanWriter was terminated explicitly by the user, process is unfinished",
                 "status": "unfinished"
             }
         elif command == "write_plan":
-            keys_to_fetch_from_state = ["plan_file_location"]
+            keys_to_fetch_from_state = ["memory_files"]
             fetched_state = self._fetch_state_attributes_by_keys(keys=keys_to_fetch_from_state)
-            plan_file_location = fetched_state["plan_file_location"]
+            plan_file_location = fetched_state["memory_files"]["plan"]
             output_payload["command_args"]["plan_file_location"] = plan_file_location
             return output_payload
         else:

PlanWriterFlow.yaml

@@ -5,7 +5,7 @@ _target_: Tachi67.PlanWriterFlowModule.PlanWriterFlow.instantiate_from_default_c
 
 input_interface:
   - "goal"
-  - "plan_file_location"
+  - "memory_files"
 
 output_interface:
   - "plan"

run_planwriter.py

@@ -50,9 +50,11 @@ if __name__ == "__main__":
     with open(plan_file_location, 'w') as file:
         pass
 
+    mem_files = {"plan": plan_file_location}
+
     input_data = {
         "goal": "create a function that adds two numbers and returns the result",
-        "plan_file_location": plan_file_location
+        "memory_files": mem_files
     }
     input_message = InputMessage.build(
         data_dict=input_data,