aitech-sd-lab/modules/dialogue_state.py
2022-06-08 08:58:25 +02:00

31 lines
816 B
Python

def default_state():
state = dict(user_action=[],
system_action=[],
belief_state={},
request_state={},
terminated=False,
history=[])
state['belief_state'] = {
"cinema": {
"book": {
"movie": "",
"task": "",
"date": "",
"hour": "",
"seat": "",
"row": ""
},
"semi": {
"name": "",
"e-mail": "",
"price": 0,
"quantity": 0,
"tickettype": "",
"payments": "",
"cancel_book_status": False,
"cancel_buy_status": False
}
}
}
return state