@dbx-tools/appkit-mastra 0.1.42 → 0.1.56

package/dist/index.d.ts

@@ -18,4 +18,3 @@ export * from "./src/config.js";
 export * from "./src/genie.js";
 export * from "./src/plugin.js";
 export { MASTRA_MODEL_OVERRIDE_KEY, MODEL_OVERRIDE_BODY_FIELDS, MODEL_OVERRIDE_HEADER, MODEL_OVERRIDE_QUERY, extractModelOverride, type ModelOverrideRequest, } from "./src/serving.js";
-export * from "./src/tools/email.js";

package/dist/index.js

@@ -18,4 +18,3 @@ export * from "./src/config.js";
 export * from "./src/genie.js";
 export * from "./src/plugin.js";
 export { MASTRA_MODEL_OVERRIDE_KEY, MODEL_OVERRIDE_BODY_FIELDS, MODEL_OVERRIDE_HEADER, MODEL_OVERRIDE_QUERY, extractModelOverride, } from "./src/serving.js";
-export * from "./src/tools/email.js";

package/dist/src/chart.js

@@ -30,7 +30,7 @@ import { commonUtils, logUtils, stringUtils } from "@dbx-tools/shared";
 import { Agent } from "@mastra/core/agent";
 import { createTool } from "@mastra/core/tools";
 import { z } from "zod";
-import { buildModel, ModelTier } from "./model.js";
+import { buildModel, ModelClass } from "./model.js";
 const log = logUtils.logger("mastra/chart");
 /* ------------------------------ constants ------------------------------ */
 /**
@@ -276,7 +276,7 @@ function getPlannerAgent(config) {
             name: "Chart Planner",
             description: "Picks chart type and axis encodings for a dataset.",
             instructions: CHART_PLANNER_INSTRUCTIONS,
-            model: ({ requestContext }) => buildModel(config, requestContext, { tier: ModelTier.Fast }),
+            model: ({ requestContext }) => buildModel(config, requestContext, { modelClass: ModelClass.ChatFast }),
         });
         plannerAgents.set(config, agent);
     }

package/dist/src/config.d.ts

@@ -4,7 +4,8 @@
  * Kept in a leaf module so `plugin.ts`, `server.ts`, `model.ts`, and
  * `memory.ts` can import them without creating a cycle.
  */
-import type { BasePluginConfig, getExecutionContext } from "@databricks/appkit";
+import type { BasePluginConfig } from "@databricks/appkit";
+import type { appkitUtils } from "@dbx-tools/shared";
 import type { AgentConfig } from "@mastra/core/agent";
 import type { PgVectorConfig, PostgresStoreConfig } from "@mastra/pg";
 import type { MastraAgentDefinition, MastraTools } from "./agents.js";
