File size: 402 Bytes
52b0da0
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
from enum import Enum

class MemoryTier(Enum):
    WORKING = 1
    SEMANTIC = 2
    EPISODIC = 3

class HierarchicalMemory:
    def __init__(self, db_path=None):
        self.data = []
    def ingest(self, text, tier, user, importance=0.5, confidence=0.8):
        self.data.append({"text": text, "tier": tier, "user": user})
    def retrieve(self, query, tier=None, k=5):
        return self.data[:k]