@dbx-tools/appkit-mastra 0.9.118 → 0.9.120

package/lib/tsconfig.tsbuildinfo

@@ -1 +1 @@
-{"root":["../src/agents.ts","../src/chart.ts","../src/chat.ts","../src/config.ts","../src/defaults.ts","../src/evaluation.ts","../src/filesystems.ts","../src/genie.ts","../src/identity.ts","../src/mcp.ts","../src/memory.ts","../src/mlflow.ts","../src/model.ts","../src/monty-sandbox.ts","../src/observability.ts","../src/openapi-tool.ts","../src/plugin.ts","../src/processors.ts","../src/remote-skills.ts","../src/rest.ts","../src/sandbox.ts","../src/server.ts","../src/serving.ts","../src/skill-paths.ts","../src/statement.ts","../src/storage-schema.ts","../src/style.ts","../src/summarize.ts","../src/telemetry.ts","../src/trace-attributes.ts","../src/validation.ts","../src/workspaces.ts","../src/writer.ts","../index.ts"],"version":"5.9.3"}
+{"root":["../src/agents.ts","../src/chart.ts","../src/chat.ts","../src/config.ts","../src/defaults.ts","../src/evaluation.ts","../src/filesystems.ts","../src/genie.ts","../src/identity.ts","../src/mcp.ts","../src/memory.ts","../src/mlflow.ts","../src/model.ts","../src/monty-sandbox.ts","../src/observability.ts","../src/openapi-tool.ts","../src/plugin.ts","../src/processors.ts","../src/remote-skills.ts","../src/rest.ts","../src/server.ts","../src/serving.ts","../src/skill-paths.ts","../src/statement.ts","../src/storage-schema.ts","../src/style.ts","../src/summarize.ts","../src/telemetry.ts","../src/trace-attributes.ts","../src/validation.ts","../src/workspaces.ts","../src/writer.ts","../index.ts"],"version":"5.9.3"}

package/package.json

@@ -34,20 +34,21 @@
   },
   "dependencies": {
     "@databricks/appkit": "0.81.0",
-    "@dbx-tools/appkit": "^0.9.118",
-    "@dbx-tools/core": "^0.9.118",
-    "@dbx-tools/databricks": "^0.9.118",
-    "@dbx-tools/fs": "^0.9.118",
-    "@dbx-tools/genie": "^0.9.118",
-    "@dbx-tools/model": "^0.9.118",
-    "@dbx-tools/model-protocol": "^0.9.118",
-    "@dbx-tools/path": "^0.9.118",
-    "@dbx-tools/postgres": "^0.9.118",
-    "@dbx-tools/shared-core": "^0.9.118",
-    "@dbx-tools/shared-fs": "^0.9.118",
-    "@dbx-tools/shared-genie": "^0.9.118",
-    "@dbx-tools/shared-mastra": "^0.9.118",
-    "@dbx-tools/shared-model": "^0.9.118",
+    "@dbx-tools/appkit": "^0.9.120",
+    "@dbx-tools/cache": "^0.9.120",
+    "@dbx-tools/core": "^0.9.120",
+    "@dbx-tools/databricks": "^0.9.120",
+    "@dbx-tools/fs": "^0.9.120",
+    "@dbx-tools/genie": "^0.9.120",
+    "@dbx-tools/model": "^0.9.120",
+    "@dbx-tools/model-protocol": "^0.9.120",
+    "@dbx-tools/path": "^0.9.120",
+    "@dbx-tools/postgres": "^0.9.120",
+    "@dbx-tools/shared-core": "^0.9.120",
+    "@dbx-tools/shared-fs": "^0.9.120",
+    "@dbx-tools/shared-genie": "^0.9.120",
+    "@dbx-tools/shared-mastra": "^0.9.120",
+    "@dbx-tools/shared-model": "^0.9.120",
     "@mastra/ai-sdk": "1.10.5",
     "@mastra/core": "1.71.0",
     "@mastra/express": "1.5.15",
@@ -68,7 +69,6 @@
     "ai": "^7.0.0",
     "express": "^5.1.0",
     "fuse.js": "^7.4.2",
-    "lru-cache": "11.5.3",
     "pg": "^8.22.0",
     "yaml": "^2.9.0",
     "zod": "4.3.6"
@@ -164,10 +164,6 @@
         "types": "./lib/src/rest.d.ts",
         "default": "./lib/src/rest.js"
       },
