Zmiana staga po dodaniu określonego wpisu do DSM

This commit is contained in:
Patryk Osiński 2024-06-10 23:35:45 +02:00
parent dfed26e955
commit 5f128e7e88
2 changed files with 11 additions and 13 deletions

View File

@ -1,4 +1,4 @@
from model.frame import Frame
from src.model.frame import Frame
import copy
import json
@ -62,17 +62,19 @@ class DialogStateMonitor:
new_order = dict()
for slot in frame.slots:
value = normalize(slot.value)
if slot.name == 'pizza' or slot.name == 'drink':
if (slot.name == 'pizza' and self.get_current_active_stage() == 'collect_food') or (slot.name == 'drink' and self.get_current_active_stage() == 'collect_drinks'):
if self.item_exists(slot.name, value) is False:
self.state['was_previous_order_invalid'] = True
return
self.state['was_previous_order_invalid'] = False
self.state['total_cost'] += self.state['constants'][slot.name][value]['price']
self.mark_current_stage_completed()
new_order[slot.name] = value
self.state['belief_state']['order'].append(new_order)
elif frame.act == 'inform/address':
elif frame.act == 'inform/address' and self.get_current_active_stage() == 'collect_address':
for slot in frame.slots:
self.state['belief_state']['address'][slot.name] = normalize(slot.value)
self.mark_current_stage_completed()
elif frame.act == 'inform/phone':
for slot in frame.slots:
self.state['belief_state']['phone'][slot.name] = normalize(slot.value)

View File

@ -9,17 +9,21 @@ assert dsm.item_exists('pizza', 'buraczana') is False
assert dsm.item_exists('drink', 'cola') is True
assert dsm.state['was_previous_order_invalid'] is False
assert dsm.get_current_active_stage() == 'collect_food'
frame1 = Frame('user', 'inform/order', [Slot('pizza', 'margarita'), Slot('sauce', 'ketchup')])
dsm.update(frame1)
assert dsm.get_current_active_stage() == 'collect_drinks'
assert dsm.get_total_cost() == 20
frame2 = Frame('user', 'inform/order', [Slot('pizza', 'tuna')])
dsm.update(frame2)
assert dsm.get_total_cost() == 60
assert dsm.get_current_active_stage() == 'collect_drinks'
assert dsm.get_total_cost() == 20 # Pizza is not added, as previous stage is closed already
frame3 = Frame('user', 'inform/order-complete', [])
dsm.update(frame3)
frame4 = Frame('user', 'inform/order', [Slot('drink', 'cola')])
dsm.update(frame4)
assert dsm.get_total_cost() == 70
assert dsm.get_current_active_stage() == 'collect_address'
assert dsm.get_total_cost() == 30
assert dsm.state['belief_state']['order'][0]['pizza'] == 'margarita'
assert dsm.state['belief_state']['order'][0]['sauce'] == 'ketchup'
@ -29,14 +33,6 @@ assert dsm.state['history'][1] == frame2
assert dsm.state['history'][2] == frame3
assert dsm.state['history'][3] == frame4
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
dsm.reset()
assert dsm.get_total_cost() == 0