File size: 831 Bytes
a3bcd92 5f83674 a3bcd92 5f83674 a3bcd92 5f83674 a3bcd92 5f83674 a3bcd92 5f83674 a3bcd92 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 |
from __future__ import annotations
"""In-memory storage used by legacy BLUX agent tests."""
from typing import Dict, List
class Memory:
"""Stores session and long-term memory entries."""
def __init__(self) -> None:
self.session_memory: List[Dict[str, str]] = []
self.long_term_memory: List[Dict[str, str]] = []
def store(self, user_input: str, user_type: str, decision: str) -> None:
entry = {
"input": user_input,
"user_type": user_type,
"decision": decision,
}
self.session_memory.append(entry)
self.long_term_memory.append(entry)
def recall_session(self) -> List[Dict[str, str]]:
return list(self.session_memory)
def recall_long_term(self) -> List[Dict[str, str]]:
return list(self.long_term_memory)
|