Incorporate feedback on PR.

OpenAIChatAtomicFlow.py

@@ -117,12 +117,12 @@ class OpenAIChatAtomicFlow(AtomicFlow):
 
         return False
 
-    def get_expected_inputs(self, data: Optional[Dict[str, Any]] = None):
+    def get_input_keys(self, data: Optional[Dict[str, Any]] = None):
         """Returns the expected inputs for the flow given the current state and, optionally, the input data"""
         if self._is_conversation_initialized():
             return ["query"]
         else:
-            return self.flow_config["expected_inputs"]
+            return self.flow_config["input_keys"]
 
     @staticmethod
     def _get_message(prompt_template, input_data: Dict[str, Any]):
@@ -146,22 +146,22 @@ class OpenAIChatAtomicFlow(AtomicFlow):
             if ra.key == key:
                 return ra
 
-    def _response_parsing(self, response: str, expected_outputs: List[str]):
-        target_annotators = [ra for _, ra in self.response_annotators.items() if ra.key in expected_outputs]
+    def _response_parsing(self, response: str, output_keys: List[str]):
+        target_annotators = [ra for _, ra in self.response_annotators.items() if ra.key in output_keys]
 
         parsed_outputs = {}
         for ra in target_annotators:
             parsed_out = ra(response)
             parsed_outputs.update(parsed_out)
 
-        if "raw_response" in expected_outputs:
+        if "raw_response" in output_keys:
             parsed_outputs["raw_response"] = response
         else:
             log.warning("The raw response is not logged because it was not requested as per the expected output.")
 
         if len(parsed_outputs) == 0:
             raise Exception(f"The output dictionary is empty. "
-                            f"None of the expected outputs: `{str(expected_outputs)}` were found.")
+                            f"None of the expected outputs: `{str(output_keys)}` were found.")
 
         return parsed_outputs
 
@@ -290,7 +290,7 @@ class OpenAIChatAtomicFlow(AtomicFlow):
         # ~~~ Response parsing ~~~
         output_data = self._response_parsing(
             response=response,
-            expected_outputs=input_data["expected_outputs"]
+            output_keys=input_data["output_keys"]
         )
         # self._state_update_dict(update_data=output_data)  # ToDo: Is this necessary? When?