@n8n/agents 0.1.0

package/dist/runtime/stream.d.ts

@@ -0,0 +1,15 @@
+import type { TextStreamPart, ToolSet } from 'ai';
+import type { StreamChunk, TokenUsage } from '../types';
+export declare function toTokenUsage(usage: {
+    inputTokens?: number;
+    outputTokens?: number;
+    totalTokens?: number;
+    inputTokenDetails?: {
+        cacheReadTokens?: number;
+        cacheWriteTokens?: number;
+    };
+    outputTokenDetails?: {
+        reasoningTokens?: number;
+    };
+} | undefined): TokenUsage | undefined;
+export declare function convertChunk(c: TextStreamPart<ToolSet>): StreamChunk | undefined;

package/dist/runtime/stream.js

@@ -0,0 +1,94 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.toTokenUsage = toTokenUsage;
+exports.convertChunk = convertChunk;
+function toTokenUsage(usage) {
+    if (!usage)
+        return undefined;
+    const result = {
+        promptTokens: usage.inputTokens ?? 0,
+        completionTokens: usage.outputTokens ?? 0,
+        totalTokens: usage.totalTokens ?? (usage.inputTokens ?? 0) + (usage.outputTokens ?? 0),
+    };
+    const cacheRead = usage.inputTokenDetails?.cacheReadTokens;
+    const cacheWrite = usage.inputTokenDetails?.cacheWriteTokens;
+    if (cacheRead || cacheWrite) {
+        result.inputTokenDetails = {
+            ...(cacheRead && { cacheRead }),
+            ...(cacheWrite && { cacheWrite }),
+        };
+    }
+    if (usage.outputTokenDetails?.reasoningTokens !== undefined) {
+        result.outputTokenDetails = { reasoning: usage.outputTokenDetails.reasoningTokens };
+    }
+    return result;
+}
+function convertChunk(c) {
+    switch (c.type) {
+        case 'text-delta':
+            return { type: 'text-delta', delta: c.text ?? '' };
+        case 'reasoning-delta':
+            return { type: 'reasoning-delta', delta: c.text ?? '' };
+        case 'tool-call':
+            return {
+                type: 'message',
+                message: {
+                    role: 'tool',
+                    content: [
+                        {
+                            type: 'tool-call',
+                            toolCallId: c.toolCallId,
+                            toolName: c.toolName ?? '',
+                            input: c.input,
+                        },
+                    ],
+                },
+            };
+        case 'tool-input-start':
+            return {
+                type: 'tool-call-delta',
+                name: c.toolName,
+            };
+        case 'tool-input-delta':
+            return {
+                type: 'tool-call-delta',
+                ...(c.delta !== undefined && { argumentsDelta: c.delta }),
+            };
+        case 'tool-result':
+            return {
+                type: 'message',
+                message: {
+                    role: 'tool',
+                    content: [
+                        {
+                            type: 'tool-result',
+                            toolCallId: c.toolCallId ?? '',
+                            toolName: c.toolName ?? '',
+                            result: c.output && 'value' in c.output ? c.output.value : null,
+                        },
+                    ],
+                },
+            };
+        case 'error':
+            return { type: 'error', error: c.error };
+        case 'finish-step': {
+            const usage = toTokenUsage(c.usage);
+            return {
+                type: 'finish',
+                finishReason: (c.finishReason ?? 'stop'),
+                ...(usage && { usage }),
+            };
+        }
+        case 'finish': {
+            const usage = toTokenUsage(c.totalUsage);
+            return {
+                type: 'finish',
+                finishReason: (c.finishReason ?? 'stop'),
+                ...(usage && { usage }),
+            };
+        }
+        default:
+            return undefined;
+    }
+}
+//# sourceMappingURL=stream.js.map

package/dist/runtime/stream.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"stream.js","sourceRoot":"","sources":["../../src/runtime/stream.ts"],"names":[],"mappings":";;AAMA,oCAiCC;AAGD,oCA6EC;AAjHD,SAAgB,YAAY,CAC3B,KAQY;IAEZ,IAAI,CAAC,KAAK;QAAE,OAAO,SAAS,CAAC;IAE7B,MAAM,MAAM,GAAe;QAC1B,YAAY,EAAE,KAAK,CAAC,WAAW,IAAI,CAAC;QACpC,gBAAgB,EAAE,KAAK,CAAC,YAAY,IAAI,CAAC;QACzC,WAAW,EAAE,KAAK,CAAC,WAAW,IAAI,CAAC,KAAK,CAAC,WAAW,IAAI,CAAC,CAAC,GAAG,CAAC,KAAK,CAAC,YAAY,IAAI,CAAC,CAAC;KACtF,CAAC;IAEF,MAAM,SAAS,GAAG,KAAK,CAAC,iBAAiB,EAAE,eAAe,CAAC;IAC3D,MAAM,UAAU,GAAG,KAAK,CAAC,iBAAiB,EAAE,gBAAgB,CAAC;IAC7D,IAAI,SAAS,IAAI,UAAU,EAAE,CAAC;QAC7B,MAAM,CAAC,iBAAiB,GAAG;YAC1B,GAAG,CAAC,SAAS,IAAI,EAAE,SAAS,EAAE,CAAC;YAC/B,GAAG,CAAC,UAAU,IAAI,EAAE,UAAU,EAAE,CAAC;SACjC,CAAC;IACH,CAAC;IAED,IAAI,KAAK,CAAC,kBAAkB,EAAE,eAAe,KAAK,SAAS,EAAE,CAAC;QAC7D,MAAM,CAAC,kBAAkB,GAAG,EAAE,SAAS,EAAE,KAAK,CAAC,kBAAkB,CAAC,eAAe,EAAE,CAAC;IACrF,CAAC;IAED,OAAO,MAAM,CAAC;AACf,CAAC;AAGD,SAAgB,YAAY,CAAC,CAA0B;IACtD,QAAQ,CAAC,CAAC,IAAI,EAAE,CAAC;QAChB,KAAK,YAAY;YAChB,OAAO,EAAE,IAAI,EAAE,YAAY,EAAE,KAAK,EAAE,CAAC,CAAC,IAAI,IAAI,EAAE,EAAE,CAAC;QAEpD,KAAK,iBAAiB;YACrB,OAAO,EAAE,IAAI,EAAE,iBAAiB,EAAE,KAAK,EAAE,CAAC,CAAC,IAAI,IAAI,EAAE,EAAE,CAAC;QAEzD,KAAK,WAAW;YACf,OAAO;gBACN,IAAI,EAAE,SAAS;gBACf,OAAO,EAAE;oBACR,IAAI,EAAE,MAAM;oBACZ,OAAO,EAAE;wBACR;4BACC,IAAI,EAAE,WAAW;4BACjB,UAAU,EAAE,CAAC,CAAC,UAAU;4BACxB,QAAQ,EAAE,CAAC,CAAC,QAAQ,IAAI,EAAE;4BAC1B,KAAK,EAAE,CAAC,CAAC,KAAkB;yBAC3B;qBACD;iBACD;aACD,CAAC;QAEH,KAAK,kBAAkB;YACtB,OAAO;gBACN,IAAI,EAAE,iBAAiB;gBACvB,IAAI,EAAE,CAAC,CAAC,QAAQ;aAChB,CAAC;QAEH,KAAK,kBAAkB;YACtB,OAAO;gBACN,IAAI,EAAE,iBAAiB;gBACvB,GAAG,CAAC,CAAC,CAAC,KAAK,KAAK,SAAS,IAAI,EAAE,cAAc,EAAE,CAAC,CAAC,KAAK,EAAE,CAAC;aACzD,CAAC;QAEH,KAAK,aAAa;YACjB,OAAO;gBACN,IAAI,EAAE,SAAS;gBACf,OAAO,EAAE;oBACR,IAAI,EAAE,MAAM;oBACZ,OAAO,EAAE;wBACR;4BACC,IAAI,EAAE,aAAa;4BACnB,UAAU,EAAE,CAAC,CAAC,UAAU,IAAI,EAAE;4BAC9B,QAAQ,EAAE,CAAC,CAAC,QAAQ,IAAI,EAAE;4BAE1B,MAAM,EAAE,CAAC,CAAC,MAAM,IAAI,OAAO,IAAI,CAAC,CAAC,MAAM,CAAC,CAAC,CAAE,CAAC,CAAC,MAAM,CAAC,KAAmB,CAAC,CAAC,CAAC,IAAI;yBAC9E;qBACD;iBACD;aACD,CAAC;QAEH,KAAK,OAAO;YACX,OAAO,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC,CAAC,KAAK,EAAE,CAAC;QAE1C,KAAK,aAAa,CAAC,CAAC,CAAC;YACpB,MAAM,KAAK,GAAG,YAAY,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC;YACpC,OAAO;gBACN,IAAI,EAAE,QAAQ;gBACd,YAAY,EAAE,CAAC,CAAC,CAAC,YAAY,IAAI,MAAM,CAAiB;gBACxD,GAAG,CAAC,KAAK,IAAI,EAAE,KAAK,EAAE,CAAC;aACvB,CAAC;QACH,CAAC;QAED,KAAK,QAAQ,CAAC,CAAC,CAAC;YACf,MAAM,KAAK,GAAG,YAAY,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC;YACzC,OAAO;gBACN,IAAI,EAAE,QAAQ;gBACd,YAAY,EAAE,CAAC,CAAC,CAAC,YAAY,IAAI,MAAM,CAAiB;gBACxD,GAAG,CAAC,KAAK,IAAI,EAAE,KAAK,EAAE,CAAC;aACvB,CAAC;QACH,CAAC;QAED;YACC,OAAO,SAAS,CAAC;IACnB,CAAC;AACF,CAAC"}

