#AI, GenAI and Agents on Databricks
You do not need to be an ML engineer, but a chief architect in 2026 who cannot architect a governed GenAI system is not credible. This is also where the platform is moving fastest β and where a customer's hardest current questions live.
#1. The argument that frames everything
"The differentiator in enterprise AI is no longer the model β it's the governed context you can feed it and the controls you can put around it."
Everyone has access to the same frontier models. What they do not have is your data, correctly permissioned, with lineage, with consistent business definitions, and with spend and safety controls at runtime. That is the entire lakehouse-for-AI argument, and it is the thesis Databricks organised its 2026 platform around: the competition moved from "who builds the best agent" to "who governs the data, context, cost and security underneath it". (Summit 2026 recap)
Learn that framing. It answers "why Databricks for AI" better than any feature list.
#2. The stack
| Layer | Component | What it does |
|---|---|---|
| Governance | Unity Catalog | Models, functions, vector indexes, volumes as governed objects with lineage |
| Runtime governance | Unity AI Gateway | Governs model/agent/tool traffic: spend visibility and hard caps, smart routing, contextual policies, PII/prompt-injection guardrails, MCP service registry, unified tracing |
| Models | Foundation Model APIs; external model endpoints | Pay-per-token or provisioned throughput; third-party models behind one governed interface |
| Serving | Mosaic AI Model Serving | Managed endpoints, autoscaling, scale-to-zero, A/B and shadow traffic |
| Retrieval | Mosaic AI Vector Search; Lakebase Search | Delta-synced vector indexes; hybrid vector + full-text retrieval in Postgres |
| Features | Feature engineering in UC | Offline/online consistency; point-in-time correctness |
| Build | Agent Bricks; Agent Framework | Build, evaluate and deploy agents; supports major agent SDKs |
| Orchestrate | Omnigent | Open-source meta-harness above agent frameworks: composition, control, collaboration |
| Lifecycle | MLflow | Tracking, model registry in UC, tracing, evaluation, prompt management |
| Analytics UX | Genie / Genie One | Natural-language analytics over governed data |
| Compute | AI Runtime (serverless GPU) | Training and deep learning without GPU infrastructure management |
#3. RAG β the architecture you must be able to draw
INGEST INDEX RETRIEVE GENERATE
ββββββββββββ parse/ ββββββββββββββ sync ββββββββββββββββ ββββββββββββββββ
β DocumentsβββββchunkββββΆβ Delta tableβββββββββββββΆβ Vector Searchββββββ β LLM endpoint β
β (volumes)β enrich β + metadata β (managed)β index β β β (FM API or β
ββββββββββββ ββββββββββββββ ββββββββββββββββ β β external) β
β β² βββββΆββββββββ¬ββββββββ
β permissions, β β β
β lineage, tags hybrid + filters β βΌ
βΌ β ββββββββββββββββ
UNITY CATALOG βββ access control applied ββββββββ β Response β
β + citations β
GOVERNED BY: Unity AI Gateway (budgets Β· routing Β· guardrails Β· tracing) ββββββββββββββββ
OBSERVED BY: MLflow tracing + evaluation Β· EVALUATED BY: LLM judges + ground-truth set
The five things that actually determine RAG quality β say these, not "we use vector search":
- Chunking strategy. Semantic/structural chunking beats fixed-size almost always. Chunk size and overlap are the highest-leverage tuning knobs and the most neglected.
- Retrieval quality, measured. Hybrid (vector + keyword) plus reranking. Measure recall@k against a labelled set before touching the prompt β most "the model is bad" complaints are retrieval failures.
- Metadata filtering and permissions. The index must respect who is asking. The failure mode that ends projects: a chatbot that cheerfully surfaces the HR folder. Filter at retrieval by the user's entitlements; never rely on the model to decline.
- Evaluation. A ground-truth question set, LLM judges for faithfulness/relevance/toxicity, tracked over time in MLflow. Without this you are shipping vibes.
- Grounding and citations. Every answer traceable to a source chunk. Non-negotiable in regulated industries and the thing that builds user trust fastest.
Asked: "Our RAG system gives bad answers. Where do you start?" "Retrieval before generation, every time. I'd take fifty real failing questions and check whether the correct chunk was in the retrieved set at all. If it wasn't, it's a chunking, embedding or filtering problem and the prompt is irrelevant. If it was there and the answer was still wrong, then it's a prompt, context-window or model problem. Teams almost always tune the prompt first and then wonder why nothing improves."
That answer is a reliable standout.
#4. Agents, and governing them
Agent Bricks provides a managed path to building, evaluating and deploying agents, with support for mainstream agent SDKs, managed memory, and sandboxed execution. Omnigent sits above agent harnesses (Claude Code, Codex, custom frameworks) as an open-source meta-harness β composition across harnesses, stateful policy control, and real-time human-agent collaboration β with a managed variant that inherits Unity Catalog and Unity AI Gateway governance. (Databricks docs, Unity AI Gateway)
The governance problem an architect must solve (and can now name concretely):
| Risk | Control |
|---|---|
| Runaway spend across teams and vendors | Unity AI Gateway: cost attribution per user/team/tool, hard spend caps, smart routing to cheaper models |
| An agent taking an unsafe action | Contextual service policies β allow / deny / require-approval on actions (push to a repo, write to sensitive storage), not just on data assets |
| Prompt injection, PII leakage, jailbreaks | Gateway guardrails on inputs and outputs |
| Tool sprawl | MCP service registry: a governed inventory of tools agents may call |
| No audit trail | Unified tracing of model and tool calls via MLflow; traces analysable downstream |
The distinction to state crisply: "Unity Catalog governs the assets β who may use this model, this tool, this table. Unity AI Gateway governs the interactions β what actually happens at runtime, what it costs, and what the agent is allowed to do. You need both; most organisations have neither and are running agents on an honour system."
#5. Evaluation β the discipline that separates pilots from production
- Offline: a curated question/answer set with expected behaviour; automated scoring on every change. Treat it as a test suite, because it is one.
- LLM judges: scoring faithfulness, relevance, harmfulness, and answer completeness at scale; calibrated against human labels on a sample.
- Online: user feedback, deflection/escalation rates, and traces β MLflow tracing gives per-request visibility into retrieval and generation steps.
- Regression gating: no prompt or model change ships without the eval suite passing. This is the single control that keeps a GenAI product stable, and almost nobody has it.
The line: "The reason most GenAI pilots don't reach production isn't model quality β it's that the team has no way to prove a change made things better. Evaluation is the thing I'd build second, right after retrieval, and before anyone touches the prompt."
#6. Natural-language analytics
Genie provides conversational analytics over governed data; Genie One (2026) extends this to a cross-application agentic interface producing documents and artefacts, with Genie Ontology as a continuously-built context layer grounding responses in enterprise knowledge, and Genie Agents turning conversations into reusable agents deployable into Slack and Teams. (Summit 2026)
What determines whether NL analytics succeeds β the honest architect's answer:
- A clean, well-modelled, well-documented gold layer. Natural language over a messy warehouse produces confident wrong answers, which is worse than no answers.
- Consistent metric definitions β this is exactly what Unity Catalog Metrics provides.
- Scoped context β point it at curated domains, not the entire catalog.
- Permissions that hold β the answer must respect the asker's entitlements.
"Genie is a forcing function for data modelling. If your semantics are ambiguous, the LLM will pick one interpretation and state it with total confidence. I'd scope it to a curated domain with governed metrics before I'd open it to the estate."
#7. Build vs buy vs fine-tune
A question you will be asked. Answer as a decision tree, in cost order:
- Prompt engineering with a strong base model β days, cheapest, try first.
- RAG β when the gap is knowledge, not behaviour. Most enterprise use cases stop here.
- Fine-tuning β when the gap is behaviour, format, tone or a domain idiom that prompting cannot reach; needs real labelled data and an eval harness, and it must be re-done as base models improve.
- Pre-training β essentially never for an enterprise, and saying so builds credibility.
"I'd need to see a RAG baseline with measured failure modes before agreeing to fine-tune. 'Fine-tune it' is usually a proposal to spend six weeks solving a retrieval problem."
#8. Questions to answer cold
- Draw a governed RAG architecture, including where permissions are enforced.
- RAG answers are poor β what do you check first, and why not the prompt?
- What does Unity AI Gateway govern that Unity Catalog does not?
- How do you stop agent spend running away across ten teams?
- When is fine-tuning justified?
- A customer wants NL analytics over their warehouse next month. What do you tell them?
- How do you evaluate a GenAI application, concretely?