AI workflow prototyping means building a rough, working version of a task an AI will perform, with its inputs, steps, approvals and failure states, and putting it in front of real users before engineering commits. You are not drawing screens. You are testing whether people understand what the AI is doing, trust the result, and can step in when it goes wrong. This guide walks through how to map the workflow, prototype it, design for errors, and test it.
TL;DR
- Start from the user's goal and the trigger that creates it, then map user actions, system actions, data and AI capabilities step by step.
- Prototype with working behavior and realistic data, not static screens. Static mockups hide the edge cases where AI workflows fail.
- Design the unhappy path as carefully as the happy one: wrong output, slow response, missing data, and decisions with real consequences.
- Keep people in control of consequential actions. Let the AI interpret and recommend, and let rules and approvals govern what actually happens.
- Test with task-based sessions and familiar data, then decide what to build, change or drop before a line of production code exists.
Table of Contents
- What AI workflow prototyping is, and how it differs from classic prototyping
- Step 1: Map the workflow from the user's goal
- Step 2: Choose what the prototype must prove
- Step 3: Pick the right fidelity
- Step 4: Design the unhappy path first
- Agentic workflow examples worth prototyping
- AI UX patterns that make the workflow understandable
- Step 5: Test the prototype with users
- A one-week prototyping plan
- Our perspective
- How we can help
- FAQ
- Sources
What AI workflow prototyping is, and how it differs from classic prototyping
A conventional software workflow follows a predefined path. The same input produces the same output, so a clickable mockup is a fair stand-in for the real thing. An AI workflow does not behave that way. IBM's February 2026 analysis separates deterministic workflows, which produce predictable results from predefined paths, from agentic workflows that reason, plan, use tools, adapt to ambiguous inputs and therefore behave probabilistically.
That difference changes what a prototype has to prove. With a normal feature, you ask whether people can find and complete the steps. With an AI workflow, you also ask:
- Does the user understand what the AI is about to do and why?
- Is the output good enough, often enough, that people keep using it?
- What happens to confidence when the output is wrong?
- Who approves what, and at which step?
A picture of a successful result answers none of these. A prototype that actually produces varied results, including bad ones, answers all of them.
This is why "AI workflow prototyping" is not the same as using AI tools to draw interfaces faster. Many pages on the topic compare tools that generate screens from prompts. That is useful for speed. This article is about the other problem: prototyping the behavior of an AI that does work on a user's behalf.
Step 1: Map the workflow from the user's goal
Before any prototype, write down the workflow. SAP's AppHaus describes a business AI design workshop in which teams define the goal, trace back to the trigger event, identify pain points, and then map the future experience step by step through user actions, system actions, required data and AI capabilities. That structure works well because it forces every AI step to justify itself against a real pain point.
A practical version for a product team:
- Goal. One sentence in the user's words. "Get a qualified lead list into my CRM without cleaning it by hand."
- Trigger. What event starts the job? A new sign-up, a file upload, a Monday morning report, a support ticket.
- Current pain. Where time, errors or abandonment happen today. Use real sessions, tickets or recordings, not guesses.
- Future steps. A table with one row per step: who acts (user, AI, system), what data is needed, what the AI is allowed to do, and what the user sees.
- Decision points. Mark every step where the AI's output changes something outside the product: sends an email, edits a record, spends money.
A short example of how the map reads for an AI that drafts replies to inbound sales inquiries:
| Step | Actor | Data needed | AI capability | What the user sees |
|---|---|---|---|---|
| Inquiry arrives | System | Form fields, company name | None | Inbox item |
| Context gathered | AI | CRM history, public company info | Retrieval, summarizing | A short brief with sources |
| Reply drafted | AI | Brief, tone rules, pricing rules | Generation | Draft with highlighted assumptions |
| Review | User | Draft, brief | None | Edit, approve or reject |
| Send and log | System | Approved draft | None | Confirmation, undo window |
The map often reveals that only two or three steps need AI at all. That is a good result. It tells you where to spend prototyping effort.
Step 2: Choose what the prototype must prove
You rarely need to prototype the whole workflow. Pick the riskiest assumption and build the smallest thing that tests it. Common choices:
- Comprehension: will users understand what the AI did and why?
- Quality: is the output good enough on the user's own data?
- Control: do users know where to step in, and do they use it?
- Value: does the result save enough time or money to be worth the effort of reviewing it?
This is the same discipline we apply to free trials: find the step where people stop, and fix that first. In an AI product, the equivalent drop-off is the moment a user sees a wrong or unexplained result and stops trusting the feature.
Step 3: Pick the right fidelity
Fidelity should follow the question, not the other way round. GitLab's design handbook describes two validation approaches for generative AI solutions: moderated low-fidelity Figma prototypes to understand workflow context, and unmoderated interactive LLM prototypes for non-linear engagement.
In practice, that gives you a ladder:
| Fidelity | What it is | Best for | Weak at |
|---|---|---|---|
| Paper or slide flow | Annotated screens of each step | Agreeing on the workflow, finding missing steps | Anything about output quality |
| Wizard-of-Oz | A person plays the AI behind the scenes | Testing expectations and trust before building anything | Scale, speed, latency |
| Prompt-driven prototype | A working LLM call with real prompts and sample data | Output quality, variation, edge cases | Integrations, permissions |
| Coded prototype | A small app with real data connections and approvals | End-to-end behavior, stakeholder buy-in | Production-grade security and scale |
Start at the lowest rung that can answer your question. Move up only when the answer depends on real behavior.
Why working behavior beats mockups
Stakeholders react differently to something that behaves. When a sponsor can type in a messy request and watch the AI handle it badly, the conversation shifts from opinions about screens to decisions about the workflow. Teams that want a hand with this stage can look at prototype creation and delivery, where interacting with functioning software and its edge cases gives stakeholders far more concrete feedback than a static review does.
Use realistic data. Fake data makes every AI look good. Real, messy, familiar data shows where it breaks.
Step 4: Design the unhappy path first
Most AI prototypes show one beautiful success. Production use is mostly the rest. GitLab's handbook notes that AI user flows must define both happy paths, where the goal is achieved, and unhappy paths, where the user recovers from unexpected or failed AI outputs.
For each AI step in your map, answer these questions and put the answers in the prototype:
- Wrong output. How does the user see it, correct it, and tell the system it was wrong?
- No output. What if the AI has too little data? Say what is missing and offer a way to supply it.
- Slow output. What does the user see while waiting? Show progress in terms of the task ("Reading 14 documents") rather than a generic spinner.
- Partial output. Can the user accept the good parts and fix the rest?
- Reversal. Is there an undo, and for how long?
- Escalation. When should the workflow stop and hand over to a person?
Keep consequential actions under control
The most useful design principle for production AI is a deliberate hybrid. In IBM's view, probabilistic agents handle interpretation and recommendations, while deterministic rules, observability, governance and human approval control consequential actions. Translated into prototype terms: the AI can read, summarize, classify and propose freely, but anything that sends, spends, deletes or changes a customer record passes through a rule or a person.