package/dist/runtime/strip-orphaned-tool-messages.d.ts

@@ -0,0 +1,2 @@
+import type { AgentDbMessage } from '../types/sdk/message';
+export declare function stripOrphanedToolMessages(messages: AgentDbMessage[]): AgentDbMessage[];

package/dist/runtime/strip-orphaned-tool-messages.js

@@ -0,0 +1,48 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.stripOrphanedToolMessages = stripOrphanedToolMessages;
+const message_1 = require("../sdk/message");
+function stripOrphanedToolMessages(messages) {
+    const callIds = new Set();
+    const resultIds = new Set();
+    for (const msg of messages) {
+        if (!(0, message_1.isLlmMessage)(msg))
+            continue;
+        for (const block of msg.content) {
+            if (block.type === 'tool-call' && block.toolCallId) {
+                callIds.add(block.toolCallId);
+            }
+            else if (block.type === 'tool-result' && block.toolCallId) {
+                resultIds.add(block.toolCallId);
+            }
+        }
+    }
+    const orphanedCallIds = new Set([...callIds].filter((id) => !resultIds.has(id)));
+    const orphanedResultIds = new Set([...resultIds].filter((id) => !callIds.has(id)));
+    if (orphanedCallIds.size === 0 && orphanedResultIds.size === 0) {
+        return messages;
+    }
+    const result = [];
+    for (const msg of messages) {
+        if (!(0, message_1.isLlmMessage)(msg)) {
+            result.push(msg);
+            continue;
+        }
+        const filtered = msg.content.filter((block) => {
+            if (block.type === 'tool-call' && block.toolCallId && orphanedCallIds.has(block.toolCallId)) {
+                return false;
+            }
+            if (block.type === 'tool-result' &&
+                block.toolCallId &&
+                orphanedResultIds.has(block.toolCallId)) {
+                return false;
+            }
+            return true;
+        });
+        if (filtered.length === 0)
+            continue;
+        result.push({ ...msg, content: filtered });
+    }
+    return result;
+}
+//# sourceMappingURL=strip-orphaned-tool-messages.js.map

