From a90535a68bb59886fd7679ed94ab0ff9a1d93ebd Mon Sep 17 00:00:00 2001 From: Patryk Date: Mon, 10 Jun 2024 00:19:13 +0200 Subject: [PATCH] =?UTF-8?q?Dodanie=20stan=C3=B3w=20i=20ogranicze=C5=84=20d?= =?UTF-8?q?o=20DSM?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- attributes.json | 11 ------ src/service/dialog_state_monitor.py | 57 +++++++++++++++++++++++------ src/test/dialog_state_monitor.py | 39 +++++++++++++------- 3 files changed, 70 insertions(+), 37 deletions(-) diff --git a/attributes.json b/attributes.json index 969da03..85dc730 100644 --- a/attributes.json +++ b/attributes.json @@ -36,16 +36,5 @@ "ingredient": ["tomato", "tuna", "onion", "cheese"], "price": 40 } - }, - "size": { - "m": { - "price_multiplier": 1 - }, - "l": { - "price_multiplier": 1.2 - }, - "xl": { - "price_multiplier": 1.4 - } } } \ No newline at end of file diff --git a/src/service/dialog_state_monitor.py b/src/service/dialog_state_monitor.py index b01e808..abebbea 100644 --- a/src/service/dialog_state_monitor.py +++ b/src/service/dialog_state_monitor.py @@ -10,19 +10,52 @@ def normalize(value): class DialogStateMonitor: def __init__(self): self.__initial_state = dict(belief_state={ - 'order': [], - 'address': {}, - 'order-complete': False, - 'phone': {}, - 'delivery': {}, - 'payment': {}, - 'time': {}, - 'name': {}, - }, - history=[]) + 'order': [], + 'address': {}, + 'order-complete': False, + 'phone': {}, + 'delivery': {}, + 'payment': {}, + 'time': {}, + 'name': {}, + }, + stages=[ + {'completed': False, 'name': 'collect_food'}, + {'completed': False, 'name': 'collect_drinks'}, + {'completed': False, 'name': 'collect_address'}, + ], + constraints={ + 'order': [ + 'sauce', + 'pizza', + ], + }, + history=[]) self.state = copy.deepcopy(self.__initial_state) - def update(self, frame: Frame): + def get_last_order_missing_fields(self) -> list[str]: + result = [] + try: + last_order = self.state['belief_state']['order'][-1] + except IndexError: + raise RuntimeError('No orders are placed') + for constraint in self.state['constraints']['order']: + if constraint not in last_order: + result.append(constraint) + return result + + def get_current_active_stage(self) -> str | None: + for stage in self.state['stages']: + if stage['completed'] is False: + return stage['name'] + + def mark_current_stage_completed(self) -> None: + for stage in self.state['stages']: + if stage['completed'] is False: + stage['completed'] = True + return + + def update(self, frame: Frame) -> None: self.state['history'].append(frame) if frame.source != 'user': return @@ -52,5 +85,5 @@ class DialogStateMonitor: for slot in frame.slots: self.state['belief_state']['name'][slot.name] = normalize(slot.value) - def reset(self): + def reset(self) -> None: self.state = copy.deepcopy(self.__initial_state) diff --git a/src/test/dialog_state_monitor.py b/src/test/dialog_state_monitor.py index 8614126..f6509d4 100644 --- a/src/test/dialog_state_monitor.py +++ b/src/test/dialog_state_monitor.py @@ -2,24 +2,35 @@ from src.service.dialog_state_monitor import DialogStateMonitor from src.model.frame import Frame from src.model.slot import Slot -dst = DialogStateMonitor() +dsm = DialogStateMonitor() frame1 = Frame('user', 'inform/order', [Slot('pizza', 'margaritta'), Slot('sauce', 'ketchup')]) +dsm.update(frame1) +assert dsm.get_last_order_missing_fields() == [] frame2 = Frame('user', 'inform/order', [Slot('pizza', 'carbonara')]) +dsm.update(frame2) +assert dsm.get_last_order_missing_fields() == ['sauce'] frame3 = Frame('user', 'inform/order-complete', []) -dst.update(frame1) -dst.update(frame2) -dst.update(frame3) +dsm.update(frame3) -assert dst.state['belief_state']['order'][0]['pizza'] == 'margaritta' -assert dst.state['belief_state']['order'][0]['sauce'] == 'ketchup' -assert dst.state['belief_state']['order-complete'] is True -assert dst.state['history'][0] == frame1 -assert dst.state['history'][1] == frame2 -assert dst.state['history'][2] == frame3 +assert dsm.state['belief_state']['order'][0]['pizza'] == 'margaritta' +assert dsm.state['belief_state']['order'][0]['sauce'] == 'ketchup' +assert dsm.state['belief_state']['order-complete'] is True +assert dsm.state['history'][0] == frame1 +assert dsm.state['history'][1] == frame2 +assert dsm.state['history'][2] == frame3 -dst.reset() +assert dsm.get_current_active_stage() == 'collect_food' +dsm.mark_current_stage_completed() +assert dsm.get_current_active_stage() == 'collect_drinks' +dsm.mark_current_stage_completed() +assert dsm.get_current_active_stage() == 'collect_address' +dsm.mark_current_stage_completed() +assert dsm.get_current_active_stage() is None -assert dst.state['belief_state']['order'] == [] -assert dst.state['belief_state']['order-complete'] is False -assert len(dst.state['history']) == 0 +dsm.reset() + +assert dsm.get_current_active_stage() == 'collect_food' +assert dsm.state['belief_state']['order'] == [] +assert dsm.state['belief_state']['order-complete'] is False +assert len(dsm.state['history']) == 0