@dbx-tools/ui-mastra 0.9.154 → 0.9.157

package/index.react.ts

@@ -0,0 +1,61 @@
+// GENERATED by projen watch - DO NOT EDIT.
+// Regenerated from the exporting modules in ./src.
+// Hand edits are overwritten on the next watch; this file is read-only.
+
+export * as bubbles from "./src/react/bubbles.tsx";
+export * as chatApprovals from "./src/react/chat-approvals.ts";
+export * as chatComposer from "./src/react/chat-composer.tsx";
+export * as chatFeedback from "./src/react/chat-feedback.ts";
+export * as chatHistory from "./src/react/chat-history.ts";
+export * as chatSessions from "./src/react/chat-sessions.ts";
+export * as chatStream from "./src/react/chat-stream.ts";
+export * as chatThreadLayout from "./src/react/chat-thread-layout.tsx";
+export * as chatTranscript from "./src/react/chat-transcript.tsx";
+export * as chatView from "./src/react/chat-view.tsx";
+export * as dataGrid from "./src/react/data-grid.tsx";
+export * as embedSlots from "./src/react/embed-slots.tsx";
+export * as exportMenu from "./src/react/export-menu.tsx";
+export * as feedbackControls from "./src/react/feedback-controls.tsx";
+export * as markdown from "./src/react/markdown.tsx";
+export * as mastraAssistant from "./src/react/mastra-assistant.tsx";
+export * as mastraChat from "./src/react/mastra-chat.tsx";
+export * as suggestionPills from "./src/react/suggestion-pills.tsx";
+export * as suggestions from "./src/react/suggestions.ts";
+export * as threadSidebar from "./src/react/thread-sidebar.tsx";
+export * as threadTabs from "./src/react/thread-tabs.tsx";
+export * as toolPill from "./src/react/tool-pill.tsx";
+export * as types from "./src/react/types.ts";
+export { AssistantBubble, UserBubble } from "./src/react/bubbles.tsx";
+export { resolveApprovalContinuation, useChatApprovals } from "./src/react/chat-approvals.ts";
+export { ChatComposer } from "./src/react/chat-composer.tsx";
+export { genieFeedbackTarget, useChatFeedback } from "./src/react/chat-feedback.ts";
+export type { GenieFeedbackTarget } from "./src/react/chat-feedback.ts";
+export { useChatHistory } from "./src/react/chat-history.ts";
+export { useChatSessions } from "./src/react/chat-sessions.ts";
+export type { ThreadSessionReader, ThreadSessionUpdater, ThreadMessageWriter } from "./src/react/chat-sessions.ts";
+export { closeOnTerminalChunk, waitForTerminalChunk, useChatStream } from "./src/react/chat-stream.ts";
+export { ChatThreadLayout } from "./src/react/chat-thread-layout.tsx";
+export { useChatTranscriptController, ChatTranscript } from "./src/react/chat-transcript.tsx";
+export type { ChatTranscriptController } from "./src/react/chat-transcript.tsx";
+export { ChatView } from "./src/react/chat-view.tsx";
+export { colorizeDelta, renderDataCell, humanizeLabel, TABLE_WRAPPER_CLASSES, DataGrid } from "./src/react/data-grid.tsx";
+export type { DataRow } from "./src/react/data-grid.tsx";
+export { MarkdownWithEmbeds } from "./src/react/embed-slots.tsx";
+export { ExportMenu } from "./src/react/export-menu.tsx";
+export { FeedbackControls } from "./src/react/feedback-controls.tsx";
+export { AssistantMarkdown, ToolMarkdown, SqlBlock, SourceBlock } from "./src/react/markdown.tsx";
+export { useMastraAssistant, MastraAssistant } from "./src/react/mastra-assistant.tsx";
+export type { MastraAssistantMode, MastraAssistantSide, MastraAssistantLauncherPosition, MastraAssistantResizeOptions, MastraAssistantLauncherOptions, MastraAssistantController, MastraAssistantProps } from "./src/react/mastra-assistant.tsx";
+export { useMastraChat, MastraChat } from "./src/react/mastra-chat.tsx";
+export type { ThreadIdChangeReason, UseMastraChatOptions, MastraChatProps } from "./src/react/mastra-chat.tsx";
+export { SuggestionPills } from "./src/react/suggestion-pills.tsx";
+export type { SuggestionPillsProps } from "./src/react/suggestion-pills.tsx";
+export { dedupeSuggestions, collectSuggestions } from "./src/react/suggestions.ts";
+export { ThreadSidebar } from "./src/react/thread-sidebar.tsx";
+export type { ThreadSidebarProps } from "./src/react/thread-sidebar.tsx";
+export { ThreadActions, ThreadTabs } from "./src/react/thread-tabs.tsx";
+export type { ThreadTabsProps, ThreadActionsProps } from "./src/react/thread-tabs.tsx";
+export { humanizeToolName, formatRawToolPayload, structuredToolPayload, detectToolSourceLanguage, toolInputPresentation, toolOutputPresentation, webSearchProgressGroups, webSearchQueryLabel, webSearchResultLabel, ToolSessionPill } from "./src/react/tool-pill.tsx";
+export type { ToolInputPresentation, WebSearchProgressGroup } from "./src/react/tool-pill.tsx";
+export type { ChatStatus, ToolEvent, ToolProgress, ChatModelOption, ChatInputMessage, QueuedSteer, FeedbackValue, FeedbackSubmission, MessageFeedback, ThreadSummary, ThreadPlacement, ChatViewProps, ApprovalDecision, PendingApproval } from "./src/react/types.ts";
+export { useChatThreadQuery } from "./src/react/use-chat-thread-query.ts";