package/dist/runtime/strip-orphaned-tool-messages.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"strip-orphaned-tool-messages.js","sourceRoot":"","sources":["../../src/runtime/strip-orphaned-tool-messages.ts"],"names":[],"mappings":";;AAmBA,8DAkDC;AArED,4CAA8C;AAmB9C,SAAgB,yBAAyB,CAAC,QAA0B;IACnE,MAAM,OAAO,GAAG,IAAI,GAAG,EAAU,CAAC;IAClC,MAAM,SAAS,GAAG,IAAI,GAAG,EAAU,CAAC;IAEpC,KAAK,MAAM,GAAG,IAAI,QAAQ,EAAE,CAAC;QAC5B,IAAI,CAAC,IAAA,sBAAY,EAAC,GAAG,CAAC;YAAE,SAAS;QACjC,KAAK,MAAM,KAAK,IAAI,GAAG,CAAC,OAAO,EAAE,CAAC;YACjC,IAAI,KAAK,CAAC,IAAI,KAAK,WAAW,IAAI,KAAK,CAAC,UAAU,EAAE,CAAC;gBACpD,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC;YAC/B,CAAC;iBAAM,IAAI,KAAK,CAAC,IAAI,KAAK,aAAa,IAAI,KAAK,CAAC,UAAU,EAAE,CAAC;gBAC7D,SAAS,CAAC,GAAG,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC;YACjC,CAAC;QACF,CAAC;IACF,CAAC;IAED,MAAM,eAAe,GAAG,IAAI,GAAG,CAAC,CAAC,GAAG,OAAO,CAAC,CAAC,MAAM,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC,SAAS,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;IACjF,MAAM,iBAAiB,GAAG,IAAI,GAAG,CAAC,CAAC,GAAG,SAAS,CAAC,CAAC,MAAM,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC,OAAO,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;IAEnF,IAAI,eAAe,CAAC,IAAI,KAAK,CAAC,IAAI,iBAAiB,CAAC,IAAI,KAAK,CAAC,EAAE,CAAC;QAChE,OAAO,QAAQ,CAAC;IACjB,CAAC;IAED,MAAM,MAAM,GAAqB,EAAE,CAAC;IAEpC,KAAK,MAAM,GAAG,IAAI,QAAQ,EAAE,CAAC;QAC5B,IAAI,CAAC,IAAA,sBAAY,EAAC,GAAG,CAAC,EAAE,CAAC;YACxB,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YACjB,SAAS;QACV,CAAC;QAED,MAAM,QAAQ,GAAG,GAAG,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,KAAqB,EAAE,EAAE;YAC7D,IAAI,KAAK,CAAC,IAAI,KAAK,WAAW,IAAI,KAAK,CAAC,UAAU,IAAI,eAAe,CAAC,GAAG,CAAC,KAAK,CAAC,UAAU,CAAC,EAAE,CAAC;gBAC7F,OAAO,KAAK,CAAC;YACd,CAAC;YACD,IACC,KAAK,CAAC,IAAI,KAAK,aAAa;gBAC5B,KAAK,CAAC,UAAU;gBAChB,iBAAiB,CAAC,GAAG,CAAC,KAAK,CAAC,UAAU,CAAC,EACtC,CAAC;gBACF,OAAO,KAAK,CAAC;YACd,CAAC;YACD,OAAO,IAAI,CAAC;QACb,CAAC,CAAC,CAAC;QAEH,IAAI,QAAQ,CAAC,MAAM,KAAK,CAAC;YAAE,SAAS;QAEpC,MAAM,CAAC,IAAI,CAAC,EAAE,GAAG,GAAG,EAAE,OAAO,EAAE,QAAQ,EAAE,CAAC,CAAC;IAC5C,CAAC;IAED,OAAO,MAAM,CAAC;AACf,CAAC"}

package/dist/runtime/title-generation.d.ts

@@ -0,0 +1,11 @@
+import type { BuiltMemory, TitleGenerationConfig } from '../types';
+import type { ModelConfig } from '../types/sdk/agent';
+import type { AgentDbMessage } from '../types/sdk/message';
+export declare function generateThreadTitle(opts: {
+    memory: BuiltMemory;
+    threadId: string;
+    resourceId: string;
+    titleConfig: TitleGenerationConfig;
+    agentModel: ModelConfig;
+    turnDelta: AgentDbMessage[];
+}): Promise<void>;

package/dist/runtime/title-generation.js

@@ -0,0 +1,56 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.generateThreadTitle = generateThreadTitle;
+const ai_1 = require("ai");
+const logger_1 = require("./logger");
+const model_factory_1 = require("./model-factory");
+const logger = (0, logger_1.createFilteredLogger)();
+const DEFAULT_TITLE_INSTRUCTIONS = [
+    '- you will generate a short title based on the first message a user begins a conversation with',
+    '- ensure it is not more than 80 characters long',
+    "- the title should be a summary of the user's message",
+    '- do not use quotes or colons',
+    '- the entire text you return will be used as the title',
+].join('\n');
+async function generateThreadTitle(opts) {
+    try {
+        const thread = await opts.memory.getThread(opts.threadId);
+        if (thread?.title)
+            return;
+        const userMessage = opts.turnDelta.find((m) => 'role' in m && m.role === 'user');
+        if (!userMessage || !('content' in userMessage))
+            return;
+        const userText = userMessage.content
+            .filter((c) => c.type === 'text' && c.text)
+            .map((c) => c.text)
+            .join(' ');
+        if (!userText)
+            return;
+        const titleModelId = opts.titleConfig.model ?? opts.agentModel;
+        const titleModel = (0, model_factory_1.createModel)(titleModelId);
+        const instructions = opts.titleConfig.instructions ?? DEFAULT_TITLE_INSTRUCTIONS;
+        const result = await (0, ai_1.generateText)({
+            model: titleModel,
+            messages: [
+                { role: 'system', content: instructions },
+                { role: 'user', content: userText },
+            ],
+        });
+        let title = result.text?.trim();
+        if (!title)
+            return;
+        title = title.replace(/<think>[\s\S]*?<\/think>/g, '').trim();
+        if (!title)
+            return;
+        await opts.memory.saveThread({
+            id: opts.threadId,
+            resourceId: opts.resourceId,
+            title,
+            metadata: thread?.metadata,
+        });
+    }
+    catch (error) {
+        logger.warn('Failed to generate thread title', { error });
+    }
+}
+//# sourceMappingURL=title-generation.js.map

package/dist/runtime/title-generation.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"title-generation.js","sourceRoot":"","sources":["../../src/runtime/title-generation.ts"],"names":[],"mappings":";;AAyBA,kDAmDC;AA5ED,2BAAkC;AAGlC,qCAAgD;AAChD,mDAA8C;AAI9C,MAAM,MAAM,GAAG,IAAA,6BAAoB,GAAE,CAAC;AAEtC,MAAM,0BAA0B,GAAG;IAClC,gGAAgG;IAChG,iDAAiD;IACjD,uDAAuD;IACvD,+BAA+B;IAC/B,wDAAwD;CACxD,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AASN,KAAK,UAAU,mBAAmB,CAAC,IASzC;IACA,IAAI,CAAC;QACJ,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAC1D,IAAI,MAAM,EAAE,KAAK;YAAE,OAAO;QAE1B,MAAM,WAAW,GAAG,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,MAAM,IAAI,CAAC,IAAI,CAAC,CAAC,IAAI,KAAK,MAAM,CAAC,CAAC;QACjF,IAAI,CAAC,WAAW,IAAI,CAAC,CAAC,SAAS,IAAI,WAAW,CAAC;YAAE,OAAO;QAExD,MAAM,QAAQ,GAAI,WAAW,CAAC,OAAkD;aAC9E,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,MAAM,IAAI,CAAC,CAAC,IAAI,CAAC;aAC1C,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAK,CAAC;aACnB,IAAI,CAAC,GAAG,CAAC,CAAC;QACZ,IAAI,CAAC,QAAQ;YAAE,OAAO;QAEtB,MAAM,YAAY,GAAG,IAAI,CAAC,WAAW,CAAC,KAAK,IAAI,IAAI,CAAC,UAAU,CAAC;QAC/D,MAAM,UAAU,GAAG,IAAA,2BAAW,EAAC,YAAY,CAAC,CAAC;QAC7C,MAAM,YAAY,GAAG,IAAI,CAAC,WAAW,CAAC,YAAY,IAAI,0BAA0B,CAAC;QAEjF,MAAM,MAAM,GAAG,MAAM,IAAA,iBAAY,EAAC;YACjC,KAAK,EAAE,UAAU;YACjB,QAAQ,EAAE;gBACT,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,YAAY,EAAE;gBACzC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,QAAQ,EAAE;aACnC;SACD,CAAC,CAAC;QAEH,IAAI,KAAK,GAAG,MAAM,CAAC,IAAI,EAAE,IAAI,EAAE,CAAC;QAChC,IAAI,CAAC,KAAK;YAAE,OAAO;QAGnB,KAAK,GAAG,KAAK,CAAC,OAAO,CAAC,2BAA2B,EAAE,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC;QAC9D,IAAI,CAAC,KAAK;YAAE,OAAO;QAEnB,MAAM,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC;YAC5B,EAAE,EAAE,IAAI,CAAC,QAAQ;YACjB,UAAU,EAAE,IAAI,CAAC,UAAU;YAC3B,KAAK;YACL,QAAQ,EAAE,MAAM,EAAE,QAAQ;SAC1B,CAAC,CAAC;IACJ,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QAChB,MAAM,CAAC,IAAI,CAAC,iCAAiC,EAAE,EAAE,KAAK,EAAE,CAAC,CAAC;IAC3D,CAAC;AACF,CAAC"}

