Implementacja DSM wraz z testami

This commit is contained in:
Patryk Osiński 2024-06-07 12:49:41 +02:00
parent 175de22c9b
commit b64eb49ba0
6 changed files with 75 additions and 16 deletions

View File

@ -1,9 +1,6 @@
{ {
"addr": null, "address": null,
"confirm": [ "order-complete": false,
"true",
"false"
],
"dough": ["thick"], "dough": ["thick"],
"drink": ["pepsi", "cola", "water"], "drink": ["pepsi", "cola", "water"],
"food": ["pizza"], "food": ["pizza"],

View File

@ -3,3 +3,4 @@ conllu==4.5.3
pandas==1.5.3 pandas==1.5.3
numpy==1.26.4 numpy==1.26.4
torch==2.3.0 torch==2.3.0
convlab==3.0.2a0

View File

@ -18,10 +18,10 @@ while True:
# print(frame) # print(frame)
# DSM # DSM
# monitor.append(frame) monitor.update(frame)
# DP # DP
# print(dialog_policy.next_dialogue_act(monitor.get_all()).act) # print(dialog_policy.next_dialogue_act(monitor.read()).act)
# NLG # NLG
act, slots = parse_frame(frame) act, slots = parse_frame(frame)

View File

@ -1,14 +1,55 @@
from model.frame import Frame from src.model.frame import Frame
from convlab.dst.dst import DST
import json
import copy
class DialogStateMonitor: class DialogStateMonitor:
dialog = [] def __init__(self, initial_state_file: str = '../attributes.json'):
DST.__init__(self)
with open(initial_state_file) as file:
self.__initial_state = json.load(file)
self.__memory = copy.deepcopy(self.__initial_state)
def append(self, frame: Frame): # def __access_memory__(self, path: str) -> str | int | float | None:
self.dialog.append(frame) # result = self.state['memory']
# for segment in path.split('.'):
# if segment not in result:
# return None
# result = result[segment]
# return result
def get_all(self) -> [Frame]: def update(self, frame: Frame):
return self.dialog if frame.source != 'user':
return
if frame.act == 'inform/order':
new_order = dict()
for slot in frame.slots:
new_order[slot.name] = slot.value
self.__memory['order'].append(new_order)
elif frame.act == 'inform/address':
for slot in frame.slots:
self.__memory['address'][slot.name] = slot.value
elif frame.act == 'inform/phone':
for slot in frame.slots:
self.__memory['phone'][slot.name] = slot.value
elif frame.act == 'inform/order-complete':
self.__memory['order-complete'] = True
elif frame.act == 'inform/delivery':
for slot in frame.slots:
self.__memory['delivery'][slot.name] = slot.value
elif frame.act == 'inform/payment':
for slot in frame.slots:
self.__memory['payment'][slot.name] = slot.value
elif frame.act == 'inform/time':
for slot in frame.slots:
self.__memory['time'][slot.name] = slot.value
elif frame.act == 'inform/name':
for slot in frame.slots:
self.__memory['name'][slot.name] = slot.value
def get_last(self) -> Frame: def read(self) -> dict:
return self.dialog[len(self.dialog) - 1] return self.__memory
def reset(self):
self.__memory = copy.deepcopy(self.__initial_state)

0
src/test/__init__.py Normal file
View File

View File

@ -0,0 +1,20 @@
from src.service.dialog_state_monitor import DialogStateMonitor
from src.model.frame import Frame
from src.model.slot import Slot
dst = DialogStateMonitor()
assert dst.read()['pizza']['capri']['price'] == 25
dst.update(Frame('user', 'inform/order', [Slot('B-pizza', 'margaritta'), Slot('B-sauce', 'ketchup')]))
dst.update(Frame('user', 'inform/order', [Slot('B-pizza', 'carbonara')]))
dst.update(Frame('user', 'inform/order-complete', []))
assert dst.read()['order'][0]['B-pizza'] == 'margaritta'
assert dst.read()['order'][0]['B-sauce'] == 'ketchup'
assert dst.read()['order-complete'] == True
dst.reset()
assert dst.read()['order'] == []
assert dst.read()['order-complete'] == False