AI Agent Memory: What to Store and What to Forget

Memory is persistent state with a friendly name. Treat it like a database feature, not a larger prompt. Every remembered fact should have a scope, source, freshness rule, and correction path.

At a glance

This diagram is a topic-specific decision aid: connect the article’s context to evidence before taking the next action.

This guide is for agent engineers designing persistent context. It helps you decide what to remember, retrieve, correct, and forget so you can leave with useful continuity without hidden authorization or stale context.

The AI agent memory design framework

Use Scope → Provenance → Retrieve → Correct as the decision path:

1. Scope: establish the task, context, or boundary before changing behavior.
2. Provenance: make the relevant evidence or control explicit.
3. Retrieve: compare the decision with a concrete example, trace, or test.
4. Correct: finish with an owner, rollback, escalation, or next action.

Decision rule: Bound memory by scope, preserve its source, retrieve for a reason, and make correction and expiry observable.

The framework is deliberately narrower than the entire subject. It gives the reader a repeatable route through this article; it does not claim that one metric, model, or control solves every production case.

Separate memory types

Working context belongs to one run. Session memory supports a bounded conversation. Long-term memory survives sessions and needs stronger consent, access, and deletion rules. Operational state—such as an approval or workflow status—should live in a typed system of record, not only in retrieved text.

type Memory = {
  value: string;
  scope: "run" | "session" | "user";
  source: string;
  observedAt: string;
  expiresAt?: string;
  confidence: "verified" | "inferred";
};

Log which memories were retrieved and which affected a decision. This makes AI agent architecture inspectable and complements prompt versioning when context changes behavior.

Design for correction

Users change their minds and systems change their data. Allow a memory to be corrected, expired, or deleted. Prefer source references over copied sensitive content. If the agent cannot explain where a fact came from, it should ask or abstain instead of presenting memory as truth.

Measure memory, not just recall

Track retrieval rate, useful-memory rate, stale-memory incidents, correction rate, context growth, and task outcome. More recalled facts are not automatically better. The target is the smallest relevant context that improves a user-visible result without creating surprising behavior.

Start with a memory contract

Before choosing a vector store or summary strategy, define what memory is allowed to remember. A memory contract should name the subject, scope, source of truth, purpose, retention period, sensitivity class, and correction path. “Remember everything about the user” is not a contract; it is an unbounded data-collection policy with unclear product value.

Separate facts from preferences, observations, inferences, and workflow state. A verified account identifier is different from a model inference that the user prefers a certain format. Store provenance and confidence, but do not turn confidence into permission. A low-confidence memory may help choose a question while never authorizing a side effect.

Choose the smallest useful scope

Run memory is appropriate for intermediate reasoning and expires with the task. Session memory can support a conversation but should not silently affect unrelated sessions. User or organization memory needs consent, access control, deletion, and review. Workflow state belongs in a system of record when it controls a business action.

type MemoryRecord = {
  key: string;
  value: string;
  scope: "run" | "session" | "user" | "organization";
  source: { type: "user" | "system" | "review"; reference: string };
  confidence: "verified" | "inferred";
  observedAt: string;
  expiresAt?: string;
};

The exact storage type will differ, but the fields should remain visible when a memory affects a decision. Include a stable reference rather than copying sensitive source content into every prompt.

Write selectively

Memory writes should pass a policy just like tool calls. Ask whether the fact will improve a future task, whether it is stable enough to keep, whether the user expects it to persist, and whether the source is authorized. Do not write every model inference or every sentence of a conversation. High-volume memory creates retrieval noise and makes deletion difficult.

Use explicit write events: memory.proposed, memory.approved, memory.rejected, memory.expired, and memory.deleted. A reviewer should be able to see why a fact entered the store. For sensitive domains, require a human or a deterministic source before promoting an inference to verified memory.

Retrieve with a reason

When memory is retrieved, record the key, scope, source, freshness, and reason it was selected. Avoid returning the entire memory store to the model. Filter by tenant, user, workflow, and time. Prefer a small set of relevant records with provenance to a large semantic search result that the model must sort out.

Measure useful-memory rate: how often a retrieved record changes or improves the task outcome. Also measure stale-memory incidents, contradictory-memory incidents, context growth, and correction rate. A higher retrieval count is not a success metric by itself.

Handle correction and forgetting

Users change their preferences, organizations change policies, and systems correct data. Every durable memory needs an update and deletion path. Expiration should be enforced by the store and checked at retrieval, not described only in a prompt. If a memory conflicts with a current source of truth, the current source should win and the conflict should be observable.