package/index.support.ts

@@ -0,0 +1,40 @@
+// GENERATED by projen watch - DO NOT EDIT.
+// Regenerated from the exporting modules in ./src.
+// Hand edits are overwritten on the next watch; this file is read-only.
+
+export * as activeRuns from "./src/support/active-runs.ts";
+export * as chartOption from "./src/support/chart-option.ts";
+export * as chartTheme from "./src/support/chart-theme.ts";
+export * as clipboard from "./src/support/clipboard.ts";
+export * as download from "./src/support/download.ts";
+export * as exportModule from "./src/support/export.ts";
+export * as genieClient from "./src/support/genie-client.ts";
+export * as mastraClient from "./src/support/mastra-client.ts";
+export * as modelSelection from "./src/support/model-selection.ts";
+export * as requestContext from "./src/support/request-context.ts";
+export * as threadLabels from "./src/support/thread-labels.ts";
+export * as threadPersistence from "./src/support/thread-persistence.ts";
+export * as threadSessions from "./src/support/thread-sessions.ts";
+export * as threadTabs from "./src/support/thread-tabs.ts";
+export * as toolEvents from "./src/support/tool-events.ts";
+export { StoredActiveRunSchema, activeRunStorageKey, readStoredActiveRuns, storeActiveRun, removeStoredActiveRun } from "./src/support/active-runs.ts";
+export type { StoredActiveRun } from "./src/support/active-runs.ts";
+export { normalizeChartOption } from "./src/support/chart-option.ts";
+export { LIGHT_CHART_CHROME, resolveChartChrome, useChartChrome } from "./src/support/chart-theme.ts";
+export type { ChartChrome } from "./src/support/chart-theme.ts";
+export { copyText } from "./src/support/clipboard.ts";
+export { downloadFile } from "./src/support/download.ts";
+export { exportChat, messageText } from "./src/support/export.ts";
+export type { ExportFormat, ExportBrand, EmbedResolver, ExportChatOptions } from "./src/support/export.ts";
+export { GenieChatClient } from "./src/support/genie-client.ts";
+export { MastraChatRequestError, MastraPluginClient, useMastraConfig, useMastraClient, useMastraMlflowExperiment, useMastraModels, useMastraDefaultModel, useMastraSuggestions, useMastraThreads, useChartFetch, useStatementFetch, useVisualizationFetch } from "./src/support/mastra-client.ts";
+export type { MastraStreamResponse, MastraMemoryThread, ByIdFetchState } from "./src/support/mastra-client.ts";
+export { defaultReasoningEffort, modelStorageKey, readStoredModel, storeSelectedModel } from "./src/support/model-selection.ts";
+export { snapshotRequestContext } from "./src/support/request-context.ts";
+export type { MastraRequestContextInput, MastraRequestContextSource } from "./src/support/request-context.ts";
+export { threadTitle, relativeTime } from "./src/support/thread-labels.ts";
+export { threadStorageKey } from "./src/support/thread-persistence.ts";
+export { DEFAULT_THREAD_SESSION_KEY, createThreadSession, isSessionRunning, lastUserMessage, interruptedTurnMessages, enqueueSteer, removeSteer, reorderSteers, sessionKey } from "./src/support/thread-sessions.ts";
+export type { SessionQueuedSteer, SessionRun, ThreadSession } from "./src/support/thread-sessions.ts";
+export { THREAD_TAB_SEED_MAX, syncThreadTabs, closeThreadTab, nextActiveThreadTab } from "./src/support/thread-tabs.ts";
+export { compactReasoningText, genieReasoningText, toolEventsFromParts, mergeToolEvents } from "./src/support/tool-events.ts";