@@ -51,7 +52,7 @@ export declare const TRACE_REQUEST_CONTEXT_KEYS: readonly string[];
 /** AppKit execution context plus the canonical user id. */
 export interface User {
     id: string;
-    executionContext: ReturnType<typeof getExecutionContext>;
+    executionContext: appkitUtils.ExecutionContextLike;
 }
 /** PgVector config with an optional Mastra store id. */
 export type MastraMemoryConfig = PgVectorConfig & {
@@ -182,9 +183,10 @@ export interface MastraPluginConfig extends BasePluginConfig {
      *
      * When unset, resolution is driven by the live Foundation Model API
      * `quality` / `speed` / `cost` scores: endpoints are classified into
-     * tiers (`classifyEndpoints`) and walked best-first (Thinking ->
-     * Balanced -> Fast), with the small built-in `FALLBACK_MODEL_IDS`
-     * list as the floor when the catalogue can't be read. Set this to
+     * chat classes (`classifyEndpoints`) and walked best-first
+     * (ChatThinking -> ChatBalanced -> ChatFast), with the small built-in
+     * `FALLBACK_MODEL_IDS` list as the floor when the catalogue can't be
+     * read. Set this to
      * pin a regulated workspace to an approved subset, or to put custom
      * endpoints in front of the auto-classified catalogue.
      */

package/dist/src/model.d.ts

@@ -9,39 +9,39 @@
  *
  * This module only adds the Mastra-specific glue. The actual model
  * selection - listing the workspace catalogue and resolving an
- * explicit name / tier / fallback chain to a real endpoint id - lives
+ * explicit name / class / fallback chain to a real endpoint id - lives
  * in `@dbx-tools/model` ({@link selectModel}) so non-Mastra consumers
  * (e.g. a job that just needs a model name) can reuse it. Here we
  * assemble the explicit ask from Mastra's request context (the
  * per-request override under {@link MASTRA_MODEL_OVERRIDE_KEY}, the
  * agent / plugin `modelId`, or `DATABRICKS_SERVING_ENDPOINT_NAME`),
- * pass the plugin's fuzzy / tier / fallback knobs through, and wrap
+ * pass the plugin's fuzzy / class / fallback knobs through, and wrap
  * the resolved id in the OpenAI-compatible provider config Mastra
  * expects. Catalogue fetches fail loud: network / auth errors
  * propagate so callers see the real SDK message.
  */
-import { type ModelTier } from "@dbx-tools/model";
+import { type ModelClass } from "@dbx-tools/model";
 import type { MastraModelConfig } from "@mastra/core/llm";
 import type { RequestContext } from "@mastra/core/request-context";
 import { type MastraPluginConfig } from "./config.js";
-export { classifyEndpoints, FALLBACK_MODEL_IDS, modelForTier, modelsForTier, ModelTier, } from "@dbx-tools/model";
+export { classifyEndpoints, FALLBACK_MODEL_IDS, ModelClass, modelForClass, modelsForClass, } from "@dbx-tools/model";
 /** Optional overrides accepted by {@link buildModel}. */
 export interface BuildModelOverrides {
     /**
      * Static model id from the agent / plugin config (string sugar on
      * `def.model` or `config.defaultModel`). Loses to the per-request
-     * override but wins over env / tier / fallback.
+     * override but wins over env / class / fallback.
      */
     modelId?: string;
     /**
-     * Capability tier to resolve when no explicit model id is supplied.
-     * Used by internal agents (e.g. the chart planner asks for
-     * {@link ModelTier.Fast}) to express intent without pinning an
+     * Chat capability class to resolve when no explicit model id is
+     * supplied. Used by internal agents (e.g. the chart planner asks for
+     * {@link ModelClass.ChatFast}) to express intent without pinning an
      * endpoint name; the live catalogue is classified and the top
-     * available model in the tier is chosen, falling back to the
-     * tier's static list when the workspace has none.
+     * available model in the class is chosen, falling back to the
+     * class's static list when the workspace has none.
      */
-    tier?: ModelTier;
+    modelClass?: ModelClass;
 }
 /**
  * Resolve a `MastraModelConfig` for the current agent step. Runs

package/dist/src/model.js

@@ -9,22 +9,22 @@
  *
  * This module only adds the Mastra-specific glue. The actual model
  * selection - listing the workspace catalogue and resolving an
- * explicit name / tier / fallback chain to a real endpoint id - lives
+ * explicit name / class / fallback chain to a real endpoint id - lives
  * in `@dbx-tools/model` ({@link selectModel}) so non-Mastra consumers
  * (e.g. a job that just needs a model name) can reuse it. Here we
  * assemble the explicit ask from Mastra's request context (the
  * per-request override under {@link MASTRA_MODEL_OVERRIDE_KEY}, the
  * agent / plugin `modelId`, or `DATABRICKS_SERVING_ENDPOINT_NAME`),
- * pass the plugin's fuzzy / tier / fallback knobs through, and wrap
+ * pass the plugin's fuzzy / class / fallback knobs through, and wrap
  * the resolved id in the OpenAI-compatible provider config Mastra
  * expects. Catalogue fetches fail loud: network / auth errors
  * propagate so callers see the real SDK message.
  */
-import { parseModelTier, selectModel } from "@dbx-tools/model";
+import { parseModelClass, selectModel } from "@dbx-tools/model";
 import { commonUtils, logUtils, netUtils, stringUtils } from "@dbx-tools/shared";
 import { MASTRA_USER_KEY } from "./config.js";
 import { MASTRA_MODEL_OVERRIDE_KEY, resolveServingConfig } from "./serving.js";
-export { classifyEndpoints, FALLBACK_MODEL_IDS, modelForTier, modelsForTier, ModelTier, } from "@dbx-tools/model";
+export { classifyEndpoints, FALLBACK_MODEL_IDS, ModelClass, modelForClass, modelsForClass, } from "@dbx-tools/model";
 /**
  * Resolve a `MastraModelConfig` for the current agent step. Runs
  * while `agent.stream` is inside the `asUser(req)` scope so tokens
@@ -48,21 +48,21 @@ export async function buildModel(config, requestContext, overrides = {}) {
         ? requestContext.get(MASTRA_MODEL_OVERRIDE_KEY)
         : undefined;
     // The override / agent default / env value can be either a concrete
-    // endpoint name or a capability tier slug ("thinking" / "balanced" /
-    // "fast"). A tier slug becomes a tier intent (let the live catalogue
-    // pick the best model in that band); anything else is an explicit
-    // name fuzzy-matched against the catalogue. An internal
-    // `overrides.tier` (e.g. the chart planner) is the floor when nothing
-    // was requested.
+    // endpoint name or a model class slug ("chat-thinking" /
+    // "chat-balanced" / "chat-fast"). A class slug becomes a class intent
+    // (let the live catalogue pick the best model in that band); anything
+    // else is an explicit name fuzzy-matched against the catalogue. An
+    // internal `overrides.modelClass` (e.g. the chart planner) is the
+    // floor when nothing was requested.
     const requested = override ?? overrides.modelId ?? process.env.DATABRICKS_SERVING_ENDPOINT_NAME;
-    const requestedTier = requested !== undefined ? parseModelTier(requested) : null;
-    const explicit = requestedTier === null ? requested : undefined;
-    const tier = requestedTier ?? overrides.tier;
+    const requestedClass = requested !== undefined ? parseModelClass(requested) : null;
+    const explicit = requestedClass === null ? requested : undefined;
+    const modelClass = requestedClass ?? overrides.modelClass;
     const { modelId, source } = await selectModel(user.executionContext.client, host, {
         ...(explicit !== undefined ? { explicit } : {}),
         fuzzy: serving.fuzzy,
         threshold: serving.threshold,
-        ...(tier !== undefined ? { tier } : {}),
+        ...(modelClass !== undefined ? { modelClass } : {}),
         fallbacks: serving.fallbacks,
         ttlMs: serving.ttlMs,
     });

package/dist/src/serving.d.ts

@@ -2,7 +2,7 @@
  * Mastra-specific glue over the generic `@dbx-tools/model` toolkit.
  *
  * The live `/serving-endpoints` catalogue access, fuzzy name
- * resolution, and tier/fallback selection all live in
+ * resolution, and class/fallback selection all live in
  * `@dbx-tools/model`; this module only adds what is specific to the
  * Mastra plugin: pulling a per-request model override off an HTTP
  * request (header / query / body) and projecting the plugin config

package/dist/src/serving.js

@@ -2,7 +2,7 @@
  * Mastra-specific glue over the generic `@dbx-tools/model` toolkit.
  *
  * The live `/serving-endpoints` catalogue access, fuzzy name
- * resolution, and tier/fallback selection all live in
+ * resolution, and class/fallback selection all live in
  * `@dbx-tools/model`; this module only adds what is specific to the
  * Mastra plugin: pulling a per-request model override off an HTTP
  * request (header / query / body) and projecting the plugin config