Upload 5 files

ContentWriterFlow.yaml

@@ -0,0 +1,72 @@
+# Abstract class, should provide more config in subclasses
+
+
+_target_: flow_modules.aiflows.ContentWriterFlowModule.ContentWriterFlow.instantiate_from_default_config
+name: "ContentWriter"
+description: "Generated content, writes content to file and update content with user interaction"
+max_rounds: 30
+
+
+input_interface:
+  - "goal"
+output_interface:
+  - "answer"
+  - "status"
+
+### Subflows specification
+subflows_config:
+  Controller:
+    _target_: ??? # Should be a subclass of aiflows.ChatAtomicFlow or something similar
+    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_: aiflows.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_: aiflows.interfaces.KeyInterface
+      additional_transformations:
+        - _target_: aiflows.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_: aiflows.interfaces.KeyInterface
+      keys_to_rename:
+        command: branch
+        command_args: branch_input_data
+      keys_to_select: ["branch", "branch_input_data"]
+    flow: Executor
+    output_interface:
+      _target_: aiflows.interfaces.KeyInterface
+      # in subclasses, should provide corresponding output name of the branching subflows
+      keys_to_rename:
+        branch_output_data.code: code
+        branch_output_data.feedback: feedback
+        branch_output_data.temp_code_file_location: temp_code_file_location
+      keys_to_delete: ["branch_output_data"]
+    reset: false

ContentWritrerFlow.py

@@ -0,0 +1,51 @@
+from typing import Dict, Any
+
+from aiflows.base_flows import CircularFlow
+from aiflows.utils import logging
+from abc import ABC, abstractmethod
+
+
+logging.set_verbosity_debug()
+log = logging.get_logger(__name__)
+
+
+class ContentWriterFlow(CircularFlow, ABC):
+    """This is an abstract class for writing content (plan, code)
+    The ContentWriterFlow is made of a controller and a branching executor.
+    Each time the controller is called, the controller decides whether to write content
+    or to finish. If the content writer executor is called, the executor will write content
+    in an interactive way, finally, the user is able to give feedback to the content, so that
+    the controller can decide whether to write content again or to finish.
+    
+    *Configuration Parameters*:
+    - `name`
+    - `description`
+    - `max_round`
+    - `subflows_config`:
+        - `Controller` (dict): The controller that decides whether to write content or to finish.
+        - `Executor` (dict): A branching flow, we configure the specific executor in the subflows of the executor.
+    - `early_exit_key`: The key of the early exit variable in the output payload of the executor.
+    - `topology`: The topology of the subflows, this describes the I/O interface instances.
+    
+    *Input Interface*:
+    - `goal`
+    
+    *Output Interface*:
+    - `answer`
+    - `status`
+    """
+    @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

README.md

@@ -1,3 +1,89 @@
----
-license: mit
----
+### Structure of ContentWriterFlow
+
+```
+               goal
+                |
+                v
+        +---------------+
+        |  Controller   | --------<<<<-----------+
+        +---------------+                        |
+                |                                |
+                | (command, command args)        |
+                |                                |
+                v                                |
+        +------------------+                     |
+        |   Executor       |  Each branch is an  |
+        | (Tree Structure) |  executor           |
+        +------------------+                     |
+                |                                ^
+                | (summary)                      |
+                |                                |
+                v                                |             
+                |                                |
+                +-> goes back to the Controller>-+
+
+```
+
+This is an abstract class. It is inherited by [CodeWriterFlow](https://huggingface.co/Tachi67/CodeWriterFlowModule) and [PlanWriterFlow](https://huggingface.co/Tachi67/PlanWriterFlowModule)
+
+# Table of Contents
+
+* [ContentWritrerFlow](#ContentWritrerFlow)
+  * [ContentWriterFlow](#ContentWritrerFlow.ContentWriterFlow)
+    * [detect\_finish\_or\_continue](#ContentWritrerFlow.ContentWriterFlow.detect_finish_or_continue)
+* [\_\_init\_\_](#__init__)
+
+<a id="ContentWritrerFlow"></a>
+
+# ContentWritrerFlow
+
+<a id="ContentWritrerFlow.ContentWriterFlow"></a>
+
+## ContentWriterFlow Objects
+
+```python
+class ContentWriterFlow(CircularFlow, ABC)
+```
+
+This is an abstract class for writing content (plan, code)
+The ContentWriterFlow is made of a controller and a branching executor.
+Each time the controller is called, the controller decides whether to write content
+or to finish. If the content writer executor is called, the executor will write content
+in an interactive way, finally, the user is able to give feedback to the content, so that
+the controller can decide whether to write content again or to finish.
+
+*Configuration Parameters*:
+- `name`
+- `description`
+- `max_round`
+- `subflows_config`:
+    - `Controller` (dict): The controller that decides whether to write content or to finish.
+    - `Executor` (dict): A branching flow, we configure the specific executor in the subflows of the executor.
+- `early_exit_key`: The key of the early exit variable in the output payload of the executor.
+- `topology`: The topology of the subflows, this describes the I/O interface instances.
+
+*Input Interface*:
+- `goal`
+
+*Output Interface*:
+- `answer`
+- `status`
+
+<a id="ContentWritrerFlow.ContentWriterFlow.detect_finish_or_continue"></a>
+
+#### detect\_finish\_or\_continue
+
+```python
+@abstractmethod
+@CircularFlow.output_msg_payload_processor
+def detect_finish_or_continue(output_payload: Dict[str, Any],
+                              src_flow) -> Dict[str, Any]
+```
+
+1. Writing content to file;
+2. Finish and early exit.
+
+<a id="__init__"></a>
+
+# \_\_init\_\_
+

__init__.py

@@ -0,0 +1,7 @@
+dependencies = [
+]
+from aiflows import flow_verse
+
+flow_verse.sync_dependencies(dependencies)
+
+from .ContentWritrerFlow import ContentWriterFlow

pip_requirements.txt

@@ -0,0 +1,15 @@
+colorama==0.4.6
+pytest==7.3.1
+pytest-cov==4.1.0
+hydra-core==1.3.2
+hydra-colorlog==1.1.0
+wrapt-timeout-decorator==1.3.12.2
+diskcache==5.6.1
+openai==1.0.0
+huggingface_hub==0.19.4
+jsonlines==3.1.0
+jinja2==3.1.2
+mock==5.0.2
+rich==12.6.0
+litellm==1.0.0
+aiflows