Update WritePlanFlow.yaml

WritePlanFlow.yaml

@@ -1,44 +1,45 @@
 name: "WritePlan"
 description: "Given goal and plan file location, generate plans to achieve the goal and write the plan to the file"
 
-input_interface:
-  - "goal"
-  - "file_location"
-
-output_interface:
-  - "plan_writer_output"
-
-subflows_config:
-  PlanGenerator:
-    _target_: Tachi67.WritePlanFlowModule.PlanGeneratorAtomicFlow.instantiate_from_default_config
-    backend:
-      api_infos: ???
-      model_name:
-        openai: gpt-4
-        azure: azure/gpt-4
-
-  PlanFileEditor:
-    _target_: Tachi67.WritePlanFlowModule.PlanFileEditorAtomicFlow.instantiate_from_default_config
-
-early_exit_key: "EARLY_EXIT"
-
-topology:
-  - goal: "Generate plans for the programming executor to achieve the task."
-    input_interface:
-      _target_: flows.interfaces.KeyInterface
-      additional_transformations:
-        - _target_: flows.data_transformations.KeyMatchInput
-    flow: PlanGenerator
-    output_interface:
-      _target_: Tachi67.WritePlanFlowModule.WritePlanFlow.detect_finish_or_continue
-    reset: false
-
-  - goal: "Write the plan to the specified file location"
-    input_interface:
-      _target_: flows.interfaces.KeyInterface
-      keys_to_select: ["file_location", "plan"]
-    flow: PlanFileEditor
-    output_interface:
-      _target_: flows.interfaces.KeyInterface
-      keys_to_select: ["plan_writer_output"]
-    reset: false
+flow:
+  _target_: Tachi67.WritePlanFlowModule.WritePlanFlow.instantiate_from_default_config
+
+  input_interface:
+    - "goal"
+    - "file_location"
+
+  output_interface:
+    - "plan_writer_output"
+
+  subflows_config:
+    PlanGenerator:
+      _target_: Tachi67.WritePlanFlowModule.PlanGeneratorAtomicFlow.instantiate_from_default_config
+      backend:
+        api_infos: ???
+        model_name:
+          openai: gpt-4
+          azure: azure/gpt-4
+
+    PlanFileEditor:
+      _target_: Tachi67.WritePlanFlowModule.PlanFileEditorAtomicFlow.instantiate_from_default_config
+
+  early_exit_key: "EARLY_EXIT"
+
+  topology:
+    - goal: "Generate plans for the programming executor to achieve the task."
+      input_interface:
+        _target_: flows.interfaces.KeyInterface
+        additional_transformations:
+          - _target_: flows.data_transformations.KeyMatchInput
+      flow: PlanGenerator
+      reset: false
+
+    - goal: "Write the plan to the specified file location"
+      input_interface:
+        _target_: flows.interfaces.KeyInterface
+        keys_to_select: ["file_location", "plan"]
+      flow: PlanFileEditor
+      output_interface:
+        _target_: flows.interfaces.KeyInterface
+        keys_to_select: ["plan_writer_output"]
+      reset: false