@dbx-tools/ui-mastra 0.9.74 → 0.9.93

package/index.ts

@@ -3,7 +3,7 @@
 // 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.74";
+export const PACKAGE_VERSION = "0.9.93";
 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";

package/package.json

@@ -29,11 +29,11 @@
     "react-dom": "~19.2.4"
   },
   "dependencies": {
-    "@dbx-tools/shared-core": "^0.9.74",
-    "@dbx-tools/shared-genie": "^0.9.74",
-    "@dbx-tools/shared-mastra": "^0.9.74",
-    "@dbx-tools/shared-model": "^0.9.74",
-    "@dbx-tools/ui": "^0.9.74",
+    "@dbx-tools/shared-core": "^0.9.93",
+    "@dbx-tools/shared-genie": "^0.9.93",
+    "@dbx-tools/shared-mastra": "^0.9.93",
+    "@dbx-tools/shared-model": "^0.9.93",
+    "@dbx-tools/ui": "^0.9.93",
     "@mastra/client-js": "1.50.0",
     "@mastra/core": "1.71.0",
     "@tanstack/react-table": "^8.21.3",
@@ -50,7 +50,7 @@
   "publishConfig": {
     "access": "public"
   },
-  "version": "0.9.74",
+  "version": "0.9.93",
   "dbxToolsConfig": {
     "tags": [
       "ui"

package/src/react/tool-pill.tsx

@@ -15,18 +15,34 @@ import type { ToolEvent, ToolProgress } from "./types.ts";
 // groups the wire events one assistant turn produced into per-sub-call
 // cards (question + numbered queries + prose answers + errors).
 
+/**
+ * Mastra registers built-in workspace tools under
+ * `WORKSPACE_TOOLS_PREFIX` (`mastra_workspace_*`). The wire id stays
+ * that branded name; session pills drop the prefix so labels read as
+ * `Execute Command` rather than `Mastra Workspace Execute Command`.
+ */
+const MASTRA_WORKSPACE_TOOL_PREFIX = "mastra_workspace_";
+
 /**
  * Turn a snake/camel tool id into a Title Cased label the user can
  * read. Genie tools land on this surface as flat ids
  * (`ask_genie`, `get_statement`, `prepare_chart`) plus per-space
- * suffixes for non-default aliases (`ask_genie_sales`).
+ * suffixes for non-default aliases (`ask_genie_sales`). Mastra
+ * workspace tools keep their `mastra_workspace_` wire prefix, which
+ * this strips before labeling.
  *
  * Examples:
  *   `ask_genie`     -> `Ask Genie`
  *   `ask_genie_sales` -> `Ask Genie Sales`
+ *   `mastra_workspace_execute_command` -> `Execute Command`
  *   `myCoolTool`    -> `My Cool Tool`
  */
-export const humanizeToolName = (toolName: string): string => stringUtils.toLabel(toolName);
+export const humanizeToolName = (toolName: string): string => {
+  const label = toolName.startsWith(MASTRA_WORKSPACE_TOOL_PREFIX)
+    ? toolName.slice(MASTRA_WORKSPACE_TOOL_PREFIX.length)
+    : toolName;
+  return stringUtils.toLabel(label);
+};
 
 /**
  * Track the freshest status label a running tool has published so the

package/src/support/shiki-plugin.ts

@@ -13,7 +13,10 @@ import type { CodeHighlighterPlugin } from "streamdown";
  * query previews); the rest cover code the assistant might emit. Kept
  * to a curated set so shiki only bundles these grammars.
  */
-const LANGUAGES: BundledLanguage[] = [
+const LANGUAGES = [
+  "text",
+  "plaintext",
+  "txt",
   "sql",
   "python",
   "typescript",
@@ -22,7 +25,7 @@ const LANGUAGES: BundledLanguage[] = [
   "bash",
   "yaml",
   "markdown",
-];
+] as const;
 
 /**
  * Single light theme. The demo has no dark-mode toggle, and a single
@@ -33,7 +36,7 @@ const LANGUAGES: BundledLanguage[] = [
 const THEME: BundledTheme = "github-light";
 
 /** Languages we can tokenize, as a set for O(1) support checks. */
-const SUPPORTED = new Set<BundledLanguage>(LANGUAGES);
+const SUPPORTED = new Set<string>(LANGUAGES);
 
 let _highlighter: HighlighterCore | null = null;
 let _loading: Promise<HighlighterCore> | null = null;
@@ -50,9 +53,9 @@ function loadHighlighter(): Promise<HighlighterCore> {
 }
 
 /** Tokenize `code` with the active theme into Streamdown's result shape. */
-function highlightTokens(h: HighlighterCore, code: string, language: BundledLanguage) {
+function highlightTokens(h: HighlighterCore, code: string, language: string) {
   const { tokens, fg, bg, rootStyle } = h.codeToTokens(code, {
-    lang: language,
+    lang: language as BundledLanguage,
     theme: THEME,
   });
   return { tokens, fg, bg, rootStyle };
@@ -70,7 +73,7 @@ const escapeHtml = stringUtils.escapeHtml;
  * supported or shiki fails to parse the snippet.
  */
 export async function highlightToHtml(code: string, language: string): Promise<string> {
-  if (!SUPPORTED.has(language as BundledLanguage)) return escapeHtml(code);
+  if (!SUPPORTED.has(language)) return escapeHtml(code);
   const h = await loadHighlighter();
   try {
     const { tokens } = h.codeToTokens(code, {
@@ -100,13 +103,12 @@ export async function highlightToHtml(code: string, language: string): Promise<s
  * block stays plaintext rather than throwing.
  */
 export function createShikiPlugin(): CodeHighlighterPlugin {
-  const isSupported = (language: string): language is BundledLanguage =>
-    SUPPORTED.has(language as BundledLanguage);
+  const isSupported = (language: string): boolean => SUPPORTED.has(language);
 
   return {
     name: "shiki",
     type: "code-highlighter",
-    getSupportedLanguages: () => LANGUAGES,
+    getSupportedLanguages: () => [...LANGUAGES],
     getThemes: () => [THEME, THEME],
     supportsLanguage: (language) => isSupported(language),
     highlight: (options, callback) => {