@dbx-tools/appkit-mastra 0.3.1 → 0.3.3

package/package.json

@@ -27,22 +27,22 @@
     "@opentelemetry/api": "^1.9.1",
     "pg": "^8.22.0",
     "zod": "^4.3.6",
-    "@dbx-tools/appkit": "0.3.1",
-    "@dbx-tools/core": "0.3.1",
-    "@dbx-tools/genie": "0.3.1",
-    "@dbx-tools/databricks": "0.3.1",
-    "@dbx-tools/shared-core": "0.3.1",
-    "@dbx-tools/model": "0.3.1",
-    "@dbx-tools/shared-genie": "0.3.1",
-    "@dbx-tools/shared-mastra": "0.3.1",
-    "@dbx-tools/shared-model": "0.3.1"
+    "@dbx-tools/appkit": "0.3.3",
+    "@dbx-tools/databricks": "0.3.3",
+    "@dbx-tools/model": "0.3.3",
+    "@dbx-tools/shared-genie": "0.3.3",
+    "@dbx-tools/shared-core": "0.3.3",
+    "@dbx-tools/core": "0.3.3",
+    "@dbx-tools/genie": "0.3.3",
+    "@dbx-tools/shared-mastra": "0.3.3",
+    "@dbx-tools/shared-model": "0.3.3"
   },
   "main": "index.ts",
   "license": "UNLICENSED",
   "publishConfig": {
     "access": "public"
   },
-  "version": "0.3.1",
+  "version": "0.3.3",
   "types": "index.ts",
   "type": "module",
   "exports": {

package/src/agents.ts

@@ -328,6 +328,13 @@ export type MastraMemoryConfigOverride = DistributiveOmit<PgVectorConfig, "id">
 export interface BuiltAgents {
   agents: Record<string, Agent>;
   defaultAgentId: string;
+  /**
+   * Static default serving-endpoint id per agent id, as resolved by
+   * {@link describeAgentDefaultModel}. `"<dynamic>"` when the model is a
+   * function decided at call time. Surfaced so `clientConfig()` can label
+   * the picker's "Server default" option with the real endpoint.
+   */
+  defaultModels: Record<string, string>;
   /**
    * Ambient tools shared across every agent (the built-in system tools
    * spread with `config.tools`). Surfaced so the optional MCP server
@@ -469,6 +476,7 @@ export async function buildAgents(opts: {
     ...(config.stripStaleCharts === false ? [] : [stripStaleChartsProcessor]),
   ];
   const agents: Record<string, Agent> = {};
+  const defaultModels: Record<string, string> = {};
   const approvalGatedByAgent: Array<{ agentId: string; toolIds: string[] }> = [];
 
   for (const [id, def] of Object.entries(definitions)) {
@@ -502,10 +510,12 @@ export async function buildAgents(opts: {
     // *static* default; per-request overrides (header / query /
     // body) and the workspace-catalogue fuzzy match still apply at
     // call time.
+    const defaultModel = describeAgentDefaultModel(config, def);
+    defaultModels[id] = defaultModel;
     log.info("agent registered", {
       id,
       name: def.name ?? id,
-      defaultModel: describeAgentDefaultModel(config, def),
+      defaultModel,
       tools: Object.keys(tools),
     });
   }
@@ -519,7 +529,7 @@ export async function buildAgents(opts: {
   assertApprovalGatedToolsHaveStorage(approvalGatedByAgent, memoryBuilder);
 
   log.info("agents ready", { ids, defaultAgentId });
-  return { agents, defaultAgentId, ambientTools };
+  return { agents, defaultAgentId, defaultModels, ambientTools };
 }
 
 /**

package/src/plugin.ts

@@ -76,7 +76,7 @@ import { attachRoutePatchMiddleware, isMastraRequestAllowed, MastraServer } from
 import { resolveServingConfig } from "./serving";
 import { fetchStatementData, STATEMENT_ROW_CAP } from "./statement";
 import { threadsRoute } from "./threads";
-import { error, log } from "@dbx-tools/shared-core";
+import { error, log, string } from "@dbx-tools/shared-core";
 import { plugin } from "@dbx-tools/appkit";
 
 const GENIE_MANIFEST = plugin.data(genie).plugin.manifest;
@@ -318,6 +318,20 @@ export class MastraPlugin extends Plugin<MastraPluginConfig> {
         .catch(next);
     });
 
+    // `GET /default-model[?agentId=]` reports the static serving-endpoint id
+    // an agent falls back to when the client pins no model, so the picker can
+    // name its "Server default" option. Agent-scoped (defaults to the default
+    // agent); `model` is null when the agent has no static default (a dynamic,
+    // call-time model) or the agent id is unknown. Registered before the
+    // catch-all, same as `/models`.
+    router.get(routes.MASTRA_ROUTES.defaultModel, (req, res) => {
+      const requested = string.firstNonEmpty(req.query["agentId"]);
+      const agentId = requested ?? this.built?.defaultAgentId ?? FALLBACK_AGENT_ID;
+      const model = this.built?.defaultModels[agentId];
+      // `"<dynamic>"` (a call-time function) has no fixed id to advertise.
+      res.json({ agentId, model: model && model !== "<dynamic>" ? model : null });
+    });
+
     // `GET /embed/:type/:id` is the single resolver for every embed
     // marker the agent emits in prose (`[chart:<id>]`,
     // `[data:<id>]`, ...). `:type` selects a resolver from the
@@ -746,6 +760,7 @@ export class MastraPlugin extends Plugin<MastraPluginConfig> {
         "/route/history",
         "/route/threads",
         "/models",
+        "/default-model",
         "/suggestions",
         "/route/feedback",
         "/embed/:type/:id",