AumCoreAI commited on
Commit
c497be7
·
verified ·
1 Parent(s): e917a82

Upload long_term_memory.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. long_term_memory.py +41 -0
long_term_memory.py ADDED
@@ -0,0 +1,41 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # long_term_memory.py - Persistent Code Memory
2
+ import json
3
+ import hashlib
4
+ from datetime import datetime
5
+
6
+ class LongTermMemory:
7
+ def __init__(self, file_path="code_memory.json"):
8
+ self.file_path = file_path
9
+ self.memory = self.load_memory()
10
+
11
+ def load_memory(self):
12
+ try:
13
+ with open(self.file_path, 'r') as f:
14
+ return json.load(f)
15
+ except FileNotFoundError:
16
+ return {"code_snippets": [], "patterns": []}
17
+
18
+ def save_memory(self):
19
+ with open(self.file_path, 'w') as f:
20
+ json.dump(self.memory, f, indent=2)
21
+
22
+ def add_code_snippet(self, code, description, language="python"):
23
+ snippet_id = hashlib.md5(code.encode()).hexdigest()[:10]
24
+ snippet = {
25
+ "id": snippet_id,
26
+ "code": code,
27
+ "description": description,
28
+ "language": language,
29
+ "timestamp": datetime.now().isoformat()
30
+ }
31
+ self.memory["code_snippets"].append(snippet)
32
+ self.save_memory()
33
+ return snippet_id
34
+
35
+ def find_similar_code(self, query):
36
+ # Simple keyword matching (can be enhanced with embeddings)
37
+ results = []
38
+ for snippet in self.memory["code_snippets"]:
39
+ if query.lower() in snippet["description"].lower():
40
+ results.append(snippet)
41
+ return results[:5]