File size: 611 Bytes
58974f8 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 |
from .base_memory import BaseMemory
class ChatMemory(BaseMemory):
def __init__(self, sys_prompt = None) -> None:
self.sys_prompt = sys_prompt
self.messages = [{"role": "system", "content": sys_prompt}] if sys_prompt else []
def append(self, message: list) -> None:
# assert
self.messages += message
def pop(self) -> None:
self.messages.pop()
def clear(self) -> None:
self.messages = [{"role": "system", "content": self.sys_prompt}]
def load(self) -> None:
pass
def save(self) -> None:
pass |