Update InterpreterAtomicFlow.py

InterpreterAtomicFlow.py

@@ -45,17 +45,19 @@ class InterpreterAtomicFlow(AtomicFlow):
         self.flow_state["language"] = language
         self.flow_state["code"] = code
 
-    def _process_input_data(self, input_data: Dict[str, Any]):
+    def _check_input(self, input_data: Dict[str, Any]):
         """
-        sanity check for input, allocate interpreter if any,
-        pass input data into flow state
-
-        should call something like self._update_flow_state()
+        Sanity check of input data
         """
-        # ~~~ Sanity check of input_data ~~~
+         # ~~~ Sanity check of input_data ~~~
         assert "language" in input_data, "attribute 'language' not in input data."
         assert "code" in input_data, "attribute 'code' not in input data."
 
+
+    def _process_input_data(self, input_data: Dict[str, Any]):
+        """
+        Allocate interpreter if any, pass input data into flow state
+        """
         # code in Jupyter notebook that starts with '!' is actually shell command.
         if input_data["language"] == "python" and input_data["code"].startswith("!"):
             input_data["language"] = "shell"
@@ -93,6 +95,7 @@ class InterpreterAtomicFlow(AtomicFlow):
     def run(
             self,
             input_data: Dict[str, Any]):
+        self._check_input(input_data)
         self._process_input_data(input_data)
         response = self._call()
         return {"interpreter_output": response}