Long-Term Knowledge Storage and Retrieval › Storing Information for Later

import json
import os


class SimpleMemory:
    def __init__(self, memory_file="assistant_memory.json"):
        self.memory_file = memory_file
        self.memory = self._load_memory()

    def _load_memory(self):
        """Load memory from file if it exists"""
        if os.path.exists(self.memory_file):
            with open(self.memory_file, "r") as f:
                return json.load(f)
        return {}

    def _save_memory(self):
        """Save memory to file"""
        with open(self.memory_file, "w") as f:
            json.dump(self.memory, f, indent=2)

    def store(self, key, value):
        """Store a fact in long-term memory"""
        self.memory[key] = value
        self._save_memory()
        return f"Remembered: {key}"

    def retrieve(self, key):
        """Retrieve a fact from long-term memory"""
        return self.memory.get(key, "I don't have that information stored.")


## Example usage
memory = SimpleMemory()
memory.store("user_birthday", "July 20")
memory.store("favorite_restaurant", "Luigi's Italian Kitchen")

## Later, even after restarting the program
print(memory.retrieve("user_birthday"))  # Output: July 20