@n8n/agents 0.3.0 → 0.5.0

package/dist/codegen/generate-agent-code.d.ts

@@ -0,0 +1,2 @@
+import type { AgentSchema } from '../types/sdk/schema';
+export declare function generateAgentCode(schema: AgentSchema, agentName: string): Promise<string>;

package/dist/codegen/generate-agent-code.js

@@ -0,0 +1,197 @@
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __importStar = (this && this.__importStar) || (function () {
+    var ownKeys = function(o) {
+        ownKeys = Object.getOwnPropertyNames || function (o) {
+            var ar = [];
+            for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
+            return ar;
+        };
+        return ownKeys(o);
+    };
+    return function (mod) {
+        if (mod && mod.__esModule) return mod;
+        var result = {};
+        if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
+        __setModuleDefault(result, mod);
+        return result;
+    };
+})();
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.generateAgentCode = generateAgentCode;
+function escapeTemplateLiteral(str) {
+    return str.replace(/\\/g, '\\\\').replace(/`/g, '\\`').replace(/\$/g, '\\$');
+}
+function escapeSingleQuote(str) {
+    return JSON.stringify(str).slice(1, -1).replace(/'/g, "\\'");
+}
+let prettierInstance;
+async function formatCode(code) {
+    prettierInstance ??= await Promise.resolve().then(() => __importStar(require('prettier')));
+    return await prettierInstance.format(code, {
+        parser: 'typescript',
+        singleQuote: true,
+        useTabs: true,
+        trailingComma: 'all',
+        printWidth: 100,
+    });
+}
+function assertAllHandled(_) {
+}
+function modelParts(model) {
+    if (model.provider && model.name) {
+        return [`.model('${escapeSingleQuote(model.provider)}', '${escapeSingleQuote(model.name)}')`];
+    }
+    if (model.name) {
+        return [`.model('${escapeSingleQuote(model.name)}')`];
+    }
+    return [];
+}
+function toolPart(tool) {
+    if (!tool.editable) {
+        return {
+            part: `.tool(new WorkflowTool('${escapeSingleQuote(tool.name)}'))`,
+            usesWorkflowTool: true,
+        };
+    }
+    const parts = [`new Tool('${escapeSingleQuote(tool.name)}')`];
+    parts.push(`.description('${escapeSingleQuote(tool.description)}')`);
+    if (tool.inputSchemaSource)
+        parts.push(`.input(${tool.inputSchemaSource})`);
+    if (tool.outputSchemaSource)
+        parts.push(`.output(${tool.outputSchemaSource})`);
+    if (tool.suspendSchemaSource)
+        parts.push(`.suspend(${tool.suspendSchemaSource})`);
+    if (tool.resumeSchemaSource)
+        parts.push(`.resume(${tool.resumeSchemaSource})`);
+    if (tool.handlerSource)
+        parts.push(`.handler(${tool.handlerSource})`);
+    if (tool.toMessageSource)
+        parts.push(`.toMessage(${tool.toMessageSource})`);
+    if (tool.requireApproval)
+        parts.push('.requireApproval()');
+    if (tool.needsApprovalFnSource)
+        parts.push(`.needsApprovalFn(${tool.needsApprovalFnSource})`);
+    return { part: `.tool(${parts.join('')})`, usesWorkflowTool: false };
+}
+function evalPart(ev) {
+    const parts = [`new Eval('${escapeSingleQuote(ev.name)}')`];
+    if (ev.description)
+        parts.push(`.description('${escapeSingleQuote(ev.description)}')`);
+    if (ev.modelId)
+        parts.push(`.model('${escapeSingleQuote(ev.modelId)}')`);
+    if (ev.credentialName)
+        parts.push(`.credential('${escapeSingleQuote(ev.credentialName)}')`);
+    if (ev.handlerSource) {
+        parts.push(ev.type === 'check' ? `.check(${ev.handlerSource})` : `.judge(${ev.handlerSource})`);
+    }
+    return `.eval(${parts.join('')})`;
+}
+function guardrailPart(g) {
+    const method = g.position === 'input' ? 'inputGuardrail' : 'outputGuardrail';
+    return `.${method}(${g.source})`;
+}
+function memoryPart(memory) {
+    if (memory.source) {
+        return `.memory(${memory.source})`;
+    }
+    return `.memory(new Memory().lastMessages(${memory.lastMessages ?? 10}))`;
+}
+function thinkingPart(thinking) {
+    const props = [];
+    if (thinking.budgetTokens !== undefined)
+        props.push(`budgetTokens: ${thinking.budgetTokens}`);
+    if (thinking.reasoningEffort)
+        props.push(`reasoningEffort: '${thinking.reasoningEffort}'`);
+    if (props.length > 0) {
+        return `.thinking('${thinking.provider}', { ${props.join(', ')} })`;
+    }
+    return `.thinking('${thinking.provider}')`;
+}
+function buildImports(schema, needsWorkflowTool) {
+    const agentImports = new Set(['Agent']);
+    if (schema.tools.some((t) => t.editable))
+        agentImports.add('Tool');
+    if (needsWorkflowTool)
+        agentImports.add('WorkflowTool');
+    if (schema.memory)
+        agentImports.add('Memory');
+    if (schema.mcp && schema.mcp.length > 0)
+        agentImports.add('McpClient');
+    if (schema.evaluations.length > 0)
+        agentImports.add('Eval');
+    const toolsNeedZod = schema.tools.some((t) => (t.inputSchemaSource?.includes('z.') ?? false) ||
+        (t.outputSchemaSource?.includes('z.') ?? false));
+    const structuredOutputNeedsZod = schema.config.structuredOutput.schemaSource?.includes('z.') ?? false;
+    let imports = `import { ${Array.from(agentImports).sort().join(', ')} } from '@n8n/agents';`;
+    if (toolsNeedZod || structuredOutputNeedsZod)
+        imports += "\nimport { z } from 'zod';";
+    return imports;
+}
+async function generateAgentCode(schema, agentName) {
+    const { model, credential, instructions, description: _description, tools, providerTools, memory, evaluations, guardrails, mcp, telemetry, checkpoint, config, ...rest } = schema;
+    assertAllHandled(rest);
+    const { thinking, toolCallConcurrency, requireToolApproval, structuredOutput, ...configRest } = config;
+    assertAllHandled(configRest);
+    const parts = [];
+    let needsWorkflowTool = false;
+    parts.push(`export default new Agent('${escapeSingleQuote(agentName)}')`);
+    parts.push(...modelParts(model));
+    if (credential)
+        parts.push(`.credential('${escapeSingleQuote(credential)}')`);
+    if (instructions)
+        parts.push(`.instructions(\`${escapeTemplateLiteral(instructions)}\`)`);
+    for (const tool of tools) {
+        const { part, usesWorkflowTool } = toolPart(tool);
+        if (usesWorkflowTool)
+            needsWorkflowTool = true;
+        parts.push(part);
+    }
+    for (const pt of providerTools) {
+        parts.push(`.providerTool(${pt.source})`);
+    }
+    if (memory)
+        parts.push(memoryPart(memory));
+    for (const ev of evaluations) {
+        parts.push(evalPart(ev));
+    }
+    for (const g of guardrails) {
+        parts.push(guardrailPart(g));
+    }
+    if (mcp && mcp.length > 0) {
+        const configs = mcp.map((s) => s.configSource).join(', ');
+        parts.push(`.mcp(new McpClient([${configs}]))`);
+    }
+    if (telemetry)
+        parts.push(`.telemetry(${telemetry.source})`);
+    if (checkpoint)
+        parts.push(`.checkpoint('${escapeSingleQuote(checkpoint)}')`);
+    if (thinking)
+        parts.push(thinkingPart(thinking));
+    if (toolCallConcurrency)
+        parts.push(`.toolCallConcurrency(${toolCallConcurrency})`);
+    if (requireToolApproval)
+        parts.push('.requireToolApproval()');
+    if (structuredOutput.enabled && structuredOutput.schemaSource) {
+        parts.push(`.structuredOutput(${structuredOutput.schemaSource})`);
+    }
+    const imports = buildImports(schema, needsWorkflowTool);
+    const raw = `${imports}\n\n${parts.join('')};\n`;
+    return await formatCode(raw);
+}
+//# sourceMappingURL=generate-agent-code.js.map

package/dist/codegen/generate-agent-code.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"generate-agent-code.js","sourceRoot":"","sources":["../../src/codegen/generate-agent-code.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AA4IA,8CA4EC;AA1MD,SAAS,qBAAqB,CAAC,GAAW;IACzC,OAAO,GAAG,CAAC,OAAO,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC,OAAO,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC,OAAO,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC;AAC9E,CAAC;AAED,SAAS,iBAAiB,CAAC,GAAW;IACrC,OAAO,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;AAC9D,CAAC;AAED,IAAI,gBAA6C,CAAC;AAMlD,KAAK,UAAU,UAAU,CAAC,IAAY;IACrC,gBAAgB,KAAK,wDAAa,UAAU,GAAC,CAAC;IAC9C,OAAO,MAAM,gBAAgB,CAAC,MAAM,CAAC,IAAI,EAAE;QAC1C,MAAM,EAAE,YAAY;QACpB,WAAW,EAAE,IAAI;QACjB,OAAO,EAAE,IAAI;QACb,aAAa,EAAE,KAAK;QACpB,UAAU,EAAE,GAAG;KACf,CAAC,CAAC;AACJ,CAAC;AAOD,SAAS,gBAAgB,CAAC,CAAwB;AAElD,CAAC;AAMD,SAAS,UAAU,CAAC,KAA2B;IAC9C,IAAI,KAAK,CAAC,QAAQ,IAAI,KAAK,CAAC,IAAI,EAAE,CAAC;QAClC,OAAO,CAAC,WAAW,iBAAiB,CAAC,KAAK,CAAC,QAAQ,CAAC,OAAO,iBAAiB,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAC/F,CAAC;IACD,IAAI,KAAK,CAAC,IAAI,EAAE,CAAC;QAChB,OAAO,CAAC,WAAW,iBAAiB,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACvD,CAAC;IACD,OAAO,EAAE,CAAC;AACX,CAAC;AAED,SAAS,QAAQ,CAAC,IAAgB;IACjC,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE,CAAC;QACpB,OAAO;YACN,IAAI,EAAE,2BAA2B,iBAAiB,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK;YAClE,gBAAgB,EAAE,IAAI;SACtB,CAAC;IACH,CAAC;IACD,MAAM,KAAK,GAAG,CAAC,aAAa,iBAAiB,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAC9D,KAAK,CAAC,IAAI,CAAC,iBAAiB,iBAAiB,CAAC,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC;IACrE,IAAI,IAAI,CAAC,iBAAiB;QAAE,KAAK,CAAC,IAAI,CAAC,UAAU,IAAI,CAAC,iBAAiB,GAAG,CAAC,CAAC;IAC5E,IAAI,IAAI,CAAC,kBAAkB;QAAE,KAAK,CAAC,IAAI,CAAC,WAAW,IAAI,CAAC,kBAAkB,GAAG,CAAC,CAAC;IAC/E,IAAI,IAAI,CAAC,mBAAmB;QAAE,KAAK,CAAC,IAAI,CAAC,YAAY,IAAI,CAAC,mBAAmB,GAAG,CAAC,CAAC;IAClF,IAAI,IAAI,CAAC,kBAAkB;QAAE,KAAK,CAAC,IAAI,CAAC,WAAW,IAAI,CAAC,kBAAkB,GAAG,CAAC,CAAC;IAC/E,IAAI,IAAI,CAAC,aAAa;QAAE,KAAK,CAAC,IAAI,CAAC,YAAY,IAAI,CAAC,aAAa,GAAG,CAAC,CAAC;IACtE,IAAI,IAAI,CAAC,eAAe;QAAE,KAAK,CAAC,IAAI,CAAC,cAAc,IAAI,CAAC,eAAe,GAAG,CAAC,CAAC;IAC5E,IAAI,IAAI,CAAC,eAAe;QAAE,KAAK,CAAC,IAAI,CAAC,oBAAoB,CAAC,CAAC;IAC3D,IAAI,IAAI,CAAC,qBAAqB;QAAE,KAAK,CAAC,IAAI,CAAC,oBAAoB,IAAI,CAAC,qBAAqB,GAAG,CAAC,CAAC;IAC9F,OAAO,EAAE,IAAI,EAAE,SAAS,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,GAAG,EAAE,gBAAgB,EAAE,KAAK,EAAE,CAAC;AACtE,CAAC;AAED,SAAS,QAAQ,CAAC,EAAc;IAC/B,MAAM,KAAK,GAAG,CAAC,aAAa,iBAAiB,CAAC,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAC5D,IAAI,EAAE,CAAC,WAAW;QAAE,KAAK,CAAC,IAAI,CAAC,iBAAiB,iBAAiB,CAAC,EAAE,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC;IACvF,IAAI,EAAE,CAAC,OAAO;QAAE,KAAK,CAAC,IAAI,CAAC,WAAW,iBAAiB,CAAC,EAAE,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;IACzE,IAAI,EAAE,CAAC,cAAc;QAAE,KAAK,CAAC,IAAI,CAAC,gBAAgB,iBAAiB,CAAC,EAAE,CAAC,cAAc,CAAC,IAAI,CAAC,CAAC;IAC5F,IAAI,EAAE,CAAC,aAAa,EAAE,CAAC;QACtB,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,KAAK,OAAO,CAAC,CAAC,CAAC,UAAU,EAAE,CAAC,aAAa,GAAG,CAAC,CAAC,CAAC,UAAU,EAAE,CAAC,aAAa,GAAG,CAAC,CAAC;IACjG,CAAC;IACD,OAAO,SAAS,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,GAAG,CAAC;AACnC,CAAC;AAED,SAAS,aAAa,CAAC,CAAkB;IACxC,MAAM,MAAM,GAAG,CAAC,CAAC,QAAQ,KAAK,OAAO,CAAC,CAAC,CAAC,gBAAgB,CAAC,CAAC,CAAC,iBAAiB,CAAC;IAC7E,OAAO,IAAI,MAAM,IAAI,CAAC,CAAC,MAAM,GAAG,CAAC;AAClC,CAAC;AAED,SAAS,UAAU,CAAC,MAAoB;IACvC,IAAI,MAAM,CAAC,MAAM,EAAE,CAAC;QACnB,OAAO,WAAW,MAAM,CAAC,MAAM,GAAG,CAAC;IACpC,CAAC;IACD,OAAO,qCAAqC,MAAM,CAAC,YAAY,IAAI,EAAE,IAAI,CAAC;AAC3E,CAAC;AAED,SAAS,YAAY,CAAC,QAAwD;IAC7E,MAAM,KAAK,GAAa,EAAE,CAAC;IAC3B,IAAI,QAAQ,CAAC,YAAY,KAAK,SAAS;QAAE,KAAK,CAAC,IAAI,CAAC,iBAAiB,QAAQ,CAAC,YAAY,EAAE,CAAC,CAAC;IAC9F,IAAI,QAAQ,CAAC,eAAe;QAAE,KAAK,CAAC,IAAI,CAAC,qBAAqB,QAAQ,CAAC,eAAe,GAAG,CAAC,CAAC;IAC3F,IAAI,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACtB,OAAO,cAAc,QAAQ,CAAC,QAAQ,QAAQ,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC;IACrE,CAAC;IACD,OAAO,cAAc,QAAQ,CAAC,QAAQ,IAAI,CAAC;AAC5C,CAAC;AAED,SAAS,YAAY,CAAC,MAAmB,EAAE,iBAA0B;IACpE,MAAM,YAAY,GAAG,IAAI,GAAG,CAAS,CAAC,OAAO,CAAC,CAAC,CAAC;IAChD,IAAI,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC;QAAE,YAAY,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;IACnE,IAAI,iBAAiB;QAAE,YAAY,CAAC,GAAG,CAAC,cAAc,CAAC,CAAC;IACxD,IAAI,MAAM,CAAC,MAAM;QAAE,YAAY,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;IAC9C,IAAI,MAAM,CAAC,GAAG,IAAI,MAAM,CAAC,GAAG,CAAC,MAAM,GAAG,CAAC;QAAE,YAAY,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC;IACvE,IAAI,MAAM,CAAC,WAAW,CAAC,MAAM,GAAG,CAAC;QAAE,YAAY,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;IAE5D,MAAM,YAAY,GAAG,MAAM,CAAC,KAAK,CAAC,IAAI,CACrC,CAAC,CAAC,EAAE,EAAE,CACL,CAAC,CAAC,CAAC,iBAAiB,EAAE,QAAQ,CAAC,IAAI,CAAC,IAAI,KAAK,CAAC;QAC9C,CAAC,CAAC,CAAC,kBAAkB,EAAE,QAAQ,CAAC,IAAI,CAAC,IAAI,KAAK,CAAC,CAChD,CAAC;IACF,MAAM,wBAAwB,GAC7B,MAAM,CAAC,MAAM,CAAC,gBAAgB,CAAC,YAAY,EAAE,QAAQ,CAAC,IAAI,CAAC,IAAI,KAAK,CAAC;IAEtE,IAAI,OAAO,GAAG,YAAY,KAAK,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,IAAI,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,wBAAwB,CAAC;IAC7F,IAAI,YAAY,IAAI,wBAAwB;QAAE,OAAO,IAAI,4BAA4B,CAAC;IACtF,OAAO,OAAO,CAAC;AAChB,CAAC;AAMM,KAAK,UAAU,iBAAiB,CAAC,MAAmB,EAAE,SAAiB;IAI7E,MAAM,EACL,KAAK,EACL,UAAU,EACV,YAAY,EACZ,WAAW,EAAE,YAAY,EACzB,KAAK,EACL,aAAa,EACb,MAAM,EACN,WAAW,EACX,UAAU,EACV,GAAG,EACH,SAAS,EACT,UAAU,EACV,MAAM,EACN,GAAG,IAAI,EACP,GAAG,MAAM,CAAC;IAIX,gBAAgB,CAAC,IAAI,CAAC,CAAC;IAEvB,MAAM,EAAE,QAAQ,EAAE,mBAAmB,EAAE,mBAAmB,EAAE,gBAAgB,EAAE,GAAG,UAAU,EAAE,GAC5F,MAAM,CAAC;IACR,gBAAgB,CAAC,UAAU,CAAC,CAAC;IAG7B,MAAM,KAAK,GAAa,EAAE,CAAC;IAC3B,IAAI,iBAAiB,GAAG,KAAK,CAAC;IAE9B,KAAK,CAAC,IAAI,CAAC,6BAA6B,iBAAiB,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC;IAC1E,KAAK,CAAC,IAAI,CAAC,GAAG,UAAU,CAAC,KAAK,CAAC,CAAC,CAAC;IAEjC,IAAI,UAAU;QAAE,KAAK,CAAC,IAAI,CAAC,gBAAgB,iBAAiB,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC;IAC9E,IAAI,YAAY;QAAE,KAAK,CAAC,IAAI,CAAC,mBAAmB,qBAAqB,CAAC,YAAY,CAAC,KAAK,CAAC,CAAC;IAE1F,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;QAC1B,MAAM,EAAE,IAAI,EAAE,gBAAgB,EAAE,GAAG,QAAQ,CAAC,IAAI,CAAC,CAAC;QAClD,IAAI,gBAAgB;YAAE,iBAAiB,GAAG,IAAI,CAAC;QAC/C,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAClB,CAAC;IAED,KAAK,MAAM,EAAE,IAAI,aAAa,EAAE,CAAC;QAChC,KAAK,CAAC,IAAI,CAAC,iBAAiB,EAAE,CAAC,MAAM,GAAG,CAAC,CAAC;IAC3C,CAAC;IAED,IAAI,MAAM;QAAE,KAAK,CAAC,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC,CAAC;IAE3C,KAAK,MAAM,EAAE,IAAI,WAAW,EAAE,CAAC;QAC9B,KAAK,CAAC,IAAI,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,CAAC;IAC1B,CAAC;IAED,KAAK,MAAM,CAAC,IAAI,UAAU,EAAE,CAAC;QAC5B,KAAK,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC,CAAC;IAC9B,CAAC;IAED,IAAI,GAAG,IAAI,GAAG,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QAC3B,MAAM,OAAO,GAAG,GAAG,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC1D,KAAK,CAAC,IAAI,CAAC,uBAAuB,OAAO,KAAK,CAAC,CAAC;IACjD,CAAC;IAED,IAAI,SAAS;QAAE,KAAK,CAAC,IAAI,CAAC,cAAc,SAAS,CAAC,MAAM,GAAG,CAAC,CAAC;IAC7D,IAAI,UAAU;QAAE,KAAK,CAAC,IAAI,CAAC,gBAAgB,iBAAiB,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC;IAC9E,IAAI,QAAQ;QAAE,KAAK,CAAC,IAAI,CAAC,YAAY,CAAC,QAAQ,CAAC,CAAC,CAAC;IACjD,IAAI,mBAAmB;QAAE,KAAK,CAAC,IAAI,CAAC,wBAAwB,mBAAmB,GAAG,CAAC,CAAC;IACpF,IAAI,mBAAmB;QAAE,KAAK,CAAC,IAAI,CAAC,wBAAwB,CAAC,CAAC;IAC9D,IAAI,gBAAgB,CAAC,OAAO,IAAI,gBAAgB,CAAC,YAAY,EAAE,CAAC;QAC/D,KAAK,CAAC,IAAI,CAAC,qBAAqB,gBAAgB,CAAC,YAAY,GAAG,CAAC,CAAC;IACnE,CAAC;IAED,MAAM,OAAO,GAAG,YAAY,CAAC,MAAM,EAAE,iBAAiB,CAAC,CAAC;IACxD,MAAM,GAAG,GAAG,GAAG,OAAO,OAAO,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,KAAK,CAAC;IACjD,OAAO,MAAM,UAAU,CAAC,GAAG,CAAC,CAAC;AAC9B,CAAC"}

package/dist/index.d.ts

@@ -13,19 +13,28 @@ export { Telemetry } from './sdk/telemetry';
 export { LangSmithTelemetry } from './integrations/langsmith';
 export type { LangSmithTelemetryConfig } from './integrations/langsmith';
 export { Agent } from './sdk/agent';
+export type { AgentBuilder, CredentialProvider, ResolvedCredential, CredentialListItem, } from './types';
 export { McpClient } from './sdk/mcp-client';
 export { Network } from './sdk/network';
 export { providerTools } from './sdk/provider-tools';
 export { verify } from './sdk/verify';
 export type { VerifyResult } from './sdk/verify';
 export type { ContentCitation, ContentFile, ContentMetadata, ContentReasoning, ContentText, ContentToolCall, ContentToolResult, Message, MessageContent, MessageRole, AgentMessage, CustomAgentMessages, AgentDbMessage, } from './types/sdk/message';
-export { toDbMessage, filterLlmMessages, isLlmMessage, } from './sdk/message';
+export type { HandlerExecutor } from './types/sdk/handler-executor';
+export type { AgentSchema, ToolSchema, MemorySchema, EvalSchema, ThinkingSchema, ProviderToolSchema, GuardrailSchema, McpServerSchema, TelemetrySchema, } from './types/sdk/schema';
+export { generateAgentCode } from './codegen/generate-agent-code';
+export { filterLlmMessages, isLlmMessage } from './sdk/message';
 export { fetchProviderCatalog } from './sdk/catalog';
+export { providerCapabilities } from './sdk/provider-capabilities';
+export type { ProviderCapability } from './sdk/provider-capabilities';
 export type { ProviderCatalog, ProviderInfo, ModelInfo, ModelCost, ModelLimits, } from './sdk/catalog';
 export { SqliteMemory } from './storage/sqlite-memory';
+export { UPDATE_WORKING_MEMORY_TOOL_NAME, WORKING_MEMORY_DEFAULT_INSTRUCTION, } from './runtime/working-memory';
 export type { SqliteMemoryConfig } from './storage/sqlite-memory';
 export { PostgresMemory } from './storage/postgres-memory';
 export type { PostgresMemoryConfig } from './storage/postgres-memory';
+export { createModel } from './runtime/model-factory';
+export { generateTitleFromMessage } from './runtime/title-generation';
 export { Workspace } from './workspace';
 export { BaseFilesystem } from './workspace';
 export { BaseSandbox } from './workspace';

package/dist/index.js

@@ -33,7 +33,7 @@ var __importStar = (this && this.__importStar) || (function () {
     };
 })();
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.ProcessHandle = exports.SandboxProcessManager = exports.createWorkspaceTools = exports.BaseSandbox = exports.BaseFilesystem = exports.Workspace = exports.PostgresMemory = exports.SqliteMemory = exports.fetchProviderCatalog = exports.isLlmMessage = exports.filterLlmMessages = exports.toDbMessage = exports.verify = exports.providerTools = exports.Network = exports.McpClient = exports.Agent = exports.LangSmithTelemetry = exports.Telemetry = exports.evals = exports.evaluate = exports.Eval = exports.Guardrail = exports.Memory = exports.Tool = exports.AgentEvent = void 0;
+exports.ProcessHandle = exports.SandboxProcessManager = exports.createWorkspaceTools = exports.BaseSandbox = exports.BaseFilesystem = exports.Workspace = exports.generateTitleFromMessage = exports.createModel = exports.PostgresMemory = exports.WORKING_MEMORY_DEFAULT_INSTRUCTION = exports.UPDATE_WORKING_MEMORY_TOOL_NAME = exports.SqliteMemory = exports.providerCapabilities = exports.fetchProviderCatalog = exports.isLlmMessage = exports.filterLlmMessages = exports.generateAgentCode = exports.verify = exports.providerTools = exports.Network = exports.McpClient = exports.Agent = exports.LangSmithTelemetry = exports.Telemetry = exports.evals = exports.evaluate = exports.Eval = exports.Guardrail = exports.Memory = exports.Tool = exports.AgentEvent = void 0;
 var types_1 = require("./types");
 Object.defineProperty(exports, "AgentEvent", { enumerable: true, get: function () { return types_1.AgentEvent; } });
 var tool_1 = require("./sdk/tool");
@@ -61,16 +61,26 @@ var provider_tools_1 = require("./sdk/provider-tools");
 Object.defineProperty(exports, "providerTools", { enumerable: true, get: function () { return provider_tools_1.providerTools; } });
 var verify_1 = require("./sdk/verify");
 Object.defineProperty(exports, "verify", { enumerable: true, get: function () { return verify_1.verify; } });
+var generate_agent_code_1 = require("./codegen/generate-agent-code");
+Object.defineProperty(exports, "generateAgentCode", { enumerable: true, get: function () { return generate_agent_code_1.generateAgentCode; } });
 var message_1 = require("./sdk/message");
-Object.defineProperty(exports, "toDbMessage", { enumerable: true, get: function () { return message_1.toDbMessage; } });
 Object.defineProperty(exports, "filterLlmMessages", { enumerable: true, get: function () { return message_1.filterLlmMessages; } });
 Object.defineProperty(exports, "isLlmMessage", { enumerable: true, get: function () { return message_1.isLlmMessage; } });
 var catalog_1 = require("./sdk/catalog");
 Object.defineProperty(exports, "fetchProviderCatalog", { enumerable: true, get: function () { return catalog_1.fetchProviderCatalog; } });
+var provider_capabilities_1 = require("./sdk/provider-capabilities");
+Object.defineProperty(exports, "providerCapabilities", { enumerable: true, get: function () { return provider_capabilities_1.providerCapabilities; } });
 var sqlite_memory_1 = require("./storage/sqlite-memory");
 Object.defineProperty(exports, "SqliteMemory", { enumerable: true, get: function () { return sqlite_memory_1.SqliteMemory; } });
+var working_memory_1 = require("./runtime/working-memory");
+Object.defineProperty(exports, "UPDATE_WORKING_MEMORY_TOOL_NAME", { enumerable: true, get: function () { return working_memory_1.UPDATE_WORKING_MEMORY_TOOL_NAME; } });
+Object.defineProperty(exports, "WORKING_MEMORY_DEFAULT_INSTRUCTION", { enumerable: true, get: function () { return working_memory_1.WORKING_MEMORY_DEFAULT_INSTRUCTION; } });
 var postgres_memory_1 = require("./storage/postgres-memory");
 Object.defineProperty(exports, "PostgresMemory", { enumerable: true, get: function () { return postgres_memory_1.PostgresMemory; } });
+var model_factory_1 = require("./runtime/model-factory");
+Object.defineProperty(exports, "createModel", { enumerable: true, get: function () { return model_factory_1.createModel; } });
+var title_generation_1 = require("./runtime/title-generation");
+Object.defineProperty(exports, "generateTitleFromMessage", { enumerable: true, get: function () { return title_generation_1.generateTitleFromMessage; } });
 var workspace_1 = require("./workspace");
 Object.defineProperty(exports, "Workspace", { enumerable: true, get: function () { return workspace_1.Workspace; } });
 var workspace_2 = require("./workspace");

package/dist/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AA2CA,iCAAqC;AAA5B,mGAAA,UAAU,OAAA;AAGnB,mCAAkC;AAAzB,4FAAA,IAAI,OAAA;AACb,uCAAsC;AAA7B,gGAAA,MAAM,OAAA;AACf,6CAA4C;AAAnC,sGAAA,SAAS,OAAA;AAClB,mCAAkC;AAAzB,4FAAA,IAAI,OAAA;AACb,2CAA0C;AAAjC,oGAAA,QAAQ,OAAA;AAEjB,uDAAuC;AACvC,6CAA4C;AAAnC,sGAAA,SAAS,OAAA;AAClB,sDAA8D;AAArD,+GAAA,kBAAkB,OAAA;AAE3B,qCAAoC;AAA3B,8FAAA,KAAK,OAAA;AACd,+CAA6C;AAApC,uGAAA,SAAS,OAAA;AAClB,yCAAwC;AAA/B,kGAAA,OAAO,OAAA;AAChB,uDAAqD;AAA5C,+GAAA,aAAa,OAAA;AACtB,uCAAsC;AAA7B,gGAAA,MAAM,OAAA;AAiBf,yCAIuB;AAHtB,sGAAA,WAAW,OAAA;AACX,4GAAA,iBAAiB,OAAA;AACjB,uGAAA,YAAY,OAAA;AAEb,yCAAqD;AAA5C,+GAAA,oBAAoB,OAAA;AAQ7B,yDAAuD;AAA9C,6GAAA,YAAY,OAAA;AAErB,6DAA2D;AAAlD,iHAAA,cAAc,OAAA;AAGvB,yCAAwC;AAA/B,sGAAA,SAAS,OAAA;AAClB,yCAA6C;AAApC,2GAAA,cAAc,OAAA;AACvB,yCAA0C;AAAjC,wGAAA,WAAW,OAAA;AACpB,yCAAmD;AAA1C,iHAAA,oBAAoB,OAAA;AAC7B,yCAAmE;AAA1D,kHAAA,qBAAqB,OAAA;AAAE,0GAAA,aAAa,OAAA"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AA2CA,iCAAqC;AAA5B,mGAAA,UAAU,OAAA;AAGnB,mCAAkC;AAAzB,4FAAA,IAAI,OAAA;AACb,uCAAsC;AAA7B,gGAAA,MAAM,OAAA;AACf,6CAA4C;AAAnC,sGAAA,SAAS,OAAA;AAClB,mCAAkC;AAAzB,4FAAA,IAAI,OAAA;AACb,2CAA0C;AAAjC,oGAAA,QAAQ,OAAA;AAEjB,uDAAuC;AACvC,6CAA4C;AAAnC,sGAAA,SAAS,OAAA;AAClB,sDAA8D;AAArD,+GAAA,kBAAkB,OAAA;AAE3B,qCAAoC;AAA3B,8FAAA,KAAK,OAAA;AAOd,+CAA6C;AAApC,uGAAA,SAAS,OAAA;AAClB,yCAAwC;AAA/B,kGAAA,OAAO,OAAA;AAChB,uDAAqD;AAA5C,+GAAA,aAAa,OAAA;AACtB,uCAAsC;AAA7B,gGAAA,MAAM,OAAA;AA6Bf,qEAAkE;AAAzD,wHAAA,iBAAiB,OAAA;AAC1B,yCAAgE;AAAvD,4GAAA,iBAAiB,OAAA;AAAE,uGAAA,YAAY,OAAA;AACxC,yCAAqD;AAA5C,+GAAA,oBAAoB,OAAA;AAC7B,qEAAmE;AAA1D,6HAAA,oBAAoB,OAAA;AAS7B,yDAAuD;AAA9C,6GAAA,YAAY,OAAA;AACrB,2DAGkC;AAFjC,iIAAA,+BAA+B,OAAA;AAC/B,oIAAA,kCAAkC,OAAA;AAGnC,6DAA2D;AAAlD,iHAAA,cAAc,OAAA;AAGvB,yDAAsD;AAA7C,4GAAA,WAAW,OAAA;AACpB,+DAAsE;AAA7D,4HAAA,wBAAwB,OAAA;AAEjC,yCAAwC;AAA/B,sGAAA,SAAS,OAAA;AAClB,yCAA6C;AAApC,2GAAA,cAAc,OAAA;AACvB,yCAA0C;AAAjC,wGAAA,WAAW,OAAA;AACpB,yCAAmD;AAA1C,iHAAA,oBAAoB,OAAA;AAC7B,yCAAmE;AAA1D,kHAAA,qBAAqB,OAAA;AAAE,0GAAA,aAAa,OAAA"}

