Clean Versions

InterpreterAtomicFlow.py

@@ -42,26 +42,13 @@ class InterpreterAtomicFlow(AtomicFlow):
     is not compatible with that of the current version of aiflows(v.0.1.7).
     """
     def __init__(self,
-                 max_output=2000,
                  **kwargs):
         super().__init__(**kwargs)
-        self.max_output = max_output
+        self.max_output = self.flow_config["max_output"]
         self._code_interpreters = {}
 
-    @classmethod
-    def instantiate_from_config(cls, config):
-        flow_config = deepcopy(config)
-
-        kwargs = {"flow_config": flow_config}
-
-        # ~~~ Instantiate flow ~~~
-        return cls(**kwargs)
-
-
     def set_up_flow_state(self):
-        """
-        class-specific flow state: language and code,
-        which describes the programming language and the code to run.
+        """ class-specific flow state: language and code, which describes the programming language and the code to run.
         """
         super().set_up_flow_state()
         self.flow_state["language"] = None
@@ -73,13 +60,16 @@ class InterpreterAtomicFlow(AtomicFlow):
         """
         updates the language and code passed from _process_input_data
         to the flow state
+        :param language: the programming language
+        :param code: the code to run
         """
         self.flow_state["language"] = language
         self.flow_state["code"] = code
 
     def _check_input(self, input_data: Dict[str, Any]):
-        """
-        Sanity check of input data
+        """ Sanity check of input data
+        :param input_data: input data
+        :type input_data: Dict[str, Any]
         """
          # ~~~ Sanity check of input_data ~~~
         assert "language" in input_data, "attribute 'language' not in input data."
@@ -87,8 +77,9 @@ class InterpreterAtomicFlow(AtomicFlow):
 
 
     def _process_input_data(self, input_data: Dict[str, Any]):
-        """
-        Allocate interpreter if any, pass input data into flow state
+        """ Allocate interpreter if any, pass input data into flow state
+        :param input_data: input data
+        :type input_data: Dict[str, Any]
         """
         # code in Jupyter notebook that starts with '!' is actually shell command.
         if input_data["language"] == "python" and input_data["code"].startswith("!"):
@@ -109,6 +100,8 @@ class InterpreterAtomicFlow(AtomicFlow):
         )
 
     def _call(self):
+        """ This method runs the code interpreter and returns the output. (runs the code interpreter and returns the output.)
+        """
         output = ""
         try:
             code_interpreter = self._code_interpreters[self.flow_state["language"]]
@@ -127,7 +120,10 @@ class InterpreterAtomicFlow(AtomicFlow):
     def run(
             self,
             input_message: FlowMessage):
-        
+        """ Run the code interpreter and return the output.
+        :param input_message: The input message of the flow.
+        :type input_message: FlowMessage
+        """
         input_data = input_message.data
         self._check_input(input_data)
         self._process_input_data(input_data)

InterpreterAtomicFlow.yaml

@@ -1,3 +1,4 @@
 name: "InterpreterAtomicFlow"
 description: "A flow that compiles and runs codes"
-_target_: flow_modules.aiflows.InterpreterFlowModule.InterpreterAtomicFlow.instantiate_from_default_config
+_target_: flow_modules.aiflows.InterpreterFlowModule.InterpreterAtomicFlow.instantiate_from_default_config
+max_output: 2000

run.py

@@ -59,6 +59,7 @@ if __name__ == "__main__":
         cl=cl,
         flow_endpoint="InterpreterAtomicFlow",
         user_id="local",
+        config_overrides= cfg
     )
     
     #6. ~~~ Get the data ~~~