package/dist/runtime/tool-adapter.d.ts

@@ -0,0 +1,23 @@
+import { type Tool as AiSdkTool } from 'ai';
+import { type BuiltProviderTool, type BuiltTool, type BuiltTelemetry } from '../types';
+import type { SubAgentUsage } from '../types/sdk/agent';
+declare const SUSPEND_BRAND: unique symbol;
+declare const AGENT_TOOL_BRAND: unique symbol;
+export interface SuspendedToolResult {
+    readonly [SUSPEND_BRAND]: true;
+    payload: unknown;
+}
+export declare function isSuspendedToolResult(value: unknown): value is SuspendedToolResult;
+export interface AgentToolResult {
+    readonly [AGENT_TOOL_BRAND]: true;
+    readonly output: unknown;
+    readonly subAgentUsage: SubAgentUsage[];
+}
+export declare function isAgentToolResult(value: unknown): value is AgentToolResult;
+export declare function createAgentToolResult(output: unknown, subAgentUsage: SubAgentUsage[]): never;
+export declare function toAiSdkProviderTools(tools?: BuiltProviderTool[]): Record<string, AiSdkTool>;
+export declare function toAiSdkTools(tools?: BuiltTool[]): Record<string, AiSdkTool>;
+export declare function executeTool(args: unknown, builtTool: BuiltTool, resumeData?: unknown, parentTelemetry?: BuiltTelemetry): Promise<unknown>;
+export declare function isInterruptible(toolName: string, toolMap: Map<string, BuiltTool>): boolean;
+export declare function buildToolMap(tools?: BuiltTool[]): Map<string, BuiltTool>;
+export {};

package/dist/runtime/tool-adapter.js

@@ -0,0 +1,104 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.isSuspendedToolResult = isSuspendedToolResult;
+exports.isAgentToolResult = isAgentToolResult;
+exports.createAgentToolResult = createAgentToolResult;
+exports.toAiSdkProviderTools = toAiSdkProviderTools;
+exports.toAiSdkTools = toAiSdkTools;
+exports.executeTool = executeTool;
+exports.isInterruptible = isInterruptible;
+exports.buildToolMap = buildToolMap;
+const ai_1 = require("ai");
+const zod_1 = require("zod");
+const zod_2 = require("../utils/zod");
+const SUSPEND_BRAND = Symbol('SuspendBrand');
+const AGENT_TOOL_BRAND = Symbol('AgentToolBrand');
+function isSuspendedToolResult(value) {
+    return typeof value === 'object' && value !== null && SUSPEND_BRAND in value;
+}
+function isAgentToolResult(value) {
+    return typeof value === 'object' && value !== null && AGENT_TOOL_BRAND in value;
+}
+function createAgentToolResult(output, subAgentUsage) {
+    const base = typeof output === 'object' && output !== null ? output : {};
+    return { ...base, [AGENT_TOOL_BRAND]: true, output, subAgentUsage };
+}
+function toAiSdkProviderTools(tools) {
+    if (!tools || tools.length === 0)
+        return {};
+    const result = {};
+    for (const t of tools) {
+        const providerTool = {
+            type: 'provider',
+            id: t.name,
+            args: t.args,
+            inputSchema: t.inputSchema ?? zod_1.z.any(),
+        };
+        result[t.name] = providerTool;
+    }
+    return result;
+}
+const fixSchema = (schema) => {
+    if (typeof schema === 'object' &&
+        schema !== null &&
+        'properties' in schema &&
+        !('type' in schema)) {
+        return { ...schema, type: 'object' };
+    }
+    return schema;
+};
+function toAiSdkTools(tools) {
+    if (!tools || tools.length === 0)
+        return {};
+    const result = {};
+    for (const t of tools) {
+        if (t.inputSchema) {
+            if ((0, zod_2.isZodSchema)(t.inputSchema)) {
+                result[t.name] = (0, ai_1.tool)({
+                    description: t.description,
+                    inputSchema: t.inputSchema,
+                    providerOptions: t.providerOptions,
+                });
+            }
+            else {
+                result[t.name] = (0, ai_1.tool)({
+                    description: t.description,
+                    inputSchema: (0, ai_1.jsonSchema)(fixSchema(t.inputSchema)),
+                    providerOptions: t.providerOptions,
+                });
+            }
+        }
+    }
+    return result;
+}
+async function executeTool(args, builtTool, resumeData, parentTelemetry) {
+    if (!builtTool.handler) {
+        throw new Error(`No handler found for tool "${builtTool.name}"`);
+    }
+    if (builtTool.suspendSchema) {
+        const ctx = {
+            suspend: async (payload) => {
+                return await Promise.resolve({ [SUSPEND_BRAND]: true, payload });
+            },
+            resumeData,
+            parentTelemetry,
+        };
+        return await builtTool.handler(args, ctx);
+    }
+    const ctx = { parentTelemetry };
+    return await builtTool.handler(args, ctx);
+}
+function isInterruptible(toolName, toolMap) {
+    const builtTool = toolMap.get(toolName);
+    return !!builtTool?.suspendSchema;
+}
+function buildToolMap(tools) {
+    const map = new Map();
+    if (tools) {
+        for (const t of tools) {
+            map.set(t.name, t);
+        }
+    }
+    return map;
+}
+//# sourceMappingURL=tool-adapter.js.map

