Add package_input_message

ChatHumanFlowModule.py

@@ -107,12 +107,13 @@ class ChatHumanFlowModule(CompositeFlow):
         :type input_message: FlowMessage
         """
         self.flow_state["assistant_outputs"].append(input_message.data["api_output"])
+        message = self.package_input_message(data=input_message.data)
 
         if self.max_rounds_reached():
             self.generate_reply()
         else:
             self.subflows["User"].get_reply(
-                input_message,
+                message,
             )
             self.flow_state["last_state"] = "User"
         
@@ -127,6 +128,7 @@ class ChatHumanFlowModule(CompositeFlow):
         :type input_message: FlowMessage
         """
         message = self.input_interface_assistant(input_message)
+        message = self.package_input_message(data=message.data)
         
         if  self.flow_state["last_state"] is None:
             self.flow_state["input_message"] = input_message

ChatHumanFlowModule.yaml

@@ -25,12 +25,14 @@ end_of_interaction:
 
 subflows_config:
   Assistant:
-    _target_: flow_modules.aiflows.ChatFlowModule.ChatAtomicFlow.instantiate_from_default_config
+    name: "Assistant"
+    description: "A flow that represents the assistant."
     flow_class_name: flow_modules.aiflows.ChatFlowModule.ChatAtomicFlow
     flow_endpoint: Assistant
     user_id: local
   User:
-    _target_: flow_modules.aiflows.HumanStandardInputFlowModule.HumanStandardInputFlow.instantiate_from_default_config
+    name: "User"
+    description: "A flow that represents the user."
     flow_class_name: flow_modules.aiflows.HumanStandardInputFlowModule.HumanStandardInputFlow
     flow_endpoint: User
     user_id: local

demo.yaml

@@ -1,5 +1,5 @@
 
-max_rounds: 2
+max_rounds: 20
 _target_: flow_modules.aiflows.ChatInteractiveFlowModule.ChatHumanFlowModule.instantiate_from_default_config
 subflows_config:
   Assistant:
@@ -34,4 +34,4 @@ subflows_config:
     input_interface:
       - "api_output"
     output_interface:
-      - "human_input"
+      - "human_input"