| class Message: | |
| def __init__(self, msg=None): | |
| self._messages = msg if msg else [] | |
| self._images = [] | |
| self.skip_next = False | |
| def add_message(self, question, answer=None): | |
| quension_msg_dict = {'from': 'human'} | |
| quension_msg_dict['value'] = question | |
| answer_msg_dict = {'from': 'gpt'} | |
| answer_msg_dict['value'] = answer | |
| self._messages.append(quension_msg_dict) | |
| self._messages.append(answer_msg_dict) | |
| def add_image(self, image, index=0): | |
| self._images.append((image, index)) | |
| def images(self): | |
| return self._images | |
| def messages(self): | |
| return self._messages | |
| def copy(self): | |
| return Message(self._messages) | |