package/dist/runtime/tool-adapter.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"tool-adapter.js","sourceRoot":"","sources":["../../src/runtime/tool-adapter.ts"],"names":[],"mappings":";;AAqCA,sDAEC;AAWD,8CAEC;AAUD,sDAGC;AAWD,oDAcC;AAoBD,oCAsBC;AAOD,kCAuBC;AAKD,0CAGC;AAGD,oCAQC;AArLD,2BAA8D;AAE9D,6BAAwB;AAUxB,sCAA2C;AAU3C,MAAM,aAAa,GAAG,MAAM,CAAC,cAAc,CAAC,CAAC;AAO7C,MAAM,gBAAgB,GAAG,MAAM,CAAC,gBAAgB,CAAC,CAAC;AAQlD,SAAgB,qBAAqB,CAAC,KAAc;IACnD,OAAO,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,KAAK,IAAI,IAAI,aAAa,IAAI,KAAK,CAAC;AAC9E,CAAC;AAWD,SAAgB,iBAAiB,CAAC,KAAc;IAC/C,OAAO,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,KAAK,IAAI,IAAI,gBAAgB,IAAI,KAAK,CAAC;AACjF,CAAC;AAUD,SAAgB,qBAAqB,CAAC,MAAe,EAAE,aAA8B;IACpF,MAAM,IAAI,GAAG,OAAO,MAAM,KAAK,QAAQ,IAAI,MAAM,KAAK,IAAI,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC;IACzE,OAAO,EAAE,GAAG,IAAI,EAAE,CAAC,gBAAgB,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,aAAa,EAAW,CAAC;AAC9E,CAAC;AAWD,SAAgB,oBAAoB,CAAC,KAA2B;IAC/D,IAAI,CAAC,KAAK,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC;QAAE,OAAO,EAAE,CAAC;IAE5C,MAAM,MAAM,GAA8B,EAAE,CAAC;IAC7C,KAAK,MAAM,CAAC,IAAI,KAAK,EAAE,CAAC;QACvB,MAAM,YAAY,GAAsB;YACvC,IAAI,EAAE,UAAU;YAChB,EAAE,EAAE,CAAC,CAAC,IAAI;YACV,IAAI,EAAE,CAAC,CAAC,IAAI;YACZ,WAAW,EAAE,CAAC,CAAC,WAAW,IAAI,OAAC,CAAC,GAAG,EAAE;SACrC,CAAC;QACF,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,YAAY,CAAC;IAC/B,CAAC;IACD,OAAO,MAAM,CAAC;AACf,CAAC;AAED,MAAM,SAAS,GAAG,CAAC,MAAmB,EAAe,EAAE;IAEtD,IACC,OAAO,MAAM,KAAK,QAAQ;QAC1B,MAAM,KAAK,IAAI;QACf,YAAY,IAAI,MAAM;QACtB,CAAC,CAAC,MAAM,IAAI,MAAM,CAAC,EAClB,CAAC;QACF,OAAO,EAAE,GAAG,MAAM,EAAE,IAAI,EAAE,QAAiB,EAAE,CAAC;IAC/C,CAAC;IACD,OAAO,MAAM,CAAC;AACf,CAAC,CAAC;AAOF,SAAgB,YAAY,CAAC,KAAmB;IAC/C,IAAI,CAAC,KAAK,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC;QAAE,OAAO,EAAE,CAAC;IAE5C,MAAM,MAAM,GAA8B,EAAE,CAAC;IAC7C,KAAK,MAAM,CAAC,IAAI,KAAK,EAAE,CAAC;QACvB,IAAI,CAAC,CAAC,WAAW,EAAE,CAAC;YACnB,IAAI,IAAA,iBAAW,EAAC,CAAC,CAAC,WAAW,CAAC,EAAE,CAAC;gBAChC,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,IAAA,SAAI,EAAC;oBACrB,WAAW,EAAE,CAAC,CAAC,WAAW;oBAC1B,WAAW,EAAE,CAAC,CAAC,WAAW;oBAC1B,eAAe,EAAE,CAAC,CAAC,eAAe;iBAClC,CAAC,CAAC;YACJ,CAAC;iBAAM,CAAC;gBACP,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,IAAA,SAAI,EAAC;oBACrB,WAAW,EAAE,CAAC,CAAC,WAAW;oBAC1B,WAAW,EAAE,IAAA,eAAU,EAAC,SAAS,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC;oBACjD,eAAe,EAAE,CAAC,CAAC,eAAe;iBAClC,CAAC,CAAC;YACJ,CAAC;QACF,CAAC;IACF,CAAC;IACD,OAAO,MAAM,CAAC;AACf,CAAC;AAOM,KAAK,UAAU,WAAW,CAChC,IAAa,EACb,SAAoB,EACpB,UAAoB,EACpB,eAAgC;IAEhC,IAAI,CAAC,SAAS,CAAC,OAAO,EAAE,CAAC;QACxB,MAAM,IAAI,KAAK,CAAC,8BAA8B,SAAS,CAAC,IAAI,GAAG,CAAC,CAAC;IAClE,CAAC;IAED,IAAI,SAAS,CAAC,aAAa,EAAE,CAAC;QAC7B,MAAM,GAAG,GAA6B;YACrC,OAAO,EAAE,KAAK,EAAE,OAAgB,EAAkB,EAAE;gBACnD,OAAO,MAAM,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC,aAAa,CAAC,EAAE,IAAI,EAAE,OAAO,EAAW,CAAC,CAAC;YAC3E,CAAC;YACD,UAAU;YACV,eAAe;SACf,CAAC;QACF,OAAO,MAAM,SAAS,CAAC,OAAO,CAAC,IAAI,EAAE,GAAG,CAAC,CAAC;IAC3C,CAAC;IAED,MAAM,GAAG,GAAgB,EAAE,eAAe,EAAE,CAAC;IAC7C,OAAO,MAAM,SAAS,CAAC,OAAO,CAAC,IAAI,EAAE,GAAG,CAAC,CAAC;AAC3C,CAAC;AAKD,SAAgB,eAAe,CAAC,QAAgB,EAAE,OAA+B;IAChF,MAAM,SAAS,GAAG,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;IACxC,OAAO,CAAC,CAAC,SAAS,EAAE,aAAa,CAAC;AACnC,CAAC;AAGD,SAAgB,YAAY,CAAC,KAAmB;IAC/C,MAAM,GAAG,GAAG,IAAI,GAAG,EAAqB,CAAC;IACzC,IAAI,KAAK,EAAE,CAAC;QACX,KAAK,MAAM,CAAC,IAAI,KAAK,EAAE,CAAC;YACvB,GAAG,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;QACpB,CAAC;IACF,CAAC;IACD,OAAO,GAAG,CAAC;AACZ,CAAC"}

package/dist/runtime/working-memory.d.ts

