@dbx-tools/ui-mastra 0.3.21 → 0.3.22

package/src/react/mastra-chat.tsx

@@ -1,13 +1,21 @@
-import {
-  feedback,
-  type MastraThread,
-} from "@dbx-tools/shared-mastra";
 import { error as errorUtil, log } from "@dbx-tools/shared-core";
+import { feedback, type MastraThread } from "@dbx-tools/shared-mastra";
+import { useBrand } from "@dbx-tools/ui-branding/react";
 import type { UIMessage } from "ai";
 import { nanoid } from "nanoid";
 import { useCallback, useEffect, useMemo, useRef, useState } from "react";
+import { ChatView } from "./chat-view";
+import { dedupeSuggestions } from "./suggestions";
+import type {
+  ApprovalDecision,
+  ChatViewProps,
+  FeedbackSubmission,
+  MessageFeedback,
+  ThreadSummary,
+  ToolEvent,
+  ToolProgress,
+} from "./types";
 import { exportChat, type EmbedResolver, type ExportFormat } from "../support/export";
-import { useBrand } from "@dbx-tools/ui-branding/react";
 import {
   useMastraClient,
   useMastraDefaultModel,
@@ -27,17 +35,6 @@ import {
   terminateRunningToolEvents,
   type ThreadSession,
 } from "../support/thread-sessions";
-import { ChatView } from "./chat-view";
-import { dedupeSuggestions } from "./suggestions";
-import type {
-  ApprovalDecision,
-  ChatViewProps,
-  FeedbackSubmission,
-  MessageFeedback,
-  ThreadSummary,
-  ToolEvent,
-  ToolProgress,
-} from "./types";
 
 // Self-contained drop-in chat. `useMastraChat` drives the conversation
 // over `@mastra/client-js`: `agent.stream()` returns a Response
@@ -88,9 +85,7 @@ const TITLE_PREVIEW_MAX = 60;
  */
 const deriveThreadTitle = (text: string): string => {
   const clean = text.trim().replace(/\s+/g, " ");
-  return clean.length > TITLE_PREVIEW_MAX
-    ? `${clean.slice(0, TITLE_PREVIEW_MAX - 1)}…`
-    : clean;
+  return clean.length > TITLE_PREVIEW_MAX ? `${clean.slice(0, TITLE_PREVIEW_MAX - 1)}…` : clean;
 };
 
 /**
@@ -155,8 +150,7 @@ const storeStoredSidebarOpen = (key: string, open: boolean): void => {
  * when absent - which is the "no feedback for this turn" signal.
  */
 const readMlflowTraceId = (stream: unknown): string | undefined => {
-  const headers = (stream as { headers?: { get?: (name: string) => string | null } })
-    ?.headers;
+  const headers = (stream as { headers?: { get?: (name: string) => string | null } })?.headers;
   const raw = headers?.get?.(feedback.MLFLOW_TRACE_ID_HEADER);
   return raw?.trim() || undefined;
 };
@@ -316,9 +310,7 @@ export const useMastraChat = (
   // without this the sidebar wouldn't show a brand-new conversation
   // until the delayed post-turn refresh. Keyed by thread id; each entry
   // is pruned once the real server row supersedes it.
-  const [optimisticThreads, setOptimisticThreads] = useState<
-    Record<string, ThreadSummary>
-  >({});
+  const [optimisticThreads, setOptimisticThreads] = useState<Record<string, ThreadSummary>>({});
   // Optimistic title overrides for threads the user just renamed, keyed
   // by thread id. Applied over the server rows in `sidebarThreads` so the
   // new name shows instantly; each entry is dropped once the server list
@@ -329,9 +321,7 @@ export const useMastraChat = (
   // first question so the row stops reading "New conversation"; dropped
   // as soon as the server's auto-generated title lands (see the prune
   // effect below). A manual rename (`renamedThreads`) still wins.
-  const [provisionalTitles, setProvisionalTitles] = useState<Record<string, string>>(
-    {},
-  );
+  const [provisionalTitles, setProvisionalTitles] = useState<Record<string, string>>({});
   // Surface the active thread in the sidebar immediately (called when its
   // first message is sent). Upserts an untitled row stamped "now" so it
   // sorts to the top; the server row replaces it on the next refresh.
@@ -395,10 +385,7 @@ export const useMastraChat = (
     [getSession, notifySessions],
   );
   const activeKey = sessionKey(activeThreadId);
-  const activeSession = useMemo(
-    () => getSession(activeKey),
-    [activeKey, getSession, sessionsTick],
-  );
+  const activeSession = useMemo(() => getSession(activeKey), [activeKey, getSession, sessionsTick]);
   const streamingThreadIds = useMemo(() => {
     const ids: string[] = [];
     for (const [id, session] of sessionsRef.current.entries()) {
@@ -469,8 +456,9 @@ export const useMastraChat = (
       let assistantReasoning = "";
       if (existing) {
         for (const part of existing.parts) {
-          if (part.type === "text") textSegments.push(part.text);
-          else if (part.type === "reasoning") {
+          if (part.type === "text") {
+            textSegments.push(part.text);
+          } else if (part.type === "reasoning") {
             assistantReasoning += (part as { text?: string }).text ?? "";
           }
         }
@@ -519,9 +507,7 @@ export const useMastraChat = (
         if (started) return;
         started = true;
         updateSession(threadId, (session) =>
-          session.status === "streaming"
-            ? session
-            : { ...session, status: "streaming" },
+          session.status === "streaming" ? session : { ...session, status: "streaming" },
         );
       };
 
@@ -600,8 +586,7 @@ export const useMastraChat = (
                   break;
                 }
                 updateSession(threadId, (session) => {
-                  const existingApprovals =
-                    session.pendingApprovalsByMessage[assistantId] ?? [];
+                  const existingApprovals = session.pendingApprovalsByMessage[assistantId] ?? [];
                   if (existingApprovals.some((a) => a.toolCallId === toolCallId)) {
                     return session;
                   }
@@ -647,9 +632,7 @@ export const useMastraChat = (
                 const toolCallId = chunk.payload?.toolCallId;
                 if (typeof toolCallId !== "string") break;
                 patchToolEvents((list) =>
-                  list.map((e) =>
-                    e.id === toolCallId ? { ...e, status: "error" } : e,
-                  ),
+                  list.map((e) => (e.id === toolCallId ? { ...e, status: "error" } : e)),
                 );
                 break;
               }
@@ -664,9 +647,7 @@ export const useMastraChat = (
                 if (!isToolProgress(output)) break;
                 patchToolEvents((list) =>
                   list.map((e) =>
-                    e.id === toolCallId
-                      ? { ...e, progress: [...(e.progress ?? []), output] }
-                      : e,
+                    e.id === toolCallId ? { ...e, progress: [...(e.progress ?? []), output] } : e,
                   ),
                 );
                 break;
@@ -752,8 +733,7 @@ export const useMastraChat = (
           ...session,
           error: errorUtil.toError(caught),
           status: "error",
-          abortController:
-            session.abortController === controller ? null : session.abortController,
+          abortController: session.abortController === controller ? null : session.abortController,
           runId: runIdRef.current,
         }));
       }
@@ -773,8 +753,7 @@ export const useMastraChat = (
       // Capture this run's thread + model at send time and pass them per
       // call, so a run keeps its own routing even if the user switches
       // threads or changes the model picker while it streams.
-      const streamThreadId =
-        threadId === DEFAULT_THREAD_SESSION_KEY ? undefined : threadId;
+      const streamThreadId = threadId === DEFAULT_THREAD_SESSION_KEY ? undefined : threadId;
       const model = modelRef.current || undefined;
       return driveStream(threadId, assistantId, (signal) => {
         const messages = history.flatMap((m) =>
@@ -880,8 +859,7 @@ export const useMastraChat = (
         toolCallId,
         runId,
       });
-      const streamThreadId =
-        activeKey === DEFAULT_THREAD_SESSION_KEY ? undefined : activeKey;
+      const streamThreadId = activeKey === DEFAULT_THREAD_SESSION_KEY ? undefined : activeKey;
       await driveStream(activeKey, assistantId, (signal) =>
         decision.approved
           ? mastraClient.approveToolCallStream(agentId, {
@@ -1103,9 +1081,7 @@ export const useMastraChat = (
       if (!title) return;
       setRenamedThreads((prev) => ({ ...prev, [threadId]: title }));
       setOptimisticThreads((prev) =>
-        prev[threadId]
-          ? { ...prev, [threadId]: { ...prev[threadId], id: threadId, title } }
-          : prev,
+        prev[threadId] ? { ...prev, [threadId]: { ...prev[threadId], id: threadId, title } } : prev,
       );
       try {
         await mastraClient.renameThread(threadId, title, { agentId });
@@ -1131,17 +1107,14 @@ export const useMastraChat = (
     const lastUserText = getSession(threadId).lastUserText;
     if (!lastUserText) return;
     const prev = getSession(threadId).messages;
-    const lastAssistant =
-      prev.length > 0 && prev.at(-1)?.role === "assistant" ? prev.at(-1) : null;
+    const lastAssistant = prev.length > 0 && prev.at(-1)?.role === "assistant" ? prev.at(-1) : null;
     const trimmed = lastAssistant ? prev.slice(0, -1) : prev;
     if (lastAssistant) {
       updateSession(threadId, (session) => {
-        const { [lastAssistant.id]: _events, ...toolEvents } =
-          session.toolEventsByMessage;
+        const { [lastAssistant.id]: _events, ...toolEvents } = session.toolEventsByMessage;
         const { [lastAssistant.id]: _approvals, ...pendingApprovals } =
           session.pendingApprovalsByMessage;
-        const { [lastAssistant.id]: _feedback, ...feedback } =
-          session.feedbackByMessage;
+        const { [lastAssistant.id]: _feedback, ...feedback } = session.feedbackByMessage;
         return {
           ...session,
           toolEventsByMessage: toolEvents,
@@ -1244,15 +1217,7 @@ export const useMastraChat = (
         historyInFlightRef.current = false;
         setIsLoadingMore(false);
       });
-  }, [
-    activeKey,
-    activeThreadId,
-    getSession,
-    mastraClient,
-    agentId,
-    updateSession,
-    writeMessages,
-  ]);
+  }, [activeKey, activeThreadId, getSession, mastraClient, agentId, updateSession, writeMessages]);
 
   // Chat export (opt-in). Resolves `[chart:<id>]` / `[data:<id>]` embeds
   // straight off the client so the export inlines the same charts /
@@ -1291,8 +1256,7 @@ export const useMastraChat = (
   const exportConversation = useCallback(
     async (format: ExportFormat) => {
       const title =
-        (activeThreadId && threads.find((t) => t.id === activeThreadId)?.title) ||
-        "Conversation";
+        (activeThreadId && threads.find((t) => t.id === activeThreadId)?.title) || "Conversation";
       try {
         await exportChat({
           messages: getSession(activeKey).messages,
@@ -1309,15 +1273,7 @@ export const useMastraChat = (
         });
       }
     },
-    [
-      exportResolver,
-      activeThreadId,
-      activeKey,
-      getSession,
-      threads,
-      exportUserLabel,
-      exportBrand,
-    ],
+    [exportResolver, activeThreadId, activeKey, getSession, threads, exportUserLabel, exportBrand],
   );
   const exportMessage = useCallback(
     async (message: UIMessage, format: ExportFormat) => {
@@ -1364,9 +1320,7 @@ export const useMastraChat = (
       try {
         const result = await mastraClient.feedback({
           traceId,
-          ...(submission.value !== undefined
-            ? { value: submission.value === "up" }
-            : {}),
+          ...(submission.value !== undefined ? { value: submission.value === "up" } : {}),
           ...(submission.comment ? { comment: submission.comment } : {}),
         });
         if (!result.ok) {
@@ -1424,9 +1378,7 @@ export const useMastraChat = (
   // changes aren't masked by a stale override.
   useEffect(() => {
     const settled = threads
-      .filter(
-        (t) => renamedThreads[t.id] !== undefined && t.title === renamedThreads[t.id],
-      )
+      .filter((t) => renamedThreads[t.id] !== undefined && t.title === renamedThreads[t.id])
       .map((t) => t.id);
     if (settled.length === 0) return;
     setRenamedThreads((prev) => {

package/src/react/suggestion-pills.tsx

@@ -20,11 +20,7 @@ export interface SuggestionPillsProps {
  * capsule shape that scales cleanly when text wraps to multiple lines.
  * Disabled when no `onSelect` is provided.
  */
-export const SuggestionPills = ({
-  questions,
-  onSelect,
-  className,
-}: SuggestionPillsProps) => {
+export const SuggestionPills = ({ questions, onSelect, className }: SuggestionPillsProps) => {
   if (questions.length === 0) return null;
   return (
     <div className={cn("flex flex-wrap gap-1.5", className)}>

package/src/react/thread-sidebar.tsx

@@ -1,3 +1,4 @@
+import { wire } from "@dbx-tools/shared-mastra";
 import {
   Button,
   Input,
@@ -7,7 +8,6 @@ import {
   TooltipTrigger,
   cn,
 } from "@dbx-tools/ui-appkit/react";
-import { wire } from "@dbx-tools/shared-mastra";
 import {
   Loader2Icon,
   MessageSquarePlusIcon,
@@ -288,9 +288,7 @@ export const ThreadSidebar = ({
                               setConfirmDeleteId((id) => (id === thread.id ? null : id))
                             }
                             aria-label={
-                              isConfirming
-                                ? "Confirm delete conversation"
-                                : "Delete conversation"
+                              isConfirming ? "Confirm delete conversation" : "Delete conversation"
                             }
                             className={cn(
                               "size-6 shrink-0",

package/src/react/tool-pill.tsx

@@ -1,3 +1,4 @@
+import { genieModel } from "@dbx-tools/shared-genie";
 import {
   Collapsible,
   CollapsibleContent,
@@ -5,7 +6,6 @@ import {
   Spinner,
   cn,
 } from "@dbx-tools/ui-appkit/react";
-import { genieModel } from "@dbx-tools/shared-genie";
 import { CheckIcon, ChevronDownIcon, XIcon } from "lucide-react";
 import { SqlBlock, ToolMarkdown } from "./markdown";
 import type { ToolEvent, ToolProgress } from "./types";
@@ -176,10 +176,7 @@ const summarizeProgress = (progress: ToolProgress[]): ToolDetailSummary => {
   // different `message_id`s can collide on `attachment_id` (Genie
   // restarts numbering per message), so the bucket map is owned by
   // the group, not the summary.
-  const bucketFor = (
-    group: MessageGroup,
-    attachmentId: string | undefined,
-  ): AttachmentBucket => {
+  const bucketFor = (group: MessageGroup, attachmentId: string | undefined): AttachmentBucket => {
     const key = attachmentId ?? ANON_ATTACHMENT_KEY;
     let bucket = group.attachments.find((b) => b.key === key);
     if (!bucket) {
@@ -284,9 +281,7 @@ const isAttachmentRenderable = (b: AttachmentBucket): boolean =>
 
 /** True when a group has any renderable content (question, attachments, or errors). */
 const isGroupRenderable = (g: MessageGroup): boolean =>
-  Boolean(g.question) ||
-  g.attachments.some(isAttachmentRenderable) ||
-  g.errors.length > 0;
+  Boolean(g.question) || g.attachments.some(isAttachmentRenderable) || g.errors.length > 0;
 
 /**
  * Body of one Genie sub-call (one `message_id` group). Pulled out
@@ -342,10 +337,7 @@ const MessageGroupBody = ({
         // SQL itself stays a nested Collapsible (default closed)
         // for the same reason: code is the heaviest content here,
         // and most readers only want a glance.
-        <Collapsible
-          key={bucket.key}
-          className="rounded border border-border/60 bg-background/40"
-        >
+        <Collapsible key={bucket.key} className="rounded border border-border/60 bg-background/40">
           <CollapsibleTrigger className="group flex w-full items-center gap-1.5 px-2 py-1 text-left text-[11px] uppercase tracking-wide text-muted-foreground hover:text-foreground">
             <ChevronDownIcon className="size-3 shrink-0 transition-transform group-data-[state=closed]:-rotate-90" />
             <span>{queryBuckets.length > 1 ? `Query ${i + 1}` : "Query"}</span>
@@ -360,10 +352,7 @@ const MessageGroupBody = ({
               {bucket.thinking.length > 0 && (
                 <ul className="flex flex-col gap-1.5 border-l-2 border-border/60 pl-3 text-muted-foreground">
                   {bucket.thinking.map((p, j) => (
-                    <li
-                      key={`think-${j}`}
-                      className="whitespace-pre-wrap break-words leading-snug"
-                    >
+                    <li key={`think-${j}`} className="whitespace-pre-wrap break-words leading-snug">
                       <span className="font-medium text-foreground/80">
                         {humanizeThoughtType(p.thought_type)}:
                       </span>{" "}
@@ -402,10 +391,7 @@ const MessageGroupBody = ({
         // when there's only one prose bucket; multiple buckets
         // (rare - typically interpretation + a follow-up question)
         // get numbered so each row is addressable.
-        <Collapsible
-          key={bucket.key}
-          className="rounded border border-border/60 bg-background/40"
-        >
+        <Collapsible key={bucket.key} className="rounded border border-border/60 bg-background/40">
           <CollapsibleTrigger className="group flex w-full items-center gap-1.5 px-2 py-1 text-left text-[11px] uppercase tracking-wide text-muted-foreground hover:text-foreground">
             <ChevronDownIcon className="size-3 shrink-0 transition-transform group-data-[state=closed]:-rotate-90" />
             <span>{proseBuckets.length > 1 ? `Answer ${i + 1}` : "Answer"}</span>
@@ -415,10 +401,7 @@ const MessageGroupBody = ({
               {bucket.thinking.length > 0 && (
                 <ul className="flex flex-col gap-1 text-[11px] text-muted-foreground">
                   {bucket.thinking.map((p, j) => (
-                    <li
-                      key={`think-${j}`}
-                      className="whitespace-pre-wrap break-words leading-snug"
-                    >
+                    <li key={`think-${j}`} className="whitespace-pre-wrap break-words leading-snug">
                       <span className="font-medium text-foreground/80">
                         {humanizeThoughtType(p.thought_type)}:
                       </span>{" "}
@@ -534,18 +517,11 @@ const ToolCallRow = ({ event }: { event: ToolEvent }) => {
 
   const header = (
     <span className="min-w-0 flex-1">
-      <span
-        className={cn(
-          "block truncate",
-          isRunning && "animate-pulse text-foreground/90",
-        )}
-      >
+      <span className={cn("block truncate", isRunning && "animate-pulse text-foreground/90")}>
         {verb}
       </span>
       {question && (
-        <span className="mt-0.5 block break-words italic text-foreground/80">
-          {question}
-        </span>
+        <span className="mt-0.5 block break-words italic text-foreground/80">{question}</span>
       )}
     </span>
   );
@@ -603,11 +579,7 @@ export const ToolSessionPill = ({ events }: { events: ToolEvent[] }) => {
   const latest = events[events.length - 1]!;
   const anyRunning = events.some((e) => e.status === "running");
   const anyError = events.some((e) => e.status === "error");
-  const tone: "running" | "error" | "done" = anyRunning
-    ? "running"
-    : anyError
-      ? "error"
-      : "done";
+  const tone: "running" | "error" | "done" = anyRunning ? "running" : anyError ? "error" : "done";
 
   // Outer header uses the tool name verb (not the live wire
   // status) per the "show the most recent tool call" contract.
@@ -654,9 +626,7 @@ export const ToolSessionPill = ({ events }: { events: ToolEvent[] }) => {
             )}
           >
             {verb}
-            {countSuffix && (
-              <span className="text-muted-foreground/70">{countSuffix}</span>
-            )}
+            {countSuffix && <span className="text-muted-foreground/70">{countSuffix}</span>}
           </span>
           {/*
            * Trailing "live" pip on the right edge while any tool is

package/src/react/types.ts

@@ -300,10 +300,7 @@ export type ChatViewProps = {
    * the bubble shows thumbs up/down plus a comment affordance. The
    * handler logs the feedback to MLflow via the plugin's feedback route.
    */
-  onFeedback?: (
-    message: UIMessage,
-    submission: FeedbackSubmission,
-  ) => void | Promise<void>;
+  onFeedback?: (message: UIMessage, submission: FeedbackSubmission) => void | Promise<void>;
 };
 
 /** Payload {@link ChatViewProps.onResolveToolApproval} receives. */

package/src/support/chart-option.ts

@@ -48,9 +48,7 @@ const isObj = object.isRecord;
 /** True when `title` (object or array) carries any non-empty `text`. */
 function hasTitleText(title: unknown): boolean {
   const entries = Array.isArray(title) ? title : [title];
-  return entries.some(
-    (t) => isObj(t) && typeof t.text === "string" && t.text.trim().length > 0,
-  );
+  return entries.some((t) => isObj(t) && typeof t.text === "string" && t.text.trim().length > 0);
 }
 
 /** Pin a title to the top-center so it clears the plot / axis names. */
@@ -109,8 +107,7 @@ function normalizeAxis(axis: Obj, pos: "x" | "y"): Obj {
 
 /** Apply {@link normalizeAxis} across an axis field (object or array). */
 function normalizeAxisField(axis: unknown, pos: "x" | "y"): unknown {
-  if (Array.isArray(axis))
-    return axis.map((a) => (isObj(a) ? normalizeAxis(a, pos) : a));
+  if (Array.isArray(axis)) return axis.map((a) => (isObj(a) ? normalizeAxis(a, pos) : a));
   return isObj(axis) ? normalizeAxis(axis, pos) : axis;
 }
 

package/src/support/export.ts

@@ -23,12 +23,8 @@
  * @module
  */
 
-import {
-  marker as markers,
-  type Chart,
-  type StatementData,
-} from "@dbx-tools/shared-mastra";
 import { string } from "@dbx-tools/shared-core";
+import { marker as markers, type Chart, type StatementData } from "@dbx-tools/shared-mastra";
 import type { UIMessage } from "ai";
 import * as echarts from "echarts";
 import { marked } from "marked";
@@ -188,9 +184,7 @@ function printViaHiddenIframe(html: string, downloadName: string): void {
 
 /** One slice of a message: prose, a chart embed, or a data embed. */
 type Segment =
-  | { kind: "text"; text: string }
-  | { kind: "chart"; id: string }
-  | { kind: "data"; id: string };
+  { kind: "text"; text: string } | { kind: "chart"; id: string } | { kind: "data"; id: string };
 
 /**
  * Split message text into prose / chart / data segments at
@@ -268,19 +262,14 @@ async function buildHtmlDocument(
 }
 
 /** Render one message's body (prose + inlined charts / tables) to HTML. */
-async function messageBodyHtml(
-  message: UIMessage,
-  resolver: EmbedResolver,
-): Promise<string> {
+async function messageBodyHtml(message: UIMessage, resolver: EmbedResolver): Promise<string> {
   const isAssistant = message.role === "assistant";
   const parts: string[] = [];
   for (const seg of splitSegments(messageText(message))) {
     if (seg.kind === "text") {
       if (!seg.text.trim()) continue;
       parts.push(
-        isAssistant
-          ? markdownToHtml(seg.text)
-          : `<p class="plain">${escapeHtml(seg.text)}</p>`,
+        isAssistant ? markdownToHtml(seg.text) : `<p class="plain">${escapeHtml(seg.text)}</p>`,
       );
       continue;
     }
@@ -320,9 +309,7 @@ async function chartSvg(resolver: EmbedResolver, id: string): Promise<string | n
       height: CHART_HEIGHT_PX,
     });
     try {
-      instance.setOption(
-        normalizeChartOption(chart.result.option) as echarts.EChartsCoreOption,
-      );
+      instance.setOption(normalizeChartOption(chart.result.option) as echarts.EChartsCoreOption);
       return instance.renderToSVGString();
     } finally {
       instance.dispose();
@@ -333,21 +320,14 @@ async function chartSvg(resolver: EmbedResolver, id: string): Promise<string | n
 }
 
 /** Resolve a data id and render its rows to an HTML table. `null` on miss. */
-async function dataTableHtml(
-  resolver: EmbedResolver,
-  id: string,
-): Promise<string | null> {
+async function dataTableHtml(resolver: EmbedResolver, id: string): Promise<string | null> {
   const data = await safeStatement(resolver, id);
   if (!data || data.rows.length === 0) return null;
-  const head = data.columns
-    .map((c) => `<th>${escapeHtml(string.toLabel(c))}</th>`)
-    .join("");
+  const head = data.columns.map((c) => `<th>${escapeHtml(string.toLabel(c))}</th>`).join("");
   const body = data.rows
     .map(
       (row) =>
-        `<tr>${data.columns
-          .map((c) => `<td>${escapeHtml(cellText(row[c]))}</td>`)
-          .join("")}</tr>`,
+        `<tr>${data.columns.map((c) => `<td>${escapeHtml(cellText(row[c]))}</td>`).join("")}</tr>`,
     )
     .join("");
   const note = data.truncated
@@ -379,10 +359,7 @@ async function buildMarkdown(
 }
 
 /** Render one message's body to Markdown (charts noted, tables as GFM). */
-async function messageBodyMarkdown(
-  message: UIMessage,
-  resolver: EmbedResolver,
-): Promise<string> {
+async function messageBodyMarkdown(message: UIMessage, resolver: EmbedResolver): Promise<string> {
   const parts: string[] = [];
   for (const seg of splitSegments(messageText(message))) {
     if (seg.kind === "text") {
@@ -405,17 +382,13 @@ function chartTitle(chart: Chart): string {
   const option = chart.result?.option as { title?: unknown } | undefined;
   const title = option?.title;
   const text =
-    (Array.isArray(title) ? title[0]?.text : (title as { text?: unknown })?.text) ??
-    undefined;
+    (Array.isArray(title) ? title[0]?.text : (title as { text?: unknown })?.text) ?? undefined;
   const label = typeof text === "string" ? text.trim() : "";
   return label || `${chart.result?.chartType ?? "chart"} chart`;
 }
 
 /** Render statement rows to a GFM table. `null` on miss / empty. */
-async function dataTableMarkdown(
-  resolver: EmbedResolver,
-  id: string,
-): Promise<string | null> {
+async function dataTableMarkdown(resolver: EmbedResolver, id: string): Promise<string | null> {
   const data = await safeStatement(resolver, id);
   if (!data || data.rows.length === 0) return null;
   const header = `| ${data.columns.map((c) => mdCell(string.toLabel(c))).join(" | ")} |`;
@@ -423,19 +396,14 @@ async function dataTableMarkdown(
   const rows = data.rows.map(
     (row) => `| ${data.columns.map((c) => mdCell(cellText(row[c]))).join(" | ")} |`,
   );
-  const note = data.truncated
-    ? `\n\n_Showing ${data.rows.length} of ${data.rowCount} rows._`
-    : "";
+  const note = data.truncated ? `\n\n_Showing ${data.rows.length} of ${data.rowCount} rows._` : "";
   return `${[header, sep, ...rows].join("\n")}${note}`;
 }
 
 /* -------------------------------- helpers -------------------------------- */
 
 /** Resolve a chart id, swallowing errors (best-effort export). */
-async function safeChart(
-  resolver: EmbedResolver,
-  id: string,
-): Promise<Chart | undefined> {
+async function safeChart(resolver: EmbedResolver, id: string): Promise<Chart | undefined> {
   try {
     return await resolver.chart(id);
   } catch {