Memory persistence and storage in Agentic AI - Practice Problems & Coding Challenges

Challenge - 5 Problems

🧠 Conceptual

intermediate

Understanding Memory Persistence in Agentic AI

Which of the following best describes memory persistence in an agentic AI system?

A The process of storing and retrieving information across multiple sessions to maintain context.

B A technique to compress data to reduce storage space without retaining any context.

C A method to erase all stored data after each interaction to protect privacy.

D The ability of the AI to remember information only during a single session without saving it.

💡 Hint

Think about how an AI keeps track of past conversations over time.

Memory persistence means the AI can save information beyond a single session, allowing it to recall past interactions and maintain context.

❓ Predict Output

intermediate

Output of Memory Storage Simulation Code

What is the output of the following Python code simulating a simple memory storage for an agentic AI?

Agentic AI

class Memory:
    def __init__(self):
        self.storage = {}
    def remember(self, key, value):
        self.storage[key] = value
    def recall(self, key):
        return self.storage.get(key, 'Not found')

memory = Memory()
memory.remember('task', 'clean the house')
print(memory.recall('task'))
print(memory.recall('deadline'))

A clean the house\nNot found

B clean the house\nNone

C Not found\nNot found

D None\nNot found

💡 Hint

Check what happens when a key is missing in the dictionary.

The method recall returns the stored value if the key exists; otherwise, it returns 'Not found'.

❓ Model Choice

advanced

Choosing a Storage Model for Long-Term Memory

Which storage model is best suited for an agentic AI that needs to store large amounts of unstructured data for long-term memory persistence?

A Temporary cache that clears data after each session.

B NoSQL document store that allows flexible schema and horizontal scaling.

C In-memory dictionary with limited size and no backup.

D Relational database with fixed schema and strict tables.

💡 Hint

Consider flexibility and scalability for unstructured data.

NoSQL document stores handle unstructured data well and scale horizontally, making them ideal for long-term memory persistence in agentic AI.

❓ Hyperparameter

advanced

Hyperparameter Affecting Memory Retention Duration

In an agentic AI system using a memory persistence mechanism, which hyperparameter directly controls how long information is retained before being discarded?

A Learning rate

B Batch size

C Memory decay rate

D Number of hidden layers

💡 Hint

Think about what controls forgetting or fading of stored information.

The memory decay rate controls how quickly stored information fades or is removed, affecting retention duration.

🔧 Debug

expert

Debugging Memory Retrieval Error

Given the following code snippet for an agentic AI's memory retrieval, what error will occur when running it?

Agentic AI

class AgentMemory:
    def __init__(self):
        self.data = {}
    def store(self, key, value):
        self.data[key] = value
    def retrieve(self, key):
        return self.data[key]

memory = AgentMemory()
memory.store('goal', 'learn AI')
print(memory.retrieve('plan'))

A TypeError

B AttributeError

C IndexError

D KeyError

💡 Hint

What happens when you try to access a dictionary key that does not exist?

Accessing a missing key in a dictionary raises a KeyError in Python.

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