Memory Systems

class AgentMemory:
    def __init__(self):
        self.short_term = []        # Context window
        self.vector_db = ChromaDB()  # Bộ nhớ dài hạn
        self.episodes = []           # Kinh nghiệm

    def remember(self, text, type="long"):
        if type == "short":
            self.short_term.append(text)
        elif type == "long":
            embedding = embed(text)
            self.vector_db.add(embedding, text)
        elif type == "episode":
            self.episodes.append(text)

    def recall(self, query, k=5):
        """Truy xuất thông tin liên quan"""
        # Bộ nhớ ngắn hạn. luôn có
        context = self.short_term[-10:]

        # Bộ nhớ dài hạn. semantic search
        relevant = self.vector_db.search(query, k=k)
        context.extend(relevant)

        # Bộ nhớ sự kiện. tìm kinh nghiệm tương tự
        similar_episodes = self.find_similar_episodes(query)
        context.extend(similar_episodes)

        return context