package/dist/runtime/agent-runtime.d.ts

@@ -1,8 +1,8 @@
 import type { ProviderOptions } from '@ai-sdk/provider-utils';
 import type { z } from 'zod';
-import type { BuiltMemory, BuiltProviderTool, BuiltTool, CheckpointStore, GenerateResult, RunOptions, SemanticRecallConfig, SerializableAgentState, StreamResult, ThinkingConfig, TitleGenerationConfig, BuiltTelemetry } from '../types';
+import type { BuiltMemory, BuiltProviderTool, BuiltTelemetry, BuiltTool, CheckpointStore, GenerateResult, RunOptions, SemanticRecallConfig, SerializableAgentState, StreamResult, ThinkingConfig, TitleGenerationConfig } from '../types';
 import { AgentEventBus } from './event-bus';
-import type { ModelConfig, ExecutionOptions } from '../types/sdk/agent';
+import type { ExecutionOptions, ModelConfig } from '../types/sdk/agent';
 import type { AgentMessage } from '../types/sdk/message';
 export interface AgentRuntimeConfig {
     name: string;
@@ -18,6 +18,7 @@ export interface AgentRuntimeConfig {
         structured: boolean;
         schema?: z.ZodObject<z.ZodRawShape>;
         scope?: 'resource' | 'thread';
+        instruction?: string;
     };
     semanticRecall?: SemanticRecallConfig;
     structuredOutput?: z.ZodType;
@@ -60,13 +61,13 @@ export declare class AgentRuntime {
     private runStreamLoop;
     private saveToMemory;
     private saveEmbeddingsForMessages;
-    private extractAndPersistWorkingMemory;
     private buildThinkingProviderOptions;
     private buildCallProviderOptions;
     private iterateToolCallsConcurrent;
     private iteratePendingToolCallsConcurrent;
     private processToolCall;
     private buildLoopContext;
+    private buildWorkingMemoryToolForRun;
     private persistSuspension;
     private cleanupRun;
     private emitTurnEnd;

package/dist/runtime/agent-runtime.js

@@ -43,7 +43,6 @@ const zod_to_json_schema_1 = require("zod-to-json-schema");
 const catalog_1 = require("../sdk/catalog");
 const message_1 = require("../sdk/message");
 const event_bus_1 = require("./event-bus");
-const logger_1 = require("./logger");
 const memory_store_1 = require("./memory-store");
 const message_list_1 = require("./message-list");
 const messages_1 = require("./messages");
@@ -56,14 +55,6 @@ const title_generation_1 = require("./title-generation");
 const tool_adapter_1 = require("./tool-adapter");
 const working_memory_1 = require("./working-memory");
 const zod_1 = require("../utils/zod");
-const logger = (0, logger_1.createFilteredLogger)();
-function isTextPart(part) {
-    return (typeof part === 'object' &&
-        part !== null &&
-        'type' in part &&
-        part.type === 'text' &&
-        'text' in part);
-}
 const MAX_LOOP_ITERATIONS = 20;
 const ajv = new ajv_1.default({ strict: false });
 const EMPTY_MESSAGE_LIST = {
@@ -199,7 +190,7 @@ class AgentRuntime {
                 limit: this.config.lastMessages ?? 10,
             });
             if (memMessages.length > 0) {
-                list.addHistory((0, strip_orphaned_tool_messages_1.stripOrphanedToolMessages)(memMessages.map(message_1.toDbMessage)));
+                list.addHistory((0, strip_orphaned_tool_messages_1.stripOrphanedToolMessages)(memMessages));
             }
         }
         if (this.config.semanticRecall && options?.persistence?.threadId) {
@@ -240,7 +231,7 @@ class AgentRuntime {
                 }
                 else {
                     recalled = allMsgs.filter((m) => {
-                        const id = 'id' in m && typeof m.id === 'string' ? m.id : undefined;
+                        const id = m.id;
                         return id !== undefined && hitIds.has(id);
                     });
                 }
@@ -258,12 +249,10 @@ class AgentRuntime {
             return;
         const { historyIds } = list.serialize();
         const historyIdSet = new Set(historyIds);
-        const newRecalled = recalled
-            .filter((m) => {
-            const id = 'id' in m && typeof m.id === 'string' ? m.id : undefined;
+        const newRecalled = recalled.filter((m) => {
+            const id = m.id;
             return !id || !historyIdSet.has(id);
-        })
-            .map(message_1.toDbMessage);
+        });
         if (newRecalled.length > 0) {
             list.addHistory(newRecalled);
         }
@@ -344,7 +333,7 @@ class AgentRuntime {
         };
     }
     async runGenerateLoop(list, options, pendingResume, runId) {
-        const { model, toolMap, aiTools, providerOptions, hasTools, outputSpec } = this.buildLoopContext(options);
+        const { model, toolMap, aiTools, providerOptions, hasTools, outputSpec } = this.buildLoopContext({ ...options, persistence: options?.persistence });
         let totalUsage;
         let lastFinishReason = 'stop';
         let structuredOutput;
@@ -435,15 +424,6 @@ class AgentRuntime {
         if (lastFinishReason === 'tool-calls') {
             throw new Error(`Agent loop exceeded ${maxIterations} iterations without reaching a stop condition`);
         }
-        if (this.config.workingMemory &&
-            this.config.memory?.saveWorkingMemory &&
-            options?.persistence) {
-            this.extractAndPersistWorkingMemory(list, {
-                threadId: options.persistence.threadId,
-                resourceId: options.persistence.resourceId,
-                scope: this.config.workingMemory?.scope ?? 'resource',
-            });
-        }
         await this.saveToMemory(list, options);
         await this.flushTelemetry(options);
         if (this.config.titleGeneration && options?.persistence?.threadId && this.config.memory) {
@@ -484,20 +464,9 @@ class AgentRuntime {
         return readable;
     }
     async runStreamLoop(list, options, writer, pendingResume, runId) {
-        const { model, toolMap, aiTools, providerOptions, hasTools, outputSpec } = this.buildLoopContext(options);
-        const wmParamsStream = this.resolveWorkingMemoryParams(options?.persistence);
-        const wmFilter = wmParamsStream?.persistFn
-            ? new working_memory_1.WorkingMemoryStreamFilter(writer, async (content) => {
-                await wmParamsStream.persistFn(content);
-            })
-            : undefined;
+        const { model, toolMap, aiTools, providerOptions, hasTools, outputSpec } = this.buildLoopContext({ ...options, persistence: options?.persistence });
         const writeChunk = async (chunk) => {
-            if (wmFilter) {
-                await wmFilter.write(chunk);
-            }
-            else {
-                await writer.write(chunk);
-            }
+            await writer.write(chunk);
         };
         let totalUsage;
         let lastFinishReason = 'stop';
@@ -507,8 +476,6 @@ class AgentRuntime {
         const closeStreamWithError = async (error, status) => {
             await this.cleanupRun(runId);
             this.updateState({ status });
-            if (wmFilter)
-                await wmFilter.flush();
             await writer.write({ type: 'error', error });
             await writer.write({ type: 'finish', finishReason: 'error' });
             await writer.close();
@@ -663,8 +630,6 @@ class AgentRuntime {
             }
             this.emitTurnEnd(newMessages, (0, runtime_helpers_1.extractToolResults)(list.responseDelta()));
         }
-        if (wmFilter)
-            await wmFilter.flush();
         const costUsage = this.applyCost(totalUsage);
         const parentCost = costUsage?.cost ?? 0;
         const subCost = collectedSubAgentUsage.reduce((sum, s) => sum + (s.usage.cost ?? 0), 0);
@@ -680,15 +645,6 @@ class AgentRuntime {
             }),
         });
         try {
-            if (this.config.workingMemory &&
-                this.config.memory?.saveWorkingMemory &&
-                options?.persistence) {
-                this.extractAndPersistWorkingMemory(list, {
-                    threadId: options.persistence.threadId,
-                    resourceId: options.persistence.resourceId,
-                    scope: this.config.workingMemory?.scope ?? 'resource',
-                });
-            }
             await this.saveToMemory(list, options);
             if (this.config.titleGeneration && options?.persistence && this.config.memory) {
                 void (0, title_generation_1.generateThreadTitle)({
@@ -756,35 +712,6 @@ class AgentRuntime {
             })),
         });
     }
-    extractAndPersistWorkingMemory(list, params) {
-        const delta = list.responseDelta();
-        for (let i = delta.length - 1; i >= 0; i--) {
-            const msg = delta[i];
-            if (!(0, message_1.isLlmMessage)(msg) || msg.role !== 'assistant')
-                continue;
-            for (const part of msg.content) {
-                if (!isTextPart(part))
-                    continue;
-                const { cleanText, workingMemory } = (0, working_memory_1.parseWorkingMemory)(part.text);
-                if (workingMemory !== null) {
-                    if (this.config.workingMemory?.structured && this.config.workingMemory.schema) {
-                        try {
-                            this.config.workingMemory.schema.parse(JSON.parse(workingMemory));
-                        }
-                        catch {
-                            part.text = cleanText;
-                            return;
-                        }
-                    }
-                    part.text = cleanText;
-                    this.config.memory.saveWorkingMemory(params, workingMemory).catch((error) => {
-                        logger.warn('Failed to persist working memory', { error });
-                    });
-                }
-                return;
-            }
-        }
-    }
     buildThinkingProviderOptions() {
         if (!this.config.thinking)
             return undefined;
@@ -1109,10 +1036,8 @@ class AgentRuntime {
             : actualResult;
         const toolResultMsg = (0, runtime_helpers_1.makeToolResultMessage)(toolCallId, toolName, modelResult);
         list.addResponse([toolResultMsg]);
-        const customToolMessage = builtTool?.toMessage?.(actualResult);
-        let customMessage;
-        if (customToolMessage) {
-            customMessage = (0, message_1.toDbMessage)(customToolMessage);
+        const customMessage = builtTool?.toMessage?.(actualResult);
+        if (customMessage) {
             list.addResponse([customMessage]);
         }
         return {
@@ -1129,12 +1054,16 @@ class AgentRuntime {
         };
     }
     buildLoopContext(execOptions) {
-        const aiTools = (0, tool_adapter_1.toAiSdkTools)(this.config.tools);
+        const wmTool = this.buildWorkingMemoryToolForRun(execOptions?.persistence);
+        const allUserTools = wmTool
+            ? [...(this.config.tools ?? []), wmTool]
+            : (this.config.tools ?? []);
+        const aiTools = (0, tool_adapter_1.toAiSdkTools)(allUserTools);
         const aiProviderTools = (0, tool_adapter_1.toAiSdkProviderTools)(this.config.providerTools);
         const allTools = { ...aiTools, ...aiProviderTools };
         return {
             model: (0, model_factory_1.createModel)(this.config.model),
-            toolMap: (0, tool_adapter_1.buildToolMap)(this.config.tools),
+            toolMap: (0, tool_adapter_1.buildToolMap)(allUserTools),
             aiTools: allTools,
             providerOptions: this.buildCallProviderOptions(execOptions?.providerOptions),
             hasTools: Object.keys(allTools).length > 0,
@@ -1143,6 +1072,16 @@ class AgentRuntime {
                 : undefined,
         };
     }
+    buildWorkingMemoryToolForRun(persistence) {
+        const wmParams = this.resolveWorkingMemoryParams(persistence);
+        if (!wmParams)
+            return undefined;
+        return (0, working_memory_1.buildWorkingMemoryTool)({
+            structured: wmParams.structured,
+            schema: wmParams.schema,
+            persist: wmParams.persistFn,
+        });
+    }
     async persistSuspension(pendingToolCalls, options, list, totalUsage, existingRunId) {
         const runId = existingRunId ?? (0, run_state_1.generateRunId)();
         const executionOptions = options?.maxIterations !== undefined ? { maxIterations: options.maxIterations } : undefined;
@@ -1211,6 +1150,7 @@ class AgentRuntime {
             template: wmParams.template,
             structured: wmParams.structured,
             state: wmState,
+            ...(wmParams.instruction !== undefined && { instruction: wmParams.instruction }),
         };
     }
     resolveWorkingMemoryParams(options) {
@@ -1239,6 +1179,7 @@ class AgentRuntime {
             template: this.config.workingMemory.template,
             structured: this.config.workingMemory.structured,
             schema: this.config.workingMemory.schema,
+            instruction: this.config.workingMemory.instruction,
         };
     }
 }