diff --git a/src/service/dialog_state_monitor.py b/src/service/dialog_state_monitor.py index 4b9d785..0282006 100644 --- a/src/service/dialog_state_monitor.py +++ b/src/service/dialog_state_monitor.py @@ -70,14 +70,18 @@ class DialogStateMonitor: if stage['name'] == "collect_food": for order in self.state['belief_state']['order']: - if len(order) > 0: - stage['completed'] = True - return + for o in order: + k, v = o + if k == 'pizza': + stage['completed'] = True + return elif stage["name"] == "collect_drinks": for order in self.state['belief_state']['order']: - if order.get("drink"): - stage['completed'] = True - return + for o in order: + k, v = o + if k == 'drink': + stage['completed'] = True + return elif stage["name"] == "collect_address": if not len(self.state['belief_state']['address']): return