-      "./sandbox": {
-        "types": "./lib/src/sandbox.d.ts",
-        "default": "./lib/src/sandbox.js"
-      },
       "./server": {
         "types": "./lib/src/server.d.ts",
         "default": "./lib/src/server.js"
@@ -223,7 +219,7 @@
       "./package.json": "./package.json"
     }
   },
-  "version": "0.9.118",
+  "version": "0.9.120",
   "types": "./lib/index.d.ts",
   "dbxToolsConfig": {
     "tags": [
@@ -316,10 +312,6 @@
       "types": "./lib/src/rest.d.ts",
       "default": "./lib/src/rest.js"
     },
-    "./sandbox": {
-      "types": "./lib/src/sandbox.d.ts",
-      "default": "./lib/src/sandbox.js"
-    },
     "./server": {
       "types": "./lib/src/server.d.ts",
       "default": "./lib/src/server.js"

package/src/agents.ts

@@ -616,7 +616,7 @@ function codeModeTransport(): Promise<CodeModeTransport> {
       return transport;
     } catch (caught) {
       codeModeLogger.warn("isolated-vm unavailable; using QuickJS", {
-        error: errorUtils.errorMessage(caught),
+        error: errorUtils.toError(caught),
       });
       return new QuickJsCodeModeTransport();
     }
@@ -767,7 +767,7 @@ export async function buildAgents(opts: {
         fallbackToTmp: config.workspaceFallbackToTmp,
         ...(config.workspaceTools !== undefined ? { tools: config.workspaceTools } : {}),
         pluginContext: context,
-        sandbox: config.sandbox === true ? "databricks" : config.sandbox,
+        sandbox: config.sandbox,
       });
       markDefaultWorkspace(workspace);
     } else if (workspace) {
@@ -854,8 +854,7 @@ function toolErrorLoggingProcessor(logger: log.Logger): OutputProcessor {
         logger.error("tool execution failed", {
           toolName: part.payload.toolName,
           toolCallId: part.payload.toolCallId,
-          error: error.message,
-          stack: error.stack,
+          error,
         });
       }
       return part;

package/src/chart.ts

@@ -516,7 +516,7 @@ async function writeChart(entry: Chart, userKey: string): Promise<void> {
   } catch (err) {
     logger.warn("write-error", {
       chartId: entry.chartId,
-      error: errorUtils.errorMessage(err),
+      error: errorUtils.toError(err),
     });
   }
 }
@@ -534,7 +534,7 @@ async function readChart(chartId: string, userKey: string): Promise<Chart | unde
   } catch (err) {
     logger.warn("read-error", {
       chartId,
-      error: errorUtils.errorMessage(err),
+      error: errorUtils.toError(err),
     });
     return undefined;
   }
