Leonydis137 commited on
Commit
86b36e2
·
verified ·
1 Parent(s): 6f19edb

Create memory.py

Browse files
Files changed (1) hide show
  1. agents/memory.py +9 -0
agents/memory.py ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ class MemoryAgent:
2
+ def __init__(self):
3
+ self.data = []
4
+
5
+ def save(self, key, value):
6
+ self.data.append((key, value))
7
+
8
+ def retrieve(self, query):
9
+ return [v for k, v in self.data if query.lower() in k.lower()]