react-o11y

@assistant-ui/react-o11y provides headless, Radix-style primitives for rendering observability spans as collapsible, indented hierarchical UIs. Use it to build trace inspectors, waterfall debug panels, or LLM call timelines on top of your own span data source.

• Headless — Zero styling opinions, bring your own CSS / Tailwind.
• Composable — Radix-style primitive parts you fully control.
• Tree-aware — Automatic depth, parent / child collapse, time range computation.
• Tap-powered — Built on the same @assistant-ui/tap reactive primitives as the runtimes.

Installation

npm install @assistant-ui/react-o11y

Quick start

Mount SpanResource with your span data, then render primitives that read from the resource's state:

"use client";

import {
  SpanPrimitive,
  SpanResource,
  type SpanData,
} from "@assistant-ui/react-o11y";
import { useAui, AuiProvider } from "@assistant-ui/store";

function SpanRow() {
  return (
    <SpanPrimitive.Root className="flex items-center gap-2 py-1">
      <SpanPrimitive.Indent />
      <SpanPrimitive.CollapseToggle className="size-4 cursor-pointer">
        ▸
      </SpanPrimitive.CollapseToggle>
      <SpanPrimitive.StatusIndicator className="size-2 rounded-full data-[span-status=running]:bg-yellow-500 data-[span-status=completed]:bg-green-500 data-[span-status=failed]:bg-red-500" />
      <SpanPrimitive.TypeBadge className="rounded bg-muted px-1.5 text-xs" />
      <SpanPrimitive.Name className="text-sm" />
      <SpanPrimitive.Children components={{ Span: SpanRow }} />
    </SpanPrimitive.Root>
  );
}

export function TraceView({ spans }: { spans: SpanData[] }) {
  const aui = useAui({ resource: SpanResource({ spans }) });

  return (
    <AuiProvider value={aui}>
      <SpanPrimitive.Children components={{ Span: SpanRow }} />
    </AuiProvider>
  );
}

SpanData is the input shape your code produces (see SpanData). SpanResource enriches it with depth, child counts, time range, and collapse state.

Anatomy

import {
  SpanPrimitive,
  SpanResource,
  SpanByIndexProvider,
} from "@assistant-ui/react-o11y";

// Top-level: mount SpanResource via useAui
const aui = useAui({ resource: SpanResource({ spans }) });

<AuiProvider value={aui}>
  {/* Render visible spans (flat-list output respecting collapsed state) */}
  <SpanPrimitive.Children components={{ Span: SpanRow }} />
</AuiProvider>;

// Each <SpanRow /> sits inside its own SpanByIndexProvider context and reads via primitives
function SpanRow() {
  return (
    <SpanPrimitive.Root>
      <SpanPrimitive.Indent />
      <SpanPrimitive.CollapseToggle />
      <SpanPrimitive.StatusIndicator />
      <SpanPrimitive.TypeBadge />
      <SpanPrimitive.Name />
    </SpanPrimitive.Root>
  );
}

SpanPrimitive.Children produces a flat list of visible spans (collapsed subtrees are excluded). Each rendered child is wrapped in a SpanByIndexProvider so its primitives resolve to that specific span's state.

API reference

SpanResource

Tap resource that ingests raw span data and exposes a tree-aware reactive state to primitives.

SpanResource({ spans }: { spans: SpanData[] }): ClientOutput<"span">;

Mount through useAui({ resource: SpanResource({ spans }) }).

The resource computes:

• depth of each span based on parent chain.
• hasChildren flag.
• timeRange (min / max across all spans).
• collapse state managed internally; toggles through primitives.

SpanData

Input shape you provide to SpanResource:

SpanState

Returned by useAuiState((s) => s.span) inside any span-scoped subtree:

type SpanState = {
  id: string;
  parentSpanId: string | null;
  name: string;
  type: string;
  status: "running" | "completed" | "failed" | "skipped";
  startedAt: number;
  endedAt: number | null;
  latencyMs: number | null;
  depth: number;
  hasChildren: boolean;
  isCollapsed: boolean;
  children: SpanItemState[];
  timeRange: { min: number; max: number };
};

SpanPrimitive.Root

Container <div> exposing span state via data attributes for styling:

Accepts all standard <div> props.

SpanPrimitive.Indent

A <div> that adds horizontal padding proportional to span depth.

Final paddingLeft = baseIndent + depth * indentPerLevel.

SpanPrimitive.CollapseToggle

A <button> that toggles the current span's collapsed state. Renders only when the span has children. Stops click propagation so it can be nested inside a clickable row.

Exposes data-collapsed for styling. Accepts all standard <button> props.

SpanPrimitive.StatusIndicator

A <span> that exposes data-span-status for status-based styling (color, icon swap, etc.). Renders no built-in glyph; you provide the visual.

<SpanPrimitive.StatusIndicator className="size-2 rounded-full data-[span-status=running]:bg-yellow-500 data-[span-status=completed]:bg-green-500 data-[span-status=failed]:bg-red-500 data-[span-status=skipped]:bg-gray-400" />

SpanPrimitive.TypeBadge

A <span> that defaults its children to span.type (override by passing custom children). Exposes data-span-type.

SpanPrimitive.Name

A <span> that defaults its children to span.name. Override by passing custom children.

SpanPrimitive.Children

Iterates over visible child spans and renders each.

Two usage modes:

Render-prop:

<SpanPrimitive.Children>
  {({ span }) => <CustomRow span={span} />}
</SpanPrimitive.Children>

Component prop (recommended for performance):

<SpanPrimitive.Children components={{ Span: SpanRow }} />

The components.Span form is memoized; pass a stable component reference to avoid re-renders on unrelated state changes.

Each child is wrapped in SpanByIndexProvider automatically, so primitives inside SpanRow read that child's state.

SpanByIndexProvider

Lower-level escape hatch. Selects one span by index and provides its scope to descendants:

<SpanByIndexProvider index={0}>
  <SpanPrimitive.Root>{/* reads span at index 0 */}</SpanPrimitive.Root>
</SpanByIndexProvider>

Most users do not need to use this directly; SpanPrimitive.Children wires it up automatically.

SpanPrimitive.ChildByIndex

Convenience component that pairs SpanByIndexProvider with a render component:

<SpanPrimitive.ChildByIndex index={0} components={{ Span: SpanRow }} />

Useful when you want explicit control over which child indices to render (e.g. for virtualization).

Styling

All primitives forward refs and accept standard DOM props (className, style, etc.). Use the data attributes for state-based styling:

[data-span-status="running"] {
  /* yellow indicator */
}
[data-span-status="failed"] {
  /* red indicator */
}
[data-collapsed="true"] svg {
  transform: rotate(0deg);
}
[data-collapsed="false"] svg {
  transform: rotate(90deg);
}

Related