Design deletion as a workflow. Remove primary records, derived embeddings, summaries, caches, and references that could reintroduce the value. Record completion without retaining the deleted content. For organizational memory, define who may request deletion and which policies apply to shared records.

Avoid hidden authorization

Memory should provide context, never unreviewed permission. A remembered statement that a user once approved an export should not approve a new export. A retrieved preference should not override current policy. Keep authorization evidence in a trusted system and require it at action time.

This principle connects memory to prompt injection defense and AI agent architecture. Memory is another input channel. If it can silently change the tool boundary, it needs the same trust classification, validation, and trace evidence as a retrieved webpage.

Evaluate memory as a change

Build cases for useful recall, irrelevant recall, stale facts, contradictory facts, deletion, and missing consent. Run the same task with memory enabled and disabled. Compare task success, unnecessary questions, tool selection, latency, tokens, cost, and user correction. The correct result may be to ignore memory.

Keep the memory policy revision and retrieval strategy in the trace. When a user reports surprising behavior, an investigator should see which records were supplied, not only the final answer. Add confirmed incidents to regression coverage, and sample successful runs to find cases where memory was overused.

Plan storage and indexing

Use a storage model that supports scoped queries, filtering, expiration, deletion, and provenance. Vector similarity can help with semantic recall but should not be the only access path. Keep structured keys for exact facts, timestamps for freshness, and tenant boundaries in the query itself. Encrypt sensitive values according to the product’s data-handling policy and minimize what enters telemetry.

Indexing is part of behavior. An embedding model change can alter which memories are retrieved even when application code is unchanged. Version embeddings and retrieval configuration, then evaluate the candidate against a fixed memory dataset. Do not delete the previous index until the new one has passed correction and deletion tests.

A practical memory lifecycle

1. Propose a fact with source and purpose.
2. Validate scope, sensitivity, consent, and expected lifetime.
3. Store the minimum useful representation.
4. Retrieve only within the current trust and access boundary.
5. Show provenance when the fact affects a meaningful decision.
6. Correct, expire, or delete through an explicit workflow.
7. Measure whether memory improved the user-visible task.

Memory is successful when users experience better continuity without losing control or understanding why the agent behaved differently. Persistent context should reduce repetition, not create a second invisible application state.

Memory review: a field guide

This extended guide applies the same production discipline to memory: scope, provenance, correction, retention, and retrieval contamination.

Start with the failure you need to prevent

A strong AI agent memory practice begins with a failure that matters to a user. Write the failure in observable terms: the task stopped, the wrong action happened, the answer lacked evidence, the reviewer could not decide, or the system consumed more resources than the outcome justified. This wording gives the team a testable target and prevents the article from drifting into a collection of attractive but disconnected recommendations.

Name the boundary

Most production incidents happen at a boundary: model to application, application to tool, retrieved data to instruction, browser action to page state, or metric to operational decision. For AI agent memory, draw that boundary in plain language before choosing an implementation. Identify what is trusted, what is untrusted, what must be validated, and what evidence crosses the boundary. A boundary that cannot be named cannot be monitored consistently.

Preserve the decision context

A final answer or status flag rarely explains a failure by itself. Preserve the inputs, versions, policies, selected evidence, attempted actions, retries, and resulting state that shaped the decision. For AI agent memory, use stable identifiers and redaction so an investigator can reconstruct the path without copying secrets into every log line. The goal is not maximal data collection; it is enough context to test a plausible explanation.

Measure the user-visible outcome

Choose a primary outcome for AI agent memory and define it before choosing a dashboard. Completion, correctness, groundedness, safe refusal, review quality, time to useful result, and cost per successful task answer different questions. Pair the outcome with failure dimensions such as model version, prompt version, cohort, provider, tool, language, and task type. An aggregate score is useful only when a person can move from the score to representative traces.

Make the unhappy path explicit

A mature AI agent memory workflow names what happens when evidence is missing, a dependency is slow, a schema changes, a permission is denied, a reviewer does not respond, or the system cannot establish confidence. These are product states, not merely exceptions. Give each state an owner, a user-facing explanation, a safe fallback, and a trace event. This is how a graceful degradation policy becomes something an operator can verify.

Use small, reviewable examples

Examples teach the boundary faster than abstract claims. Keep a representative AI agent memory example small enough to inspect, but include the field that changes the decision. Show the expected result and one plausible failure. Explain which part is illustrative and which part should be adapted to the repository. Avoid examples that imply a guarantee the implementation cannot make; precision is more persuasive than spectacle.