Celigo, describing its agent builder, explains that it implemented guardrails as independent input and output controls that sit outside the agent's core reasoning, so that safety mechanisms stay intact if the agent fails and the interface can remain clear. Even in a prototype, show where those controls live. If a reviewer cannot see the boundary between what the AI decides and what a rule enforces, the design is not ready.
Agentic workflow examples worth prototyping
The word "agentic" gets stretched. A useful test: the system takes more than one step toward a goal, chooses between tools or paths on its own, and you can point to the moment a person approves or overrides it. Examples that fit across B2B SaaS:
- Lead and inquiry triage. The AI reads an incoming request, pulls CRM and public context, drafts a reply, and routes it. The user approves each draft until the team trusts a category enough to send without review.
- Trial setup assistant. The AI asks two or three questions, then configures the workspace with sample data, integrations and a first project. The user sees a preview and confirms before anything is written. This is where AI most directly serves activation, because setup that comes before the first result is where trials lose people. Our guide to onboarding flows that get users to a first result covers the underlying patterns.
- Support resolution. The AI classifies a ticket, searches the knowledge base, proposes an answer and, for refunds or account changes, stops for human approval.
- Report and insight generation. The AI assembles a weekly summary from several data sources, flags anomalies, and links each claim to its source row.
- Back-office reconciliation. The AI matches records, proposes corrections and queues exceptions for a reviewer.
For each, the prototype should show four things: the task the AI performs, what the user can steer, which steps require approval, and how an error is recovered. For more on how teams build and measure these, see our pieces on AI agents in SaaS and workflow automation for SaaS.
AI UX patterns that make the workflow understandable
These patterns recur in well-designed AI products. Use them as a checklist when you review a prototype, then test whether they work for your users.
Before the AI acts
- State the plan. Show what the AI intends to do in plain language, in order. Users steer better when they can see the intent.
- Offer scoped controls. Let users limit sources, tone, record types or spending before the run, not after.
- Set expectations on time and quality. If the run takes a minute, say so.
While and after it acts
- Show sources and assumptions. Each claim links to the data it came from. Highlight what the AI guessed.
- Make review fast. Present the output so a reviewer can decide quickly: the proposed change, the evidence, and three clear actions (approve, edit, reject). Approving with edits means a reviewer does not have to restart the whole process.
- Keep a visible history. A log of what the AI did, when, and with which input, so anyone can trace a result.
When something goes wrong
- Undo and versioning. Reversible by default, with a stated window.
- Plain-language errors. Say what failed and what the user can do next.
- Feedback that matters. A "wrong" button that visibly changes what happens next is worth more than a generic thumbs-down.
Step 5: Test the prototype with users
Prototype testing for AI differs from usability testing of a normal flow because the output varies between sessions. Plan for that.
Use task-based sessions. Give participants a real job, such as "get these 30 inquiries answered", rather than a tour of features. Ask them to think aloud. One practical guide for product managers recommends task-based experiments, thinking aloud, and production-like tests with familiar real data in place of generic or fake data.
Test the backing workflow too. Oracle's guidance on testing agentic apps says to test both the app's user-facing experience and the workflows behind it, with approval criteria that read like business rules. Write your acceptance criteria that way: "A refund over the approval limit is never issued without a named approver" is testable. "The AI should be careful" is not.
Vary the output quality on purpose. Run one session where the AI is right nearly every time, and another where it makes visible mistakes. Watch what users do when a result is wrong. Do they notice? Do they know how to fix it? Do they keep using the feature afterward? This reveals where confidence breaks, and which fallback messages you need.

