chat-restaruacja/src/service/dialog_policy.py

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", [])