Add support for multiple human_message input keys.

OpenAIChatAtomicFlow.py

@@ -103,7 +103,7 @@ class OpenAIChatAtomicFlow(AtomicFlow):
     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 [self.flow_config["default_human_input_key"]]
+            return self.flow_config["default_human_input_keys"]
         else:
             return self.flow_config["input_keys"]
 

OpenAIChatAtomicFlow.yaml

@@ -34,7 +34,8 @@ human_message_prompt_template:
   input_variables:
     - "query"
   template_format: jinja2
-default_human_input_key: "query"
+default_human_input_keys:
+  - "query"
 input_data_transformations: []
 
 query_message_prompt_template: