@dbx-tools/appkit-mastra 0.4.1 → 0.5.0

package/src/agents.ts

@@ -26,14 +26,14 @@ import { createTool } from "@mastra/core/tools";
 import type { Workspace } from "@mastra/core/workspace";
 import type { PgVectorConfig, PostgresStoreConfig } from "@mastra/pg";
 
-import { buildRenderDataTool } from "./chart";
-import type { MastraPluginConfig } from "./config";
-import { buildGenieToolkitProvider, resolveGenieSpaces } from "./genie";
-import type { MemoryBuilder } from "./memory";
-import { buildModel } from "./model";
-import { ResultProcessor, stripStaleChartsProcessor } from "./processors";
-import { buildSummarizeTool } from "./summarize";
-import { createWorkspace } from "./workspaces";
+import { buildRenderDataTool } from "./chart.ts";
+import type { MastraPluginConfig } from "./config.ts";
+import { buildGenieToolkitProvider, resolveGenieSpaces } from "./genie.ts";
+import type { MemoryBuilder } from "./memory.ts";
+import { buildModel } from "./model.ts";
+import { ResultProcessor, stripStaleChartsProcessor } from "./processors.ts";
+import { buildSummarizeTool } from "./summarize.ts";
+import { createWorkspace } from "./workspaces.ts";
 
 /**
  * Tool record accepted by every Mastra `Agent.tools` field and by the

package/src/chart.ts

@@ -36,8 +36,8 @@ import type { RequestContext } from "@mastra/core/request-context";
 import { createTool } from "@mastra/core/tools";
 import { z } from "zod";
 
-import { resolveUserKey, type MastraPluginConfig } from "./config";
-import { buildModel } from "./model";
+import { resolveUserKey, type MastraPluginConfig } from "./config.ts";
+import { buildModel } from "./model.ts";
 
 const logger = log.logger("mastra/chart");
 

package/src/config.ts

@@ -22,8 +22,8 @@ import type { RequestContext } from "@mastra/core/request-context";
 import { MASTRA_RESOURCE_ID_KEY, MASTRA_THREAD_ID_KEY } from "@mastra/core/request-context";
 import type { PgVectorConfig, PostgresStoreConfig } from "@mastra/pg";
 
-import type { MastraAgentDefinition, MastraTools } from "./agents";
-import type { GenieSpacesConfig } from "./genie";
+import type { MastraAgentDefinition, MastraTools } from "./agents.ts";
+import type { GenieSpacesConfig } from "./genie.ts";
 
 /**
  * `RequestContext` key under which {@link MastraServer} stores the

package/src/genie.ts

@@ -54,12 +54,12 @@ import { MASTRA_THREAD_ID_KEY } from "@mastra/core/request-context";
 import { createTool } from "@mastra/core/tools";
 import { z } from "zod";
 
-import type { MastraTools } from "./agents";
-import { chartPlannerRequestSchema, prepareChart } from "./chart";
-import { MASTRA_USER_KEY, resolveUserKey } from "./config";
-import type { MastraPluginConfig, User } from "./config";
-import { fetchStatementData } from "./statement";
-import { safeWrite } from "./writer";
+import type { MastraTools } from "./agents.ts";
+import { chartPlannerRequestSchema, prepareChart } from "./chart.ts";
+import { MASTRA_USER_KEY, resolveUserKey } from "./config.ts";
+import type { MastraPluginConfig, User } from "./config.ts";
+import { fetchStatementData } from "./statement.ts";
+import { safeWrite } from "./writer.ts";
 
 const logger = log.logger("mastra/genie");
 

package/src/history.ts

@@ -32,7 +32,7 @@ import { MASTRA_RESOURCE_ID_KEY, MASTRA_THREAD_ID_KEY } from "@mastra/core/reque
 import type { ContextWithMastra } from "@mastra/core/server";
 import { registerApiRoute } from "@mastra/core/server";
 
-import { clampPerPage, parseIntParam } from "./pagination";
+import { clampPerPage, parseIntParam } from "./pagination.ts";
 
 const logger = log.logger("mastra/history");
 

package/src/mcp.ts

@@ -15,8 +15,8 @@
 import type { Agent } from "@mastra/core/agent";
 import { MCPServer } from "@mastra/mcp";
 
-import type { MastraTools } from "./agents";
-import type { MastraMcpConfig, MastraPluginConfig } from "./config";
+import type { MastraTools } from "./agents.ts";
+import type { MastraMcpConfig, MastraPluginConfig } from "./config.ts";
 
 /** MCP server version advertised when the caller doesn't pin one. */
 const DEFAULT_MCP_VERSION = "1.0.0";

