| 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 | |