package/index.ts

@@ -3,82 +3,4 @@
 // Hand edits are overwritten on the next watch; this file is read-only.
 
 export const PACKAGE_IDENTIFIER = "@dbx-tools/ui-mastra";
-export const PACKAGE_VERSION = "0.9.154";
-export * as reactBubbles from "./src/react/bubbles.tsx";
-export * as reactChatApprovals from "./src/react/chat-approvals.ts";
-export * as reactChatComposer from "./src/react/chat-composer.tsx";
-export * as reactChatFeedback from "./src/react/chat-feedback.ts";
-export * as reactChatHistory from "./src/react/chat-history.ts";
-export * as reactChatSessions from "./src/react/chat-sessions.ts";
-export * as reactChatStream from "./src/react/chat-stream.ts";
-export * as reactChatThreadLayout from "./src/react/chat-thread-layout.tsx";
-export * as reactChatTranscript from "./src/react/chat-transcript.tsx";
-export * as reactChatView from "./src/react/chat-view.tsx";
-export * as reactDataGrid from "./src/react/data-grid.tsx";
-export * as reactEmbedSlots from "./src/react/embed-slots.tsx";
-export * as reactExportMenu from "./src/react/export-menu.tsx";
-export * as reactFeedbackControls from "./src/react/feedback-controls.tsx";
-export * as reactMarkdown from "./src/react/markdown.tsx";
-export * as reactMastraAssistant from "./src/react/mastra-assistant.tsx";
-export * as reactMastraChat from "./src/react/mastra-chat.tsx";
-export * as reactSuggestionPills from "./src/react/suggestion-pills.tsx";
-export * as reactSuggestions from "./src/react/suggestions.ts";
-export * as reactThreadSidebar from "./src/react/thread-sidebar.tsx";
-export * as reactThreadTabs from "./src/react/thread-tabs.tsx";
-export * as reactToolPill from "./src/react/tool-pill.tsx";
-export * as reactTypes from "./src/react/types.ts";
-export * as supportActiveRuns from "./src/support/active-runs.ts";
-export * as supportChartOption from "./src/support/chart-option.ts";
-export * as supportChartTheme from "./src/support/chart-theme.ts";
-export * as supportClipboard from "./src/support/clipboard.ts";
-export * as supportDownload from "./src/support/download.ts";
-export * as supportExport from "./src/support/export.ts";
-export * as supportGenieClient from "./src/support/genie-client.ts";
-export * as supportMastraClient from "./src/support/mastra-client.ts";
-export * as supportModelSelection from "./src/support/model-selection.ts";
-export * as supportRequestContext from "./src/support/request-context.ts";
-export * as supportThreadLabels from "./src/support/thread-labels.ts";
-export * as supportThreadPersistence from "./src/support/thread-persistence.ts";
-export * as supportThreadSessions from "./src/support/thread-sessions.ts";
-export * as supportThreadTabs from "./src/support/thread-tabs.ts";
-export * as supportToolEvents from "./src/support/tool-events.ts";
-export { AssistantBubble, UserBubble } from "./src/react/bubbles.tsx";
-export { ChatComposer } from "./src/react/chat-composer.tsx";
-export type { GenieFeedbackTarget } from "./src/react/chat-feedback.ts";
-export type { ThreadSessionReader, ThreadSessionUpdater, ThreadMessageWriter } from "./src/react/chat-sessions.ts";
-export { ChatThreadLayout } from "./src/react/chat-thread-layout.tsx";
-export { useChatTranscriptController, ChatTranscript } from "./src/react/chat-transcript.tsx";
-export type { ChatTranscriptController } from "./src/react/chat-transcript.tsx";
-export { ChatView } from "./src/react/chat-view.tsx";
-export { TABLE_WRAPPER_CLASSES, DataGrid } from "./src/react/data-grid.tsx";
-export type { DataRow } from "./src/react/data-grid.tsx";
-export { MarkdownWithEmbeds } from "./src/react/embed-slots.tsx";
-export { ExportMenu } from "./src/react/export-menu.tsx";
-export { FeedbackControls } from "./src/react/feedback-controls.tsx";
-export { AssistantMarkdown, ToolMarkdown, SqlBlock, SourceBlock } from "./src/react/markdown.tsx";
-export type { MastraAssistantMode, MastraAssistantSide, MastraAssistantLauncherPosition, MastraAssistantResizeOptions, MastraAssistantLauncherOptions, MastraAssistantController, MastraAssistantProps } from "./src/react/mastra-assistant.tsx";
-export { useMastraChat, MastraChat } from "./src/react/mastra-chat.tsx";
-export type { ThreadIdChangeReason, UseMastraChatOptions, MastraChatProps } from "./src/react/mastra-chat.tsx";
-export { SuggestionPills } from "./src/react/suggestion-pills.tsx";
-export type { SuggestionPillsProps } from "./src/react/suggestion-pills.tsx";
-export { dedupeSuggestions, collectSuggestions } from "./src/react/suggestions.ts";
-export { ThreadSidebar } from "./src/react/thread-sidebar.tsx";
-export type { ThreadSidebarProps } from "./src/react/thread-sidebar.tsx";
-export { ThreadActions, ThreadTabs } from "./src/react/thread-tabs.tsx";
-export type { ThreadTabsProps, ThreadActionsProps } from "./src/react/thread-tabs.tsx";
-export { humanizeToolName, ToolSessionPill } from "./src/react/tool-pill.tsx";
-export type { ToolInputPresentation, WebSearchProgressGroup } from "./src/react/tool-pill.tsx";
-export type { ChatStatus, ToolEvent, ToolProgress, ChatModelOption, ChatInputMessage, QueuedSteer, FeedbackValue, FeedbackSubmission, MessageFeedback, ThreadSummary, ThreadPlacement, ChatViewProps, ApprovalDecision, PendingApproval } from "./src/react/types.ts";
-export { StoredActiveRunSchema, activeRunStorageKey } from "./src/support/active-runs.ts";
-export type { StoredActiveRun } from "./src/support/active-runs.ts";
-export { LIGHT_CHART_CHROME } from "./src/support/chart-theme.ts";
-export type { ChartChrome } from "./src/support/chart-theme.ts";
-export type { ExportFormat, ExportBrand, EmbedResolver, ExportChatOptions } from "./src/support/export.ts";
-export { GenieChatClient } from "./src/support/genie-client.ts";
-export { MastraChatRequestError, MastraPluginClient, useMastraConfig, useMastraClient, useMastraMlflowExperiment, useMastraModels, useMastraDefaultModel, useMastraSuggestions, useMastraThreads, useChartFetch, useStatementFetch, useVisualizationFetch } from "./src/support/mastra-client.ts";
-export type { MastraStreamResponse, MastraMemoryThread, ByIdFetchState } from "./src/support/mastra-client.ts";
-export { defaultReasoningEffort, modelStorageKey, readStoredModel, storeSelectedModel } from "./src/support/model-selection.ts";
-export type { MastraRequestContextInput, MastraRequestContextSource } from "./src/support/request-context.ts";
-export { DEFAULT_THREAD_SESSION_KEY } from "./src/support/thread-sessions.ts";
-export type { SessionQueuedSteer, SessionRun, ThreadSession } from "./src/support/thread-sessions.ts";
-export { THREAD_TAB_SEED_MAX } from "./src/support/thread-tabs.ts";
+export const PACKAGE_VERSION = "0.9.157";