Test variation deliberately

The happy path is only one member of the AI agent memory distribution. Add cases for empty input, ambiguous intent, stale state, partial failure, duplicate work, adversarial content, long context, and a dependency that returns a valid-looking but wrong result. Test cohorts and release versions separately. When behavior is probabilistic, record the evidence and evaluator version so a changed score can be distinguished from a changed system.

Roll out in stages

Use shadow mode, a narrow cohort, explicit comparison, and a rollback condition when changing AI agent memory. Shadow mode reveals what the system would have done without changing user behavior. A cohort exposes regressions hidden by averages. A rollback condition turns caution into an operational decision. Keep the release, prompt, model, data, and configuration identifiers together so the comparison remains meaningful after deployment.

Create a learning loop

Production feedback should become a queue of explainable cases for AI agent memory, not a pile of screenshots or vague tickets. Attach the user report, trace, expected behavior, actual behavior, and disposition. Classify the mechanism, add a compact regression case when appropriate, and record whether the fix changed code, data, prompt, policy, or operations. This closes the loop between observability and engineering instead of treating incidents as isolated surprises.

Review the economics and ownership

Every AI agent memory decision has an operating cost. Include reviewer time, evaluator calls, storage, retries, provider spend, and the cost of a wrong action when they matter. Assign an owner for quality, reliability, security, and budget signals. A workflow is not production-ready because it can run; it is ready when the organization can understand, afford, and improve it.

Build the runbook around evidence

The runbook for AI agent memory should begin with the trace or artifact an operator opens, then provide a short decision tree. If the parent task is missing, investigate ingestion. If the task exists but the child span is missing, inspect propagation. If evidence is present but the outcome is wrong, compare versions and evaluator results. Keep the first response steps concrete and link to the source of truth rather than asking an operator to search the whole repository.

Know when not to automate

Automation is not always the correct endpoint for AI agent memory. If the consequence is irreversible, the evidence is incomplete, or the policy cannot be expressed safely, stop and route to a person or a trusted system. A bounded refusal is often a better product result than a confident action. Document the boundary so future optimization does not quietly remove the control that made the workflow acceptable.

A practical review sequence

Review AI agent memory in this order: define the task, identify boundaries, preserve evidence, measure the outcome, exercise failures, test variation, stage the rollout, and assign ownership. The sequence matters because a polished implementation with no observable outcome is difficult to validate, while a clear contract makes later instrumentation and testing much cheaper.

Final field checklist

Before shipping AI agent memory, confirm that the happy path and degraded paths are documented, the rendered page or workflow is accessible, examples are safe to copy, identifiers are stable, sensitive values are redacted, links point to canonical content, and a real person can investigate one representative failure. Then run the repository checks and review the exact diff. Quality is the combination of useful behavior, explainable evidence, and a release process that can recover.

Questions to ask in review

Ask whether the current AI agent memory design makes the next failure easier or harder to explain. Where is the source of truth? Which value is user-controlled? Which decision is irreversible? What happens when a dependency returns a partial result? Which version can be rolled back? These questions expose missing contracts faster than adding another dashboard or another prompt instruction.

The smallest useful implementation

Start with the smallest implementation that creates evidence for AI agent memory. One parent run, a few stable fields, one representative fixture, and one clear outcome are often enough to reveal the largest gap. Expand only when a real investigation needs another dimension. This keeps instrumentation affordable and prevents the team from confusing volume of telemetry with quality of understanding.

Keep the page and system honest

Documentation about AI agent memory should describe the behavior a reader can actually reproduce. Mark illustrative examples clearly, keep names consistent, and avoid claims that depend on an unmentioned environment. On the rendered page, make the important artifact easy to inspect: use a labelled code block, a readable trace scenario, or a compact decision table. The presentation is part of the teaching contract.

Accessibility is a reliability feature

A AI agent memory workflow should remain understandable when a reader cannot hover, drag, or run the example. Provide a static explanation, meaningful labels, visible focus, keyboard access, and mobile-safe overflow. The same principle applies in production: a trace should expose text and structured fields, not require an operator to infer meaning from color or animation alone.

Signals that justify a change

Change AI agent memory when evidence shows a repeated failure mechanism, a measurable user benefit, or an unacceptable operational cost. Record the baseline, candidate, cohort, and decision owner. If the change improves one dimension while harming another, state the trade-off instead of hiding it in an average. This creates a durable record future agents can use when the same decision returns.

Closing principle

