Compare commits
3 Commits
3d820e8d6d
...
a850690dc9
Author | SHA1 | Date | |
---|---|---|---|
a850690dc9 | |||
2b6a8b01a6 | |||
79bf27aca4 |
123
attributes.json
123
attributes.json
@ -1,40 +1,91 @@
|
|||||||
{
|
{
|
||||||
"dough": ["thick"],
|
"dough": [
|
||||||
"drink": ["pepsi", "cola", "water"],
|
"thick"
|
||||||
"food": ["pizza"],
|
],
|
||||||
"meat": ["chicken", "ham", "tuna"],
|
"drink": {
|
||||||
"sauce": ["garlic", "1000w"],
|
"pepsi": {
|
||||||
"ingredient": {
|
"price": 10
|
||||||
"chicken": {},
|
|
||||||
"tuna": {},
|
|
||||||
"pineapple": {},
|
|
||||||
"onion": {},
|
|
||||||
"cheese": {},
|
|
||||||
"tomato": {},
|
|
||||||
"ham": {},
|
|
||||||
"pepper": {}
|
|
||||||
},
|
},
|
||||||
"menu": ["capri", "margarita", "hawajska", "barcelona", "tuna"],
|
"cola": {
|
||||||
"pizza": {
|
"price": 10
|
||||||
"capri": {
|
},
|
||||||
"ingredient": ["tomato", "ham", "mushrooms", "cheese"],
|
"water": {
|
||||||
"price": 25
|
"price": 5
|
||||||
},
|
|
||||||
"margarita": {
|
|
||||||
"ingredient": ["tomato", "cheese"],
|
|
||||||
"price": 20
|
|
||||||
},
|
|
||||||
"hawajska": {
|
|
||||||
"ingredient": ["tomato", "pineapple", "chicken", "cheese"],
|
|
||||||
"price": 30
|
|
||||||
},
|
|
||||||
"barcelona": {
|
|
||||||
"ingredient": ["tomato", "onion", "ham", "pepper", "cheese"],
|
|
||||||
"price": 40
|
|
||||||
},
|
|
||||||
"tuna": {
|
|
||||||
"ingredient": ["tomato", "tuna", "onion", "cheese"],
|
|
||||||
"price": 40
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
},
|
||||||
|
"food": [
|
||||||
|
"pizza"
|
||||||
|
],
|
||||||
|
"meat": [
|
||||||
|
"chicken",
|
||||||
|
"ham",
|
||||||
|
"tuna"
|
||||||
|
],
|
||||||
|
"sauce": [
|
||||||
|
"garlic",
|
||||||
|
"1000w"
|
||||||
|
],
|
||||||
|
"ingredient": {
|
||||||
|
"chicken": {},
|
||||||
|
"tuna": {},
|
||||||
|
"pineapple": {},
|
||||||
|
"onion": {},
|
||||||
|
"cheese": {},
|
||||||
|
"tomato": {},
|
||||||
|
"ham": {},
|
||||||
|
"pepper": {}
|
||||||
|
},
|
||||||
|
"menu": [
|
||||||
|
"capri",
|
||||||
|
"margarita",
|
||||||
|
"hawajska",
|
||||||
|
"barcelona",
|
||||||
|
"tuna"
|
||||||
|
],
|
||||||
|
"pizza": {
|
||||||
|
"capri": {
|
||||||
|
"ingredient": [
|
||||||
|
"tomato",
|
||||||
|
"ham",
|
||||||
|
"mushrooms",
|
||||||
|
"cheese"
|
||||||
|
],
|
||||||
|
"price": 25
|
||||||
|
},
|
||||||
|
"margarita": {
|
||||||
|
"ingredient": [
|
||||||
|
"tomato",
|
||||||
|
"cheese"
|
||||||
|
],
|
||||||
|
"price": 20
|
||||||
|
},
|
||||||
|
"hawajska": {
|
||||||
|
"ingredient": [
|
||||||
|
"tomato",
|
||||||
|
"pineapple",
|
||||||
|
"chicken",
|
||||||
|
"cheese"
|
||||||
|
],
|
||||||
|
"price": 30
|
||||||
|
},
|
||||||
|
"barcelona": {
|
||||||
|
"ingredient": [
|
||||||
|
"tomato",
|
||||||
|
"onion",
|
||||||
|
"ham",
|
||||||
|
"pepper",
|
||||||
|
"cheese"
|
||||||
|
],
|
||||||
|
"price": 40
|
||||||
|
},
|
||||||
|
"tuna": {
|
||||||
|
"ingredient": [
|
||||||
|
"tomato",
|
||||||
|
"tuna",
|
||||||
|
"onion",
|
||||||
|
"cheese"
|
||||||
|
],
|
||||||
|
"price": 40
|
||||||
|
}
|
||||||
|
}
|
||||||
}
|
}
|
@ -44,7 +44,10 @@ class DialogStateMonitor:
|
|||||||
stage['completed'] = True
|
stage['completed'] = True
|
||||||
return
|
return
|
||||||
|
|
||||||
def pizza_exists(self, name: str) -> bool:
|
def item_exists(self, type: str, name: str) -> bool:
|
||||||
|
return normalize(name) in self.state['constants'][type]
|
||||||
|
|
||||||
|
def drink_exists(self, name: str) -> bool:
|
||||||
return normalize(name) in self.state['constants']['pizza']
|
return normalize(name) in self.state['constants']['pizza']
|
||||||
|
|
||||||
def get_total_cost(self) -> int:
|
def get_total_cost(self) -> int:
|
||||||
@ -58,12 +61,12 @@ class DialogStateMonitor:
|
|||||||
new_order = dict()
|
new_order = dict()
|
||||||
for slot in frame.slots:
|
for slot in frame.slots:
|
||||||
value = normalize(slot.value)
|
value = normalize(slot.value)
|
||||||
if slot.name == 'pizza':
|
if slot.name == 'pizza' or slot.name == 'drink':
|
||||||
if self.pizza_exists(value) is False:
|
if self.item_exists(slot.name, value) is False:
|
||||||
self.state['was_previous_order_invalid'] = True
|
self.state['was_previous_order_invalid'] = True
|
||||||
return
|
return
|
||||||
self.state['was_previous_order_invalid'] = False
|
self.state['was_previous_order_invalid'] = False
|
||||||
self.state['total_cost'] += self.state['constants']['pizza'][value]['price']
|
self.state['total_cost'] += self.state['constants'][slot.name][value]['price']
|
||||||
new_order[slot.name] = value
|
new_order[slot.name] = value
|
||||||
self.state['belief_state']['order'].append(new_order)
|
self.state['belief_state']['order'].append(new_order)
|
||||||
elif frame.act == 'inform/address':
|
elif frame.act == 'inform/address':
|
||||||
|
@ -4,7 +4,9 @@ from src.model.slot import Slot
|
|||||||
|
|
||||||
dsm = DialogStateMonitor()
|
dsm = DialogStateMonitor()
|
||||||
|
|
||||||
assert dsm.pizza_exists('capri') is True
|
assert dsm.item_exists('pizza', 'capri') is True
|
||||||
|
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.state['was_previous_order_invalid'] is False
|
||||||
|
|
||||||
frame1 = Frame('user', 'inform/order', [Slot('pizza', 'margarita'), Slot('sauce', 'ketchup')])
|
frame1 = Frame('user', 'inform/order', [Slot('pizza', 'margarita'), Slot('sauce', 'ketchup')])
|
||||||
@ -15,6 +17,9 @@ dsm.update(frame2)
|
|||||||
assert dsm.get_total_cost() == 60
|
assert dsm.get_total_cost() == 60
|
||||||
frame3 = Frame('user', 'inform/order-complete', [])
|
frame3 = Frame('user', 'inform/order-complete', [])
|
||||||
dsm.update(frame3)
|
dsm.update(frame3)
|
||||||
|
frame4 = Frame('user', 'inform/order', [Slot('drink', 'cola')])
|
||||||
|
dsm.update(frame4)
|
||||||
|
assert dsm.get_total_cost() == 70
|
||||||
|
|
||||||
assert dsm.state['belief_state']['order'][0]['pizza'] == 'margarita'
|
assert dsm.state['belief_state']['order'][0]['pizza'] == 'margarita'
|
||||||
assert dsm.state['belief_state']['order'][0]['sauce'] == 'ketchup'
|
assert dsm.state['belief_state']['order'][0]['sauce'] == 'ketchup'
|
||||||
@ -22,7 +27,7 @@ assert dsm.state['belief_state']['order-complete'] is True
|
|||||||
assert dsm.state['history'][0] == frame1
|
assert dsm.state['history'][0] == frame1
|
||||||
assert dsm.state['history'][1] == frame2
|
assert dsm.state['history'][1] == frame2
|
||||||
assert dsm.state['history'][2] == frame3
|
assert dsm.state['history'][2] == frame3
|
||||||
|
assert dsm.state['history'][3] == frame4
|
||||||
|
|
||||||
assert dsm.get_current_active_stage() == 'collect_food'
|
assert dsm.get_current_active_stage() == 'collect_food'
|
||||||
dsm.mark_current_stage_completed()
|
dsm.mark_current_stage_completed()
|
||||||
@ -42,7 +47,7 @@ assert len(dsm.state['history']) == 0
|
|||||||
|
|
||||||
dsm.reset()
|
dsm.reset()
|
||||||
|
|
||||||
frame1 = Frame('user', 'inform/order', [Slot('pizza', 'buraczna')])
|
frame1 = Frame('user', 'inform/order', [Slot('pizza', 'buraczana')])
|
||||||
dsm.update(frame1)
|
dsm.update(frame1)
|
||||||
assert dsm.state['was_previous_order_invalid'] is True
|
assert dsm.state['was_previous_order_invalid'] is True
|
||||||
assert dsm.state['belief_state']['order'] == []
|
assert dsm.state['belief_state']['order'] == []
|
||||||
|
Loading…
Reference in New Issue
Block a user