Skip to content

Persisting an agent's intermediate memory or structured state

GenAI applications often need a persistent store to retain conversation history, agent memory, or structured intermediate outputs across sessions and requests. On Databricks, this is typically implemented using Delta tables in Unity Catalog, or lower-latency key-value/document stores integrated via Feature Serving or external online stores, so state survives beyond a single model invocation.

1 · Learn the must-know

  • Delta tables in Unity Catalog are the standard Databricks-native option for durable storage of chat history, session state, or structured memory, giving ACID transactions, governance, and lineage.
  • For low-latency read/write access needed in real-time agent interactions, Databricks Online Tables (backed by Feature Serving infrastructure) can expose Delta-managed data as a fast key-value store.
  • Unity Catalog governs access control and lineage for any persistent datastore used by an application, so tables storing memory or intermediate state should be registered as UC-managed assets rather than ad hoc storage.
  • Structured intermediate information (e.g., tool call outputs, retrieved context, agent scratchpad data) should be written using a defined schema so it can be queried, audited, and reused across pipeline runs.
  • Storing conversation/session memory outside the model call is essential for multi-turn agent applications, since the serving endpoint itself is stateless between invocations.
  • When persisting data from a Model Serving endpoint, writes should go through a governed path (e.g., a Databricks job, Lakehouse app backend, or feature-serving write path) rather than embedding storage credentials directly in the model artifact.

2 · Check your understanding

Check this objectiveFree · always available

A Generative AI Engineer is building a multi-turn support assistant deployed as a Databricks App. Returning users expect the assistant to recall context from conversations that took place days earlier. The current implementation keeps conversation history only in a Python variable inside the notebook session, so all history is lost whenever the app restarts. The engineer needs a durable, queryable store for each user's conversation history across restarts. Which approach should the engineer implement?

Your objective map0 tried · 0 answered correctly · 56 untouched

What you have tried across Databricks GenAI Engineer's objectives, not a readiness score.

Design Applications10.71% of the exam*0 of 6 tried
Data Preparation14.29% of the exam*0 of 8 tried
Application Development23.21% of the exam*0 of 13 tried
Assembling and Deploying Applications26.79% of the exam*0 of 15 tried
Governance7.14% of the exam*0 of 4 tried
Evaluation and Monitoring17.86% of the exam*0 of 10 tried

* Our estimate. Databricks publishes no section weights.

3 · Keep going