The best AI agent memory practice makes good behavior easier to produce and bad behavior easier to see. It gives the user a useful outcome, gives the operator an explainable trace, and gives the team a safe way to improve the system. That is the standard to apply when reviewing both the implementation and this article.

Turn principles into an operating artifact

A principle becomes useful when someone can apply it during a real change. For this topic, create a compact artifact: a trace field list, a test fixture, a rollout table, a budget rule, a review form, or a runbook. Keep the artifact close to the decision it supports. If it lives in a separate document with no owner, it will become stale while the system continues to change.

Compare like with like

When evaluating a change, compare the same task distribution, evidence policy, model or release metadata, and outcome definition. A baseline gathered from easy cases cannot prove an improvement on difficult cases. Preserve the sampling rule and the exclusions. When the data is sparse, say so; a small honest sample is more useful than a precise-looking number that hides its uncertainty.

Make failure visible to the right person

Visibility is not the same as exposure. A user needs a clear explanation and safe next step. An operator needs a trace, timing, versions, and likely cause. An engineer needs a reproducible fixture. A security owner needs the trust boundary and containment event. Design the evidence for the person who must act, and avoid copying sensitive content into places where no action is possible.

Review changes as a system

A prompt, model, schema, retrieval index, browser selector, evaluator, and dashboard threshold can each change behavior. Review the connected system rather than approving one file in isolation. Record what changed, which outcomes could move, which tests cover the risk, and how to revert. This practice keeps small changes from becoming untraceable production behavior.

The standard for completion

Completion means more than a passing happy-path test. A reader should understand the contract, an operator should be able to investigate a failure, and the team should have a bounded path to improve or roll back the behavior. Apply that standard to the article too: the examples should render clearly, the interaction should teach something, the image should have a job, and the validation command should pass.

Keep the decision reversible

When the evidence is uncertain, prefer a change that can be measured and undone. Record the baseline, define the stop condition, and make the owner explicit. This is especially important for memory design because a locally attractive improvement can shift risk into quality, safety, cost, or trust. Reversibility is not hesitation; it is how a team learns without making every experiment a permanent incident.

Share the result clearly

A good handoff states what changed, what did not change, which traces or examples support the conclusion, and what remains unknown. Use the same vocabulary in the article, dashboard, runbook, and release note. Consistent language reduces translation work and makes future investigations faster.

Keep the workflow maintainable

A workflow that works only because one person remembers its hidden assumptions is not ready to scale. Put the contract, owner, evidence fields, and rollback path next to the implementation. Revisit them after incidents and material dependency changes. This keeps the system understandable when the original author is not available.

Review the edges

The most revealing cases are often at the edges of the distribution: the longest task, the oldest record, the least common cohort, the first retry, the empty result, or the most consequential action. Include those cases in review deliberately. They are where a simple average most often lies.

Put this guidanceWhat to Store and What to Forget into practice

A small release checklist

• Define the user-visible outcome and the boundary this article does not cover.
• Capture the evidence needed to explain movement, including the relevant version, cohort, and failure class.
• Assign one owner for the next bounded action and write the stop condition before changing production behavior.

Decision rule: Prefer the smallest change that produces inspectable evidence. If the example cannot be tested, label it illustrative and verify the surrounding assumptions before adapting it.

Frequently asked questions

Recommended next reads

Continue with ai agent architecture, prompt versioning and prompt management, AI agent observability. Choose the one that matches the next decision in your workflow rather than opening every guide at once.

The ai agent memory design operating model

Put ai agent memory design into practice

A useful article should help a reader move from an idea to a safe production decision. Start by stating the user-visible outcome, then identify the inputs, the owner, the evidence, and the terminal states. Do not begin with a tool or framework. Begin with the decision the system must support and the cost of getting that decision wrong. For ai agent memory design, use this as a concrete review criterion: connect the decision to its evidence, owner, and recovery path.

Define the contract

Write the input and output contract before tuning the model. Name required fields, acceptable values, missing-data behavior, and the response a user receives when the system cannot proceed. A schema validates shape, but it does not prove meaning or authorization. Follow parsing with semantic checks, policy checks, and an explicit fallback. For ai agent memory design, use this as a concrete review criterion: connect the decision to its evidence, owner, and recovery path.

Separate proposal from authority

A model can classify, extract, summarize, rank, or propose an action. Deterministic code should own permissions, limits, accounting, retention, and irreversible side effects. This boundary makes the system easier to test and easier to explain after an incident. If a model suggestion is rejected, route it to a visible state instead of silently rewriting it. For ai agent memory design, use this as a concrete review criterion: connect the decision to its evidence, owner, and recovery path.