@@ -0,0 +1,21 @@
+import type { z } from 'zod';
+import type { StreamChunk } from '../types';
+type ZodObjectSchema = z.ZodObject<z.ZodRawShape>;
+export declare function parseWorkingMemory(text: string): {
+    cleanText: string;
+    workingMemory: string | null;
+};
+export declare function buildWorkingMemoryInstruction(template: string, structured: boolean): string;
+export declare function templateFromSchema(schema: ZodObjectSchema): string;
+type PersistFn = (content: string) => Promise<void>;
+export declare class WorkingMemoryStreamFilter {
+    private writer;
+    private persist;
+    private state;
+    private buffer;
+    private pendingText;
+    constructor(writer: WritableStreamDefaultWriter<StreamChunk>, persist: PersistFn);
+    write(chunk: StreamChunk): Promise<void>;
+    flush(): Promise<void>;
+}
+export {};

package/dist/runtime/working-memory.js

@@ -0,0 +1,138 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.WorkingMemoryStreamFilter = void 0;
+exports.parseWorkingMemory = parseWorkingMemory;
+exports.buildWorkingMemoryInstruction = buildWorkingMemoryInstruction;
+exports.templateFromSchema = templateFromSchema;
+const logger_1 = require("./logger");
+const logger = (0, logger_1.createFilteredLogger)();
+const OPEN_TAG = '<working_memory>';
+const CLOSE_TAG = '</working_memory>';
+function parseWorkingMemory(text) {
+    const openIdx = text.indexOf(OPEN_TAG);
+    if (openIdx === -1)
+        return { cleanText: text, workingMemory: null };
+    const closeIdx = text.indexOf(CLOSE_TAG, openIdx);
+    if (closeIdx === -1)
+        return { cleanText: text, workingMemory: null };
+    const contentStart = openIdx + OPEN_TAG.length;
+    const rawContent = text.slice(contentStart, closeIdx);
+    const workingMemory = rawContent.replace(/^\n/, '').replace(/\n$/, '');
+    const before = text.slice(0, openIdx).replace(/\n$/, '');
+    const after = text.slice(closeIdx + CLOSE_TAG.length).replace(/^\n/, '');
+    const cleanText = (before + (after ? '\n' + after : '')).trim();
+    return { cleanText, workingMemory };
+}
+function buildWorkingMemoryInstruction(template, structured) {
+    const format = structured
+        ? 'Emit the updated state as valid JSON matching the schema'
+        : 'Update the template with any new information learned';
+    return [
+        '',
+        '## Working Memory',
+        '',
+        'You have persistent working memory that survives across conversations.',
+        'The current state will be shown to you in a system message.',
+        'IMPORTANT: Always respond to the user first with your normal reply.',
+        `Then, at the very end of your response, emit your updated working memory inside ${OPEN_TAG}...${CLOSE_TAG} tags on a new line.`,
+        `${format}. If nothing changed, emit the current state unchanged.`,
+        'The working memory block must be the last thing in your response, after your reply to the user.',
+        '',
+        'Current template:',
+        '```',
+        template,
+        '```',
+    ].join('\n');
+}
+function templateFromSchema(schema) {
+    const obj = {};
+    for (const [key, field] of Object.entries(schema.shape)) {
+        const desc = field.description;
+        obj[key] = desc ?? '';
+    }
+    return JSON.stringify(obj, null, 2);
+}
+class WorkingMemoryStreamFilter {
+    constructor(writer, persist) {
+        this.state = 'normal';
+        this.buffer = '';
+        this.pendingText = '';
+        this.writer = writer;
+        this.persist = persist;
+    }
+    async write(chunk) {
+        if (chunk.type !== 'text-delta') {
+            await this.writer.write(chunk);
+            return;
+        }
+        this.pendingText += chunk.delta;
+        while (this.pendingText.length > 0) {
+            if (this.state === 'normal') {
+                const openIdx = this.pendingText.indexOf(OPEN_TAG);
+                if (openIdx === -1) {
+                    const lastLt = this.pendingText.lastIndexOf('<');
+                    if (lastLt !== -1 &&
+                        this.pendingText.length - lastLt < OPEN_TAG.length &&
+                        OPEN_TAG.startsWith(this.pendingText.slice(lastLt))) {
+                        if (lastLt > 0) {
+                            await this.writer.write({
+                                type: 'text-delta',
+                                delta: this.pendingText.slice(0, lastLt),
+                            });
+                        }
+                        this.pendingText = this.pendingText.slice(lastLt);
+                    }
+                    else {
+                        await this.writer.write({ type: 'text-delta', delta: this.pendingText });
+                        this.pendingText = '';
+                    }
+                    break;
+                }
+                if (openIdx > 0) {
+                    await this.writer.write({
+                        type: 'text-delta',
+                        delta: this.pendingText.slice(0, openIdx),
+                    });
+                }
+                this.state = 'inside';
+                this.pendingText = this.pendingText.slice(openIdx + OPEN_TAG.length);
+                this.buffer = '';
+            }
+            else {
+                const closeIdx = this.pendingText.indexOf(CLOSE_TAG);
+                if (closeIdx === -1) {
+                    const lastLt = this.pendingText.lastIndexOf('<');
+                    if (lastLt !== -1 &&
+                        this.pendingText.length - lastLt < CLOSE_TAG.length &&
+                        CLOSE_TAG.startsWith(this.pendingText.slice(lastLt))) {
+                        this.buffer += this.pendingText.slice(0, lastLt);
+                        this.pendingText = this.pendingText.slice(lastLt);
+                    }
+                    else {
+                        this.buffer += this.pendingText;
+                        this.pendingText = '';
+                    }
+                    break;
+                }
+                this.buffer += this.pendingText.slice(0, closeIdx);
+                this.pendingText = this.pendingText.slice(closeIdx + CLOSE_TAG.length);
+                this.state = 'normal';
+                const content = this.buffer.replace(/^\n/, '').replace(/\n$/, '');
+                this.persist(content).catch((error) => {
+                    logger.warn('Failed to persist working memory', { error });
+                });
+                this.buffer = '';
+            }
+        }
+    }
+    async flush() {
+        if (this.state === 'normal' && this.pendingText.length > 0) {
+            await this.writer.write({ type: 'text-delta', delta: this.pendingText });
+        }
+        this.pendingText = '';
+        this.buffer = '';
+        this.state = 'normal';
+    }
+}
+exports.WorkingMemoryStreamFilter = WorkingMemoryStreamFilter;
+//# sourceMappingURL=working-memory.js.map