package/src/memory.ts

@@ -36,10 +36,10 @@ import { Memory } from "@mastra/memory";
 import { PgVector, PostgresStore } from "@mastra/pg";
 import { Pool, type PoolConfig } from "pg";
 
-import type { MastraAgentDefinition, MastraMemoryConfigOverride } from "./agents";
-import type { MastraPluginConfig } from "./config";
-import { agentStorageSchemaName } from "./storage-schema";
-import { summaryModel, TITLE_INSTRUCTIONS } from "./summarize";
+import type { MastraAgentDefinition, MastraMemoryConfigOverride } from "./agents.ts";
+import type { MastraPluginConfig } from "./config.ts";
+import { agentStorageSchemaName } from "./storage-schema.ts";
+import { summaryModel, TITLE_INSTRUCTIONS } from "./summarize.ts";
 
 const logger = log.logger("mastra/memory");
 

package/src/mlflow.ts

@@ -22,7 +22,7 @@
 import { appkit } from "@dbx-tools/appkit";
 import { async, error, log } from "@dbx-tools/shared-core";
 import { feedback } from "@dbx-tools/shared-mastra";
-import { databricksFetch, readResponseJson, readResponseText } from "./rest";
+import { databricksFetch, readResponseJson, readResponseText } from "./rest.ts";
 
 const logger = log.logger("mastra/mlflow");
 

package/src/model.ts

@@ -30,9 +30,9 @@ import { model } from "@dbx-tools/shared-model";
 import type { MastraModelConfig } from "@mastra/core/llm";
 import type { RequestContext } from "@mastra/core/request-context";
 
-import { MASTRA_USER_KEY, type MastraPluginConfig, type User } from "./config";
-import { MASTRA_MODEL_OVERRIDE_KEY, resolveServingConfig } from "./serving";
-import { rewriteServingBody, rewriteServingResponseBody } from "./serving-sanitize";
+import { MASTRA_USER_KEY, type MastraPluginConfig, type User } from "./config.ts";
+import { rewriteServingBody, rewriteServingResponseBody } from "./serving-sanitize.ts";
+import { MASTRA_MODEL_OVERRIDE_KEY, resolveServingConfig } from "./serving.ts";
 
 type ModelClass = model.ModelClass;
 const { parseModelClass } = classes;

package/src/observability.ts

@@ -38,7 +38,7 @@ import { log } from "@dbx-tools/shared-core";
 import { Observability } from "@mastra/observability";
 import { OtelBridge } from "@mastra/otel-bridge";
 
-import { TRACE_REQUEST_CONTEXT_KEYS } from "./config";
+import { TRACE_REQUEST_CONTEXT_KEYS } from "./config.ts";
 
 const logger = log.logger("mastra/observability");
 

package/src/plugin.ts

@@ -74,32 +74,32 @@ import type { RequestContext } from "@mastra/core/request-context";
 import express from "express";
 import type { Pool } from "pg";
 
