From cdcf0ebfc9d6f59c1d7595485a722315d475c871 Mon Sep 17 00:00:00 2001
From: kb <krzboj@st.amu.edu.pl>
Date: Thu, 13 Jun 2024 15:56:23 +0200
Subject: [PATCH] Tests

---
 src/service/dialog_state_monitor.py | 14 ++++++++------
 1 file changed, 8 insertions(+), 6 deletions(-)

diff --git a/src/service/dialog_state_monitor.py b/src/service/dialog_state_monitor.py
index 4b9d785..9f7ce74 100644
--- a/src/service/dialog_state_monitor.py
+++ b/src/service/dialog_state_monitor.py
@@ -70,14 +70,16 @@ class DialogStateMonitor:
 
             if stage['name'] == "collect_food":
                 for order in self.state['belief_state']['order']:
-                    if len(order) > 0:
-                        stage['completed'] = True
-                        return
+                    for k, v in order:
+                        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 k, v in order:
+                        if k == 'drink':
+                            stage['completed'] = True
+                            return
             elif stage["name"] == "collect_address":
                 if not len(self.state['belief_state']['address']):
                     return