package/dist/runtime/working-memory.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"working-memory.js","sourceRoot":"","sources":["../../src/runtime/working-memory.ts"],"names":[],"mappings":";;;AAgBA,gDAmBC;AAKD,sEAqBC;AAKD,gDAOC;AAtED,qCAAgD;AAEhD,MAAM,MAAM,GAAG,IAAA,6BAAoB,GAAE,CAAC;AAItC,MAAM,QAAQ,GAAG,kBAAkB,CAAC;AACpC,MAAM,SAAS,GAAG,mBAAmB,CAAC;AAMtC,SAAgB,kBAAkB,CAAC,IAAY;IAI9C,MAAM,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;IACvC,IAAI,OAAO,KAAK,CAAC,CAAC;QAAE,OAAO,EAAE,SAAS,EAAE,IAAI,EAAE,aAAa,EAAE,IAAI,EAAE,CAAC;IAEpE,MAAM,QAAQ,GAAG,IAAI,CAAC,OAAO,CAAC,SAAS,EAAE,OAAO,CAAC,CAAC;IAClD,IAAI,QAAQ,KAAK,CAAC,CAAC;QAAE,OAAO,EAAE,SAAS,EAAE,IAAI,EAAE,aAAa,EAAE,IAAI,EAAE,CAAC;IAErE,MAAM,YAAY,GAAG,OAAO,GAAG,QAAQ,CAAC,MAAM,CAAC;IAC/C,MAAM,UAAU,GAAG,IAAI,CAAC,KAAK,CAAC,YAAY,EAAE,QAAQ,CAAC,CAAC;IACtD,MAAM,aAAa,GAAG,UAAU,CAAC,OAAO,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC,OAAO,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;IAEvE,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,OAAO,CAAC,CAAC,OAAO,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;IACzD,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,QAAQ,GAAG,SAAS,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;IACzE,MAAM,SAAS,GAAG,CAAC,MAAM,GAAG,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,GAAG,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;IAEhE,OAAO,EAAE,SAAS,EAAE,aAAa,EAAE,CAAC;AACrC,CAAC;AAKD,SAAgB,6BAA6B,CAAC,QAAgB,EAAE,UAAmB;IAClF,MAAM,MAAM,GAAG,UAAU;QACxB,CAAC,CAAC,0DAA0D;QAC5D,CAAC,CAAC,sDAAsD,CAAC;IAE1D,OAAO;QACN,EAAE;QACF,mBAAmB;QACnB,EAAE;QACF,wEAAwE;QACxE,6DAA6D;QAC7D,qEAAqE;QACrE,mFAAmF,QAAQ,MAAM,SAAS,sBAAsB;QAChI,GAAG,MAAM,yDAAyD;QAClE,iGAAiG;QACjG,EAAE;QACF,mBAAmB;QACnB,KAAK;QACL,QAAQ;QACR,KAAK;KACL,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AACd,CAAC;AAKD,SAAgB,kBAAkB,CAAC,MAAuB;IACzD,MAAM,GAAG,GAA2B,EAAE,CAAC;IACvC,KAAK,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC;QACzD,MAAM,IAAI,GAAG,KAAK,CAAC,WAAW,CAAC;QAC/B,GAAG,CAAC,GAAG,CAAC,GAAG,IAAI,IAAI,EAAE,CAAC;IACvB,CAAC;IACD,OAAO,IAAI,CAAC,SAAS,CAAC,GAAG,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC;AACrC,CAAC;AASD,MAAa,yBAAyB;IAWrC,YAAY,MAAgD,EAAE,OAAkB;QANxE,UAAK,GAAwB,QAAQ,CAAC;QAEtC,WAAM,GAAG,EAAE,CAAC;QAEZ,gBAAW,GAAG,EAAE,CAAC;QAGxB,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;IACxB,CAAC;IAED,KAAK,CAAC,KAAK,CAAC,KAAkB;QAC7B,IAAI,KAAK,CAAC,IAAI,KAAK,YAAY,EAAE,CAAC;YACjC,MAAM,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;YAC/B,OAAO;QACR,CAAC;QAED,IAAI,CAAC,WAAW,IAAI,KAAK,CAAC,KAAK,CAAC;QAEhC,OAAO,IAAI,CAAC,WAAW,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACpC,IAAI,IAAI,CAAC,KAAK,KAAK,QAAQ,EAAE,CAAC;gBAC7B,MAAM,OAAO,GAAG,IAAI,CAAC,WAAW,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;gBACnD,IAAI,OAAO,KAAK,CAAC,CAAC,EAAE,CAAC;oBAEpB,MAAM,MAAM,GAAG,IAAI,CAAC,WAAW,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC;oBACjD,IACC,MAAM,KAAK,CAAC,CAAC;wBACb,IAAI,CAAC,WAAW,CAAC,MAAM,GAAG,MAAM,GAAG,QAAQ,CAAC,MAAM;wBAClD,QAAQ,CAAC,UAAU,CAAC,IAAI,CAAC,WAAW,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,EAClD,CAAC;wBAEF,IAAI,MAAM,GAAG,CAAC,EAAE,CAAC;4BAChB,MAAM,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC;gCACvB,IAAI,EAAE,YAAY;gCAClB,KAAK,EAAE,IAAI,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC,EAAE,MAAM,CAAC;6BACxC,CAAC,CAAC;wBACJ,CAAC;wBACD,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC,WAAW,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;oBACnD,CAAC;yBAAM,CAAC;wBAEP,MAAM,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,IAAI,EAAE,YAAY,EAAE,KAAK,EAAE,IAAI,CAAC,WAAW,EAAE,CAAC,CAAC;wBACzE,IAAI,CAAC,WAAW,GAAG,EAAE,CAAC;oBACvB,CAAC;oBACD,MAAM;gBACP,CAAC;gBAED,IAAI,OAAO,GAAG,CAAC,EAAE,CAAC;oBACjB,MAAM,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC;wBACvB,IAAI,EAAE,YAAY;wBAClB,KAAK,EAAE,IAAI,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC,EAAE,OAAO,CAAC;qBACzC,CAAC,CAAC;gBACJ,CAAC;gBACD,IAAI,CAAC,KAAK,GAAG,QAAQ,CAAC;gBACtB,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC,WAAW,CAAC,KAAK,CAAC,OAAO,GAAG,QAAQ,CAAC,MAAM,CAAC,CAAC;gBACrE,IAAI,CAAC,MAAM,GAAG,EAAE,CAAC;YAClB,CAAC;iBAAM,CAAC;gBAEP,MAAM,QAAQ,GAAG,IAAI,CAAC,WAAW,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;gBACrD,IAAI,QAAQ,KAAK,CAAC,CAAC,EAAE,CAAC;oBAErB,MAAM,MAAM,GAAG,IAAI,CAAC,WAAW,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC;oBACjD,IACC,MAAM,KAAK,CAAC,CAAC;wBACb,IAAI,CAAC,WAAW,CAAC,MAAM,GAAG,MAAM,GAAG,SAAS,CAAC,MAAM;wBACnD,SAAS,CAAC,UAAU,CAAC,IAAI,CAAC,WAAW,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,EACnD,CAAC;wBACF,IAAI,CAAC,MAAM,IAAI,IAAI,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC,EAAE,MAAM,CAAC,CAAC;wBACjD,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC,WAAW,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;oBACnD,CAAC;yBAAM,CAAC;wBACP,IAAI,CAAC,MAAM,IAAI,IAAI,CAAC,WAAW,CAAC;wBAChC,IAAI,CAAC,WAAW,GAAG,EAAE,CAAC;oBACvB,CAAC;oBACD,MAAM;gBACP,CAAC;gBACD,IAAI,CAAC,MAAM,IAAI,IAAI,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC,EAAE,QAAQ,CAAC,CAAC;gBACnD,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC,WAAW,CAAC,KAAK,CAAC,QAAQ,GAAG,SAAS,CAAC,MAAM,CAAC,CAAC;gBACvE,IAAI,CAAC,KAAK,GAAG,QAAQ,CAAC;gBACtB,MAAM,OAAO,GAAG,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC,OAAO,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;gBAClE,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,KAAK,CAAC,CAAC,KAAc,EAAE,EAAE;oBAC9C,MAAM,CAAC,IAAI,CAAC,kCAAkC,EAAE,EAAE,KAAK,EAAE,CAAC,CAAC;gBAC5D,CAAC,CAAC,CAAC;gBACH,IAAI,CAAC,MAAM,GAAG,EAAE,CAAC;YAClB,CAAC;QACF,CAAC;IACF,CAAC;IAED,KAAK,CAAC,KAAK;QACV,IAAI,IAAI,CAAC,KAAK,KAAK,QAAQ,IAAI,IAAI,CAAC,WAAW,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC5D,MAAM,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,IAAI,EAAE,YAAY,EAAE,KAAK,EAAE,IAAI,CAAC,WAAW,EAAE,CAAC,CAAC;QAC1E,CAAC;QAED,IAAI,CAAC,WAAW,GAAG,EAAE,CAAC;QACtB,IAAI,CAAC,MAAM,GAAG,EAAE,CAAC;QACjB,IAAI,CAAC,KAAK,GAAG,QAAQ,CAAC;IACvB,CAAC;CACD;AApGD,8DAoGC"}

