from data.frigate_store import FRIGATE_MEMORY

class Memory:
    def __init__(self):
        self.knowledge = {}
        for k, v in FRIGATE_MEMORY.items():
            self.learn(k, v[:500])  # pre-load knowledge

    def retrieve(self, query):
        hits = []
        for k, v in self.knowledge.items():
            if query.lower() in v.lower():
                hits.append(v[:300])
        return hits[:5]

    def learn(self, key, value):
        self.knowledge[key] = value