fromsentence_transformersimportSentenceTransformerfromnuonuo.hippocampusimportHippocampalMemorymodel=SentenceTransformer('all-MiniLM-L6-v2',device='cuda')memory=HippocampalMemory(embed_dim=384)defemb(text):returnmodel.encode([text],convert_to_tensor=True,normalize_embeddings=True,device='cuda')[0]# Store with paraphrase augmentationmemory.store(emb("The database is slow"),emb("Check missing indexes"),cue_variants=[emb("DB performance terrible"),emb("Database crawling")],metadata={"target":"Check missing indexes"})# Recallresults=memory.recall(emb("DB is really slow today"),top_k=3)chain=memory.recall_chain(emb("DB is really slow today"),hops=3)# Save/Loadmemory.save("hippocampus.pt")