diff --git a/src/service/dialog_state_monitor.py b/src/service/dialog_state_monitor.py index 1bdbeea..91f7a59 100644 --- a/src/service/dialog_state_monitor.py +++ b/src/service/dialog_state_monitor.py @@ -147,6 +147,13 @@ class DialogStateMonitor: new_order[slot.name].append(value) else: new_order[slot.name] = [value] + if slot.name == "drink": + if new_order.get("drink"): + new_order[slot.name].append(value) + else: + new_order[slot.name] = [value] + else: + new_order[slot.name] = value if len(new_order) > 0: self.state['belief_state']['order'].append(new_order) self.complete_stage_if_valid('collect_food')