package/dist/sdk/agent.d.ts

@@ -0,0 +1,88 @@
+import type { ProviderOptions } from '@ai-sdk/provider-utils';
+import { z } from 'zod';
+import type { Eval } from './eval';
+import type { McpClient } from './mcp-client';
+import { Memory } from './memory';
+import { Telemetry } from './telemetry';
+import { AgentRuntime } from '../runtime/agent-runtime';
+import type { AgentEvent, AgentEventHandler, AgentMiddleware, BuiltAgent, BuiltEval, BuiltGuardrail, BuiltMemory, BuiltProviderTool, BuiltTool, BuiltTelemetry, CheckpointStore, ExecutionOptions, GenerateResult, MemoryConfig, ModelConfig, Provider, RunOptions, SerializableAgentState, StreamResult, ThinkingConfigFor, ResumeOptions } from '../types';
+import type { AgentMessage } from '../types/sdk/message';
+import type { Workspace } from '../workspace/workspace';
+type ToolParameter = BuiltTool | {
+    build(): BuiltTool;
+};
+export declare class Agent implements BuiltAgent {
+    readonly name: string;
+    private modelId?;
+    private modelConfigObj?;
+    private instructionProviderOpts?;
+    private instructionsText?;
+    private tools;
+    private providerTools;
+    private memoryConfig?;
+    private inputGuardrails;
+    private outputGuardrails;
+    private agentEvals;
+    private outputSchema?;
+    private checkpointStore?;
+    private thinkingConfig?;
+    private credentialName?;
+    private resolvedKey?;
+    private runtime?;
+    private concurrencyValue?;
+    private telemetryBuilder?;
+    private telemetryConfig?;
+    private middlewares;
+    private requireToolApprovalValue;
+    private mcpClients;
+    private buildPromise;
+    private eventBus;
+    private workspaceInstance?;
+    constructor(name: string);
+    model(providerOrIdOrConfig: string | ModelConfig, modelName?: string): this;
+    instructions(text: string, options?: {
+        providerOptions?: ProviderOptions;
+    }): this;
+    tool(t: ToolParameter | ToolParameter[]): this;
+    providerTool(builtProviderTool: BuiltProviderTool): this;
+    memory(m: MemoryConfig | Memory | BuiltMemory): this;
+    middleware(m: AgentMiddleware): this;
+    inputGuardrail(g: BuiltGuardrail | {
+        build(): BuiltGuardrail;
+    }): this;
+    outputGuardrail(g: BuiltGuardrail | {
+        build(): BuiltGuardrail;
+    }): this;
+    eval(e: Eval | BuiltEval | {
+        ensureBuilt(): BuiltEval;
+    }): this;
+    checkpoint(storage: 'memory' | CheckpointStore): this;
+    credential(name: string): this;
+    protected get declaredCredential(): string | undefined;
+    protected set resolvedApiKey(key: string);
+    structuredOutput(schema: z.ZodType): this;
+    thinking<P extends Provider>(_provider: P, config?: ThinkingConfigFor<P>): this;
+    telemetry(t: Telemetry | BuiltTelemetry): this;
+    protected get declaredTelemetry(): Telemetry | undefined;
+    toolCallConcurrency(n: number): this;
+    requireToolApproval(): this;
+    workspace(ws: Workspace): this;
+    mcp(client: McpClient): this;
+    get evaluations(): BuiltEval[];
+    on(event: AgentEvent, handler: AgentEventHandler): void;
+    asTool(description: string): BuiltTool;
+    getState(): SerializableAgentState;
+    abort(): void;
+    generate(input: AgentMessage[] | string, options?: RunOptions & ExecutionOptions): Promise<GenerateResult>;
+    stream(input: AgentMessage[] | string, options?: RunOptions & ExecutionOptions): Promise<StreamResult>;
+    resume(method: 'generate', data: unknown, options: ResumeOptions & ExecutionOptions): Promise<GenerateResult>;
+    resume(method: 'stream', data: unknown, options: ResumeOptions & ExecutionOptions): Promise<StreamResult>;
+    approve(method: 'generate', options: ResumeOptions & ExecutionOptions): Promise<GenerateResult>;
+    approve(method: 'stream', options: ResumeOptions & ExecutionOptions): Promise<StreamResult>;
+    deny(method: 'generate', options: ResumeOptions & ExecutionOptions): Promise<GenerateResult>;
+    deny(method: 'stream', options: ResumeOptions & ExecutionOptions): Promise<StreamResult>;
+    private ensureBuilt;
+    private toMessages;
+    protected build(): Promise<AgentRuntime>;
+}
+export {};