Build representative cases

Create a small case set before claiming that an implementation works. Include ordinary examples, empty inputs, ambiguous requests, stale data, conflicting evidence, malformed outputs, permission boundaries, and provider failures. Label the expected user outcome and the unacceptable failure. Keep the cases versioned so a later change can be compared against the same baseline. For ai agent memory design, use this as a concrete review criterion: connect the decision to its evidence, owner, and recovery path.

Validate the full path

Unit tests are valuable for parsers, transitions, policies, and retry rules, but they do not prove the workflow works end to end. Add scenario tests that exercise retrieval, model calls, tool calls, timeouts, duplicate events, approvals, and partial completion. Preserve the trace for each important scenario and check both the final result and the intermediate evidence. For ai agent memory design, use this as a concrete review criterion: connect the decision to its evidence, owner, and recovery path.

Make retries safe

Every external write needs a stable identity, a bounded retry policy, and a reconciliation path. A timeout is not proof that an action failed. Record unknown outcomes explicitly, query the provider when possible, and prevent duplicate work with idempotency keys. Never let a generic retry loop decide whether a financial, permission, or communication side effect is safe. For ai agent memory design, use this as a concrete review criterion: connect the decision to its evidence, owner, and recovery path.

Design human review

Human review should be specific and actionable. Show the proposed action, normalized arguments, evidence, policy version, expected consequence, and expiration. Bind approval to that exact proposal. If a material field changes, invalidate the approval. Record the reviewer, decision, reason, and result so rejection patterns become improvements rather than repeated surprises. For ai agent memory design, use this as a concrete review criterion: connect the decision to its evidence, owner, and recovery path.

Instrument useful evidence

A trace should connect the original request to each meaningful stage and the final outcome. Capture versions, latency, cost, status, error category, tool arguments after redaction, and the evidence used for the decision. Avoid collecting sensitive payloads merely because they are available. Store references or structured summaries when raw content is not necessary for diagnosis. For ai agent memory design, use this as a concrete review criterion: connect the decision to its evidence, owner, and recovery path.

Measure outcomes

Completion rate is not enough. Measure correctness, user acceptance, review rate, repair rate, latency, cost, retries, and the frequency of unsafe or unauthorized proposals. Segment the results by version, model, customer cohort, input class, and failure mechanism. A higher automation rate can be a regression when it creates more corrections or delays discovery of mistakes. For ai agent memory design, use this as a concrete review criterion: connect the decision to its evidence, owner, and recovery path.

Roll out in stages

Use observation, shadow mode, suggestions, prepared drafts, approved execution, and only then broader automation. Give each stage an owner, cohort, success threshold, and rollback condition. Preserve the manual path until the automated path has demonstrated normal and degraded behavior. Keep in-flight work recoverable when a release is paused. For ai agent memory design, use this as a concrete review criterion: connect the decision to its evidence, owner, and recovery path.

Operate failure states

Make waiting, degraded, unknown, failed, cancelled, and completed distinct states. Tell the user whether the system is waiting for data, approval, a provider, a retry window, or manual repair. Give every waiting state an expiration or escalation path. Silent queues and ambiguous success messages are operational defects, not cosmetic issues. For ai agent memory design, use this as a concrete review criterion: connect the decision to its evidence, owner, and recovery path.

Review changes

Prompts, models, schemas, retrieval indexes, tools, providers, and policies are all dependencies. Version them and record their versions with each run. When behavior changes, compare a representative cohort with the previous baseline. Review quality, safety, latency, cost, and correction work together; optimizing one dimension can damage another. For ai agent memory design, use this as a concrete review criterion: connect the decision to its evidence, owner, and recovery path.

Example decision record

For a meaningful release, write a short record with the goal, affected workflow, expected improvement, known risks, evidence set, rollout stage, owner, rollback command, and review date. Link it to representative traces and the test cases that protect the most important boundary. This record turns a vague confidence claim into something another engineer can inspect. For ai agent memory design, use this as a concrete review criterion: connect the decision to its evidence, owner, and recovery path.

Practical checklist

Before launch, confirm that the workflow has a named owner, a clear contract, policy enforcement, safe side effects, representative tests, useful traces, and a recovery path. After launch, sample successful and failed runs, inspect corrections, update the case set, and retire metrics that do not change a decision. The goal is not maximal automation; it is a dependable user outcome that the team can explain and improve. For ai agent memory design, use this as a concrete review criterion: connect the decision to its evidence, owner, and recovery path.