@dbx-tools/appkit-mastra 0.3.44 → 0.4.0

package/lib/index.d.ts

@@ -0,0 +1,71 @@
+export * as agents from "./src/agents.js";
+export * as chart from "./src/chart.js";
+export * as config from "./src/config.js";
+export * as defaults from "./src/defaults.js";
+export * as filesystems from "./src/filesystems.js";
+export * as genie from "./src/genie.js";
+export * as history from "./src/history.js";
+export * as mcp from "./src/mcp.js";
+export * as memory from "./src/memory.js";
+export * as mlflow from "./src/mlflow.js";
+export * as model from "./src/model.js";
+export * as observability from "./src/observability.js";
+export * as pagination from "./src/pagination.js";
+export * as plugin from "./src/plugin.js";
+export * as processors from "./src/processors.js";
+export * as rest from "./src/rest.js";
+export * as server from "./src/server.js";
+export * as serving from "./src/serving.js";
+export * as servingSanitize from "./src/serving-sanitize.js";
+export * as statement from "./src/statement.js";
+export * as storageSchema from "./src/storage-schema.js";
+export * as summarize from "./src/summarize.js";
+export * as threads from "./src/threads.js";
+export * as validation from "./src/validation.js";
+export * as workspaces from "./src/workspaces.js";
+export * as writer from "./src/writer.js";
+export { tool, createAgent, FALLBACK_AGENT_ID, DEFAULT_AGENT_MAX_STEPS, DEFAULT_STYLE_INSTRUCTIONS, buildAgents, approvalGatedToolIds } from "./src/agents.js";
+export type { MastraTools, AppKitToolOptions, ToolkitOptions, MastraPluginToolkitProvider, MastraPlugins, MastraToolsFn, MastraAgentWorkspaceResolver, MastraAgentDefinition, MastraStorageConfigOverride, MastraMemoryConfigOverride, BuiltAgents } from "./src/agents.js";
+export { chartPlanSchema, chartPlannerRequestSchema, prepareChart, fetchChart, buildRenderDataTool } from "./src/chart.js";
+export type { ChartPlannerRequest, PrepareChartOptions, FetchChartOptions } from "./src/chart.js";
+export { MASTRA_USER_KEY, MASTRA_USER_NAME_KEY, MASTRA_USER_EMAIL_KEY, MASTRA_REQUEST_ID_KEY, MASTRA_SCOPES_KEY, TRACE_REQUEST_CONTEXT_KEYS, executionContextUserId, resolveUserKey, MASTRA_CONFIG_SCHEMA } from "./src/config.js";
+export type { User, MastraMemoryConfig, MastraMcpConfig, MastraPluginConfig } from "./src/config.js";
+export { modelCatalogueDefaults, genieSuggestionDefaults, statementDataDefaults, chartFetchDefaults, feedbackWriteDefaults } from "./src/defaults.js";
+export { normalizeDatabricksBasePath, isDbfsPath, isWorkspaceFilesPath, resolveDatabricksAbsolutePath, toDatabricksWorkspacePath, DatabricksWorkspaceFilesystem, emptyFilesystem } from "./src/filesystems.js";
+export type { DatabricksMkdirsMode, DatabricksWorkspaceFilesystemOptions } from "./src/filesystems.js";
+export { DEFAULT_GENIE_ALIAS, GENIE_INSTRUCTIONS, normalizeGenieSpaces, resolveGenieSpaces, buildGenieTools, buildGenieToolkitProvider, collectSpaceSuggestions } from "./src/genie.js";
+export type { GenieSpaceConfig, GenieSpacesConfig } from "./src/genie.js";
+export { loadHistory, clearHistory, historyRoute } from "./src/history.js";
+export type { LoadHistoryOptions, ClearHistoryOptions, HistoryRouteOptions } from "./src/history.js";
+export { buildMcpServer } from "./src/mcp.js";
+export type { ResolvedMcp } from "./src/mcp.js";
+export { createServicePrincipalPool, needsLakebase, createMemoryBuilder, MemoryBuilder } from "./src/memory.js";
+export { mlflowEnabled, resolveFeedbackEnabled, logFeedback } from "./src/mlflow.js";
+export type { LogFeedbackParams } from "./src/mlflow.js";
+export { buildModel } from "./src/model.js";
+export type { BuildModelOverrides } from "./src/model.js";
+export { isOtlpTracingConfigured, buildObservability } from "./src/observability.js";
+export type { BuildObservabilityOptions } from "./src/observability.js";
+export { clampPerPage, parseIntParam } from "./src/pagination.js";
+export type { PerPageBounds } from "./src/pagination.js";
+export { MastraPlugin, mastra } from "./src/plugin.js";
+export { stripStaleChartsProcessor, ResultProcessor } from "./src/processors.js";
+export { databricksFetch, readResponseText, readResponseJson } from "./src/rest.js";
+export type { DatabricksFetchInit } from "./src/rest.js";
+export { stampRequestContextUser, createRequestContext, MastraServer, isMastraRequestAllowed, attachRoutePatchMiddleware } from "./src/server.js";
+export type { MastraApiGateOptions } from "./src/server.js";
+export { MASTRA_MODEL_OVERRIDE_KEY, extractModelOverride, resolveServingConfig } from "./src/serving.js";
+export type { ModelOverrideRequest } from "./src/serving.js";
+export { rewriteServingBody, stripReasoningFromServingMessages, repairAssistantPrefill, rewriteServingResponseBody, flattenChoiceMessageContent } from "./src/serving-sanitize.js";
+export type { ServingChatMessage } from "./src/serving-sanitize.js";
+export { STATEMENT_ROW_CAP, fetchStatementData } from "./src/statement.js";
+export { agentStorageSchemaName } from "./src/storage-schema.js";
+export { TITLE_INSTRUCTIONS, summaryModel, summarizeText, buildSummarizeTool } from "./src/summarize.js";
+export type { SummarizeOptions } from "./src/summarize.js";
+export { listThreads, deleteThread, renameThread, threadsRoute } from "./src/threads.js";
+export type { ListThreadsOptions, DeleteThreadOptions, RenameThreadOptions, ThreadsRouteOptions } from "./src/threads.js";
+export { invalidFields } from "./src/validation.js";
+export type { SchemaIssues } from "./src/validation.js";
+export { createWorkspace } from "./src/workspaces.js";
+export type { CreateWorkspaceOptions } from "./src/workspaces.js";
+export { safeWrite } from "./src/writer.js";

package/lib/index.js

