@tangle-network/ui 11.0.0 → 11.1.0

package/src/chat/agent-timeline.test.tsx

@@ -126,6 +126,40 @@ describe("AgentTimeline tool actions", () => {
     ).toBeInTheDocument()
   })
 
+  it("renders every row and no toggle when collapseAfter is unset", () => {
+    const items: AgentTimelineItem[] = Array.from({ length: 5 }, (_, i) => ({
+      id: `s-${i}`,
+      kind: "status",
+      label: `Step ${i}`,
+    }))
+    render(<AgentTimeline items={items} />)
+    expect(screen.getByText("Step 4")).toBeInTheDocument()
+    expect(
+      screen.queryByRole("button", { name: /more step/i }),
+    ).not.toBeInTheDocument()
+  })
+
+  it("collapses to the first N spine rows behind a toggle and reveals the rest", async () => {
+    const user = userEvent.setup()
+    const items: AgentTimelineItem[] = Array.from({ length: 5 }, (_, i) => ({
+      id: `s-${i}`,
+      kind: "status",
+      label: `Step ${i}`,
+    }))
+    render(<AgentTimeline items={items} collapseAfter={2} />)
+
+    // First 2 visible, the other 3 hidden behind the toggle
+    expect(screen.getByText("Step 0")).toBeInTheDocument()
+    expect(screen.getByText("Step 1")).toBeInTheDocument()
+    expect(screen.queryByText("Step 4")).not.toBeInTheDocument()
+
+    const toggle = screen.getByRole("button", { name: /show 3 more steps/i })
+    await user.click(toggle)
+
+    expect(screen.getByText("Step 4")).toBeInTheDocument()
+    expect(screen.getByRole("button", { name: /show less/i })).toBeInTheDocument()
+  })
+
   it("renders the source part's real input in the expanded detail", async () => {
     const user = userEvent.setup()
     const probePart: ToolPart = {

package/src/chat/agent-timeline.tsx

@@ -1,4 +1,4 @@
-import { type KeyboardEvent, type ReactNode } from "react";
+import { useState, type KeyboardEvent, type ReactNode } from "react";
 import {
   AlertTriangle,
   CheckCircle2,
@@ -88,6 +88,9 @@ export interface AgentTimelineProps {
   /** Optional actions rendered beside each tool item (e.g. "open in artifacts").
    *  Receives the source tool part carried on the item. */
   renderToolActions?: (part: ToolPart) => ReactNode;
+  /** When set, collapse the timeline to the first N spine rows behind a
+   *  "Show N more steps" toggle. Omit to always show every row. */
+  collapseAfter?: number;
 }
 
 const TONE_STYLES: Record<AgentTimelineTone, { dot: string; card: string; text: string; icon: typeof Info }> = {
@@ -130,17 +133,24 @@ function formatTime(date: Date): string {
 interface AgentTimelineRowProps {
   isLast: boolean;
   accentClassName: string;
+  /** Dot top-offset. Defaults to aligning with the first text line; card rows
+   * pass CARD_ROW_DOT to center the dot on the card header instead. */
+  dotClassName?: string;
   children: ReactNode;
 }
 
-function AgentTimelineRow({ isLast, accentClassName, children }: AgentTimelineRowProps) {
+// Centers the dot on a RunRowShell card header (2.5rem: h-6 badge + py-2),
+// so tool and reasoning rows read as dot-aligned rather than top-anchored.
+const CARD_ROW_DOT = "mt-[calc((2.5rem-var(--timeline-dot-size))/2)]";
+
+function AgentTimelineRow({ isLast, accentClassName, dotClassName = "mt-2", children }: AgentTimelineRowProps) {
   return (
     <div className="grid grid-cols-[1.25rem_minmax(0,1fr)] gap-x-4">
       <div className="relative flex justify-center">
         {!isLast && (
           <span className="absolute top-4 bottom-[-0.75rem] left-1/2 w-px -translate-x-1/2 bg-border" />
         )}
-        <span className={cn("relative mt-2 h-[var(--timeline-dot-size)] w-[var(--timeline-dot-size)] rounded-full ring-4 ring-[var(--bg-root)]", accentClassName)} />
+        <span className={cn("relative h-[var(--timeline-dot-size)] w-[var(--timeline-dot-size)] rounded-full ring-4 ring-[var(--bg-root)]", dotClassName, accentClassName)} />
       </div>
       <div className="min-w-0 pb-3">{children}</div>
     </div>
@@ -245,7 +255,10 @@ export function AgentTimeline({
   emptyState,
   className,
   renderToolActions,
+  collapseAfter,
 }: AgentTimelineProps) {
+  const [expanded, setExpanded] = useState(false);
+
   if (items.length === 0 && !isThinking) {
     return emptyState ? (
       <div className={cn("flex h-full items-center justify-center p-4", className)}>
@@ -258,13 +271,41 @@ export function AgentTimeline({
     ? [...items, { id: "__thinking__", kind: "custom", content: <ThinkingIndicator /> }]
     : items;
 
-  // Determine which items participate in the timeline connector (non-user-message items)
-  // User messages are rendered outside the timeline grid
-  const timelineItems = renderedItems.filter((item) => !(item.kind === "message" && item.role === "user"));
+  const isUserMessage = (item: AgentTimelineItem) =>
+    item.kind === "message" && item.role === "user";
+
+  // Items on the vertical connector (user messages render off-spine).
+  const timelineItems = renderedItems.filter((item) => !isUserMessage(item));
+
+  const limit = collapseAfter ?? Infinity;
+  const collapsible = timelineItems.length > limit;
+  const collapsed = collapsible && !expanded;
+
+  // While collapsed, keep only the first `limit` spine rows (and any user
+  // messages that precede them); the rest hides behind the toggle.
+  let renderList = renderedItems;
+  let hiddenCount = 0;
+  if (collapsed) {
+    const visible: AgentTimelineItem[] = [];
+    let spineCount = 0;
+    for (const item of renderedItems) {
+      if (spineCount >= limit) break;
+      visible.push(item);
+      if (!isUserMessage(item)) spineCount += 1;
+    }
+    renderList = visible;
+    hiddenCount = timelineItems.length - limit;
+  }
+
+  const visibleSpine = renderList.filter((item) => !isUserMessage(item));
+  // When a toggle row follows, no real row is last — the toggle owns the tail.
+  const lastSpineItem = collapsible
+    ? undefined
+    : visibleSpine[visibleSpine.length - 1];
 
   return (
     <div className={cn("mx-auto w-full max-w-5xl px-4 py-4", className)}>
-      {renderedItems.map((item, index) => {
+      {renderList.map((item) => {
         // User messages: right-aligned bubble, off-spine — needs its own vertical
         // rhythm so it doesn't sit flush against the status/tool/agent row below.
         if (item.kind === "message" && item.role === "user") {
@@ -275,8 +316,7 @@ export function AgentTimeline({
           );
         }
 
-        const timelineIndex = timelineItems.indexOf(item);
-        const isLast = timelineIndex === timelineItems.length - 1;
+        const isLast = item === lastSpineItem;
 
         if (item.kind === "message") {
           return (
@@ -288,7 +328,7 @@ export function AgentTimeline({
 
         if (item.kind === "tool") {
           return (
-            <AgentTimelineRow key={item.id} isLast={isLast} accentClassName="bg-[var(--border-hover)]">
+            <AgentTimelineRow key={item.id} isLast={isLast} accentClassName="bg-[var(--border-hover)]" dotClassName={CARD_ROW_DOT}>
               <ToolCallStep
                 type={item.call.type}
                 label={item.call.label}
@@ -305,7 +345,7 @@ export function AgentTimeline({
 
         if (item.kind === "tool_group") {
           return (
-            <AgentTimelineRow key={item.id} isLast={isLast} accentClassName="bg-[var(--border-hover)]">
+            <AgentTimelineRow key={item.id} isLast={isLast} accentClassName="bg-[var(--border-hover)]" dotClassName={CARD_ROW_DOT}>
               <ToolCallGroup title={item.title}>
                 {item.calls.map((call, callIndex) => {
                   const part = item.parts?.[callIndex];
@@ -354,11 +394,24 @@ export function AgentTimeline({
 
         // custom
         return (
-          <AgentTimelineRow key={item.id} isLast={isLast} accentClassName="bg-[var(--border-hover)]">
+          <AgentTimelineRow key={item.id} isLast={isLast} accentClassName="bg-[var(--border-hover)]" dotClassName={CARD_ROW_DOT}>
             {(item as AgentTimelineCustomItem).content}
           </AgentTimelineRow>
         );
       })}
+      {collapsible ? (
+        <AgentTimelineRow isLast accentClassName="bg-[var(--border-hover)]">
+          <button
+            type="button"
+            onClick={() => setExpanded((value) => !value)}
+            className="text-xs font-medium text-muted-foreground transition-colors hover:text-foreground"
+          >
+            {collapsed
+              ? `Show ${hiddenCount} more step${hiddenCount === 1 ? "" : "s"}`
+              : "Show less"}
+          </button>
+        </AgentTimelineRow>
+      ) : null}
     </div>
   );
 }

package/src/chat/chat-container.tsx

@@ -63,6 +63,9 @@ export interface ChatContainerProps {
   /** Optional actions rendered beside individual tool items in the `timeline`
    *  presentation, which has no run/message grouping (hence part-only). */
   renderTimelineToolActions?: (part: ToolPart) => ReactNode;
+  /** In the `timeline` presentation, collapse to the first N spine rows behind
+   *  a "Show N more steps" toggle. Omit to always show every row. */
+  collapseTimelineAfter?: number;
 }
 
 const OPENUI_NODE_TYPES = new Set([
@@ -373,6 +376,7 @@ export const ChatContainer = memo(
     renderUserMessageActions,
     renderToolActions,
     renderTimelineToolActions,
+    collapseTimelineAfter,
   }: ChatContainerProps) => {
     const scrollRef = useRef<HTMLDivElement>(null);
 
@@ -414,6 +418,7 @@ export const ChatContainer = memo(
                 items={timeline.items}
                 isThinking={timeline.showThinking}
                 renderToolActions={renderTimelineToolActions}
+                collapseAfter={collapseTimelineAfter}
               />
             </div>
           ) : (

package/src/run/expandable-content.test.tsx

@@ -0,0 +1,52 @@
+import { render, screen } from "@testing-library/react";
+import userEvent from "@testing-library/user-event";
+import { afterEach, beforeEach, describe, expect, it, vi } from "vitest";
+import { ExpandableContent } from "./expandable-content";
+
+describe("ExpandableContent", () => {
+  let scrollHeightSpy: ReturnType<typeof vi.spyOn> | undefined;
+
+  afterEach(() => {
+    scrollHeightSpy?.mockRestore();
+    scrollHeightSpy = undefined;
+  });
+
+  it("renders no toggle when content fits within the cap", () => {
+    render(
+      <ExpandableContent collapsedMaxPx={180}>
+        <div>short</div>
+      </ExpandableContent>,
+    );
+    expect(
+      screen.queryByRole("button", { name: /show more/i }),
+    ).not.toBeInTheDocument();
+  });
+
+  describe("when content overflows", () => {
+    beforeEach(() => {
+      scrollHeightSpy = vi
+        .spyOn(HTMLElement.prototype, "scrollHeight", "get")
+        .mockReturnValue(999);
+    });
+
+    it("clamps with a Show more toggle that expands and collapses", async () => {
+      const user = userEvent.setup();
+      render(
+        <ExpandableContent collapsedMaxPx={180}>
+          <div>tall content</div>
+        </ExpandableContent>,
+      );
+
+      const showMore = screen.getByRole("button", { name: /show more/i });
+      await user.click(showMore);
+      expect(
+        screen.getByRole("button", { name: /show less/i }),
+      ).toBeInTheDocument();
+
+      await user.click(screen.getByRole("button", { name: /show less/i }));
+      expect(
+        screen.getByRole("button", { name: /show more/i }),
+      ).toBeInTheDocument();
+    });
+  });
+});

package/src/run/expandable-content.tsx

@@ -0,0 +1,73 @@
+import { useEffect, useRef, useState, type ReactNode } from "react";
+import { cn } from "../lib/utils";
+
+export interface ExpandableContentProps {
+  children: ReactNode;
+  /** Height (px) the content is clamped to while collapsed. */
+  collapsedMaxPx?: number;
+  /**
+   * Gradient from-color for the bottom fade, so it blends with the surface
+   * behind the content (e.g. `from-muted`, `from-card`, `from-[var(--surface-danger-bg)]`).
+   */
+  fadeClassName?: string;
+  className?: string;
+}
+
+/**
+ * Clamps tall content to a fixed height with a bottom fade and a
+ * "Show more / Show less" toggle. Content shorter than the cap renders
+ * as-is with no toggle.
+ */
+export function ExpandableContent({
+  children,
+  collapsedMaxPx = 180,
+  fadeClassName = "from-muted",
+  className,
+}: ExpandableContentProps) {
+  const [expanded, setExpanded] = useState(false);
+  const [overflowing, setOverflowing] = useState(false);
+  const contentRef = useRef<HTMLDivElement>(null);
+
+  useEffect(() => {
+    const el = contentRef.current;
+    if (!el) return;
+    const measure = () =>
+      setOverflowing(el.scrollHeight > collapsedMaxPx + 4);
+    measure();
+    if (typeof ResizeObserver === "undefined") return;
+    const observer = new ResizeObserver(measure);
+    observer.observe(el);
+    return () => observer.disconnect();
+  }, [collapsedMaxPx]);
+
+  const clamped = overflowing && !expanded;
+
+  return (
+    <div className={className}>
+      <div
+        ref={contentRef}
+        className="relative overflow-hidden"
+        style={clamped ? { maxHeight: collapsedMaxPx } : undefined}
+      >
+        {children}
+        {clamped ? (
+          <div
+            className={cn(
+              "pointer-events-none absolute inset-x-0 bottom-0 h-10 bg-gradient-to-t to-transparent",
+              fadeClassName,
+            )}
+          />
+        ) : null}
+      </div>
+      {overflowing ? (
+        <button
+          type="button"
+          onClick={() => setExpanded((value) => !value)}
+          className="mt-1 text-[11px] font-medium text-muted-foreground transition-colors hover:text-foreground"
+        >
+          {expanded ? "Show less" : "Show more"}
+        </button>
+      ) : null}
+    </div>
+  );
+}

package/src/run/expanded-tool-detail.tsx

@@ -17,6 +17,7 @@ import { WebSearchPreview } from "../tool-previews/web-search-preview";
 import { QuestionPreview } from "../tool-previews/question-preview";
 import { DiffPreview } from "../tool-previews/diff-preview";
 import { cn } from "../lib/utils";
+import { ExpandableContent } from "./expandable-content";
 
 export interface ExpandedToolDetailProps {
   part: ToolPart;
@@ -146,15 +147,17 @@ export const ExpandedToolDetail = memo(({ part }: ExpandedToolDetailProps) => {
               Output
             </span>
           </div>
-          <CodeBlock
-            code={
-              outputStr.length > 2000
-                ? outputStr.slice(0, 2000) + "\n...(truncated)"
-                : outputStr
-            }
-            language="json"
-            className="rounded-none border-0"
-          />
+          <ExpandableContent fadeClassName="from-card">
+            <CodeBlock
+              code={
+                outputStr.length > 2000
+                  ? outputStr.slice(0, 2000) + "\n...(truncated)"
+                  : outputStr
+              }
+              language="json"
+              className="rounded-none border-0"
+            />
+          </ExpandableContent>
         </div>
       )}
 
@@ -167,9 +170,11 @@ export const ExpandedToolDetail = memo(({ part }: ExpandedToolDetailProps) => {
               Error
             </span>
           </div>
-          <pre className="bg-[var(--surface-danger-bg)] p-3 text-xs font-mono whitespace-pre-wrap break-all text-[var(--surface-danger-text)]">
-            {error}
-          </pre>
+          <ExpandableContent fadeClassName="from-[var(--surface-danger-bg)]">
+            <pre className="bg-[var(--surface-danger-bg)] p-3 text-xs font-mono whitespace-pre-wrap break-all text-[var(--surface-danger-text)]">
+              {error}
+            </pre>
+          </ExpandableContent>
         </div>
       )}
 

package/src/run/inline-thinking-item.stories.tsx

@@ -1,6 +1,7 @@
 import type { Meta, StoryObj } from '@storybook/react'
 import { InlineThinkingItem } from './inline-thinking-item'
-import type { ReasoningPart } from '../types/parts'
+import { InlineToolItem } from './inline-tool-item'
+import type { ReasoningPart, ToolPart } from '../types/parts'
 
 const NOW = Date.now()
 
@@ -122,6 +123,34 @@ export const EmptyText: Story = {
   ),
 }
 
+const bashTool: ToolPart = {
+  type: 'tool',
+  id: 'tool-shell-demo',
+  tool: 'bash',
+  state: {
+    status: 'completed',
+    input: { command: 'npx tsc --noEmit' },
+    output: '✓ No type errors',
+    time: { start: NOW - 2040, end: NOW - 200 },
+  },
+}
+
+// Reasoning and tool rows adjacent — they must read as one family (shared
+// badge geometry, trailing status/duration, chevron), not two designs.
+export const UnifiedWithToolRow: Story = {
+  name: 'Unified shell (reasoning + tool)',
+  render: () => (
+    <div className="p-6 max-w-2xl space-y-2">
+      <InlineThinkingItem
+        part={thinkingComplete}
+        defaultOpen={false}
+        autoCollapse={false}
+      />
+      <InlineToolItem part={bashTool} />
+    </div>
+  ),
+}
+
 export const AllStates: Story = {
   render: () => (
     <div className="p-6 max-w-2xl space-y-3">

package/src/run/inline-thinking-item.tsx

@@ -1,11 +1,10 @@
 import { memo, useEffect, useRef, useState } from "react";
-import * as Collapsible from "@radix-ui/react-collapsible";
-import { Brain, ChevronRight } from "lucide-react";
-import { cn } from "../lib/utils";
-import { formatDuration, truncateText } from "../utils/format";
+import { Brain } from "lucide-react";
+import { truncateText } from "../utils/format";
 import type { ReasoningPart } from "../types/parts";
 import { Markdown } from "../markdown/markdown";
-import { LiveDuration } from "./run-item-primitives";
+import { RunRowShell } from "./run-row-shell";
+import { ExpandableContent } from "./expandable-content";
 
 export interface InlineThinkingItemProps {
   part: ReasoningPart;
@@ -28,7 +27,8 @@ export const InlineThinkingItem = memo(
 
     const startTime = part.time?.start;
     const endTime = part.time?.end;
-    const durationMs = startTime && endTime ? endTime - startTime : undefined;
+    const durationMs =
+      startTime != null && endTime != null ? endTime - startTime : undefined;
     const isActive = startTime != null && endTime == null;
     const preview = part.text ? truncateText(part.text, 120) : undefined;
 
@@ -50,70 +50,34 @@ export const InlineThinkingItem = memo(
     }, [autoCollapse, durationMs, isActive]);
 
     return (
-      <Collapsible.Root open={open} onOpenChange={setOpen}>
-        <div
-          className={cn(
-            "overflow-hidden rounded-[var(--radius-lg)] border bg-card transition-colors",
-            isActive
-              ? "border-[var(--border-accent)]"
-              : "border-border hover:border-[var(--border-accent)]",
-            className,
-          )}
-        >
-          <Collapsible.Trigger asChild>
-            <button
-              className="flex w-full items-center gap-2.5 px-3 py-2 text-left text-sm cursor-pointer"
-            >
-              <div
-                className={cn(
-                  "flex h-6 w-6 shrink-0 items-center justify-center rounded-[var(--radius-sm)] border",
-                  isActive
-                    ? "border-[var(--border-accent)] bg-[var(--accent-surface-soft)] text-primary"
-                    : "border-border bg-muted text-muted-foreground",
-                )}
-              >
-                <Brain className={cn("h-3 w-3 shrink-0", isActive && "animate-pulse")} />
-              </div>
-
-              <p className="min-w-0 flex-1 truncate font-[var(--font-sans)] text-muted-foreground">
-                {preview ?? (isActive ? "Thinking…" : "Reasoning")}
-              </p>
-
-              <div className="flex shrink-0 items-center gap-2">
-                {isActive && startTime ? <LiveDuration startTime={startTime} /> : null}
-                {!isActive && durationMs != null ? (
-                  <span className="shrink-0 text-xs tabular-nums text-muted-foreground">
-                    {formatDuration(durationMs)}
-                  </span>
-                ) : null}
-                <ChevronRight
-                  className={cn(
-                    "h-3 w-3 text-muted-foreground transition-transform shrink-0",
-                    open && "rotate-90",
-                  )}
-                />
-              </div>
-            </button>
-          </Collapsible.Trigger>
-
-          <Collapsible.Content className="overflow-hidden data-[state=open]:animate-slideDown data-[state=closed]:animate-slideUp">
-            {part.text ? (
-              <div
-                className={cn(
-                  "max-h-60 overflow-y-auto border-t border-border bg-muted px-3 py-3 text-sm leading-relaxed text-muted-foreground",
-                  contentClassName,
-                )}
-              >
-                <Markdown>{part.text}</Markdown>
-              </div>
-            ) : (
-              <div className="border-t border-border bg-muted px-3 py-2.5 text-xs text-muted-foreground">
-                No reasoning text provided.
-              </div>
-            )}
-          </Collapsible.Content>
-        </div>
-      </Collapsible.Root>
+      <RunRowShell
+        icon={<Brain className="h-3.5 w-3.5" />}
+        title={isActive ? "Thinking…" : "Reasoning"}
+        description={preview}
+        status={isActive ? "running" : "idle"}
+        startTime={startTime}
+        durationMs={durationMs}
+        open={open}
+        onOpenChange={setOpen}
+        className={className}
+      >
+        {part.text ? (
+          <ExpandableContent
+            fadeClassName="from-[var(--md3-surface-container)]"
+            className={
+              contentClassName ? `px-3 py-3 ${contentClassName}` : "px-3 py-3"
+            }
+          >
+            <div className="text-sm leading-relaxed text-foreground">
+              <Markdown>{part.text}</Markdown>
+            </div>
+          </ExpandableContent>
+        ) : (
+          <div className="px-3 py-2.5 text-xs text-muted-foreground">
+            No reasoning text provided.
+          </div>
+        )}
+      </RunRowShell>
     );
   },
 );