-import { buildAgents, FALLBACK_AGENT_ID, type BuiltAgents } from "./agents";
-import { fetchChart } from "./chart";
-import { MASTRA_CONFIG_SCHEMA, resolveUserKey, type MastraPluginConfig } from "./config";
+import { buildAgents, FALLBACK_AGENT_ID, type BuiltAgents } from "./agents.ts";
+import { fetchChart } from "./chart.ts";
+import { MASTRA_CONFIG_SCHEMA, resolveUserKey, type MastraPluginConfig } from "./config.ts";
 import {
   chartFetchDefaults,
   feedbackWriteDefaults,
   genieSuggestionDefaults,
   modelCatalogueDefaults,
   statementDataDefaults,
-} from "./defaults";
-import { collectSpaceSuggestions, resolveGenieSpaces } from "./genie";
-import { historyRoute } from "./history";
-import { buildMcpServer, type ResolvedMcp } from "./mcp";
-import { createMemoryBuilder, createServicePrincipalPool, needsLakebase } from "./memory";
-import { logFeedback, resolveFeedbackEnabled } from "./mlflow";
-import { buildObservability } from "./observability";
+} from "./defaults.ts";
+import { collectSpaceSuggestions, resolveGenieSpaces } from "./genie.ts";
+import { historyRoute } from "./history.ts";
+import { buildMcpServer, type ResolvedMcp } from "./mcp.ts";
+import { createMemoryBuilder, createServicePrincipalPool, needsLakebase } from "./memory.ts";
+import { logFeedback, resolveFeedbackEnabled } from "./mlflow.ts";
+import { buildObservability } from "./observability.ts";
 import {
   attachRoutePatchMiddleware,
   createRequestContext,
   isMastraRequestAllowed,
   MastraServer,
-} from "./server";
-import { resolveServingConfig } from "./serving";
-import { fetchStatementData, STATEMENT_ROW_CAP } from "./statement";
-import { threadsRoute } from "./threads";
-import { invalidFields } from "./validation";
+} from "./server.ts";
+import { resolveServingConfig } from "./serving.ts";
+import { fetchStatementData, STATEMENT_ROW_CAP } from "./statement.ts";
+import { threadsRoute } from "./threads.ts";
+import { invalidFields } from "./validation.ts";
 
 const GENIE_MANIFEST = plugin.data(genie).plugin.manifest;
 const LAKEBASE_MANIFEST = plugin.data(lakebase).plugin.manifest;

package/src/server.ts

@@ -28,10 +28,14 @@ import {
   MASTRA_USER_NAME_KEY,
   type MastraPluginConfig,
   type User,
-} from "./config";
-import { resolveFeedbackEnabled } from "./mlflow";
+} from "./config.ts";
+import { resolveFeedbackEnabled } from "./mlflow.ts";
 
-import { extractModelOverride, MASTRA_MODEL_OVERRIDE_KEY, resolveServingConfig } from "./serving";
+import {
+  extractModelOverride,
+  MASTRA_MODEL_OVERRIDE_KEY,
+  resolveServingConfig,
+} from "./serving.ts";
 /**
  * OpenTelemetry's sentinel for "no valid trace" - 32 zero hex chars.
  * `trace.getActiveSpan()` returns a non-recording span with this id

package/src/serving.ts

@@ -15,7 +15,7 @@ import { serving as nodeServing } from "@dbx-tools/model";
 import { string } from "@dbx-tools/shared-core";
 import { override } from "@dbx-tools/shared-mastra";
 
-import type { MastraPluginConfig } from "./config";
+import type { MastraPluginConfig } from "./config.ts";
 
 /**
  * `RequestContext` key under which {@link MastraServer} stores the

package/src/summarize.ts

@@ -27,8 +27,8 @@ import type { RequestContext } from "@mastra/core/request-context";
 import { createTool } from "@mastra/core/tools";
 import { z } from "zod";
 
-import type { MastraPluginConfig } from "./config";
-import { buildModel } from "./model";
+import type { MastraPluginConfig } from "./config.ts";
+import { buildModel } from "./model.ts";
 
 /** Fast / small chat tier used for both titling and summaries. */
 const SUMMARY_MODEL_CLASS = model.ModelClass.ChatFast;

package/src/threads.ts

@@ -36,8 +36,8 @@ import { MASTRA_RESOURCE_ID_KEY, MASTRA_THREAD_ID_KEY } from "@mastra/core/reque
 import type { ContextWithMastra } from "@mastra/core/server";
 import { registerApiRoute } from "@mastra/core/server";
 
-import { clampPerPage, parseIntParam } from "./pagination";
-import { invalidFields } from "./validation";
+import { clampPerPage, parseIntParam } from "./pagination.ts";
+import { invalidFields } from "./validation.ts";
 
 const logger = log.logger("mastra/threads");
 

package/src/workspaces.ts

@@ -21,8 +21,8 @@ import {
   type WorkspaceFilesystem,
 } from "@mastra/core/workspace";
 
-import { MASTRA_SCOPES_KEY, MASTRA_USER_EMAIL_KEY, MASTRA_USER_KEY, type User } from "./config";
-import { DatabricksWorkspaceFilesystem, emptyFilesystem } from "./filesystems";
+import { MASTRA_SCOPES_KEY, MASTRA_USER_EMAIL_KEY, MASTRA_USER_KEY, type User } from "./config.ts";
+import { DatabricksWorkspaceFilesystem, emptyFilesystem } from "./filesystems.ts";
 
 /* ------------------------------ constants ------------------------------ */