Write the decision before you test. For each assumption, define in advance what result means build, change or drop. For example: "If fewer than most participants catch a wrong draft before approving, the review step needs redesign before we build." Setting the bar first stops the team from reading every result as a success.
What to measure
| Question | Signal to watch |
|---|---|
| Do users understand the AI's role? | They can explain, unprompted, what it did and what it did not do |
| Is the output useful? | Share of outputs accepted without edits, accepted with edits, rejected |
| Is review efficient? | Time to decide on each item |
| Does trust survive errors? | Whether users continue to use the feature after a visible mistake |
| Is control clear? | Whether users find and use approval, undo and steer controls |
Treat these as prototype signals, not forecasts. They tell you what to fix before building, and what to watch once the feature ships.
A one-week prototyping plan
Teams often assume this takes a quarter. A focused first pass can be compact:
- Day 1. Workshop the goal, trigger, pain points and future steps. Agree on the single riskiest assumption.
- Day 2. Draft the workflow map and mark every decision point. Collect real sample data.
- Days 3 and 4. Build the smallest working prototype: prompts, sample data, a review screen, an undo.
- Day 5. Run five task-based sessions, record decisions, and write up what to build, change or drop.
The point is a decision, not a demo. At the end you should know which AI step earns its place, what controls it needs, and what to put in the spec.
Our perspective
Most teams we meet have more AI ideas than they can fund, and they pick the one with the best demo. We think the better test is which workflow pays for itself once real users, real data and real mistakes are involved.
That is why we prototype behavior before we design screens. A working prototype exposes the step where a user stops trusting the AI, the approval that is missing, and the setup that delays the first useful result. These are the same drop-offs we find when we go through a free trial as a buyer with a stopwatch: the product asks for effort before it delivers a win.
So we map the workflow, name the one idea worth building, prototype it with realistic data, and tie every design change to observed user behavior rather than a hunch. Then we hand your team specs and front-end it can ship. If an AI feature is meant to lift activation or upgrades, we judge it by that number.
Edijs
How we can help
If you have an AI feature in mind, or a trial where users leave before the first result, we start with AI + UX Discovery: we map the workflow and name the one idea that pays for itself. From there we can move into AI agents and workflows, starting with a working prototype, then specs and front-end your team can ship. You can see the full range on our services page.

FAQ
How is AI workflow prototyping different from AI-assisted prototyping?
AI-assisted prototyping uses AI tools to produce screens or code faster. AI workflow prototyping tests the behavior of an AI that performs a task for the user: its steps, approvals, errors and the user's trust. You can use AI tools for the first to speed up the second, but they answer different questions.
Do I need engineers to prototype an AI workflow?
Not at the start. A slide flow or a Wizard-of-Oz session, where a person plays the AI, tests comprehension and trust without code. You need working prompts and real data when the question is output quality, and a coded prototype when you need to see end-to-end behavior with approvals and integrations.
How many users do I need to test with?
For a first pass, a handful of participants doing real tasks is usually enough to expose the biggest problems in the workflow, because the same breakdowns tend to repeat. Add more sessions when you compare variants or when different user roles perform the task differently.
Which steps should always require human approval?
Any step that sends something outside the product, spends money, deletes data or changes a customer record. The AI can propose these actions, but a rule or a person should authorize them. The right threshold depends on the risk in your product, so define it in business-rule language and test it.
Sources
- Building and evaluating AI agents that work in the real world, IBM
- A Guide for Designing with AI, The GitLab Handbook
- Business AI Design Workshop, SAP AppHaus
- Inside the Design of Celigo's Agent Builder, Celigo
- ATLAS: Test an Agentic App and its agents thoroughly, Oracle
- AI Prototyping: The Ultimate Guide For Product Managers, Product Compass
