@n8n/instance-ai 1.2.1 → 1.3.0

package/dist/tools/workspace.tool.js

@@ -0,0 +1,278 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.createWorkspaceTool = createWorkspaceTool;
+const tools_1 = require("@mastra/core/tools");
+const api_types_1 = require("@n8n/api-types");
+const nanoid_1 = require("nanoid");
+const zod_1 = require("zod");
+const sanitize_mcp_schemas_1 = require("../agent/sanitize-mcp-schemas");
+const workflowIdField = zod_1.z.string().describe('Workflow ID');
+const workflowNameField = zod_1.z
+    .string()
+    .optional()
+    .describe('Workflow name (for confirmation message)');
+const projectIdField = zod_1.z.string().describe('Project ID');
+const folderIdField = zod_1.z.string().describe('Folder ID');
+const folderNameField = zod_1.z.string().optional().describe('Folder name (for confirmation message)');
+const listProjectsAction = zod_1.z.object({
+    action: zod_1.z.literal('list-projects').describe('List all projects accessible to the current user'),
+});
+const listTagsAction = zod_1.z.object({
+    action: zod_1.z.literal('list-tags').describe('List all available tags'),
+});
+const tagWorkflowAction = zod_1.z.object({
+    action: zod_1.z
+        .literal('tag-workflow')
+        .describe('Assign tags to a workflow, creating missing tags automatically'),
+    workflowId: workflowIdField,
+    workflowName: workflowNameField,
+    tags: zod_1.z.array(zod_1.z.string()).min(1).describe('Tag names to assign to the workflow'),
+});
+const cleanupTestExecutionsAction = zod_1.z.object({
+    action: zod_1.z
+        .literal('cleanup-test-executions')
+        .describe('Delete manual/test execution records for a workflow'),
+    workflowId: workflowIdField,
+    workflowName: workflowNameField,
+    olderThanHours: zod_1.z
+        .number()
+        .optional()
+        .describe('Only delete executions older than this many hours. Defaults to 1.'),
+});
+const listFoldersAction = zod_1.z.object({
+    action: zod_1.z.literal('list-folders').describe('List folders in a project'),
+    projectId: projectIdField,
+});
+const createFolderAction = zod_1.z.object({
+    action: zod_1.z.literal('create-folder').describe('Create a new folder in a project'),
+    name: zod_1.z.string().describe('Name for the new folder'),
+    projectId: projectIdField,
+    parentFolderId: zod_1.z
+        .string()
+        .optional()
+        .describe('ID of the parent folder for nesting. Omit for root-level folder.'),
+});
+const deleteFolderAction = zod_1.z.object({
+    action: zod_1.z.literal('delete-folder').describe('Delete a folder from a project'),
+    folderId: folderIdField,
+    folderName: folderNameField,
+    projectId: projectIdField,
+    transferToFolderId: zod_1.z
+        .string()
+        .optional()
+        .describe('ID of a folder to move contents into before deletion. If omitted, contents are flattened to project root and archived.'),
+    transferToFolderName: zod_1.z
+        .string()
+        .optional()
+        .describe('Name of the transfer folder (for confirmation message)'),
+});
+const moveWorkflowToFolderAction = zod_1.z.object({
+    action: zod_1.z.literal('move-workflow-to-folder').describe('Move a workflow into a folder'),
+    workflowId: workflowIdField,
+    workflowName: workflowNameField,
+    folderId: folderIdField,
+    folderName: folderNameField,
+});
+const suspendSchema = zod_1.z.object({
+    requestId: zod_1.z.string(),
+    message: zod_1.z.string(),
+    severity: api_types_1.instanceAiConfirmationSeveritySchema,
+});
+const resumeSchema = zod_1.z.object({
+    approved: zod_1.z.boolean(),
+});
+function buildInputSchema(context) {
+    const baseActions = [
+        listProjectsAction,
+        listTagsAction,
+        tagWorkflowAction,
+        cleanupTestExecutionsAction,
+    ];
+    if (context.workspaceService?.listFolders) {
+        return (0, sanitize_mcp_schemas_1.sanitizeInputSchema)(zod_1.z.discriminatedUnion('action', [
+            ...baseActions,
+            listFoldersAction,
+            createFolderAction,
+            deleteFolderAction,
+            moveWorkflowToFolderAction,
+        ]));
+    }
+    return (0, sanitize_mcp_schemas_1.sanitizeInputSchema)(zod_1.z.discriminatedUnion('action', [...baseActions]));
+}
+function extractCtx(ctx) {
+    return {
+        resumeData: ctx?.agent?.resumeData,
+        suspend: ctx?.agent?.suspend,
+    };
+}
+async function handleListProjects(context) {
+    const projects = await context.workspaceService.listProjects();
+    return { projects };
+}
+async function handleListTags(context) {
+    const tags = await context.workspaceService.listTags();
+    return { tags };
+}
+async function handleTagWorkflow(context, input, ctx) {
+    const { resumeData, suspend } = extractCtx(ctx);
+    if (context.permissions?.tagWorkflow === 'blocked') {
+        return { appliedTags: [], denied: true, reason: 'Action blocked by admin' };
+    }
+    const needsApproval = context.permissions?.tagWorkflow !== 'always_allow';
+    if (needsApproval && (resumeData === undefined || resumeData === null)) {
+        await suspend?.({
+            requestId: (0, nanoid_1.nanoid)(),
+            message: `Tag workflow "${input.workflowName ?? input.workflowId}" (ID: ${input.workflowId}) with [${input.tags.join(', ')}]?`,
+            severity: 'info',
+        });
+        return { appliedTags: [] };
+    }
+    if (resumeData !== undefined && resumeData !== null && !resumeData.approved) {
+        return { appliedTags: [], denied: true, reason: 'User denied the action' };
+    }
+    const appliedTags = await context.workspaceService.tagWorkflow(input.workflowId, input.tags);
+    return { appliedTags };
+}
+async function handleCleanupTestExecutions(context, input, ctx) {
+    const { resumeData, suspend } = extractCtx(ctx);
+    if (context.permissions?.cleanupTestExecutions === 'blocked') {
+        return { deletedCount: 0, denied: true, reason: 'Action blocked by admin' };
+    }
+    const needsApproval = context.permissions?.cleanupTestExecutions !== 'always_allow';
+    if (needsApproval && (resumeData === undefined || resumeData === null)) {
+        const hours = input.olderThanHours ?? 1;
+        await suspend?.({
+            requestId: (0, nanoid_1.nanoid)(),
+            message: `Delete test executions for workflow "${input.workflowName ?? input.workflowId}" older than ${hours} hour(s)?`,
+            severity: 'warning',
+        });
+        return { deletedCount: 0 };
+    }
+    if (resumeData !== undefined && resumeData !== null && !resumeData.approved) {
+        return { deletedCount: 0, denied: true, reason: 'User denied the action' };
+    }
+    const result = await context.workspaceService.cleanupTestExecutions(input.workflowId, {
+        olderThanHours: input.olderThanHours,
+    });
+    return { deletedCount: result.deletedCount };
+}
+async function handleListFolders(context, input) {
+    const folders = await context.workspaceService.listFolders(input.projectId);
+    return { folders };
+}
+async function handleCreateFolder(context, input, ctx) {
+    const { resumeData, suspend } = extractCtx(ctx);
+    if (context.permissions?.createFolder === 'blocked') {
+        return {
+            id: '',
+            name: '',
+            parentFolderId: null,
+            denied: true,
+            reason: 'Action blocked by admin',
+        };
+    }
+    const needsApproval = context.permissions?.createFolder !== 'always_allow';
+    if (needsApproval && (resumeData === undefined || resumeData === null)) {
+        await suspend?.({
+            requestId: (0, nanoid_1.nanoid)(),
+            message: `Create folder "${input.name}" in project "${input.projectId}"?`,
+            severity: 'info',
+        });
+        return { id: '', name: '', parentFolderId: null };
+    }
+    if (resumeData !== undefined && resumeData !== null && !resumeData.approved) {
+        return {
+            id: '',
+            name: '',
+            parentFolderId: null,
+            denied: true,
+            reason: 'User denied the action',
+        };
+    }
+    const folder = await context.workspaceService.createFolder(input.name, input.projectId, input.parentFolderId);
+    return { ...folder };
+}
+async function handleDeleteFolder(context, input, ctx) {
+    const { resumeData, suspend } = extractCtx(ctx);
+    if (context.permissions?.deleteFolder === 'blocked') {
+        return { success: false, denied: true, reason: 'Action blocked by admin' };
+    }
+    const needsApproval = context.permissions?.deleteFolder !== 'always_allow';
+    if (needsApproval && (resumeData === undefined || resumeData === null)) {
+        const transferNote = input.transferToFolderId
+            ? ` Contents will be moved to folder "${input.transferToFolderName ?? input.transferToFolderId}".`
+            : ' Contents will be flattened to project root and archived.';
+        await suspend?.({
+            requestId: (0, nanoid_1.nanoid)(),
+            message: `Delete folder "${input.folderName ?? input.folderId}"?${transferNote}`,
+            severity: 'destructive',
+        });
+        return { success: false };
+    }
+    if (resumeData !== undefined && resumeData !== null && !resumeData.approved) {
+        return { success: false, denied: true, reason: 'User denied the action' };
+    }
+    await context.workspaceService.deleteFolder(input.folderId, input.projectId, input.transferToFolderId);
+    return { success: true };
+}
+async function handleMoveWorkflowToFolder(context, input, ctx) {
+    const { resumeData, suspend } = extractCtx(ctx);
+    if (context.permissions?.moveWorkflowToFolder === 'blocked') {
+        return { success: false, denied: true, reason: 'Action blocked by admin' };
+    }
+    const needsApproval = context.permissions?.moveWorkflowToFolder !== 'always_allow';
+    if (needsApproval && (resumeData === undefined || resumeData === null)) {
+        await suspend?.({
+            requestId: (0, nanoid_1.nanoid)(),
+            message: `Move workflow "${input.workflowName ?? input.workflowId}" to folder "${input.folderName ?? input.folderId}"?`,
+            severity: 'info',
+        });
+        return { success: false };
+    }
+    if (resumeData !== undefined && resumeData !== null && !resumeData.approved) {
+        return { success: false, denied: true, reason: 'User denied the action' };
+    }
+    await context.workspaceService.moveWorkflowToFolder(input.workflowId, input.folderId);
+    return { success: true };
+}
+function createWorkspaceTool(context) {
+    if (!context.workspaceService) {
+        return (0, tools_1.createTool)({
+            id: 'workspace',
+            description: 'Workspace management is not available in this environment.',
+            inputSchema: zod_1.z.object({ action: zod_1.z.string() }),
+            execute: async () => {
+                return { error: 'Workspace service is not available in this environment.' };
+            },
+        });
+    }
+    const inputSchema = buildInputSchema(context);
+    return (0, tools_1.createTool)({
+        id: 'workspace',
+        description: 'Manage workspace resources — projects, tags, folders, and execution cleanup.',
+        inputSchema,
+        suspendSchema,
+        resumeSchema,
+        execute: async (input, ctx) => {
+            switch (input.action) {
+                case 'list-projects':
+                    return await handleListProjects(context);
+                case 'list-tags':
+                    return await handleListTags(context);
+                case 'tag-workflow':
+                    return await handleTagWorkflow(context, input, ctx);
+                case 'cleanup-test-executions':
+                    return await handleCleanupTestExecutions(context, input, ctx);
+                case 'list-folders':
+                    return await handleListFolders(context, input);
+                case 'create-folder':
+                    return await handleCreateFolder(context, input, ctx);
+                case 'delete-folder':
+                    return await handleDeleteFolder(context, input, ctx);
+                case 'move-workflow-to-folder':
+                    return await handleMoveWorkflowToFolder(context, input, ctx);
+            }
+        },
+    });
+}
+//# sourceMappingURL=workspace.tool.js.map

