core/memory/chat_memory.py

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