12 lines
342 B
Python
12 lines
342 B
Python
from model.frame import Frame
|
|
|
|
class DialogPolicy:
|
|
def __init__(self, frames: list[Frame]) -> None:
|
|
self.frames = frames
|
|
|
|
def next_dialogue_act(self) -> Frame:
|
|
if self.frames[-1].act == "welcomemsg":
|
|
return Frame("system", "welcomemsg", [])
|
|
else:
|
|
return Frame("system", "canthelp", [])
|