package/dist/tools/workspace.tool.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"workspace.tool.js","sourceRoot":"","sources":["../../src/tools/workspace.tool.ts"],"names":[],"mappings":";;AAyWA,kDA0CC;AAhZD,8CAAgD;AAChD,8CAAsE;AACtE,mCAAgC;AAChC,6BAAwB;AAExB,wEAAoE;AAKpE,MAAM,eAAe,GAAG,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,aAAa,CAAC,CAAC;AAC3D,MAAM,iBAAiB,GAAG,OAAC;KACzB,MAAM,EAAE;KACR,QAAQ,EAAE;KACV,QAAQ,CAAC,0CAA0C,CAAC,CAAC;AACvD,MAAM,cAAc,GAAG,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC;AACzD,MAAM,aAAa,GAAG,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,WAAW,CAAC,CAAC;AACvD,MAAM,eAAe,GAAG,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,wCAAwC,CAAC,CAAC;AAIjG,MAAM,kBAAkB,GAAG,OAAC,CAAC,MAAM,CAAC;IACnC,MAAM,EAAE,OAAC,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC,QAAQ,CAAC,kDAAkD,CAAC;CAC/F,CAAC,CAAC;AAEH,MAAM,cAAc,GAAG,OAAC,CAAC,MAAM,CAAC;IAC/B,MAAM,EAAE,OAAC,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC,QAAQ,CAAC,yBAAyB,CAAC;CAClE,CAAC,CAAC;AAEH,MAAM,iBAAiB,GAAG,OAAC,CAAC,MAAM,CAAC;IAClC,MAAM,EAAE,OAAC;SACP,OAAO,CAAC,cAAc,CAAC;SACvB,QAAQ,CAAC,gEAAgE,CAAC;IAC5E,UAAU,EAAE,eAAe;IAC3B,YAAY,EAAE,iBAAiB;IAC/B,IAAI,EAAE,OAAC,CAAC,KAAK,CAAC,OAAC,CAAC,MAAM,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,qCAAqC,CAAC;CAChF,CAAC,CAAC;AAEH,MAAM,2BAA2B,GAAG,OAAC,CAAC,MAAM,CAAC;IAC5C,MAAM,EAAE,OAAC;SACP,OAAO,CAAC,yBAAyB,CAAC;SAClC,QAAQ,CAAC,qDAAqD,CAAC;IACjE,UAAU,EAAE,eAAe;IAC3B,YAAY,EAAE,iBAAiB;IAC/B,cAAc,EAAE,OAAC;SACf,MAAM,EAAE;SACR,QAAQ,EAAE;SACV,QAAQ,CAAC,mEAAmE,CAAC;CAC/E,CAAC,CAAC;AAEH,MAAM,iBAAiB,GAAG,OAAC,CAAC,MAAM,CAAC;IAClC,MAAM,EAAE,OAAC,CAAC,OAAO,CAAC,cAAc,CAAC,CAAC,QAAQ,CAAC,2BAA2B,CAAC;IACvE,SAAS,EAAE,cAAc;CACzB,CAAC,CAAC;AAEH,MAAM,kBAAkB,GAAG,OAAC,CAAC,MAAM,CAAC;IACnC,MAAM,EAAE,OAAC,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC,QAAQ,CAAC,kCAAkC,CAAC;IAC/E,IAAI,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,yBAAyB,CAAC;IACpD,SAAS,EAAE,cAAc;IACzB,cAAc,EAAE,OAAC;SACf,MAAM,EAAE;SACR,QAAQ,EAAE;SACV,QAAQ,CAAC,kEAAkE,CAAC;CAC9E,CAAC,CAAC;AAEH,MAAM,kBAAkB,GAAG,OAAC,CAAC,MAAM,CAAC;IACnC,MAAM,EAAE,OAAC,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC,QAAQ,CAAC,gCAAgC,CAAC;IAC7E,QAAQ,EAAE,aAAa;IACvB,UAAU,EAAE,eAAe;IAC3B,SAAS,EAAE,cAAc;IACzB,kBAAkB,EAAE,OAAC;SACnB,MAAM,EAAE;SACR,QAAQ,EAAE;SACV,QAAQ,CACR,wHAAwH,CACxH;IACF,oBAAoB,EAAE,OAAC;SACrB,MAAM,EAAE;SACR,QAAQ,EAAE;SACV,QAAQ,CAAC,wDAAwD,CAAC;CACpE,CAAC,CAAC;AAEH,MAAM,0BAA0B,GAAG,OAAC,CAAC,MAAM,CAAC;IAC3C,MAAM,EAAE,OAAC,CAAC,OAAO,CAAC,yBAAyB,CAAC,CAAC,QAAQ,CAAC,+BAA+B,CAAC;IACtF,UAAU,EAAE,eAAe;IAC3B,YAAY,EAAE,iBAAiB;IAC/B,QAAQ,EAAE,aAAa;IACvB,UAAU,EAAE,eAAe;CAC3B,CAAC,CAAC;AAIH,MAAM,aAAa,GAAG,OAAC,CAAC,MAAM,CAAC;IAC9B,SAAS,EAAE,OAAC,CAAC,MAAM,EAAE;IACrB,OAAO,EAAE,OAAC,CAAC,MAAM,EAAE;IACnB,QAAQ,EAAE,gDAAoC;CAC9C,CAAC,CAAC;AAEH,MAAM,YAAY,GAAG,OAAC,CAAC,MAAM,CAAC;IAC7B,QAAQ,EAAE,OAAC,CAAC,OAAO,EAAE;CACrB,CAAC,CAAC;AAIH,SAAS,gBAAgB,CAAC,OAA0B;IACnD,MAAM,WAAW,GAAG;QACnB,kBAAkB;QAClB,cAAc;QACd,iBAAiB;QACjB,2BAA2B;KAClB,CAAC;IAEX,IAAI,OAAO,CAAC,gBAAgB,EAAE,WAAW,EAAE,CAAC;QAC3C,OAAO,IAAA,0CAAmB,EACzB,OAAC,CAAC,kBAAkB,CAAC,QAAQ,EAAE;YAC9B,GAAG,WAAW;YACd,iBAAiB;YACjB,kBAAkB;YAClB,kBAAkB;YAClB,0BAA0B;SAC1B,CAAC,CACF,CAAC;IACH,CAAC;IAED,OAAO,IAAA,0CAAmB,EAAC,OAAC,CAAC,kBAAkB,CAAC,QAAQ,EAAE,CAAC,GAAG,WAAW,CAAC,CAAC,CAAC,CAAC;AAC9E,CAAC;AASD,SAAS,UAAU,CAAC,GAA4D;IAC/E,OAAO;QACN,UAAU,EAAE,GAAG,EAAE,KAAK,EAAE,UAAwB;QAChD,OAAO,EAAE,GAAG,EAAE,KAAK,EAAE,OAAoB;KACzC,CAAC;AACH,CAAC;AAID,KAAK,UAAU,kBAAkB,CAAC,OAA0B;IAC3D,MAAM,QAAQ,GAAG,MAAM,OAAO,CAAC,gBAAiB,CAAC,YAAY,EAAE,CAAC;IAChE,OAAO,EAAE,QAAQ,EAAE,CAAC;AACrB,CAAC;AAED,KAAK,UAAU,cAAc,CAAC,OAA0B;IACvD,MAAM,IAAI,GAAG,MAAM,OAAO,CAAC,gBAAiB,CAAC,QAAQ,EAAE,CAAC;IACxD,OAAO,EAAE,IAAI,EAAE,CAAC;AACjB,CAAC;AAED,KAAK,UAAU,iBAAiB,CAC/B,OAA0B,EAC1B,KAAiD,EACjD,GAA4D;IAE5D,MAAM,EAAE,UAAU,EAAE,OAAO,EAAE,GAAG,UAAU,CAAC,GAAG,CAAC,CAAC;IAEhD,IAAI,OAAO,CAAC,WAAW,EAAE,WAAW,KAAK,SAAS,EAAE,CAAC;QACpD,OAAO,EAAE,WAAW,EAAE,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,yBAAyB,EAAE,CAAC;IAC7E,CAAC;IAED,MAAM,aAAa,GAAG,OAAO,CAAC,WAAW,EAAE,WAAW,KAAK,cAAc,CAAC;IAG1E,IAAI,aAAa,IAAI,CAAC,UAAU,KAAK,SAAS,IAAI,UAAU,KAAK,IAAI,CAAC,EAAE,CAAC;QACxE,MAAM,OAAO,EAAE,CAAC;YACf,SAAS,EAAE,IAAA,eAAM,GAAE;YACnB,OAAO,EAAE,iBAAiB,KAAK,CAAC,YAAY,IAAI,KAAK,CAAC,UAAU,UAAU,KAAK,CAAC,UAAU,WAAW,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI;YAC9H,QAAQ,EAAE,MAAe;SACzB,CAAC,CAAC;QAEH,OAAO,EAAE,WAAW,EAAE,EAAE,EAAE,CAAC;IAC5B,CAAC;IAGD,IAAI,UAAU,KAAK,SAAS,IAAI,UAAU,KAAK,IAAI,IAAI,CAAC,UAAU,CAAC,QAAQ,EAAE,CAAC;QAC7E,OAAO,EAAE,WAAW,EAAE,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,wBAAwB,EAAE,CAAC;IAC5E,CAAC;IAGD,MAAM,WAAW,GAAG,MAAM,OAAO,CAAC,gBAAiB,CAAC,WAAW,CAAC,KAAK,CAAC,UAAU,EAAE,KAAK,CAAC,IAAI,CAAC,CAAC;IAC9F,OAAO,EAAE,WAAW,EAAE,CAAC;AACxB,CAAC;AAED,KAAK,UAAU,2BAA2B,CACzC,OAA0B,EAC1B,KAA4D,EAC5D,GAA4D;IAE5D,MAAM,EAAE,UAAU,EAAE,OAAO,EAAE,GAAG,UAAU,CAAC,GAAG,CAAC,CAAC;IAEhD,IAAI,OAAO,CAAC,WAAW,EAAE,qBAAqB,KAAK,SAAS,EAAE,CAAC;QAC9D,OAAO,EAAE,YAAY,EAAE,CAAC,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,yBAAyB,EAAE,CAAC;IAC7E,CAAC;IAED,MAAM,aAAa,GAAG,OAAO,CAAC,WAAW,EAAE,qBAAqB,KAAK,cAAc,CAAC;IAGpF,IAAI,aAAa,IAAI,CAAC,UAAU,KAAK,SAAS,IAAI,UAAU,KAAK,IAAI,CAAC,EAAE,CAAC;QACxE,MAAM,KAAK,GAAG,KAAK,CAAC,cAAc,IAAI,CAAC,CAAC;QACxC,MAAM,OAAO,EAAE,CAAC;YACf,SAAS,EAAE,IAAA,eAAM,GAAE;YACnB,OAAO,EAAE,wCAAwC,KAAK,CAAC,YAAY,IAAI,KAAK,CAAC,UAAU,gBAAgB,KAAK,WAAW;YACvH,QAAQ,EAAE,SAAkB;SAC5B,CAAC,CAAC;QACH,OAAO,EAAE,YAAY,EAAE,CAAC,EAAE,CAAC;IAC5B,CAAC;IAGD,IAAI,UAAU,KAAK,SAAS,IAAI,UAAU,KAAK,IAAI,IAAI,CAAC,UAAU,CAAC,QAAQ,EAAE,CAAC;QAC7E,OAAO,EAAE,YAAY,EAAE,CAAC,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,wBAAwB,EAAE,CAAC;IAC5E,CAAC;IAGD,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,gBAAiB,CAAC,qBAAqB,CAAC,KAAK,CAAC,UAAU,EAAE;QACtF,cAAc,EAAE,KAAK,CAAC,cAAc;KACpC,CAAC,CAAC;IACH,OAAO,EAAE,YAAY,EAAE,MAAM,CAAC,YAAY,EAAE,CAAC;AAC9C,CAAC;AAED,KAAK,UAAU,iBAAiB,CAC/B,OAA0B,EAC1B,KAAiD;IAEjD,MAAM,OAAO,GAAG,MAAM,OAAO,CAAC,gBAAiB,CAAC,WAAY,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC;IAC9E,OAAO,EAAE,OAAO,EAAE,CAAC;AACpB,CAAC;AAED,KAAK,UAAU,kBAAkB,CAChC,OAA0B,EAC1B,KAAkD,EAClD,GAA4D;IAE5D,MAAM,EAAE,UAAU,EAAE,OAAO,EAAE,GAAG,UAAU,CAAC,GAAG,CAAC,CAAC;IAEhD,IAAI,OAAO,CAAC,WAAW,EAAE,YAAY,KAAK,SAAS,EAAE,CAAC;QACrD,OAAO;YACN,EAAE,EAAE,EAAE;YACN,IAAI,EAAE,EAAE;YACR,cAAc,EAAE,IAAI;YACpB,MAAM,EAAE,IAAI;YACZ,MAAM,EAAE,yBAAyB;SACjC,CAAC;IACH,CAAC;IAED,MAAM,aAAa,GAAG,OAAO,CAAC,WAAW,EAAE,YAAY,KAAK,cAAc,CAAC;IAG3E,IAAI,aAAa,IAAI,CAAC,UAAU,KAAK,SAAS,IAAI,UAAU,KAAK,IAAI,CAAC,EAAE,CAAC;QACxE,MAAM,OAAO,EAAE,CAAC;YACf,SAAS,EAAE,IAAA,eAAM,GAAE;YACnB,OAAO,EAAE,kBAAkB,KAAK,CAAC,IAAI,iBAAiB,KAAK,CAAC,SAAS,IAAI;YACzE,QAAQ,EAAE,MAAe;SACzB,CAAC,CAAC;QAEH,OAAO,EAAE,EAAE,EAAE,EAAE,EAAE,IAAI,EAAE,EAAE,EAAE,cAAc,EAAE,IAAI,EAAE,CAAC;IACnD,CAAC;IAGD,IAAI,UAAU,KAAK,SAAS,IAAI,UAAU,KAAK,IAAI,IAAI,CAAC,UAAU,CAAC,QAAQ,EAAE,CAAC;QAC7E,OAAO;YACN,EAAE,EAAE,EAAE;YACN,IAAI,EAAE,EAAE;YACR,cAAc,EAAE,IAAI;YACpB,MAAM,EAAE,IAAI;YACZ,MAAM,EAAE,wBAAwB;SAChC,CAAC;IACH,CAAC;IAGD,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,gBAAiB,CAAC,YAAa,CAC3D,KAAK,CAAC,IAAI,EACV,KAAK,CAAC,SAAS,EACf,KAAK,CAAC,cAAc,CACpB,CAAC;IACF,OAAO,EAAE,GAAG,MAAM,EAAE,CAAC;AACtB,CAAC;AAED,KAAK,UAAU,kBAAkB,CAChC,OAA0B,EAC1B,KAAkD,EAClD,GAA4D;IAE5D,MAAM,EAAE,UAAU,EAAE,OAAO,EAAE,GAAG,UAAU,CAAC,GAAG,CAAC,CAAC;IAEhD,IAAI,OAAO,CAAC,WAAW,EAAE,YAAY,KAAK,SAAS,EAAE,CAAC;QACrD,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,yBAAyB,EAAE,CAAC;IAC5E,CAAC;IAED,MAAM,aAAa,GAAG,OAAO,CAAC,WAAW,EAAE,YAAY,KAAK,cAAc,CAAC;IAG3E,IAAI,aAAa,IAAI,CAAC,UAAU,KAAK,SAAS,IAAI,UAAU,KAAK,IAAI,CAAC,EAAE,CAAC;QACxE,MAAM,YAAY,GAAG,KAAK,CAAC,kBAAkB;YAC5C,CAAC,CAAC,sCAAsC,KAAK,CAAC,oBAAoB,IAAI,KAAK,CAAC,kBAAkB,IAAI;YAClG,CAAC,CAAC,2DAA2D,CAAC;QAC/D,MAAM,OAAO,EAAE,CAAC;YACf,SAAS,EAAE,IAAA,eAAM,GAAE;YACnB,OAAO,EAAE,kBAAkB,KAAK,CAAC,UAAU,IAAI,KAAK,CAAC,QAAQ,KAAK,YAAY,EAAE;YAChF,QAAQ,EAAE,aAAsB;SAChC,CAAC,CAAC;QAEH,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC;IAC3B,CAAC;IAGD,IAAI,UAAU,KAAK,SAAS,IAAI,UAAU,KAAK,IAAI,IAAI,CAAC,UAAU,CAAC,QAAQ,EAAE,CAAC;QAC7E,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,wBAAwB,EAAE,CAAC;IAC3E,CAAC;IAGD,MAAM,OAAO,CAAC,gBAAiB,CAAC,YAAa,CAC5C,KAAK,CAAC,QAAQ,EACd,KAAK,CAAC,SAAS,EACf,KAAK,CAAC,kBAAkB,CACxB,CAAC;IACF,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC;AAC1B,CAAC;AAED,KAAK,UAAU,0BAA0B,CACxC,OAA0B,EAC1B,KAA4D,EAC5D,GAA4D;IAE5D,MAAM,EAAE,UAAU,EAAE,OAAO,EAAE,GAAG,UAAU,CAAC,GAAG,CAAC,CAAC;IAEhD,IAAI,OAAO,CAAC,WAAW,EAAE,oBAAoB,KAAK,SAAS,EAAE,CAAC;QAC7D,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,yBAAyB,EAAE,CAAC;IAC5E,CAAC;IAED,MAAM,aAAa,GAAG,OAAO,CAAC,WAAW,EAAE,oBAAoB,KAAK,cAAc,CAAC;IAGnF,IAAI,aAAa,IAAI,CAAC,UAAU,KAAK,SAAS,IAAI,UAAU,KAAK,IAAI,CAAC,EAAE,CAAC;QACxE,MAAM,OAAO,EAAE,CAAC;YACf,SAAS,EAAE,IAAA,eAAM,GAAE;YACnB,OAAO,EAAE,kBAAkB,KAAK,CAAC,YAAY,IAAI,KAAK,CAAC,UAAU,gBAAgB,KAAK,CAAC,UAAU,IAAI,KAAK,CAAC,QAAQ,IAAI;YACvH,QAAQ,EAAE,MAAe;SACzB,CAAC,CAAC;QAEH,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC;IAC3B,CAAC;IAGD,IAAI,UAAU,KAAK,SAAS,IAAI,UAAU,KAAK,IAAI,IAAI,CAAC,UAAU,CAAC,QAAQ,EAAE,CAAC;QAC7E,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,wBAAwB,EAAE,CAAC;IAC3E,CAAC;IAGD,MAAM,OAAO,CAAC,gBAAiB,CAAC,oBAAqB,CAAC,KAAK,CAAC,UAAU,EAAE,KAAK,CAAC,QAAQ,CAAC,CAAC;IACxF,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC;AAC1B,CAAC;AAID,SAAgB,mBAAmB,CAAC,OAA0B;IAC7D,IAAI,CAAC,OAAO,CAAC,gBAAgB,EAAE,CAAC;QAC/B,OAAO,IAAA,kBAAU,EAAC;YACjB,EAAE,EAAE,WAAW;YACf,WAAW,EAAE,4DAA4D;YACzE,WAAW,EAAE,OAAC,CAAC,MAAM,CAAC,EAAE,MAAM,EAAE,OAAC,CAAC,MAAM,EAAE,EAAE,CAAC;YAE7C,OAAO,EAAE,KAAK,IAAI,EAAE;gBACnB,OAAO,EAAE,KAAK,EAAE,yDAAyD,EAAE,CAAC;YAC7E,CAAC;SACD,CAAC,CAAC;IACJ,CAAC;IAED,MAAM,WAAW,GAAG,gBAAgB,CAAC,OAAO,CAAC,CAAC;IAE9C,OAAO,IAAA,kBAAU,EAAC;QACjB,EAAE,EAAE,WAAW;QACf,WAAW,EAAE,8EAA8E;QAC3F,WAAW;QACX,aAAa;QACb,YAAY;QACZ,OAAO,EAAE,KAAK,EAAE,KAAY,EAAE,GAAG,EAAE,EAAE;YACpC,QAAQ,KAAK,CAAC,MAAM,EAAE,CAAC;gBACtB,KAAK,eAAe;oBACnB,OAAO,MAAM,kBAAkB,CAAC,OAAO,CAAC,CAAC;gBAC1C,KAAK,WAAW;oBACf,OAAO,MAAM,cAAc,CAAC,OAAO,CAAC,CAAC;gBACtC,KAAK,cAAc;oBAClB,OAAO,MAAM,iBAAiB,CAAC,OAAO,EAAE,KAAK,EAAE,GAAG,CAAC,CAAC;gBACrD,KAAK,yBAAyB;oBAC7B,OAAO,MAAM,2BAA2B,CAAC,OAAO,EAAE,KAAK,EAAE,GAAG,CAAC,CAAC;gBAC/D,KAAK,cAAc;oBAClB,OAAO,MAAM,iBAAiB,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC;gBAChD,KAAK,eAAe;oBACnB,OAAO,MAAM,kBAAkB,CAAC,OAAO,EAAE,KAAK,EAAE,GAAG,CAAC,CAAC;gBACtD,KAAK,eAAe;oBACnB,OAAO,MAAM,kBAAkB,CAAC,OAAO,EAAE,KAAK,EAAE,GAAG,CAAC,CAAC;gBACtD,KAAK,yBAAyB;oBAC7B,OAAO,MAAM,0BAA0B,CAAC,OAAO,EAAE,KAAK,EAAE,GAAG,CAAC,CAAC;YAC/D,CAAC;QACF,CAAC;KACD,CAAC,CAAC;AACJ,CAAC"}