package/package.json

@@ -22,19 +22,22 @@
     "@types/react-dom": "^19.2.2",
     "react": "~19.2.4",
     "react-dom": "~19.2.4",
+    "react-router-dom": "^7.6.2",
     "typescript": "^5.9.3"
   },
   "peerDependencies": {
     "react": "~19.2.4",
-    "react-dom": "~19.2.4"
+    "react-dom": "~19.2.4",
+    "react-router-dom": "^7.6.2"
   },
   "dependencies": {
     "@databricks/appkit-ui": "0.81.0",
-    "@dbx-tools/shared-core": "^0.9.154",
-    "@dbx-tools/shared-genie": "^0.9.154",
-    "@dbx-tools/shared-mastra": "^0.9.154",
-    "@dbx-tools/shared-model": "^0.9.154",
-    "@dbx-tools/ui": "^0.9.154",
+    "@dbx-tools/shared-auth": "^0.9.157",
+    "@dbx-tools/shared-core": "^0.9.157",
+    "@dbx-tools/shared-genie": "^0.9.157",
+    "@dbx-tools/shared-mastra": "^0.9.157",
+    "@dbx-tools/shared-model": "^0.9.157",
+    "@dbx-tools/ui": "^0.9.157",
     "@mastra/client-js": "1.50.0",
     "@mastra/core": "1.71.0",
     "@tanstack/react-table": "^8.21.3",
@@ -47,11 +50,13 @@
     "streamdown": "^2.5.0",
     "zod": "4.3.6"
   },
