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)