AI workflows
that follow your rules.

import { expr, workflow } from "rustyflows";
import { pricing } from "./pricing-rules";

const ctx = expr.context();
const approvedPct = ctx.approval.approvedPct.or(ctx.quote_decision.proposedPct);
const finalTotal = ctx.price.listTotalCents.discount(approvedPct);
const emailPayload = expr.object({
  to: ctx.input.customerEmail,
  cc: ctx.input.salesRep.or(ctx.approval.approver),
  subject: ctx.crm.quoteId.format("Quote % is ready"),
  body: ctx.reply.message,
  quoteId: ctx.crm.quoteId,
  quote: ctx.final_quote,
});

export const quoteAndBuy = workflow("quote-and-buy")
  .input({
    customerMessage: "string",
    customerEmail: "string",
    salesRep: "string",
  })
  .policy({
    maxRunCostUsd: 2,
    allowedModelProviders: ["anthropic"],
  })

  // Model extracts intent — schema-checked, nothing else gets through
  .generate("request", {
    agent: "product-advisor",
    prompt: "What does the customer need? {{input.customerMessage}}",
    output: { productNeed: "string", units: "number", requestedPct: "number" },
  })

  // Tool params are expression objects, not TypeScript callbacks
  .tool("catalog", {
    toolId: "catalog.search",
    params: expr.object({ query: ctx.request.productNeed }),
    output: { sku: "string", title: "string", unitPriceCents: "number" },
  })
  .tool("price", {
    toolId: "catalog.price",
    params: expr.object({
      sku: ctx.catalog.sku,
      units: ctx.request.units,
    }),
    output: { listTotalCents: "number", currency: "string" },
  })

  // Pricing writes ctx.quote_decision through the same expression IR
  .ruleset("quote_decision", pricing)

  // Gate condition and display fields are Rust-evaluated expressions
  .gate("approval", {
    type: "approval",
    when: ctx.quote_decision.decision.eq("needs_sales_approval"),
    show: expr.object({
      sku: ctx.catalog.sku,
      title: ctx.catalog.title,
      requestedPct: ctx.request.requestedPct,
      proposedPct: ctx.quote_decision.proposedPct,
      proposedTotalCents: ctx.price.listTotalCents.discount(ctx.quote_decision.proposedPct),
      reasons: ctx.quote_decision.reasons,
    }),
    output: { approved: "boolean", approvedPct: "number", approver: "string", note: "string" },
  })

  .map("final_quote", expr.object({
    sku: ctx.catalog.sku,
    title: ctx.catalog.title,
    finalDiscountPct: approvedPct,
    finalTotalCents: finalTotal,
    reasons: ctx.quote_decision.reasons,
  }))
  .tool("crm", {
    toolId: "crm.create-quote",
    params: ctx.final_quote,
    output: { quoteId: "string" },
  })
  .generate("reply", {
    agent: "sales-assistant",
    prompt: "Present quote {{crm.quoteId}}: {{final_quote}}. Why: {{quote_decision.reasons}}",
    output: { message: "string" },
  })
  .tool("send_email", {
    toolId: "email.send-quote",
    params: emailPayload,
    output: { status: '"sent" | "queued"' },
  });