@@ -629,7 +629,7 @@ async function runPrepareChart(chartId: string, opts: PrepareChartOptions): Prom
       elapsedMs: Date.now() - startedAt,
     });
   } catch (err) {
-    logger.warn("error", { chartId, error: errorUtils.errorMessage(err) });
+    logger.warn("error", { chartId, error: errorUtils.toError(err) });
     // The entry's `error` is rendered in the chat, so only an AppKitError's
     // own message travels; anything else could carry upstream provider detail.
     const errText = err instanceof AppKitError ? err.message : CHART_FAILED_MESSAGE;

package/src/chat.ts

@@ -133,7 +133,7 @@ export function agentChatRoutes() {
 function serializeChatStreamError(error: unknown): string {
   if (isStaleMastraResumeError(error)) {
     logger.warn("ignored stale mastra resume", {
-      error: errorUtils.errorMessage(error),
+      error: errorUtils.toError(error),
     });
     return STALE_MASTRA_RESUME_STREAM_TEXT;
   }

package/src/config.ts

@@ -20,14 +20,17 @@ import type { BrandContext } from "@dbx-tools/shared-core";
 import type { AgentConfig } from "@mastra/core/agent";
 import type { RequestContext } from "@mastra/core/request-context";
 import { MASTRA_RESOURCE_ID_KEY, MASTRA_THREAD_ID_KEY } from "@mastra/core/request-context";
-import type { WorkspaceToolsConfig } from "@mastra/core/workspace";
+import type {
+  WorkspaceSandbox,
+  WorkspaceSandboxResolver,
+  WorkspaceToolsConfig,
+} from "@mastra/core/workspace";
 import type { PgVectorConfig, PostgresStoreConfig } from "@mastra/pg";
 
 import type { AnyMastraAgentDefinition, MastraTools } from "./agents.ts";
 import type { GenieSpacesConfig } from "./genie.ts";
 import { IDENTITY_MODES, type MastraIdentityMode } from "./identity.ts";
-import type { RemoteSkillsOption } from "./remote-skills.ts";
-import type { DatabricksWorkspaceSandboxOptions } from "./sandbox.ts";
+import type { LocalSkillMount, RemoteSkillsOption } from "./remote-skills.ts";
 
 /**
  * `RequestContext` key under which {@link MastraServer} stores the
@@ -216,30 +219,34 @@ export interface MastraPluginConfig extends BasePluginConfig {
   /** AI SDK provider-options namespace. Defaults to `"openai"`; no env fallback. */
   providerId?: string;
   /**
-   * PostgresStore for Mastra threads/messages. `true` reuses the
-   * `lakebase` plugin's pool; an object opens a dedicated store.
+   * PostgresStore for Mastra threads/messages. `true` captures the
+   * `lakebase` plugin's existing service-principal pool; an object opens a
+   * separately configured store.
    */
   storage?: boolean | PostgresStoreConfig;
   /**
-   * PgVector store for Mastra memory recall. `true` reuses the
-   * `lakebase` plugin's pool; an object opens a dedicated store.
+   * PgVector store for Mastra memory recall. `true` captures the
+   * `lakebase` plugin's existing service-principal pool; an object opens a
+   * separately configured store.
    */
   memory?: boolean | MastraMemoryConfig;
   /**
-   * Expose executable tools from each agent definition through native Mastra
-   * Code Mode. Defaults on whenever the agent has executable tools. The plugin
-   * selects Mastra's isolated-vm transport once during startup and falls back
-   * to Mastra QuickJS when the native transport is unavailable. Set `false` to
+   * Expose executable tools available to the current request through native
+   * Mastra Code Mode. Defaults on whenever the request has executable tools.
+   * The plugin selects Mastra's isolated-vm transport once during startup and
+   * falls back to Mastra QuickJS when the native transport is unavailable.
+   * These transports do not execute through the workspace command sandbox;
+   * SSH is used only when an exposed SSH-backed tool is called. Set `false` to
    * opt out.
    */
   codeMode?: boolean | MastraCodeModeConfig;
   /**
    * Sandbox for auto-created agent workspaces. Defaults to Monty. `false`
-   * disables command execution; `true`, `"databricks"`, or an object
-   * selects/configures Databricks Sandbox. An agent with an explicit custom
-   * `workspace` keeps that workspace and its sandbox.
+   * disables command execution; `true` or `"monty"` selects Monty; an explicit
+   * Mastra sandbox or resolver supplies a remote transport. An agent with an
+   * explicit custom `workspace` keeps that workspace and its sandbox.
    */
-  sandbox?: boolean | "databricks" | "monty" | DatabricksWorkspaceSandboxOptions;
+  sandbox?: boolean | "monty" | WorkspaceSandbox | WorkspaceSandboxResolver;
   /** Add user-scoped `/tmp` only when no configured workspace path mounts writable. */
   workspaceFallbackToTmp?: boolean;
   /** Native Mastra workspace tool availability, approval, and hook configuration. */
@@ -634,10 +641,10 @@ export interface MastraPluginConfig extends BasePluginConfig {
    * without it, a URL source is fetched directly. A source that resolves
    * through neither fails startup unless `failOnError: false`.
    *
-   * Provisioned skills are written to the Databricks user's Assistant skills
-   * folder (`/Workspace/Users/<email>/.assistant/skills`, the personal skill
-   * target) so they persist and are picked up by the built-in Assistant-skills
-   * mount. With no writable workspace, they go to a local temp dir instead.
+   * AI Tools is cached daily with `@dbx-tools/cache` and served to every
+   * auto-created workspace from the shared `/.databricks/skills` memory mount.
+   * Other sources are written to the Databricks user's Assistant skills folder
+   * when writable, with a read-only local-memory fallback.
    *
    * @example
    * ```ts
@@ -651,6 +658,8 @@ export interface MastraPluginConfig extends BasePluginConfig {
    * ```
    */
   remoteSkills?: RemoteSkillsOption;
+  /** Read-only local skill trees mounted into every default agent workspace. */
+  localSkillMounts?: LocalSkillMount[];
 }
 
 /**
@@ -677,27 +686,27 @@ export const MASTRA_CONFIG_SCHEMA: ConfigSchema = {
     remoteSkills: {
       type: ["string", "array", "object"],
       description:
-        'Remote Agent-Skill sources provisioned at startup: "aitools" for Databricks\' own skill set (read from the public databricks/databricks-agent-skills repo, no CLI needed), a GitHub owner/repo, a git/GitLab URL, or a direct SKILL.md/archive URL. Non-aitools sources prefer the optional `skills` npm CLI, else a direct fetch; fails startup on error unless failOnError:false. Written to the Databricks user Assistant skills folder, or read-only local backing storage exposed at stable virtual workspace mounts.',
+        'Remote Agent-Skill sources provisioned at startup: "aitools" for Databricks\' own skill set (daily @dbx-tools/cache snapshot served from the shared in-memory /.databricks/skills mount), a GitHub owner/repo, a git/GitLab URL, or a direct SKILL.md/archive URL. Non-aitools sources prefer the optional `skills` npm CLI, else a direct fetch; fails startup on error unless failOnError:false. Other sources write to the Databricks user Assistant skills folder or a read-only local-memory fallback.',
     },
     storage: {
       type: ["boolean", "object"],
       description:
-        "PostgresStore for Mastra threads / messages. `true` reuses the `lakebase` plugin's pool, an object opens a dedicated store. Auto-enabled when the `lakebase` plugin is registered.",
+        "PostgresStore for Mastra threads / messages. `true` captures the `lakebase` plugin's existing service-principal pool; an object opens a separately configured store. Auto-enabled when the `lakebase` plugin is registered.",
     },
     memory: {
       type: ["boolean", "object"],
       description:
-        "PgVector store for Mastra semantic recall. `true` reuses the `lakebase` plugin's pool, an object opens a dedicated store. Auto-enabled when the `lakebase` plugin is registered.",
+        "PgVector store for Mastra semantic recall. `true` captures the `lakebase` plugin's existing service-principal pool; an object opens a separately configured store. Auto-enabled when the `lakebase` plugin is registered.",
     },
     codeMode: {
       type: ["boolean", "object"],
       description:
-        "Expose all executable tools available to the current request, including native workspace tools, through dynamically generated Mastra Code Mode for sandboxed TypeScript batching, aggregation, and arithmetic. The plugin selects Mastra isolated-vm once during startup and falls back to Mastra QuickJS when native initialization is unavailable. False disables it.",
+        "Expose all executable tools available to the current request through native Mastra Code Mode for TypeScript batching, aggregation, and arithmetic. The plugin uses Mastra isolated-vm with the existing QuickJS fallback; neither transport runs through the workspace command sandbox, so SSH is used only when an SSH-backed tool is called. False disables it.",
     },
     sandbox: {
       type: ["boolean", "string", "object"],
       description:
-        'Command sandbox for auto-created agent workspaces. Defaults to Monty. False disables command execution; true, "databricks", or an object selects/configures Databricks Sandbox.',
+        'Command sandbox for auto-created agent workspaces. Defaults to Monty. False disables command execution; true or "monty" selects Monty; an explicit sandbox or resolver supplies another transport.',
     },
     workspaceFallbackToTmp: {
       type: "boolean",

package/src/defaults.ts

@@ -53,9 +53,9 @@ const STATEMENT_CACHE_TTL_SEC = 5 * 60;
 export const modelCatalogueDefaults: ExecuteSettings = {
   default: {
     // Cache disabled here because `listServingEndpoints` already memoizes the
-    // catalogue through the same `CacheManager`, keyed by workspace host with
-    // `config.modelCacheTtlMs`. That entry is shared across identities; a
-    // second interceptor cache would duplicate it once per user.
+    // catalogue in the shared model cache, keyed by workspace host and trusted
+    // credential identity with `config.modelCacheTtlMs`. A second interceptor
+    // cache would duplicate it once per user.
     cache: { enabled: false },
     // Retry enabled: listing endpoints is an idempotent GET, and the listing
     // is on the chat UI's load path, so a transient 5xx should not blank the

package/src/genie.ts

@@ -45,6 +45,7 @@ import {
 } from "@databricks/appkit";
 import type { WorkspaceClient } from "@databricks/appkit";
 import { pluginRegistry } from "@dbx-tools/appkit";
+import { createCache } from "@dbx-tools/cache/cache";
 import { chat, space as genieSpace } from "@dbx-tools/genie";
 import { errorUtils, log, stringUtils } from "@dbx-tools/shared-core";
 import { genieModel, type GenieMessage } from "@dbx-tools/shared-genie";
@@ -52,7 +53,6 @@ import type { StartedEvent } from "@dbx-tools/shared-mastra";
 import type { RequestContext } from "@mastra/core/request-context";
 import { MASTRA_THREAD_ID_KEY } from "@mastra/core/request-context";
 import { createTool, type ToolExecutionContext } from "@mastra/core/tools";
-import { LRUCache } from "lru-cache";
 import { z } from "zod";
 
 import type { MastraTools } from "./agents.ts";
@@ -248,7 +248,7 @@ async function readCachedConversationId(cacheKey: string | undefined): Promise<s
     return v ?? undefined;
   } catch (err) {
     logger.warn("conversation-cache:read-error", {
-      error: errorUtils.errorMessage(err),
+      error: errorUtils.toError(err),
     });
     return undefined;
   }
@@ -272,7 +272,7 @@ async function saveCachedConversationId(
     });
   } catch (err) {
     logger.warn("conversation-cache:write-error", {
-      error: errorUtils.errorMessage(err),
+      error: errorUtils.toError(err),
     });
   }
 }
@@ -284,7 +284,7 @@ async function evictCachedConversationId(cacheKey: string | undefined): Promise<
     await CacheManager.getInstanceSync().delete(cacheKey);
   } catch (err) {
     logger.warn("conversation-cache:delete-error", {
-      error: errorUtils.errorMessage(err),
+      error: errorUtils.toError(err),
     });
   }
 }
@@ -515,7 +515,7 @@ function buildAskGenieTool(opts: {
             logger.warn("conversation-cache:stale, resetting", {
               spaceId,
               conversationId: rejectedConversationId,
-              error: errorUtils.errorMessage(err),
+              error: errorUtils.toError(err),
             });
             if (ownsReusableConversation) {
               await evictCachedConversationId(cacheKey);
@@ -1091,19 +1091,19 @@ const SUGGESTION_LIMIT = 6;
 const SUGGESTION_CACHE_TTL_MS = 10 * 60_000;
 
 /** Bounded process-local cache of parsed sample questions by Genie space id. */
-const _suggestionCache = new LRUCache<string, string[]>({
-  max: 128,
+const _suggestionCache = createCache({
+  maxEntries: 128,
   ttl: SUGGESTION_CACHE_TTL_MS,
 });
 
 /** Read a space's cached questions, or `undefined` on miss / expiry. */
-function readSuggestionCache(spaceId: string): string[] | undefined {
-  return _suggestionCache.get(spaceId);
+function readSuggestionCache(spaceId: string): Promise<string[] | undefined> {
+  return _suggestionCache.get<string[]>(spaceId);
 }
 
 /** Cache a space's parsed questions for {@link SUGGESTION_CACHE_TTL_MS}. */
-function writeSuggestionCache(spaceId: string, questions: string[]): void {
-  _suggestionCache.set(spaceId, questions);
+async function writeSuggestionCache(spaceId: string, questions: string[]): Promise<void> {
+  await _suggestionCache.set(spaceId, questions);
 }
 
 /**
@@ -1115,7 +1115,9 @@ function writeSuggestionCache(spaceId: string, questions: string[]): void {
  * questions and a multi-space app round-trips breadth-first up to the
  * cap. A per-space fetch failure degrades to "no questions for that
  * space" (logged, not thrown) so one unreachable space never blanks
- * the whole list. Returns `[]` when no spaces are configured.
+ * the whole list. When `signal` aborts (the HTTP caller disconnected),
+ * collection stops silently and returns what was merged so far.
+ * Returns `[]` when no spaces are configured.
  */
 export async function collectSpaceSuggestions(opts: {
   spaces: Record<string, GenieSpaceConfig>;
@@ -1128,7 +1130,7 @@ export async function collectSpaceSuggestions(opts: {
   const seen = new Set<string>();
 
   for (const { spaceId } of Object.values(opts.spaces)) {
-    let questions = readSuggestionCache(spaceId);
+    let questions = await readSuggestionCache(spaceId);
     if (!questions) {
       try {
         const space = await genieSpace.getGenieSpace(spaceId, {
@@ -1136,11 +1138,12 @@ export async function collectSpaceSuggestions(opts: {
           ...(opts.signal ? { context: opts.signal } : {}),
         });
         questions = genieSpace.genieSampleQuestions(space);
-        writeSuggestionCache(spaceId, questions);
+        await writeSuggestionCache(spaceId, questions);
       } catch (err) {
+        if (opts.signal?.aborted) return merged;
         logger.warn("suggestions:fetch-error", {
           spaceId,
-          error: errorUtils.errorMessage(err),
+          error: errorUtils.toError(err),
         });
         questions = [];
       }

package/src/memory.ts

@@ -32,9 +32,7 @@
  * @module
  */
 
-import { getUsernameWithApiLookup } from "@databricks/appkit";
 import { migration as appkitMigration } from "@dbx-tools/appkit";
-import { postgresConnectionOptions } from "@dbx-tools/postgres";
 import { hash, log } from "@dbx-tools/shared-core";
 import type { AgentMemoryOption } from "@mastra/core/agent";
 import {
@@ -45,7 +43,7 @@ import {
 import { fastembed } from "@mastra/fastembed";
 import { Memory } from "@mastra/memory";
 import { PgVector, PostgresStore } from "@mastra/pg";
-import { Pool, type PoolConfig } from "pg";
+import type { Pool } from "pg";
 
 import type { AnyMastraAgentDefinition, MastraMemoryConfigOverride } from "./agents.ts";
 import type { MastraPluginConfig } from "./config.ts";
@@ -132,34 +130,6 @@ export function withSoftStorageInit(store: PostgresStore): PostgresStore {
   return store;
 }
 
-/**
- * Build a dedicated **service-principal** Lakebase pool for Mastra
- * memory from the lakebase plugin's resolved SP pg config.
- *
- * The plugin's `exports().pool` is a `RoutingPool` that switches to
- * the per-user (OBO) pool whenever a query runs inside an `asUser`
- * scope - exactly the context the mastra plugin establishes around
- * every chat turn. Memory (threads / messages + semantic recall) must
- * instead always act as the app service principal: it owns the
- * auto-created `mastra_*` schemas (a per-user role usually can't
- * `CREATE SCHEMA`) and is shared across users, so it cannot inherit a
- * request's OBO identity.
- *
- * `pgConfig` must be the plugin's `exports().getPgConfig()` evaluated
- * **outside** any `asUser` scope (i.e. during setup), so it carries
- * the SP connection target, OAuth token-refresh `password` callback,
- * and any `lakebase({ pool })` tuning overrides - all of which this
- * pool inherits. See the call site in `plugin.ts`.
- */
-export async function createServicePrincipalPool(pgConfig: PoolConfig): Promise<Pool> {
-  // `getPgConfig()` resolves the SP username synchronously from
-  // `PGUSER` / `DATABRICKS_CLIENT_ID`; fall back to the async API
-  // lookup (e.g. local dev authenticating via PAT) so the pool always
-  // has an identity to connect with.
-  const user = pgConfig.user ?? (await getUsernameWithApiLookup());
-  return new Pool(postgresConnectionOptions({ ...pgConfig, user }));
-}
-
 /** Effective per-knob setting after the plugin/agent cascade. */
 type StorageSetting = AnyMastraAgentDefinition["storage"];
 type MemorySetting = AnyMastraAgentDefinition["memory"];
@@ -179,7 +149,7 @@ export function needsLakebase(config: MastraPluginConfig): boolean {
 
 /**
  * Construct a per-agent {@link Memory} factory bound to the supplied
- * service-principal pool (see {@link createServicePrincipalPool}).
+ * service-principal pool.
  * Caches the shared `PgVector` singleton (built on first need) so each
  * agent build is O(1) after the first.
  */

package/src/mlflow.ts

@@ -366,7 +366,7 @@ export async function initializeAppMlflowTraceInfo(client?: WorkspaceClient): Pr
     return true;
   } catch (err) {
     logger.warn("Databricks Apps MLflow trace-info promotion disabled", {
-      error: errorUtils.errorMessage(err),
+      error: errorUtils.toError(err),
     });
     return false;
   }
@@ -424,7 +424,7 @@ export function persistAppMlflowTraceInfo(params: AppMlflowTraceInfo): void {
     .catch((err) => {
       logger.warn("Databricks Apps MLflow trace-info promotion failed", {
         traceId: params.traceId,
-        error: errorUtils.errorMessage(err),
+        error: errorUtils.toError(err),
       });
     })
     .finally(() => pendingAppMlflowTraceInfo.delete(pending));
@@ -466,7 +466,7 @@ async function detectedDirectMlflowTraceLocation(
   } catch (err) {
     logger.warn("direct MLflow UC trace-location detection failed", {
       experimentId,
-      error: errorUtils.errorMessage(err),
+      error: errorUtils.toError(err),
     });
     return undefined;
   }
@@ -530,7 +530,7 @@ export async function initializeDirectMlflowTracing(): Promise<boolean> {
     return true;
   } catch (err) {
     logger.warn("direct MLflow tracing disabled", {
-      error: errorUtils.errorMessage(err),
+      error: errorUtils.toError(err),
     });
     return false;
   }
@@ -693,18 +693,10 @@ async function resolveExperimentId(client: WorkspaceClient): Promise<string | un
 }
 
 /** Parameters for {@link logFeedback}. */
-export interface LogFeedbackParams {
-  /** MLflow trace id the assessment attaches to (`tr-<hex>`). */
-  traceId: string;
-  /** Assessment name; defaults per whether a value or a comment is sent. */
-  name?: string;
-  /** Thumbs / rating / label value. Omit for a comment-only submission. */
-  value?: boolean | number | string;
-  /** Freeform comment: the rationale alongside a value, or the value itself when none. */
-  comment?: string;
+export type LogFeedbackParams = feedback.MastraFeedbackRequest & {
   /** Identity the feedback is attributed to (user email / id). */
   sourceId?: string;
-}
+};
 
 /**
  * Log a HUMAN feedback assessment to a trace. Returns the created
@@ -751,7 +743,7 @@ export async function logFeedback(
       logger.warn("feedback request failed", {
         traceId: params.traceId,
         assessmentTraceId,
-        error: errorUtils.errorMessage(err),
+        error: errorUtils.toError(err),
       });
       return undefined;
     }

package/src/monty-sandbox.ts

@@ -68,7 +68,7 @@ const montyContext = functionUtils.memoize(async (): Promise<MontyContext> => {
     } catch (error) {
       errors.push(errorUtils.toError(error));
       logger.debug(`${target.name} unavailable`, {
-        error: errorUtils.errorMessage(error),
+        error: errorUtils.toError(error),
       });
     }
   }