@@ -0,0 +1,56 @@
+// GENERATED by projen watch - DO NOT EDIT.
+// Regenerated from the exporting modules in ./src.
+// Hand edits are overwritten on the next watch; this file is read-only.
+export * as agents from "./src/agents.js";
+export * as chart from "./src/chart.js";
+export * as config from "./src/config.js";
+export * as defaults from "./src/defaults.js";
+export * as filesystems from "./src/filesystems.js";
+export * as genie from "./src/genie.js";
+export * as history from "./src/history.js";
+export * as mcp from "./src/mcp.js";
+export * as memory from "./src/memory.js";
+export * as mlflow from "./src/mlflow.js";
+export * as model from "./src/model.js";
+export * as observability from "./src/observability.js";
+export * as pagination from "./src/pagination.js";
+export * as plugin from "./src/plugin.js";
+export * as processors from "./src/processors.js";
+export * as rest from "./src/rest.js";
+export * as server from "./src/server.js";
+export * as serving from "./src/serving.js";
+export * as servingSanitize from "./src/serving-sanitize.js";
+export * as statement from "./src/statement.js";
+export * as storageSchema from "./src/storage-schema.js";
+export * as summarize from "./src/summarize.js";
+export * as threads from "./src/threads.js";
+export * as validation from "./src/validation.js";
+export * as workspaces from "./src/workspaces.js";
+export * as writer from "./src/writer.js";
+export { tool, createAgent, FALLBACK_AGENT_ID, DEFAULT_AGENT_MAX_STEPS, DEFAULT_STYLE_INSTRUCTIONS, buildAgents, approvalGatedToolIds } from "./src/agents.js";
+export { chartPlanSchema, chartPlannerRequestSchema, prepareChart, fetchChart, buildRenderDataTool } from "./src/chart.js";
+export { MASTRA_USER_KEY, MASTRA_USER_NAME_KEY, MASTRA_USER_EMAIL_KEY, MASTRA_REQUEST_ID_KEY, MASTRA_SCOPES_KEY, TRACE_REQUEST_CONTEXT_KEYS, executionContextUserId, resolveUserKey, MASTRA_CONFIG_SCHEMA } from "./src/config.js";
+export { modelCatalogueDefaults, genieSuggestionDefaults, statementDataDefaults, chartFetchDefaults, feedbackWriteDefaults } from "./src/defaults.js";
+export { normalizeDatabricksBasePath, isDbfsPath, isWorkspaceFilesPath, resolveDatabricksAbsolutePath, toDatabricksWorkspacePath, DatabricksWorkspaceFilesystem, emptyFilesystem } from "./src/filesystems.js";
+export { DEFAULT_GENIE_ALIAS, GENIE_INSTRUCTIONS, normalizeGenieSpaces, resolveGenieSpaces, buildGenieTools, buildGenieToolkitProvider, collectSpaceSuggestions } from "./src/genie.js";
+export { loadHistory, clearHistory, historyRoute } from "./src/history.js";
+export { buildMcpServer } from "./src/mcp.js";
+export { createServicePrincipalPool, needsLakebase, createMemoryBuilder, MemoryBuilder } from "./src/memory.js";
+export { mlflowEnabled, resolveFeedbackEnabled, logFeedback } from "./src/mlflow.js";
+export { buildModel } from "./src/model.js";
+export { isOtlpTracingConfigured, buildObservability } from "./src/observability.js";
+export { clampPerPage, parseIntParam } from "./src/pagination.js";
+export { MastraPlugin, mastra } from "./src/plugin.js";
+export { stripStaleChartsProcessor, ResultProcessor } from "./src/processors.js";
+export { databricksFetch, readResponseText, readResponseJson } from "./src/rest.js";
+export { stampRequestContextUser, createRequestContext, MastraServer, isMastraRequestAllowed, attachRoutePatchMiddleware } from "./src/server.js";
+export { MASTRA_MODEL_OVERRIDE_KEY, extractModelOverride, resolveServingConfig } from "./src/serving.js";
+export { rewriteServingBody, stripReasoningFromServingMessages, repairAssistantPrefill, rewriteServingResponseBody, flattenChoiceMessageContent } from "./src/serving-sanitize.js";
+export { STATEMENT_ROW_CAP, fetchStatementData } from "./src/statement.js";
+export { agentStorageSchemaName } from "./src/storage-schema.js";
+export { TITLE_INSTRUCTIONS, summaryModel, summarizeText, buildSummarizeTool } from "./src/summarize.js";
+export { listThreads, deleteThread, renameThread, threadsRoute } from "./src/threads.js";
+export { invalidFields } from "./src/validation.js";
+export { createWorkspace } from "./src/workspaces.js";
+export { safeWrite } from "./src/writer.js";
+//# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJmaWxlIjoiaW5kZXguanMiLCJzb3VyY2VSb290IjoiIiwic291cmNlcyI6WyIuLi9pbmRleC50cyJdLCJuYW1lcyI6W10sIm1hcHBpbmdzIjoiQUFBQSwyQ0FBMkM7QUFDM0MsbURBQW1EO0FBQ25ELHdFQUF3RTtBQUV4RSxPQUFPLEtBQUssTUFBTSxNQUFNLGNBQWMsQ0FBQztBQUN2QyxPQUFPLEtBQUssS0FBSyxNQUFNLGFBQWEsQ0FBQztBQUNyQyxPQUFPLEtBQUssTUFBTSxNQUFNLGNBQWMsQ0FBQztBQUN2QyxPQUFPLEtBQUssUUFBUSxNQUFNLGdCQUFnQixDQUFDO0FBQzNDLE9BQU8sS0FBSyxXQUFXLE1BQU0sbUJBQW1CLENBQUM7QUFDakQsT0FBTyxLQUFLLEtBQUssTUFBTSxhQUFhLENBQUM7QUFDckMsT0FBTyxLQUFLLE9BQU8sTUFBTSxlQUFlLENBQUM7QUFDekMsT0FBTyxLQUFLLEdBQUcsTUFBTSxXQUFXLENBQUM7QUFDakMsT0FBTyxLQUFLLE1BQU0sTUFBTSxjQUFjLENBQUM7QUFDdkMsT0FBTyxLQUFLLE1BQU0sTUFBTSxjQUFjLENBQUM7QUFDdkMsT0FBTyxLQUFLLEtBQUssTUFBTSxhQUFhLENBQUM7QUFDckMsT0FBTyxLQUFLLGFBQWEsTUFBTSxxQkFBcUIsQ0FBQztBQUNyRCxPQUFPLEtBQUssVUFBVSxNQUFNLGtCQUFrQixDQUFDO0FBQy9DLE9BQU8sS0FBSyxNQUFNLE1BQU0sY0FBYyxDQUFDO0FBQ3ZDLE9BQU8sS0FBSyxVQUFVLE1BQU0sa0JBQWtCLENBQUM7QUFDL0MsT0FBTyxLQUFLLElBQUksTUFBTSxZQUFZLENBQUM7QUFDbkMsT0FBTyxLQUFLLE1BQU0sTUFBTSxjQUFjLENBQUM7QUFDdkMsT0FBTyxLQUFLLE9BQU8sTUFBTSxlQUFlLENBQUM7QUFDekMsT0FBTyxLQUFLLGVBQWUsTUFBTSx3QkFBd0IsQ0FBQztBQUMxRCxPQUFPLEtBQUssU0FBUyxNQUFNLGlCQUFpQixDQUFDO0FBQzdDLE9BQU8sS0FBSyxhQUFhLE1BQU0sc0JBQXNCLENBQUM7QUFDdEQsT0FBTyxLQUFLLFNBQVMsTUFBTSxpQkFBaUIsQ0FBQztBQUM3QyxPQUFPLEtBQUssT0FBTyxNQUFNLGVBQWUsQ0FBQztBQUN6QyxPQUFPLEtBQUssVUFBVSxNQUFNLGtCQUFrQixDQUFDO0FBQy9DLE9BQU8sS0FBSyxVQUFVLE1BQU0sa0JBQWtCLENBQUM7QUFDL0MsT0FBTyxLQUFLLE1BQU0sTUFBTSxjQUFjLENBQUM7QUFDdkMsT0FBTyxFQUFFLElBQUksRUFBRSxXQUFXLEVBQUUsaUJBQWlCLEVBQUUsdUJBQXVCLEVBQUUsMEJBQTBCLEVBQUUsV0FBVyxFQUFFLG9CQUFvQixFQUFFLE1BQU0sY0FBYyxDQUFDO0FBRTVKLE9BQU8sRUFBRSxlQUFlLEVBQUUseUJBQXlCLEVBQUUsWUFBWSxFQUFFLFVBQVUsRUFBRSxtQkFBbUIsRUFBRSxNQUFNLGFBQWEsQ0FBQztBQUV4SCxPQUFPLEVBQUUsZUFBZSxFQUFFLG9CQUFvQixFQUFFLHFCQUFxQixFQUFFLHFCQUFxQixFQUFFLGlCQUFpQixFQUFFLDBCQUEwQixFQUFFLHNCQUFzQixFQUFFLGNBQWMsRUFBRSxvQkFBb0IsRUFBRSxNQUFNLGNBQWMsQ0FBQztBQUVoTyxPQUFPLEVBQUUsc0JBQXNCLEVBQUUsdUJBQXVCLEVBQUUscUJBQXFCLEVBQUUsa0JBQWtCLEVBQUUscUJBQXFCLEVBQUUsTUFBTSxnQkFBZ0IsQ0FBQztBQUNuSixPQUFPLEVBQUUsMkJBQTJCLEVBQUUsVUFBVSxFQUFFLG9CQUFvQixFQUFFLDZCQUE2QixFQUFFLHlCQUF5QixFQUFFLDZCQUE2QixFQUFFLGVBQWUsRUFBRSxNQUFNLG1CQUFtQixDQUFDO0FBRTVNLE9BQU8sRUFBRSxtQkFBbUIsRUFBRSxrQkFBa0IsRUFBRSxvQkFBb0IsRUFBRSxrQkFBa0IsRUFBRSxlQUFlLEVBQUUseUJBQXlCLEVBQUUsdUJBQXVCLEVBQUUsTUFBTSxhQUFhLENBQUM7QUFFckwsT0FBTyxFQUFFLFdBQVcsRUFBRSxZQUFZLEVBQUUsWUFBWSxFQUFFLE1BQU0sZUFBZSxDQUFDO0FBRXhFLE9BQU8sRUFBRSxjQUFjLEVBQUUsTUFBTSxXQUFXLENBQUM7QUFFM0MsT0FBTyxFQUFFLDBCQUEwQixFQUFFLGFBQWEsRUFBRSxtQkFBbUIsRUFBRSxhQUFhLEVBQUUsTUFBTSxjQUFjLENBQUM7QUFDN0csT0FBTyxFQUFFLGFBQWEsRUFBRSxzQkFBc0IsRUFBRSxXQUFXLEVBQUUsTUFBTSxjQUFjLENBQUM7QUFFbEYsT0FBTyxFQUFFLFVBQVUsRUFBRSxNQUFNLGFBQWEsQ0FBQztBQUV6QyxPQUFPLEVBQUUsdUJBQXVCLEVBQUUsa0JBQWtCLEVBQUUsTUFBTSxxQkFBcUIsQ0FBQztBQUVsRixPQUFPLEVBQUUsWUFBWSxFQUFFLGFBQWEsRUFBRSxNQUFNLGtCQUFrQixDQUFDO0FBRS9ELE9BQU8sRUFBRSxZQUFZLEVBQUUsTUFBTSxFQUFFLE1BQU0sY0FBYyxDQUFDO0FBQ3BELE9BQU8sRUFBRSx5QkFBeUIsRUFBRSxlQUFlLEVBQUUsTUFBTSxrQkFBa0IsQ0FBQztBQUM5RSxPQUFPLEVBQUUsZUFBZSxFQUFFLGdCQUFnQixFQUFFLGdCQUFnQixFQUFFLE1BQU0sWUFBWSxDQUFDO0FBRWpGLE9BQU8sRUFBRSx1QkFBdUIsRUFBRSxvQkFBb0IsRUFBRSxZQUFZLEVBQUUsc0JBQXNCLEVBQUUsMEJBQTBCLEVBQUUsTUFBTSxjQUFjLENBQUM7QUFFL0ksT0FBTyxFQUFFLHlCQUF5QixFQUFFLG9CQUFvQixFQUFFLG9CQUFvQixFQUFFLE1BQU0sZUFBZSxDQUFDO0FBRXRHLE9BQU8sRUFBRSxrQkFBa0IsRUFBRSxpQ0FBaUMsRUFBRSxzQkFBc0IsRUFBRSwwQkFBMEIsRUFBRSwyQkFBMkIsRUFBRSxNQUFNLHdCQUF3QixDQUFDO0FBRWhMLE9BQU8sRUFBRSxpQkFBaUIsRUFBRSxrQkFBa0IsRUFBRSxNQUFNLGlCQUFpQixDQUFDO0FBQ3hFLE9BQU8sRUFBRSxzQkFBc0IsRUFBRSxNQUFNLHNCQUFzQixDQUFDO0FBQzlELE9BQU8sRUFBRSxrQkFBa0IsRUFBRSxZQUFZLEVBQUUsYUFBYSxFQUFFLGtCQUFrQixFQUFFLE1BQU0saUJBQWlCLENBQUM7QUFFdEcsT0FBTyxFQUFFLFdBQVcsRUFBRSxZQUFZLEVBQUUsWUFBWSxFQUFFLFlBQVksRUFBRSxNQUFNLGVBQWUsQ0FBQztBQUV0RixPQUFPLEVBQUUsYUFBYSxFQUFFLE1BQU0sa0JBQWtCLENBQUM7QUFFakQsT0FBTyxFQUFFLGVBQWUsRUFBRSxNQUFNLGtCQUFrQixDQUFDO0FBRW5ELE9BQU8sRUFBRSxTQUFTLEVBQUUsTUFBTSxjQUFjLENBQUMiLCJzb3VyY2VzQ29udGVudCI6WyIvLyBHRU5FUkFURUQgYnkgcHJvamVuIHdhdGNoIC0gRE8gTk9UIEVESVQuXG4vLyBSZWdlbmVyYXRlZCBmcm9tIHRoZSBleHBvcnRpbmcgbW9kdWxlcyBpbiAuL3NyYy5cbi8vIEhhbmQgZWRpdHMgYXJlIG92ZXJ3cml0dGVuIG9uIHRoZSBuZXh0IHdhdGNoOyB0aGlzIGZpbGUgaXMgcmVhZC1vbmx5LlxuXG5leHBvcnQgKiBhcyBhZ2VudHMgZnJvbSBcIi4vc3JjL2FnZW50c1wiO1xuZXhwb3J0ICogYXMgY2hhcnQgZnJvbSBcIi4vc3JjL2NoYXJ0XCI7XG5leHBvcnQgKiBhcyBjb25maWcgZnJvbSBcIi4vc3JjL2NvbmZpZ1wiO1xuZXhwb3J0ICogYXMgZGVmYXVsdHMgZnJvbSBcIi4vc3JjL2RlZmF1bHRzXCI7XG5leHBvcnQgKiBhcyBmaWxlc3lzdGVtcyBmcm9tIFwiLi9zcmMvZmlsZXN5c3RlbXNcIjtcbmV4cG9ydCAqIGFzIGdlbmllIGZyb20gXCIuL3NyYy9nZW5pZVwiO1xuZXhwb3J0ICogYXMgaGlzdG9yeSBmcm9tIFwiLi9zcmMvaGlzdG9yeVwiO1xuZXhwb3J0ICogYXMgbWNwIGZyb20gXCIuL3NyYy9tY3BcIjtcbmV4cG9ydCAqIGFzIG1lbW9yeSBmcm9tIFwiLi9zcmMvbWVtb3J5XCI7XG5leHBvcnQgKiBhcyBtbGZsb3cgZnJvbSBcIi4vc3JjL21sZmxvd1wiO1xuZXhwb3J0ICogYXMgbW9kZWwgZnJvbSBcIi4vc3JjL21vZGVsXCI7XG5leHBvcnQgKiBhcyBvYnNlcnZhYmlsaXR5IGZyb20gXCIuL3NyYy9vYnNlcnZhYmlsaXR5XCI7XG5leHBvcnQgKiBhcyBwYWdpbmF0aW9uIGZyb20gXCIuL3NyYy9wYWdpbmF0aW9uXCI7XG5leHBvcnQgKiBhcyBwbHVnaW4gZnJvbSBcIi4vc3JjL3BsdWdpblwiO1xuZXhwb3J0ICogYXMgcHJvY2Vzc29ycyBmcm9tIFwiLi9zcmMvcHJvY2Vzc29yc1wiO1xuZXhwb3J0ICogYXMgcmVzdCBmcm9tIFwiLi9zcmMvcmVzdFwiO1xuZXhwb3J0ICogYXMgc2VydmVyIGZyb20gXCIuL3NyYy9zZXJ2ZXJcIjtcbmV4cG9ydCAqIGFzIHNlcnZpbmcgZnJvbSBcIi4vc3JjL3NlcnZpbmdcIjtcbmV4cG9ydCAqIGFzIHNlcnZpbmdTYW5pdGl6ZSBmcm9tIFwiLi9zcmMvc2VydmluZy1zYW5pdGl6ZVwiO1xuZXhwb3J0ICogYXMgc3RhdGVtZW50IGZyb20gXCIuL3NyYy9zdGF0ZW1lbnRcIjtcbmV4cG9ydCAqIGFzIHN0b3JhZ2VTY2hlbWEgZnJvbSBcIi4vc3JjL3N0b3JhZ2Utc2NoZW1hXCI7XG5leHBvcnQgKiBhcyBzdW1tYXJpemUgZnJvbSBcIi4vc3JjL3N1bW1hcml6ZVwiO1xuZXhwb3J0ICogYXMgdGhyZWFkcyBmcm9tIFwiLi9zcmMvdGhyZWFkc1wiO1xuZXhwb3J0ICogYXMgdmFsaWRhdGlvbiBmcm9tIFwiLi9zcmMvdmFsaWRhdGlvblwiO1xuZXhwb3J0ICogYXMgd29ya3NwYWNlcyBmcm9tIFwiLi9zcmMvd29ya3NwYWNlc1wiO1xuZXhwb3J0ICogYXMgd3JpdGVyIGZyb20gXCIuL3NyYy93cml0ZXJcIjtcbmV4cG9ydCB7IHRvb2wsIGNyZWF0ZUFnZW50LCBGQUxMQkFDS19BR0VOVF9JRCwgREVGQVVMVF9BR0VOVF9NQVhfU1RFUFMsIERFRkFVTFRfU1RZTEVfSU5TVFJVQ1RJT05TLCBidWlsZEFnZW50cywgYXBwcm92YWxHYXRlZFRvb2xJZHMgfSBmcm9tIFwiLi9zcmMvYWdlbnRzXCI7XG5leHBvcnQgdHlwZSB7IE1hc3RyYVRvb2xzLCBBcHBLaXRUb29sT3B0aW9ucywgVG9vbGtpdE9wdGlvbnMsIE1hc3RyYVBsdWdpblRvb2xraXRQcm92aWRlciwgTWFzdHJhUGx1Z2lucywgTWFzdHJhVG9vbHNGbiwgTWFzdHJhQWdlbnRXb3Jrc3BhY2VSZXNvbHZlciwgTWFzdHJhQWdlbnREZWZpbml0aW9uLCBNYXN0cmFTdG9yYWdlQ29uZmlnT3ZlcnJpZGUsIE1hc3RyYU1lbW9yeUNvbmZpZ092ZXJyaWRlLCBCdWlsdEFnZW50cyB9IGZyb20gXCIuL3NyYy9hZ2VudHNcIjtcbmV4cG9ydCB7IGNoYXJ0UGxhblNjaGVtYSwgY2hhcnRQbGFubmVyUmVxdWVzdFNjaGVtYSwgcHJlcGFyZUNoYXJ0LCBmZXRjaENoYXJ0LCBidWlsZFJlbmRlckRhdGFUb29sIH0gZnJvbSBcIi4vc3JjL2NoYXJ0XCI7XG5leHBvcnQgdHlwZSB7IENoYXJ0UGxhbm5lclJlcXVlc3QsIFByZXBhcmVDaGFydE9wdGlvbnMsIEZldGNoQ2hhcnRPcHRpb25zIH0gZnJvbSBcIi4vc3JjL2NoYXJ0XCI7XG5leHBvcnQgeyBNQVNUUkFfVVNFUl9LRVksIE1BU1RSQV9VU0VSX05BTUVfS0VZLCBNQVNUUkFfVVNFUl9FTUFJTF9LRVksIE1BU1RSQV9SRVFVRVNUX0lEX0tFWSwgTUFTVFJBX1NDT1BFU19LRVksIFRSQUNFX1JFUVVFU1RfQ09OVEVYVF9LRVlTLCBleGVjdXRpb25Db250ZXh0VXNlcklkLCByZXNvbHZlVXNlcktleSwgTUFTVFJBX0NPTkZJR19TQ0hFTUEgfSBmcm9tIFwiLi9zcmMvY29uZmlnXCI7XG5leHBvcnQgdHlwZSB7IFVzZXIsIE1hc3RyYU1lbW9yeUNvbmZpZywgTWFzdHJhTWNwQ29uZmlnLCBNYXN0cmFQbHVnaW5Db25maWcgfSBmcm9tIFwiLi9zcmMvY29uZmlnXCI7XG5leHBvcnQgeyBtb2RlbENhdGFsb2d1ZURlZmF1bHRzLCBnZW5pZVN1Z2dlc3Rpb25EZWZhdWx0cywgc3RhdGVtZW50RGF0YURlZmF1bHRzLCBjaGFydEZldGNoRGVmYXVsdHMsIGZlZWRiYWNrV3JpdGVEZWZhdWx0cyB9IGZyb20gXCIuL3NyYy9kZWZhdWx0c1wiO1xuZXhwb3J0IHsgbm9ybWFsaXplRGF0YWJyaWNrc0Jhc2VQYXRoLCBpc0RiZnNQYXRoLCBpc1dvcmtzcGFjZUZpbGVzUGF0aCwgcmVzb2x2ZURhdGFicmlja3NBYnNvbHV0ZVBhdGgsIHRvRGF0YWJyaWNrc1dvcmtzcGFjZVBhdGgsIERhdGFicmlja3NXb3Jrc3BhY2VGaWxlc3lzdGVtLCBlbXB0eUZpbGVzeXN0ZW0gfSBmcm9tIFwiLi9zcmMvZmlsZXN5c3RlbXNcIjtcbmV4cG9ydCB0eXBlIHsgRGF0YWJyaWNrc01rZGlyc01vZGUsIERhdGFicmlja3NXb3Jrc3BhY2VGaWxlc3lzdGVtT3B0aW9ucyB9IGZyb20gXCIuL3NyYy9maWxlc3lzdGVtc1wiO1xuZXhwb3J0IHsgREVGQVVMVF9HRU5JRV9BTElBUywgR0VOSUVfSU5TVFJVQ1RJT05TLCBub3JtYWxpemVHZW5pZVNwYWNlcywgcmVzb2x2ZUdlbmllU3BhY2VzLCBidWlsZEdlbmllVG9vbHMsIGJ1aWxkR2VuaWVUb29sa2l0UHJvdmlkZXIsIGNvbGxlY3RTcGFjZVN1Z2dlc3Rpb25zIH0gZnJvbSBcIi4vc3JjL2dlbmllXCI7XG5leHBvcnQgdHlwZSB7IEdlbmllU3BhY2VDb25maWcsIEdlbmllU3BhY2VzQ29uZmlnIH0gZnJvbSBcIi4vc3JjL2dlbmllXCI7XG5leHBvcnQgeyBsb2FkSGlzdG9yeSwgY2xlYXJIaXN0b3J5LCBoaXN0b3J5Um91dGUgfSBmcm9tIFwiLi9zcmMvaGlzdG9yeVwiO1xuZXhwb3J0IHR5cGUgeyBMb2FkSGlzdG9yeU9wdGlvbnMsIENsZWFySGlzdG9yeU9wdGlvbnMsIEhpc3RvcnlSb3V0ZU9wdGlvbnMgfSBmcm9tIFwiLi9zcmMvaGlzdG9yeVwiO1xuZXhwb3J0IHsgYnVpbGRNY3BTZXJ2ZXIgfSBmcm9tIFwiLi9zcmMvbWNwXCI7XG5leHBvcnQgdHlwZSB7IFJlc29sdmVkTWNwIH0gZnJvbSBcIi4vc3JjL21jcFwiO1xuZXhwb3J0IHsgY3JlYXRlU2VydmljZVByaW5jaXBhbFBvb2wsIG5lZWRzTGFrZWJhc2UsIGNyZWF0ZU1lbW9yeUJ1aWxkZXIsIE1lbW9yeUJ1aWxkZXIgfSBmcm9tIFwiLi9zcmMvbWVtb3J5XCI7XG5leHBvcnQgeyBtbGZsb3dFbmFibGVkLCByZXNvbHZlRmVlZGJhY2tFbmFibGVkLCBsb2dGZWVkYmFjayB9IGZyb20gXCIuL3NyYy9tbGZsb3dcIjtcbmV4cG9ydCB0eXBlIHsgTG9nRmVlZGJhY2tQYXJhbXMgfSBmcm9tIFwiLi9zcmMvbWxmbG93XCI7XG5leHBvcnQgeyBidWlsZE1vZGVsIH0gZnJvbSBcIi4vc3JjL21vZGVsXCI7XG5leHBvcnQgdHlwZSB7IEJ1aWxkTW9kZWxPdmVycmlkZXMgfSBmcm9tIFwiLi9zcmMvbW9kZWxcIjtcbmV4cG9ydCB7IGlzT3RscFRyYWNpbmdDb25maWd1cmVkLCBidWlsZE9ic2VydmFiaWxpdHkgfSBmcm9tIFwiLi9zcmMvb2JzZXJ2YWJpbGl0eVwiO1xuZXhwb3J0IHR5cGUgeyBCdWlsZE9ic2VydmFiaWxpdHlPcHRpb25zIH0gZnJvbSBcIi4vc3JjL29ic2VydmFiaWxpdHlcIjtcbmV4cG9ydCB7IGNsYW1wUGVyUGFnZSwgcGFyc2VJbnRQYXJhbSB9IGZyb20gXCIuL3NyYy9wYWdpbmF0aW9uXCI7XG5leHBvcnQgdHlwZSB7IFBlclBhZ2VCb3VuZHMgfSBmcm9tIFwiLi9zcmMvcGFnaW5hdGlvblwiO1xuZXhwb3J0IHsgTWFzdHJhUGx1Z2luLCBtYXN0cmEgfSBmcm9tIFwiLi9zcmMvcGx1Z2luXCI7XG5leHBvcnQgeyBzdHJpcFN0YWxlQ2hhcnRzUHJvY2Vzc29yLCBSZXN1bHRQcm9jZXNzb3IgfSBmcm9tIFwiLi9zcmMvcHJvY2Vzc29yc1wiO1xuZXhwb3J0IHsgZGF0YWJyaWNrc0ZldGNoLCByZWFkUmVzcG9uc2VUZXh0LCByZWFkUmVzcG9uc2VKc29uIH0gZnJvbSBcIi4vc3JjL3Jlc3RcIjtcbmV4cG9ydCB0eXBlIHsgRGF0YWJyaWNrc0ZldGNoSW5pdCB9IGZyb20gXCIuL3NyYy9yZXN0XCI7XG5leHBvcnQgeyBzdGFtcFJlcXVlc3RDb250ZXh0VXNlciwgY3JlYXRlUmVxdWVzdENvbnRleHQsIE1hc3RyYVNlcnZlciwgaXNNYXN0cmFSZXF1ZXN0QWxsb3dlZCwgYXR0YWNoUm91dGVQYXRjaE1pZGRsZXdhcmUgfSBmcm9tIFwiLi9zcmMvc2VydmVyXCI7XG5leHBvcnQgdHlwZSB7IE1hc3RyYUFwaUdhdGVPcHRpb25zIH0gZnJvbSBcIi4vc3JjL3NlcnZlclwiO1xuZXhwb3J0IHsgTUFTVFJBX01PREVMX09WRVJSSURFX0tFWSwgZXh0cmFjdE1vZGVsT3ZlcnJpZGUsIHJlc29sdmVTZXJ2aW5nQ29uZmlnIH0gZnJvbSBcIi4vc3JjL3NlcnZpbmdcIjtcbmV4cG9ydCB0eXBlIHsgTW9kZWxPdmVycmlkZVJlcXVlc3QgfSBmcm9tIFwiLi9zcmMvc2VydmluZ1wiO1xuZXhwb3J0IHsgcmV3cml0ZVNlcnZpbmdCb2R5LCBzdHJpcFJlYXNvbmluZ0Zyb21TZXJ2aW5nTWVzc2FnZXMsIHJlcGFpckFzc2lzdGFudFByZWZpbGwsIHJld3JpdGVTZXJ2aW5nUmVzcG9uc2VCb2R5LCBmbGF0dGVuQ2hvaWNlTWVzc2FnZUNvbnRlbnQgfSBmcm9tIFwiLi9zcmMvc2VydmluZy1zYW5pdGl6ZVwiO1xuZXhwb3J0IHR5cGUgeyBTZXJ2aW5nQ2hhdE1lc3NhZ2UgfSBmcm9tIFwiLi9zcmMvc2VydmluZy1zYW5pdGl6ZVwiO1xuZXhwb3J0IHsgU1RBVEVNRU5UX1JPV19DQVAsIGZldGNoU3RhdGVtZW50RGF0YSB9IGZyb20gXCIuL3NyYy9zdGF0ZW1lbnRcIjtcbmV4cG9ydCB7IGFnZW50U3RvcmFnZVNjaGVtYU5hbWUgfSBmcm9tIFwiLi9zcmMvc3RvcmFnZS1zY2hlbWFcIjtcbmV4cG9ydCB7IFRJVExFX0lOU1RSVUNUSU9OUywgc3VtbWFyeU1vZGVsLCBzdW1tYXJpemVUZXh0LCBidWlsZFN1bW1hcml6ZVRvb2wgfSBmcm9tIFwiLi9zcmMvc3VtbWFyaXplXCI7XG5leHBvcnQgdHlwZSB7IFN1bW1hcml6ZU9wdGlvbnMgfSBmcm9tIFwiLi9zcmMvc3VtbWFyaXplXCI7XG5leHBvcnQgeyBsaXN0VGhyZWFkcywgZGVsZXRlVGhyZWFkLCByZW5hbWVUaHJlYWQsIHRocmVhZHNSb3V0ZSB9IGZyb20gXCIuL3NyYy90aHJlYWRzXCI7XG5leHBvcnQgdHlwZSB7IExpc3RUaHJlYWRzT3B0aW9ucywgRGVsZXRlVGhyZWFkT3B0aW9ucywgUmVuYW1lVGhyZWFkT3B0aW9ucywgVGhyZWFkc1JvdXRlT3B0aW9ucyB9IGZyb20gXCIuL3NyYy90aHJlYWRzXCI7XG5leHBvcnQgeyBpbnZhbGlkRmllbGRzIH0gZnJvbSBcIi4vc3JjL3ZhbGlkYXRpb25cIjtcbmV4cG9ydCB0eXBlIHsgU2NoZW1hSXNzdWVzIH0gZnJvbSBcIi4vc3JjL3ZhbGlkYXRpb25cIjtcbmV4cG9ydCB7IGNyZWF0ZVdvcmtzcGFjZSB9IGZyb20gXCIuL3NyYy93b3Jrc3BhY2VzXCI7XG5leHBvcnQgdHlwZSB7IENyZWF0ZVdvcmtzcGFjZU9wdGlvbnMgfSBmcm9tIFwiLi9zcmMvd29ya3NwYWNlc1wiO1xuZXhwb3J0IHsgc2FmZVdyaXRlIH0gZnJvbSBcIi4vc3JjL3dyaXRlclwiO1xuIl19