package/dist/tracing/langsmith-tracing.d.ts

@@ -51,6 +51,7 @@ export declare function mergeTraceRunInputs(run: InstanceAiTraceRun | undefined,
 export declare function buildAgentTraceInputs(options: AgentTraceInputOptions): Record<string, unknown>;
 export declare function withTraceParentContext<T>(parentRun: RunTree | undefined, fn: () => Promise<T>): Promise<T>;
 export declare function withCurrentTraceSpan<T>(options: CurrentTraceSpanOptions<T>, fn: () => Promise<T>): Promise<T>;
+export declare function createTraceReplayOnlyContext(): InstanceAiTraceContext;
 export declare function createInstanceAiTraceContext(options: CreateInstanceAiTraceContextOptions): Promise<InstanceAiTraceContext | undefined>;
 export declare function continueInstanceAiTraceContext(existingContext: InstanceAiTraceContext, options: CreateInstanceAiTraceContextOptions): Promise<InstanceAiTraceContext>;
 export declare function createDetachedSubAgentTraceContext(options: CreateDetachedSubAgentTraceContextOptions): Promise<InstanceAiTraceContext | undefined>;

package/dist/tracing/langsmith-tracing.js

@@ -9,6 +9,7 @@ exports.mergeTraceRunInputs = mergeTraceRunInputs;
 exports.buildAgentTraceInputs = buildAgentTraceInputs;
 exports.withTraceParentContext = withTraceParentContext;
 exports.withCurrentTraceSpan = withCurrentTraceSpan;
+exports.createTraceReplayOnlyContext = createTraceReplayOnlyContext;
 exports.createInstanceAiTraceContext = createInstanceAiTraceContext;
 exports.continueInstanceAiTraceContext = continueInstanceAiTraceContext;
 exports.createDetachedSubAgentTraceContext = createDetachedSubAgentTraceContext;
@@ -16,6 +17,7 @@ const tools_1 = require("@mastra/core/tools");
 const langsmith_1 = require("langsmith");
 const traceable_1 = require("langsmith/traceable");
 const node_async_hooks_1 = require("node:async_hooks");
+const trace_replay_1 = require("./trace-replay");
 const stream_helpers_1 = require("../utils/stream-helpers");
 const DEFAULT_PROJECT_NAME = 'instance-ai';
 const DEFAULT_TAGS = ['instance-ai'];
@@ -534,7 +536,7 @@ function createTraceContext(projectName, traceKind, rootRun, actorRun, getProxyH
             traceClients.delete(run.traceId);
         }
     };
