Memory persistence and storage in Agentic AI

Introduction

Memory persistence and storage help AI agents remember important information over time, so they can make better decisions and learn from past experiences.

When you want an AI agent to recall previous conversations or actions.

When storing user preferences to personalize future interactions.

When saving learned knowledge to improve performance over time.

When you need to keep track of long-term goals or plans.

When recovering from interruptions without losing progress.

Syntax

Agentic AI

memory = PersistentMemory(storage_path='memory.json')

# Save data
memory.save(key='user_info', value={'name': 'Alice', 'age': 30})

# Load data
user_info = memory.load(key='user_info')

PersistentMemory is a simple example class that saves data to a file or database.

Use unique keys to store and retrieve different pieces of information.

Examples

Save the last visited page in the session data.

Agentic AI

memory.save(key='session_data', value={'last_page': 'home'})

Load and print user information previously saved.

Agentic AI

user_data = memory.load(key='user_info')
print(user_data)

Remove stored session data when no longer needed.

Agentic AI

memory.delete(key='session_data')

Sample Model

This program creates a simple persistent memory using a JSON file. It saves user info, loads and prints it, deletes it, then shows that the data is gone.

Agentic AI

class PersistentMemory:
    def __init__(self, storage_path):
        import json
        self.storage_path = storage_path
        try:
            with open(self.storage_path, 'r') as f:
                self.data = json.load(f)
        except FileNotFoundError:
            self.data = {}

    def save(self, key, value):
        self.data[key] = value
        with open(self.storage_path, 'w') as f:
            import json
            json.dump(self.data, f)

    def load(self, key):
        return self.data.get(key, None)

    def delete(self, key):
        if key in self.data:
            del self.data[key]
            with open(self.storage_path, 'w') as f:
                import json
                json.dump(self.data, f)

# Create memory instance
memory = PersistentMemory('memory.json')

# Save user info
memory.save('user_info', {'name': 'Alice', 'age': 30})

# Load and print user info
user = memory.load('user_info')
print('Loaded user info:', user)

# Delete user info
memory.delete('user_info')

# Try to load deleted info
deleted_user = memory.load('user_info')
print('After deletion:', deleted_user)

Important Notes

Persistent storage means data stays saved even if the program stops or the computer restarts.

Use simple file formats like JSON for easy reading and writing of memory data.

Always handle cases where requested data might not exist to avoid errors.

Summary

Memory persistence lets AI agents keep important info over time.

Use keys to save, load, and delete data in persistent storage.

Simple file formats like JSON make storing memory easy and reliable.

Practice

1. What is the main purpose of memory persistence in agentic AI systems?

easy

A. To keep important information available over time

B. To speed up the AI's calculations

C. To reduce the size of the AI model

D. To improve the AI's visual recognition

Solution

1. Step 1: Understand memory persistence concept

   Memory persistence means saving data so it stays available even after the AI stops running.

2. Step 2: Identify the purpose in AI context

   This helps AI remember important info across sessions, not just during one run.

3. Final Answer:

   To keep important information available over time -> Option A

4. Quick Check:

   Memory persistence = keep info over time [OK]

Hint: Memory persistence means saving info to use later [OK]

Common Mistakes:

• Confusing persistence with faster processing
• Thinking it reduces model size
• Mixing it with unrelated AI tasks

2. Which of the following is the correct way to save data in a JSON file for memory persistence?

easy

A. open('memory.json', 'a') and load data with json.load()

B. open('memory.json', 'r') and write data

C. open('memory.json', 'x') and read data

D. open('memory.json', 'w') and dump data with json.dump()

Solution

1. Step 1: Identify file mode for writing JSON

   To save data, we open the file in write mode ('w').

2. Step 2: Use json.dump() to write data

   json.dump() writes Python data to the file in JSON format.

3. Final Answer:

   open('memory.json', 'w') and dump data with json.dump() -> Option D

4. Quick Check:

   Write mode + json.dump() = save JSON [OK]

Hint: Use 'w' mode and json.dump() to save JSON data [OK]

Common Mistakes:

• Using 'r' mode to write data
• Confusing json.load() with saving
• Using 'x' mode incorrectly for reading

3. Given this code snippet for loading memory data, what will be the output if the file contains {'key': 'value'}?

import json
with open('memory.json', 'r') as f:
    data = json.load(f)
print(data['key'])

medium

A. key

B. value

C. None

D. Error: KeyError

Solution

1. Step 1: Understand json.load() output

   json.load() reads JSON and converts it to a Python dictionary.

2. Step 2: Access dictionary value by key

   data['key'] accesses the value 'value' stored under 'key'.

3. Final Answer:

   value -> Option B

4. Quick Check:

   data['key'] = 'value' [OK]

Hint: json.load() returns dict; access keys normally [OK]

Common Mistakes:

• Expecting the key name as output
• Confusing key with value
• Assuming None or error without checking file content

4. This code tries to save data but causes an error. What is the problem?

import json
data = {'name': 'AI Agent'}
file = open('memory.json', 'r')
json.dump(data, file)
file.close()

medium

A. Missing import statement for json

B. json.dump() requires a string, not dict

C. File opened in read mode, cannot write

D. File not closed before writing

Solution

1. Step 1: Check file open mode

   The file is opened with 'r' (read) mode, which does not allow writing.

2. Step 2: Understand json.dump() needs writable file

   json.dump() writes data, so the file must be opened in 'w' or 'a' mode.

3. Final Answer:

   File opened in read mode, cannot write -> Option C

4. Quick Check:

   Write requires 'w' mode, not 'r' [OK]

Hint: Open file with 'w' to write JSON data [OK]

Common Mistakes:

• Using 'r' mode when writing
• Forgetting to close the file
• Misunderstanding json.dump() input

5. You want your AI agent to remember user preferences across sessions using JSON storage. Which approach best ensures data is saved and loaded correctly?

hard

A. Save preferences with json.dump() in 'w' mode; load with json.load() in 'r' mode

B. Save preferences by appending text; load by reading lines manually

C. Save preferences in a plain text file without JSON; load by parsing strings

D. Save preferences only in memory variables without writing to file

Solution

1. Step 1: Choose reliable save method

   json.dump() with 'w' mode writes structured data safely to file.

2. Step 2: Choose matching load method

   json.load() with 'r' mode reads the structured data back correctly.

3. Step 3: Avoid unreliable or volatile methods

   Appending text or plain text parsing risks errors; memory-only loses data after session.

4. Final Answer:

   Save preferences with json.dump() in 'w' mode; load with json.load() in 'r' mode -> Option A

5. Quick Check:

   Use json.dump/load with correct modes for persistence [OK]

Hint: Use json.dump/load with 'w' and 'r' modes for safe persistence [OK]

Common Mistakes:

• Appending text without JSON format
• Not saving data to file at all
• Parsing plain text manually risking errors