package/lib/src/agents.d.ts

@@ -0,0 +1,347 @@
+/**
+ * Agent registration for the Mastra AppKit plugin.
+ *
+ * Mirrors the shape of the AppKit `agents` plugin (`config.agents` map
+ * of {@link MastraAgentDefinition}, dual-form `tools` accepting a plain
+ * record or a `(plugins) => tools` callback). Resolves each definition
+ * into a Mastra `Agent` instance during plugin setup; user-supplied
+ * tool callbacks are invoked exactly once with a typed
+ * {@link MastraPlugins} map built from registered sibling plugins.
+ *
+ * When no agents are registered the plugin falls back to a single
+ * built-in analyst so the bare `mastra()` call still mounts a working
+ * streamable agent for demos.
+ *
+ * @module
+ */
+import { plugin } from "@dbx-tools/appkit";
+import { log } from "@dbx-tools/shared-core";
+import type { AgentConfig, ToolsInput } from "@mastra/core/agent";
+import { Agent } from "@mastra/core/agent";
+import type { Tool } from "@mastra/core/tools";
+import type { Workspace } from "@mastra/core/workspace";
+import type { PgVectorConfig, PostgresStoreConfig } from "@mastra/pg";
+import type { MastraPluginConfig } from "./config.js";
+import type { MemoryBuilder } from "./memory.js";
+/**
+ * Tool record accepted by every Mastra `Agent.tools` field and by the
+ * `tools(plugins)` callback on {@link MastraAgentDefinition}.
+ *
+ * Alias of Mastra's `ToolsInput`, so it already accepts:
+ *
+ * - Mastra tools built with {@link createTool} (or `new Tool(...)`)
+ * - Mastra tools built with the AppKit-shaped {@link tool} wrapper
+ *   below
+ * - Vercel AI SDK tools (`tool({ ... })` from `ai`)
+ * - Provider-defined tools (e.g. `openai.tools.webSearch(...)`)
+ *
+ * Existing tool libraries drop in as-is - nothing in this package
+ * forces a rebuild.
+ */
+export type MastraTools = ToolsInput;
+/** Re-export of Mastra's native `createTool` for full-feature access. */
+export { createTool } from "@mastra/core/tools";
+/**
+ * AppKit-shaped tool factory. Lets users mix-and-match tools across
+ * AppKit's `agents` plugin and `mastra` with a single import:
+ *
+ * ```ts
+ * import { tool } from "@dbx-tools/appkit-mastra";
+ * import { z } from "zod";
+ *
+ * get_weather: tool({
+ *   description: "Weather",
+ *   schema: z.object({ city: z.string() }),
+ *   execute: async ({ city }) => `Sunny in ${city}`,
+ * }),
+ * ```
+ *
+ * Maps onto Mastra's `createTool`:
+ *
+ * - `description` -> `description` (required)
+ * - `schema` -> `inputSchema` (optional)
+ * - `execute(input)` -> `execute(input, ctx)` - Mastra already calls
+ *   the first arg with the parsed inputs, so the body shape is
+ *   identical. The Mastra `context` arg is forwarded as the second
+ *   parameter when the caller declares it.
+ * - `id`: optional. Defaults to a stable identifier derived from
+ *   `description` (slugified, with a short hash suffix for
+ *   uniqueness). Pass an explicit `id` when you need a stable string
+ *   for tracing or MCP exposure.
+ *
+ * Reach for {@link createTool} when you need Mastra-only fields
+ * (`outputSchema`, `suspendSchema`, `requireApproval`, `mcp`, etc.).
+ */
+export declare function tool(opts: AppKitToolOptions): Tool;
+/**
+ * Input shape for the AppKit-style {@link tool} factory. A trimmed
+ * subset of Mastra's `createTool` options that mirrors the
+ * `@databricks/appkit/beta` `tool({ description, schema, execute })`
+ * signature.
+ *
+ * Generics are intentionally absent - inference flows through the
+ * caller's `schema` (typically a Zod object), and the `execute` body
+ * destructures naturally from that. Reach for {@link createTool} when
+ * you need the fully-typed input/output schemas wired explicitly.
+ */
+export interface AppKitToolOptions {
+    /** Optional stable identifier; auto-derived from `description` when omitted. */
+    id?: string;
+    /** Human-readable description shown to the model. Required. */
+    description: string;
+    /**
+     * Optional input schema (any Standard Schema instance, e.g. Zod).
+     * Maps to Mastra's `inputSchema`; passed through to the model
+     * verbatim.
+     */
+    schema?: unknown;
+    /**
+     * Execute body. First arg is the parsed input (typed off `schema`
+     * when supplied), second arg is the full Mastra execution context
+     * (request context, abort signal, mastra instance) if you need it.
+     */
+    execute: (input: any, context?: unknown) => unknown;
+}
+/**
+ * Identity helper that brands a definition as a Mastra agent. Mirrors
+ * AppKit's `createAgent(def)` so the registration shape matches:
+ *
+ * ```ts
+ * const support = createAgent({
+ *   instructions: "...",
+ *   model: "databricks-claude-sonnet-4-6",
+ *   tools(plugins) { return { ... }; },
+ * });
+ * ```
+ *
+ * Returns the definition unchanged - the wrapper exists only to anchor
+ * type inference and to match the AppKit API surface.
+ */
+export declare function createAgent<T extends MastraAgentDefinition>(def: T): T;
+/**
+ * Filter / rename options accepted by every plugin's `.toolkit()`
+ * method. Mirrors AppKit's `ToolkitOptions` verbatim so options pass
+ * through unchanged - the underlying AppKit plugin does the filtering
+ * and we just adapt the resulting entries into Mastra tools.
+ */
+export interface ToolkitOptions {
+    /**
+     * Key prefix prepended to every tool name. AppKit's default is
+     * `${pluginName}.` when omitted; pass an explicit `""` to drop it.
+     */
+    prefix?: string;
+    /** Allowlist of local tool names. */
+    only?: string[];
+    /** Denylist of local tool names. */
+    except?: string[];
+    /** Remap specific local names to different keys. */
+    rename?: Record<string, string>;
+}
+/**
+ * Toolkit provider shape every entry in the {@link MastraPlugins} map
+ * exposes. Identical to AppKit's `PluginToolkitProvider` - any AppKit
+ * plugin that implements the standard `ToolProvider` interface
+ * (`getAgentTools` + `executeAgentTool` + `toolkit`) is reachable
+ * through this surface automatically.
+ */
+export interface MastraPluginToolkitProvider {
+    /**
+     * Returns a Mastra-shaped tools record adapted from the plugin's
+     * agent tools. Each tool dispatches back through the plugin's
+     * `executeAgentTool` so OBO auth and telemetry spans stay intact.
+     */
+    toolkit(opts?: ToolkitOptions): MastraTools;
+}
+/**
+ * Plugin map handed to the function form of
+ * {@link MastraAgentDefinition.tools}. Mirrors AppKit's `Plugins`
+ * type exactly: a string-keyed record where every value exposes
+ * `.toolkit(opts)`.
+ *
+ * Implemented as a runtime Proxy that auto-discovers any registered
+ * AppKit plugin implementing the standard `ToolProvider` interface
+ * (`analytics`, `files`, `lakebase`, `genie`, plus any third-party
+ * plugin that does the same). Unknown names resolve to `undefined`
+ * at runtime, so guard with `?.` and `?? {}` when spreading from a
+ * plugin that may not be registered in every environment.
+ *
+ * @example
+ * ```ts
+ * createAgent({
+ *   instructions: "...",
+ *   tools(plugins) {
+ *     return {
+ *       ...plugins.analytics.toolkit(),
+ *       ...plugins.files.toolkit({ only: ["uploads.read"] }),
+ *       get_weather: tool({
+ *         description: "Weather",
+ *         schema: z.object({ city: z.string() }),
+ *         execute: async ({ city }) => `Sunny in ${city}`,
+ *       }),
+ *     };
+ *   },
+ * });
+ * ```
+ */
+export type MastraPlugins = Record<string, MastraPluginToolkitProvider>;
+/** Function form of {@link MastraAgentDefinition.tools}. */
+export type MastraToolsFn = (plugins: MastraPlugins) => MastraTools | Promise<MastraTools>;
+/** Function form of {@link MastraAgentDefinition.workspace}. */
+export type MastraAgentWorkspaceResolver = () => Workspace | undefined;
+/**
+ * A code-defined Mastra agent. Mirrors the shape AppKit's `agents`
+ * plugin uses for `AgentDefinition`. The registry key under
+ * `config.agents` is the `agentId` the client streams against; `name`
+ * is purely informational (defaults to the key).
+ */
+export interface MastraAgentDefinition {
+    /** Display name used as `Agent.name`. Defaults to the registry key. */
+    name?: string;
+    /** Optional long-form description; surfaced as `Agent.description`. */
+    description?: string;
+    /** System prompt body. */
+    instructions: string;
+    /**
+     * Per-agent model override.
+     *
+     * - `undefined` (default): falls back to the workspace
+     *   `/serving-endpoints` resolver that {@link buildModel} configures
+     *   from the per-request `WorkspaceClient`.
+     * - `string`: shorthand for "use the default resolver but swap the
+     *   `modelId`" (e.g. `"databricks-meta-llama-3-3-70b-instruct"`).
+     * - Any other Mastra `DynamicArgument<MastraModelConfig>`: passed
+     *   straight through to `Agent.model`. Use this when you need full
+     *   control over auth or providerId.
+     */
+    model?: AgentConfig["model"] | string;
+    /**
+     * Per-agent tool record. Either a plain map or a callback that
+     * receives the typed {@link MastraPlugins} sibling-plugin index and
+     * returns a map. The callback runs exactly once at agent setup; the
+     * result is cached for the agent's lifetime.
+     */
+    tools?: MastraTools | MastraToolsFn;
+    /**
+     * Per-agent semantic recall (PgVector) override. Cascades from
+     * `config.memory`; the agent value wins when set.
+     *
+     * - `undefined` (default): inherit `config.memory`. When that's
+     *   enabled, the agent **shares the plugin-level singleton `PgVector`
+     *   instance** (cross-agent semantic recall across the same index).
+     * - `false`: disable semantic recall for this agent only.
+     * - `true`: enable using the shared singleton (same as default when
+     *   plugin memory is enabled; useful to opt in when plugin disabled).
+     * - {@link MastraMemoryConfig} object: dedicated `PgVector` for this
+     *   agent (private recall index). Bypasses the shared singleton.
+     */
+    memory?: boolean | MastraMemoryConfigOverride;
+    /**
+     * Per-agent thread/message storage (`PostgresStore`) override.
+     * Cascades from `config.storage`; the agent value wins when set.
+     *
+     * - `undefined` (default): inherit `config.storage`. When that's
+     *   enabled, the agent gets its **own per-agent `PostgresStore`**
+     *   keyed by {@link agentStorageSchemaName} so threads and
+     *   messages stay isolated between agents in the same database.
+     * - `false`: disable storage for this agent only (purely in-memory).
+     * - `true`: enable with the per-agent default schema.
+     * - {@link MastraStorageConfigOverride} object: dedicated
+     *   `PostgresStore` config (custom schema, connection, etc.).
+     */
+    storage?: boolean | MastraStorageConfigOverride;
+    /**
+     * Mastra {@link Workspace} for this agent (filesystem, sandbox, or other
+     * providers). Assistant skills from Databricks workspace paths are wired
+     * via {@link createWorkspace}.
+     */
+    workspace?: Workspace | MastraAgentWorkspaceResolver;
+}
+/**
+ * Distributive `Omit` so unions in `PostgresStoreConfig` /
+ * `PgVectorConfig` keep their discriminants after the override types
+ * strip `id`. The built-in `Omit` collapses unions to one shape with
+ * common fields only, which loses the connection-style discriminants.
+ */
+type DistributiveOmit<T, K extends keyof never> = T extends unknown ? Omit<T, K> : never;
+/**
+ * `PostgresStoreConfig` minus `id` - per-agent overrides accept any
+ * Mastra-supported storage shape. `id` is filled in automatically
+ * from the agent registry key so traces stay stable.
+ */
+export type MastraStorageConfigOverride = DistributiveOmit<PostgresStoreConfig, "id"> & {
+    id?: string;
+};
+/**
+ * `PgVectorConfig` minus `id` - per-agent overrides accept any
+ * Mastra-supported vector shape. `id` is filled in automatically
+ * from the agent registry key.
+ */
+export type MastraMemoryConfigOverride = DistributiveOmit<PgVectorConfig, "id"> & {
+    id?: string;
+};
+/** Output of {@link buildAgents}: resolved agents plus the default 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. Consumed by the plugin's
+     * `GET /default-model` route handler so the client's model picker can label
+     * its default option with the real endpoint's humanized name.
+     */
+    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
+     * can re-expose them when {@link MastraMcpConfig.tools} is enabled.
+     */
+    ambientTools: MastraTools;
+}
+/** Fallback agent id used when `config.agents` is omitted entirely. */
+export declare 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 declare const DEFAULT_AGENT_MAX_STEPS = 25;
+/**
+ * Style guardrails appended to every agent's `instructions` to curb
+ * common LLM-isms (em dashes, emojis, sycophantic openers, excessive
+ * hedging, throwaway closers). Appended rather than prepended so the
+ * agent's role/context comes first; the model's recency bias then
+ * helps the style rules dominate the response surface.
+ *
+ * Override globally via {@link MastraPluginConfig.styleInstructions}
+ * (pass `false` to disable entirely, or a string to replace).
+ */
+export declare const DEFAULT_STYLE_INSTRUCTIONS: string;
+/**
+ * Resolve every entry in `config.agents` into a Mastra `Agent`
+ * instance. When `config.agents` is omitted the plugin registers a
+ * single built-in `default` analyst so the bare `mastra()` call still
+ * yields a working agent.
+ *
+ * Per-agent tool callbacks are invoked once with a typed
+ * {@link MastraPlugins} index built from registered sibling plugins
+ * (currently `genie`; extend `MastraPlugins` to surface more).
+ *
+ * @throws when `config.defaultAgent` is set to an id that isn't in the
+ *   resolved registry; this is a wiring bug, not a runtime condition.
+ */
+export declare function buildAgents(opts: {
+    config: MastraPluginConfig;
+    context: plugin.PluginContextLike | undefined;
+    memoryBuilder?: MemoryBuilder;
+    log: log.Logger;
+}): Promise<BuiltAgents>;
+/**
+ * Tool ids on `tools` that are approval-gated (`requireApproval: true`).
+ * Keys are used as a fallback when a tool omits an explicit `id`.
+ */
+export declare function approvalGatedToolIds(tools: MastraTools): string[];