+  "main": "index.ts",
   "license": "Apache-2.0",
   "publishConfig": {
     "access": "public"
   },
-  "version": "0.9.154",
+  "version": "0.9.157",
+  "types": "index.ts",
   "dbxToolsConfig": {
     "tags": [
       "ui"
@@ -59,14 +64,20 @@
   },
   "type": "module",
   "exports": {
-    "./react": "./src/react/index.ts",
+    ".": "./index.ts",
+    "./react": "./index.react.ts",
+    "./support": "./index.support.ts",
     "./styles.css": "./src/styles.css",
-    "./*": "./src/*/index.ts",
     "./package.json": "./package.json"
   },
   "files": [
-    "index.ts",
+    "index*.ts",
     "src"
   ],
+  "peerDependenciesMeta": {
+    "react-router-dom": {
+      "optional": true
+    }
+  },
   "//": "~~ Generated by projen. To modify, edit .projenrc.js and run \"bunx projen\"."
 }

package/src/react/_assistant-response-status.tsx

@@ -1,4 +1,4 @@
-import { Spinner } from "@dbx-tools/ui/react";
+import { Spinner } from "@databricks/appkit-ui/react";
 
 import type { ChatStatus } from "./types.ts";
 

package/src/react/_external-link-dialog.tsx

@@ -8,7 +8,7 @@ import {
   AlertDialogHeader,
   AlertDialogTitle,
   Button,
-} from "@dbx-tools/ui/react";
+} from "@databricks/appkit-ui/react";
 import { CheckIcon, CopyIcon, ExternalLinkIcon } from "lucide-react";
 import { useEffect, useMemo, useRef, useState } from "react";
 import type { LinkSafetyModalProps } from "streamdown";

package/src/react/_model-selector.tsx

@@ -1,5 +1,3 @@
-import { stringUtils } from "@dbx-tools/shared-core";
-import type { ReasoningEffort } from "@dbx-tools/shared-model";
 import {
   Button,
   DropdownMenu,
@@ -17,7 +15,9 @@ import {
   Slider,
   Spinner,
   useIsMobile,
-} from "@dbx-tools/ui/react";
+} from "@databricks/appkit-ui/react";
+import { toLabel } from "@dbx-tools/shared-core";
+import type { ReasoningEffort } from "@dbx-tools/shared-model";
 import { CheckIcon, ChevronDownIcon, ChevronLeftIcon, ChevronRightIcon } from "lucide-react";
 import { useMemo, useState, type ReactNode } from "react";
 import type { ChatModelOption, ChatViewProps } from "./types.ts";
