@dbx-tools/appkit-mastra 0.1.25 → 0.1.27

package/package.json

@@ -9,13 +9,13 @@
     }
   },
   "name": "@dbx-tools/appkit-mastra",
-  "version": "0.1.25",
+  "version": "0.1.27",
   "dependencies": {
     "@databricks/sdk-experimental": "^0.17",
-    "@dbx-tools/appkit-mastra-shared": "0.1.25",
-    "@dbx-tools/genie": "0.1.25",
-    "@dbx-tools/genie-shared": "0.1.25",
-    "@dbx-tools/shared": "0.1.25",
+    "@dbx-tools/appkit-mastra-shared": "0.1.27",
+    "@dbx-tools/genie": "0.1.27",
+    "@dbx-tools/genie-shared": "0.1.27",
+    "@dbx-tools/shared": "0.1.27",
     "@mastra/ai-sdk": "^1",
     "@mastra/core": "^1",
     "@mastra/express": "^1",
@@ -44,13 +44,13 @@
     "src"
   ],
   "license": "Apache-2.0",
-  "homepage": "https://github.com/reggie-db/dbx-tools-appkit#readme",
+  "homepage": "https://github.com/reggie-db/dbx-tools-js#readme",
   "bugs": {
-    "url": "https://github.com/reggie-db/dbx-tools-appkit/issues"
+    "url": "https://github.com/reggie-db/dbx-tools-js/issues"
   },
   "repository": {
     "type": "git",
-    "url": "git+https://github.com/reggie-db/dbx-tools-appkit.git",
+    "url": "git+https://github.com/reggie-db/dbx-tools-js.git",
     "directory": "packages/appkit-mastra"
   }
 }

package/src/agents.ts

@@ -20,12 +20,12 @@ import { createTool } from "@mastra/core/tools";
 import type { Tool } from "@mastra/core/tools";
 import type { PgVectorConfig, PostgresStoreConfig } from "@mastra/pg";
 
-import { buildRenderDataTool } from "./chart.js";
 import type { MastraPluginConfig } from "./config.js";
 import { buildGenieToolkitProvider, resolveGenieSpaces } from "./genie.js";
 import type { MemoryBuilder } from "./memory.js";
 import { buildModel, FALLBACK_MODEL_IDS } from "./model.js";
 import { stripStaleChartsProcessor } from "./processors/strip-stale-charts.js";
+import { buildRenderDataTool } from "./chart.js";
 
 /**
  * Tool record accepted by every Mastra `Agent.tools` field and by the
@@ -323,6 +323,17 @@ export interface BuiltAgents {
 /** Fallback agent id used when `config.agents` is omitted entirely. */
 export const FALLBACK_AGENT_ID = "default";
 
+/**
+ * Default per-turn step ceiling applied to every registered agent
+ * when {@link MastraPluginConfig.agentMaxSteps} is unset. Sized to
+ * fit a decomposed Genie turn (grounding + several `ask_genie`
+ * calls + `prepare_chart` per dataset + the final-text reply) with
+ * headroom for the model to chain a couple of follow-ups before
+ * answering - well above Mastra's own `agent.generate` default of
+ * 5, which would cut multi-step orchestration off mid-loop.
+ */
+export const DEFAULT_AGENT_MAX_STEPS = 25;
+
 const FALLBACK_AGENT_INSTRUCTIONS = `You are a data analyst. The user will ask questions about
 business metrics and may share personal preferences you should remember across turns.
 
@@ -420,7 +431,7 @@ export async function buildAgents(opts: {
   const style = resolveStyleInstructions(config);
   // Default-on protection against the model copying turn-scoped
   // chartIds from prior assistant tool results into the new
-  // turn's `[[chart:<id>]]` markers. Opt out per-plugin via
+  // turn's `[chart:<id>]` markers. Opt out per-plugin via
   // `config.stripStaleCharts: false`.
   const inputProcessors =
     config.stripStaleCharts === false ? [] : [stripStaleChartsProcessor];
@@ -435,6 +446,9 @@ export async function buildAgents(opts: {
       ...(def.description !== undefined ? { description: def.description } : {}),
       instructions: composeInstructions(def.instructions, style),
       model: resolveModel(config, def.model),
+      defaultOptions: {
+        maxSteps: config.agentMaxSteps ?? DEFAULT_AGENT_MAX_STEPS,
+      },
       tools,
       ...(memory ? { memory } : {}),
       ...(inputProcessors.length > 0 ? { inputProcessors } : {}),