motionmemorymvp / motion_memory.py
Tpayne101's picture
Create motion_memory.py
4aa868c verified
raw
history blame contribute delete
803 Bytes
import json
from pathlib import Path
MEMORY_FILE = Path("data/memory.json")
# Load memory
def load_memory():
if MEMORY_FILE.exists():
with open(MEMORY_FILE, "r") as f:
return json.load(f)
return {}
# Save memory
def save_memory(memory):
with open(MEMORY_FILE, "w") as f:
json.dump(memory, f, indent=2)
# Remember a message/response for an agent
def remember(agent_id, message, response):
memory = load_memory()
if agent_id not in memory:
memory[agent_id] = []
memory[agent_id].append({
"message": message,
"response": response
})
save_memory(memory)
# Recall last n interactions
def recall(agent_id, n=5):
memory = load_memory()
if agent_id not in memory:
return []
return memory[agent_id][-n:]