Upload 3 files

ContentWriterFlow.yaml

@@ -0,0 +1,70 @@
+# Abstract class, should provide more config in subclasses
+name: "ContentWriter"
+description: "Generated content, writes content to file and update content with user interaction"
+max_rounds: 30
+
+
+input_interface:
+  - "goal"
+  # in subclass, should provide key: file_location e.g. plan_file_location, code_file_location
+output_interface:
+  - "answer"
+  - "status"
+
+### Subflows specification
+subflows_config:
+  Controller:
+    _target_: Tachi67.ControllerAtomicFlowModule.ControllerAtomicFlow.instantiate_from_default_config
+    finish:
+      description: "Signal that the objective has been satisfied, and returns the answer to the user."
+      input_args: ["answer"]
+    backend:
+      api_infos: ???
+      model_name:
+        openai: gpt-4
+        azure: azure/gpt-4
+    # In subclasses, should provide specific command and command args.
+# E.g.,
+#    commands:
+#      wiki_search:
+#        description: "Performs a search on Wikipedia."
+#        input_args: ["search_term"]
+
+
+  Executor:
+    _target_: flows.base_flows.BranchingFlow.instantiate_from_default_config
+    # In subclasses, should provide what are the branches of executors
+# E.g.,
+#    subflows_config:
+#      wiki_search:
+#        _target_: .WikiSearchAtomicFlow.instantiate_from_default_config
+
+
+early_exit_key: "EARLY_EXIT"
+
+topology:
+  - goal: "Select the next action and prepare the input for the executor."
+    input_interface:
+      _target_: flows.interfaces.KeyInterface
+      additional_transformations:
+        - _target_: flows.data_transformations.KeyMatchInput
+    flow: Controller
+    output_interface:
+      # In subclasses, should provide name of function name of output processor
+      #_target_: ControllerExecutorFlow.detect_finish_or_continue
+    reset: false
+
+  - goal: "Execute the action specified by the Controller."
+    input_interface:
+      _target_: flows.interfaces.KeyInterface
+      keys_to_rename:
+        command: branch
+        command_args: branch_input_data
+      keys_to_select: ["branch", "branch_input_data"]
+    flow: Executor
+    output_interface:
+      _target_: flows.interfaces.KeyInterface
+      keys_to_rename:
+        branch_output_data: observation
+      keys_to_select: ["observation"]
+    reset: false

ContentWritrerFlow.py

@@ -0,0 +1,27 @@
+from typing import Dict, Any
+
+from flows.base_flows import CircularFlow
+from flows.utils import logging
+from abc import ABC, abstractmethod
+
+
+logging.set_verbosity_debug()
+log = logging.get_logger(__name__)
+
+
+class ContentWriterFlow(CircularFlow, ABC):
+    @abstractmethod
+    def _on_reach_max_round(self):
+        """
+        should update flow state dictionary about the output variables and status.
+        """
+        pass
+
+    @abstractmethod
+    @CircularFlow.output_msg_payload_processor
+    def detect_finish_or_continue(self, output_payload: Dict[str, Any], src_flow) -> Dict[str, Any]:
+        """
+        1. Writing content to file;
+        2. Finish and early exit.
+        """
+        pass

__init__.py

@@ -0,0 +1,8 @@
+dependencies = [
+    {"url": "Tachi67/ControllerAtomicFlowModule", "revision": "main"},
+]
+from flows import flow_verse
+
+flow_verse.sync_dependencies(dependencies)
+
+from .ContentWritrerFlow import ContentWriterFlow