memory/short_term.py

class ShortTermMemory:

    def __init__(self, max_len=5):
        self.max_len = max_len
        self.buffer = []

    def add(self, item):
        """
        item can be a dictionary:
        {
            "text": str,
            "embedding": tensor,
            "intent": str,
            "timestamp": float
        }
        """
        self.buffer.append(item)

        if len(self.buffer) > self.max_len:
            self.buffer.pop(0)

    def get(self):
        return list(self.buffer)

    def last(self):
        if len(self.buffer) == 0:
            return None
        return self.buffer[-1]

    def clear(self):
        self.buffer.clear()

    def __len__(self):
        return len(self.buffer)