@@ -38,7 +38,7 @@ interface ModelSelectorProps {
 
 /** Convert a normalized family token into a compact menu label. */
 function familyLabel(family: string): string {
-  return family.length <= 3 ? family.toUpperCase() : stringUtils.toLabel(family);
+  return family.length <= 3 ? family.toUpperCase() : toLabel(family);
 }
 
 /** Resolve the label shown for a model endpoint. */
@@ -57,7 +57,7 @@ const ReasoningControl = ({
   onChange: NonNullable<ChatViewProps["onReasoningEffortChange"]>;
 }) => {
   const selectedIndex = value === undefined ? 0 : Math.max(0, efforts.indexOf(value) + 1);
-  const selectedLabel = value === undefined ? "Auto" : stringUtils.toLabel(value);
+  const selectedLabel = value === undefined ? "Auto" : toLabel(value);
 
   return (
     <div className="space-y-2 px-2 py-1.5">

package/src/react/bubbles.tsx

@@ -8,12 +8,12 @@ import {
   Item,
   ItemContent,
   ItemMedia,
-  RecordPreview,
   Tooltip,
   TooltipContent,
   TooltipTrigger,
   cn,
-} from "@dbx-tools/ui/react";
+} from "@databricks/appkit-ui/react";
+import { RecordPreview } from "@dbx-tools/ui/react";
 import type { UIMessage } from "ai";
 import {
   BugIcon,
@@ -39,13 +39,13 @@ import { ToolSessionPill, humanizeToolName } from "./tool-pill.tsx";
 import type {
   ApprovalDecision,
   ChatStatus,
-  ExportFormat,
   FeedbackSubmission,
   FeedbackValue,
   PendingApproval,
   ToolEvent,
 } from "./types.ts";
 import { copyText } from "../support/clipboard.ts";
+import type { ExportFormat } from "../support/export.ts";
 import { mergeToolEvents, toolEventsFromParts } from "../support/tool-events.ts";
 
 // User / assistant message bubbles plus the inline approval card and

package/src/react/chat-composer.tsx

@@ -17,7 +17,7 @@ import {
   TooltipContent,
   TooltipTrigger,
   cn,
-} from "@dbx-tools/ui/react";
+} from "@databricks/appkit-ui/react";
 import type { FileUIPart } from "ai";
 import {
   GripVerticalIcon,

package/src/react/chat-feedback.ts

@@ -1,6 +1,6 @@
-import { errorUtils, log } from "@dbx-tools/shared-core";
-import { genieModel } from "@dbx-tools/shared-genie";
-import type { FeedbackMode } from "@dbx-tools/shared-mastra/feedback";
+import { log, toError } from "@dbx-tools/shared-core";
+import { GenieMessageSchema } from "@dbx-tools/shared-genie";
+import type { FeedbackMode } from "@dbx-tools/shared-mastra";
 import type { UIMessage } from "ai";
 import { useCallback, useRef } from "react";
 import type { ThreadSessionUpdater } from "./chat-sessions.ts";
@@ -33,7 +33,7 @@ export function genieFeedbackTarget(message: UIMessage): GenieFeedbackTarget | u
       typeof event.output === "object" && event.output !== null && "message" in event.output
         ? event.output.message
         : undefined;
-    const parsed = genieModel.GenieMessageSchema.safeParse(output);
+    const parsed = GenieMessageSchema.safeParse(output);
     if (!parsed.success) continue;
     const genieMessage = parsed.data;
     const messageId = genieMessage.message_id ?? genieMessage.id;
@@ -107,7 +107,7 @@ export function useChatFeedback({
         logger.error("feedback error", {
           feedbackMode,
           ...(traceId ? { traceId } : {}),
-          error: errorUtils.toError(error),
+          error: toError(error),
         });
       }
     },

package/src/react/chat-history.ts

@@ -1,4 +1,4 @@
-import { errorUtils, log } from "@dbx-tools/shared-core";
+import { log, toError } from "@dbx-tools/shared-core";
 import type { UIMessage } from "ai";
 import { useCallback, useEffect, useRef, useState } from "react";
 import { toChronologicalUiMessages } from "./_history-messages.ts";
@@ -116,7 +116,7 @@ export function useChatHistory({
           .catch((error: unknown) => {
             if ((error as { name?: string }).name === "AbortError") throw error;
             logger.warn("suspended-run discovery error", {
-              error: errorUtils.toError(error),
+              error: toError(error),
             });
             return { runs: [], total: 0 };
           });
@@ -140,7 +140,7 @@ export function useChatHistory({
       .catch((error: unknown) => {
         if (cancelled || (error as { name?: string }).name === "AbortError") return;
         logger.error("history load error", {
-          error: errorUtils.toError(error),
+          error: toError(error),
         });
         updateSession(threadId, (current) => ({
           ...current,
@@ -181,7 +181,7 @@ export function useChatHistory({
       .catch((error: unknown) => {
         logger.error("history load-more error", {
           page,
-          error: errorUtils.toError(error),
+          error: toError(error),
         });
         updateSession(threadId, (current) => ({
           ...current,

package/src/react/chat-stream.ts

@@ -1,6 +1,5 @@
-import { errorUtils, log, stringUtils } from "@dbx-tools/shared-core";
-import { feedback } from "@dbx-tools/shared-mastra";
-import { isStaleMastraResumeError } from "@dbx-tools/shared-mastra/resume";
+import { log, toError, trimToUndefined } from "@dbx-tools/shared-core";
+import { MLFLOW_TRACE_ID_HEADER, isStaleMastraResumeError } from "@dbx-tools/shared-mastra";
 import { readUIMessageStream, type UIMessage, type UIMessageChunk } from "ai";
 import { useCallback } from "react";
 import type {
@@ -15,7 +14,7 @@ const logger = log.logger("ui-mastra/chat");
 /** Read the MLflow trace id captured by the server on a stream response. */
 const readMlflowTraceId = (stream: unknown): string | undefined => {
   const headers = (stream as { headers?: { get?: (name: string) => string | null } })?.headers;
-  return stringUtils.trimToUndefined(headers?.get?.(feedback.MLFLOW_TRACE_ID_HEADER));
+  return trimToUndefined(headers?.get?.(MLFLOW_TRACE_ID_HEADER));
 };
 
 class StreamAborted extends Error {}
@@ -139,7 +138,7 @@ export function useChatStream({ getSession, updateSession, writeMessages }: UseC
         if (error instanceof StreamAborted || signal.aborted) return;
         if (isStaleMastraResumeError(error)) {
           logger.warn("ignored stale mastra resume", {
-            error: errorUtils.toError(error),
+            error: toError(error),
           });
           return;
         }

package/src/react/chat-thread-layout.tsx

@@ -5,7 +5,7 @@ import {
   TooltipProvider,
   TooltipTrigger,
   cn,
-} from "@dbx-tools/ui/react";
+} from "@databricks/appkit-ui/react";
 import { PanelLeftIcon, PanelRightIcon } from "lucide-react";
 import React, { useEffect, useRef, useState } from "react";
 import { ThreadSidebar, type ThreadSidebarProps } from "./thread-sidebar.tsx";

package/src/react/chat-transcript.tsx

@@ -1,4 +1,3 @@
-import { errorUtils } from "@dbx-tools/shared-core";
 import {
   Alert,
   AlertDescription,
@@ -10,7 +9,8 @@ import {
   EmptyMedia,
   EmptyTitle,
   Spinner,
-} from "@dbx-tools/ui/react";
+} from "@databricks/appkit-ui/react";
+import { errorMessage } from "@dbx-tools/shared-core";
 import { isToolUIPart } from "ai";
 import { ArrowDownIcon, MessageSquareIcon, RefreshCwIcon, TriangleAlertIcon } from "lucide-react";
 import React, { useCallback, useEffect, useLayoutEffect, useRef, useState } from "react";
@@ -298,7 +298,7 @@ export const ChatTranscript = ({
                   <AlertTitle>Something went wrong</AlertTitle>
                   <AlertDescription>
                     {error
-                      ? errorUtils.errorMessage(error)
+                      ? errorMessage(error)
                       : "The assistant ran into an error. Please try again."}
                   </AlertDescription>
                 </Alert>

package/src/react/data-grid.tsx

@@ -1,4 +1,3 @@
-import { stringUtils } from "@dbx-tools/shared-core";
 import {
   Button,
   DropdownMenu,
@@ -12,7 +11,8 @@ import {
   TableHeader,
   TableRow,
   cn,
-} from "@dbx-tools/ui/react";
+} from "@databricks/appkit-ui/react";
+import { type TokenizeOptions, toLabel } from "@dbx-tools/shared-core";
 import {
   flexRender,
   getCoreRowModel,
@@ -83,8 +83,8 @@ export function renderDataCell(value: unknown): React.ReactNode {
  * column id, accessor key, and CSV header, so only the on-screen label
  * is prettified.
  */
-export function humanizeLabel(value: string, options?: stringUtils.TokenizeOptions): string {
-  return stringUtils.toLabel(value, options);
+export function humanizeLabel(value: string, options?: TokenizeOptions): string {
+  return toLabel(value, options);
 }
 
 /**

package/src/react/embed-slots.tsx

@@ -1,6 +1,10 @@
-import { GenieQueryVisualization } from "@databricks/appkit-ui/react";
-import { markers, type ParsedMarker } from "@dbx-tools/shared-mastra";
-import { Spinner } from "@dbx-tools/ui/react";
+import { GenieQueryVisualization, Spinner } from "@databricks/appkit-ui/react";
+import {
+  type ParsedMarker,
+  isUuid,
+  parseMarkers,
+  stripIncompleteMarkerTail,
+} from "@dbx-tools/shared-mastra";
 import { ClockIcon } from "lucide-react";
 import { lazy, Suspense, useMemo, useRef } from "react";
 import { DataGrid, humanizeLabel } from "./data-grid.tsx";
@@ -209,7 +213,7 @@ type RenderSegment =
  * known, renderable type resolves to a slot.
  */
 const markerSegment = (marker: ParsedMarker): RenderSegment => {
-  if (!markers.isUuid(marker.id)) return { kind: "text", text: "" };
+  if (!isUuid(marker.id)) return { kind: "text", text: "" };
   switch (marker.type) {
     case "chart":
       return { kind: "chart", chartId: marker.id };
@@ -228,7 +232,7 @@ const splitTextWithEmbeds = (text: string): RenderSegment[] => {
   // `parseMarkers` yields hits in source order with no overlaps (one
   // regex pass), so the spans splice in directly - no sort or
   // overlap guard needed.
-  for (const marker of markers.parseMarkers(text)) {
+  for (const marker of parseMarkers(text)) {
     if (marker.start > lastIdx) {
       segments.push({ kind: "text", text: text.slice(lastIdx, marker.start) });
     }
@@ -273,7 +277,7 @@ export const MarkdownWithEmbeds = ({
   text: string;
   streaming?: boolean;
 }) => {
-  const segments = splitTextWithEmbeds(markers.stripIncompleteMarkerTail(text));
+  const segments = splitTextWithEmbeds(stripIncompleteMarkerTail(text));
   return (
     <>
       {segments.map((seg, i) => {

package/src/react/export-menu.tsx

@@ -7,7 +7,7 @@ import {
   Tooltip,
   TooltipContent,
   TooltipTrigger,
-} from "@dbx-tools/ui/react";
+} from "@databricks/appkit-ui/react";
 import { DownloadIcon, FileTextIcon } from "lucide-react";
 import type { ExportFormat } from "../support/export.ts";
 

package/src/react/feedback-controls.tsx

@@ -12,7 +12,7 @@ import {
   TooltipContent,
   TooltipTrigger,
   cn,
-} from "@dbx-tools/ui/react";
+} from "@databricks/appkit-ui/react";
 import { MessageSquareTextIcon, ThumbsDownIcon, ThumbsUpIcon } from "lucide-react";
 import { useState } from "react";
 import type { FeedbackSubmission, FeedbackValue } from "./types.ts";

package/src/react/markdown.tsx

@@ -11,11 +11,9 @@ import {
   Tooltip,
   TooltipContent,
   TooltipTrigger,
-  createShikiPlugin,
-  highlightToHtml,
-  useHighlightedHtml,
   cn,
-} from "@dbx-tools/ui/react";
+} from "@databricks/appkit-ui/react";
+import { createShikiPlugin, highlightToHtml, useHighlightedHtml } from "@dbx-tools/ui/react";
 import { CheckIcon, CopyIcon } from "lucide-react";
 import React, { useEffect, useRef, useState } from "react";
 import { Streamdown } from "streamdown";

package/src/react/mastra-assistant.tsx

@@ -5,7 +5,7 @@ import {
   TooltipProvider,
   TooltipTrigger,
   cn,
-} from "@dbx-tools/ui/react";
+} from "@databricks/appkit-ui/react";
 import { BotIcon, MessageCircleIcon, XIcon } from "lucide-react";
 import {
   createContext,