Compare commits

...

2 Commits

Author SHA1 Message Date
kb
ad0b4cf739 Merge branch 'master' of https://git.wmi.amu.edu.pl/s495726/chat-restaruacja 2024-06-13 15:50:18 +02:00
kb
3fb60acd4f Test 2024-06-13 15:49:31 +02:00

View File

@ -34,12 +34,12 @@ class DialogStateMonitor:
},
total_cost=0,
stages=[
{'completed': False, 'name': 'collect_food', "confirmed": False},
# {'completed': False, 'name': 'more_food', "confirmed": False},
{'completed': False, 'name': 'collect_drinks', "confirmed": False},
# {'completed': False, 'name': 'more_drinks', "confirmed": False},
{'completed': False, 'name': 'collect_address', "confirmed": False},
{'completed': False, 'name': 'collect_phone', "confirmed": False},
{'completed': False, 'name': 'collect_food'},
{'completed': False, 'name': 'more_food'},
{'completed': False, 'name': 'collect_drinks'},
{'completed': False, 'name': 'more_drinks'},
{'completed': False, 'name': 'collect_address'},
{'completed': False, 'name': 'collect_phone'},
],
was_previous_order_invalid=False,
was_system_act_processed=False,
@ -125,7 +125,7 @@ class DialogStateMonitor:
if frame.source != 'user':
return
if frame.act == 'inform/order':
new_order = dict()
new_order = list()
for slot in frame.slots:
value = normalize(slot.value)
slot = self.slot_augmentation(slot, value)
@ -133,7 +133,6 @@ class DialogStateMonitor:
continue
if not self.value_valid(slot.name, value):
continue
print("Coca colę?")
stage_name = self.get_current_active_stage()['name']
is_collect_food = (slot.name == 'pizza' and stage_name == 'collect_food')
is_collect_drinks = (slot.name == 'drink' and stage_name == 'collect_drinks')
@ -143,7 +142,7 @@ class DialogStateMonitor:
return
self.state['was_previous_order_invalid'] = False
self.state['total_cost'] += self.state['constants'][slot.name][value]['price']
new_order[slot.name] = value
new_order.append((slot.name, value))
if len(new_order) > 0:
self.state['belief_state']['order'].append(new_order)
self.complete_stage_if_valid('collect_food')