-    return {
+    const ctx = {
         projectName,
         traceKind,
         rootRun,
@@ -546,8 +548,18 @@ function createTraceContext(projectName, traceKind, rootRun, actorRun, getProxyH
         finishRun,
         failRun,
         toHeaders: (run) => hydrateRunTree(run).toHeaders(),
-        wrapTools: (tools, traceOptions) => wrapTools(tools, traceOptions),
+        wrapTools: (tools, traceOptions) => {
+            if (ctx.replayMode === 'replay' && ctx.traceIndex && ctx.idRemapper) {
+                return replayWrapTools(tools, ctx.traceIndex, ctx.idRemapper, traceOptions);
+            }
+            if (ctx.replayMode === 'record' && ctx.traceWriter) {
+                return recordWrapTools(tools, ctx.traceWriter, traceOptions);
+            }
+            return wrapTools(tools, traceOptions);
+        },
+        replayMode: 'off',
     };
+    return ctx;
 }
 function createRunStateFromTree(tree) {
     const parentRunId = tree.parent_run?.id ?? tree.parent_run_id;
@@ -656,6 +668,162 @@ function wrapTools(tools, options) {
     }
     return wrapped;
 }
+function replayWrapTool(tool, traceIndex, idRemapper, agentRole) {
+    return (0, tools_1.createTool)({
+        id: tool.id,
+        description: tool.description,
+        inputSchema: tool.inputSchema,
+        outputSchema: tool.outputSchema,
+        suspendSchema: tool.suspendSchema,
+        resumeSchema: tool.resumeSchema,
+        requestContextSchema: tool.requestContextSchema,
+        execute: async (input, context) => {
+            const event = traceIndex.next(agentRole, tool.id);
+            const remappedInput = idRemapper.remapInput(input);
+            const realOutput = await tool.execute(remappedInput, context);
+            idRemapper.learn(event.output, realOutput);
+            return realOutput;
+        },
+        mastra: tool.mastra,
+        requireApproval: tool.requireApproval,
+        providerOptions: tool.providerOptions,
+        toModelOutput: tool.toModelOutput,
+        mcp: tool.mcp,
+        onInputStart: tool.onInputStart,
+        onInputDelta: tool.onInputDelta,
+        onInputAvailable: tool.onInputAvailable,
+        onOutput: tool.onOutput,
+    });
+}
+function pureReplayWrapTool(tool, traceIndex, idRemapper, agentRole) {
+    return (0, tools_1.createTool)({
+        id: tool.id,
+        description: tool.description,
+        inputSchema: tool.inputSchema,
+        outputSchema: tool.outputSchema,
+        suspendSchema: tool.suspendSchema,
+        resumeSchema: tool.resumeSchema,
+        requestContextSchema: tool.requestContextSchema,
+        execute: async (_input, _context) => {
+            const event = traceIndex.next(agentRole, tool.id);
+            return await Promise.resolve(idRemapper.remapOutput(event.output));
+        },
+        mastra: tool.mastra,
+        requireApproval: tool.requireApproval,
+        providerOptions: tool.providerOptions,
+        toModelOutput: tool.toModelOutput,
+        mcp: tool.mcp,
+        onInputStart: tool.onInputStart,
+        onInputDelta: tool.onInputDelta,
+        onInputAvailable: tool.onInputAvailable,
+        onOutput: tool.onOutput,
+    });
+}
+function replayWrapTools(tools, traceIndex, idRemapper, options) {
+    const agentRole = options?.agentRole ?? 'orchestrator';
+    const wrapped = {};
+    const entries = Object.entries(tools);
+    for (const [name, tool] of entries) {
+        if (!isTraceableMastraTool(tool)) {
+            wrapped[name] = tools[name];
+            continue;
+        }
+        if (trace_replay_1.PURE_REPLAY_TOOLS.has(tool.id)) {
+            wrapped[name] = pureReplayWrapTool(tool, traceIndex, idRemapper, agentRole);
+        }
+        else {
+            wrapped[name] = replayWrapTool(tool, traceIndex, idRemapper, agentRole);
+        }
+    }
+    return wrapped;
+}
+function recordWrapTool(tool, traceWriter, agentRole, traceOptions) {
+    const traced = wrapToolExecute(tool, traceOptions);
+    return (0, tools_1.createTool)({
+        id: traced.id,
+        description: traced.description,
+        inputSchema: traced.inputSchema,
+        outputSchema: traced.outputSchema,
+        suspendSchema: traced.suspendSchema,
+        resumeSchema: traced.resumeSchema,
+        requestContextSchema: traced.requestContextSchema,
+        execute: async (input, context) => {
+            const resumeData = context?.agent?.resumeData;
+            const inputRecord = (input ?? {});
+            const result = await traced.execute(input, context);
+            const outputRecord = (result ?? {});
+            if (resumeData !== undefined && resumeData !== null) {
+                traceWriter.recordToolResume(agentRole, tool.id, inputRecord, outputRecord, resumeData);
+            }
+            else if (context?.agent?.suspend && outputRecord.denied === true) {
+                traceWriter.recordToolSuspend(agentRole, tool.id, inputRecord, outputRecord, {});
+            }
+            else {
+                traceWriter.recordToolCall(agentRole, tool.id, inputRecord, outputRecord);
+            }
+            return result;
+        },
+        mastra: traced.mastra,
+        requireApproval: traced.requireApproval,
+        providerOptions: traced.providerOptions,
+        toModelOutput: traced.toModelOutput,
+        mcp: traced.mcp,
+        onInputStart: traced.onInputStart,
+        onInputDelta: traced.onInputDelta,
+        onInputAvailable: traced.onInputAvailable,
+        onOutput: traced.onOutput,
+    });
+}
+function recordWrapTools(tools, traceWriter, options) {
+    const agentRole = options?.agentRole ?? 'orchestrator';
+    const wrapped = {};
+    const entries = Object.entries(tools);
+    for (const [name, tool] of entries) {
+        if (!isTraceableMastraTool(tool)) {
+            wrapped[name] = tools[name];
+            continue;
+        }
+        wrapped[name] = recordWrapTool(tool, traceWriter, agentRole, options);
+    }
+    return wrapped;
+}
+function createTraceReplayOnlyContext() {
+    const stubRun = {
+        id: 'stub',
+        name: 'stub',
+        runType: 'chain',
+        projectName: '',
+        startTime: Date.now(),
+        traceId: 'stub',
+        dottedOrder: '',
+        executionOrder: 0,
+        childExecutionOrder: 0,
+    };
+    const ctx = {
+        projectName: '',
+        traceKind: 'message_turn',
+        rootRun: stubRun,
+        actorRun: stubRun,
+        messageRun: stubRun,
+        orchestratorRun: stubRun,
+        startChildRun: async () => await Promise.resolve(stubRun),
+        withRunTree: async (_run, fn) => await fn(),
+        finishRun: async () => { },
+        failRun: async () => { },
+        toHeaders: () => ({}),
+        wrapTools: (tools, traceOptions) => {
+            if (ctx.replayMode === 'replay' && ctx.traceIndex && ctx.idRemapper) {
+                return replayWrapTools(tools, ctx.traceIndex, ctx.idRemapper, traceOptions);
+            }
+            if (ctx.replayMode === 'record' && ctx.traceWriter) {
+                return recordWrapTools(tools, ctx.traceWriter, traceOptions);
+            }
+            return tools;
+        },
+        replayMode: 'off',
+    };
+    return ctx;
+}
 async function createRun(options) {
     const runTree = new langsmith_1.RunTree({
         name: options.name,