@temporalio/ai-sdk 1.14.0

package/LICENSE

@@ -0,0 +1,21 @@
+The MIT License
+
+Copyright (c) 2021-2025 Temporal Technologies Inc. All rights reserved.
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in
+all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
+THE SOFTWARE.

package/lib/activities.d.ts

@@ -0,0 +1,46 @@
+import type { LanguageModelV2CallOptions, LanguageModelV2CallWarning, LanguageModelV2Content, LanguageModelV2FinishReason, LanguageModelV2ResponseMetadata, LanguageModelV2Usage, ProviderV2, SharedV2Headers, SharedV2ProviderMetadata } from '@ai-sdk/provider';
+import type { ToolCallOptions } from 'ai';
+import type { McpClientFactories } from './mcp';
+export interface InvokeModelArgs {
+    modelId: string;
+    options: LanguageModelV2CallOptions;
+}
+export interface InvokeModelResult {
+    content: Array<LanguageModelV2Content>;
+    finishReason: LanguageModelV2FinishReason;
+    usage: LanguageModelV2Usage;
+    providerMetadata?: SharedV2ProviderMetadata;
+    request?: {
+        body?: unknown;
+    };
+    response?: LanguageModelV2ResponseMetadata & {
+        headers?: SharedV2Headers;
+        body?: unknown;
+    };
+    warnings: Array<LanguageModelV2CallWarning>;
+}
+export interface ListToolResult {
+    description?: string;
+    inputSchema: any;
+}
+export interface ListToolArgs {
+    clientArgs?: any;
+}
+export interface CallToolArgs {
+    clientArgs?: any;
+    name: string;
+    args: any;
+    options: ToolCallOptions;
+}
+/**
+ * Creates Temporal activities for AI model invocation using the provided AI SDK provider.
+ * These activities allow workflows to call AI models while maintaining Temporal's
+ * execution guarantees and replay safety.
+ *
+ * @param provider The AI SDK provider to use for model invocations
+ * @param mcpClientFactories A mapping of server names to functions to create mcp clients
+ * @returns An object containing the activity functions
+ *
+ * @experimental The AI SDK integration is an experimental feature; APIs may change without notice.
+ */
+export declare function createActivities(provider: ProviderV2, mcpClientFactories?: McpClientFactories): object;

package/lib/activities.js

@@ -0,0 +1,69 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.createActivities = createActivities;
+const common_1 = require("@temporalio/common");
+/**
+ * Creates Temporal activities for AI model invocation using the provided AI SDK provider.
+ * These activities allow workflows to call AI models while maintaining Temporal's
+ * execution guarantees and replay safety.
+ *
+ * @param provider The AI SDK provider to use for model invocations
+ * @param mcpClientFactories A mapping of server names to functions to create mcp clients
+ * @returns An object containing the activity functions
+ *
+ * @experimental The AI SDK integration is an experimental feature; APIs may change without notice.
+ */
+function createActivities(provider, mcpClientFactories) {
+    let activities = {
+        async invokeModel(args) {
+            const model = provider.languageModel(args.modelId);
+            return await model.doGenerate(args.options);
+        },
+    };
+    if (mcpClientFactories !== undefined) {
+        Object.entries(mcpClientFactories).forEach(([name, func]) => {
+            activities = {
+                ...activities,
+                ...activitiesForName(name, func),
+            };
+        });
+    }
+    return activities;
+}
+function activitiesForName(name, mcpClientFactory) {
+    async function listToolsActivity(args) {
+        const mcpClient = await mcpClientFactory(args.clientArgs);
+        try {
+            const tools = await mcpClient.tools();
+            return Object.fromEntries(Object.entries(tools).map(([k, v]) => [
+                k,
+                {
+                    description: v.description,
+                    inputSchema: v.inputSchema,
+                },
+            ]));
+        }
+        finally {
+            await mcpClient.close();
+        }
+    }
+    async function callToolActivity(args) {
+        const mcpClient = await mcpClientFactory(args.clientArgs);
+        try {
+            const tools = await mcpClient.tools();
+            const tool = tools[args.name];
+            if (tool === undefined) {
+                throw common_1.ApplicationFailure.retryable(`Tool ${args.name} not found.`);
+            }
+            return tool.execute(args.args, args.options);
+        }
+        finally {
+            await mcpClient.close();
+        }
+    }
+    return {
+        [name + '-listTools']: listToolsActivity,
+        [name + '-callTool']: callToolActivity,
+    };
+}
+//# sourceMappingURL=activities.js.map

package/lib/activities.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"activities.js","sourceRoot":"","sources":["../src/activities.ts"],"names":[],"mappings":";;AAyDA,4CAgBC;AA7DD,+CAAwD;AAkCxD;;;;;;;;;;GAUG;AACH,SAAgB,gBAAgB,CAAC,QAAoB,EAAE,kBAAuC;IAC5F,IAAI,UAAU,GAAG;QACf,KAAK,CAAC,WAAW,CAAC,IAAqB;YACrC,MAAM,KAAK,GAAG,QAAQ,CAAC,aAAa,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;YACnD,OAAO,MAAM,KAAK,CAAC,UAAU,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAC9C,CAAC;KACF,CAAC;IACF,IAAI,kBAAkB,KAAK,SAAS,EAAE,CAAC;QACrC,MAAM,CAAC,OAAO,CAAC,kBAAkB,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,EAAE,IAAI,CAAC,EAAE,EAAE;YAC1D,UAAU,GAAG;gBACX,GAAG,UAAU;gBACb,GAAG,iBAAiB,CAAC,IAAI,EAAE,IAAI,CAAC;aACjC,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC;IACD,OAAO,UAAU,CAAC;AACpB,CAAC;AAED,SAAS,iBAAiB,CAAC,IAAY,EAAE,gBAAkC;IACzE,KAAK,UAAU,iBAAiB,CAAC,IAAkB;QACjD,MAAM,SAAS,GAAG,MAAM,gBAAgB,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QAC1D,IAAI,CAAC;YACH,MAAM,KAAK,GAAG,MAAM,SAAS,CAAC,KAAK,EAAE,CAAC;YAEtC,OAAO,MAAM,CAAC,WAAW,CACvB,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC;gBACpC,CAAC;gBACD;oBACE,WAAW,EAAE,CAAC,CAAC,WAAW;oBAC1B,WAAW,EAAE,CAAC,CAAC,WAAW;iBAC3B;aACF,CAAC,CACH,CAAC;QACJ,CAAC;gBAAS,CAAC;YACT,MAAM,SAAS,CAAC,KAAK,EAAE,CAAC;QAC1B,CAAC;IACH,CAAC;IACD,KAAK,UAAU,gBAAgB,CAAC,IAAkB;QAChD,MAAM,SAAS,GAAG,MAAM,gBAAgB,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QAC1D,IAAI,CAAC;YACH,MAAM,KAAK,GAAG,MAAM,SAAS,CAAC,KAAK,EAAE,CAAC;YACtC,MAAM,IAAI,GAAG,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAC9B,IAAI,IAAI,KAAK,SAAS,EAAE,CAAC;gBACvB,MAAM,2BAAkB,CAAC,SAAS,CAAC,QAAQ,IAAI,CAAC,IAAI,aAAa,CAAC,CAAC;YACrE,CAAC;YACD,OAAO,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC;QAC/C,CAAC;gBAAS,CAAC;YACT,MAAM,SAAS,CAAC,KAAK,EAAE,CAAC;QAC1B,CAAC;IACH,CAAC;IACD,OAAO;QACL,CAAC,IAAI,GAAG,YAAY,CAAC,EAAE,iBAAiB;QACxC,CAAC,IAAI,GAAG,WAAW,CAAC,EAAE,gBAAgB;KACvC,CAAC;AACJ,CAAC"}

package/lib/index.d.ts

@@ -0,0 +1,4 @@
+import './load-polyfills';
+export * from './mcp';
+export * from './plugin';
+export * from './provider';

package/lib/index.js

@@ -0,0 +1,22 @@
+"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 __exportStar = (this && this.__exportStar) || function(m, exports) {
+    for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+// eslint-disable-next-line import/no-unassigned-import
+require("./load-polyfills");
+__exportStar(require("./mcp"), exports);
+__exportStar(require("./plugin"), exports);
+__exportStar(require("./provider"), exports);
+//# sourceMappingURL=index.js.map

package/lib/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;AAAA,uDAAuD;AACvD,4BAA0B;AAE1B,wCAAsB;AACtB,2CAAyB;AACzB,6CAA2B"}

package/lib/load-polyfills.d.ts

@@ -0,0 +1 @@
+export {};

package/lib/load-polyfills.js

@@ -0,0 +1,19 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+const headers_polyfill_1 = require("headers-polyfill");
+const workflow_1 = require("@temporalio/workflow");
+if ((0, workflow_1.inWorkflowContext)()) {
+    // Apply Headers polyfill
+    if (typeof globalThis.Headers === 'undefined') {
+        globalThis.Headers = headers_polyfill_1.Headers;
+    }
+    // eslint-disable-next-line @typescript-eslint/no-require-imports,import/no-unassigned-import
+    require('web-streams-polyfill/polyfill');
+    // Attach the polyfill as a Global function
+    if (!('structuredClone' in globalThis)) {
+        // eslint-disable-next-line @typescript-eslint/no-require-imports,import/no-unassigned-import
+        const structuredClone = require('@ungap/structured-clone');
+        globalThis.structuredClone = structuredClone.default;
+    }
+}
+//# sourceMappingURL=load-polyfills.js.map

package/lib/load-polyfills.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"load-polyfills.js","sourceRoot":"","sources":["../src/load-polyfills.ts"],"names":[],"mappings":";;AAAA,uDAA2C;AAC3C,mDAAyD;AAEzD,IAAI,IAAA,4BAAiB,GAAE,EAAE,CAAC;IACxB,yBAAyB;IACzB,IAAI,OAAO,UAAU,CAAC,OAAO,KAAK,WAAW,EAAE,CAAC;QAC9C,UAAU,CAAC,OAAO,GAAG,0BAAO,CAAC;IAC/B,CAAC;IAED,6FAA6F;IAC7F,OAAO,CAAC,+BAA+B,CAAC,CAAC;IACzC,2CAA2C;IAC3C,IAAI,CAAC,CAAC,iBAAiB,IAAI,UAAU,CAAC,EAAE,CAAC;QACvC,6FAA6F;QAC7F,MAAM,eAAe,GAAG,OAAO,CAAC,yBAAyB,CAAC,CAAC;QAC3D,UAAU,CAAC,eAAe,GAAG,eAAe,CAAC,OAAO,CAAC;IACvD,CAAC;AACH,CAAC"}

package/lib/mcp.d.ts

@@ -0,0 +1,31 @@
+import type { ToolSet } from 'ai';
+import type { experimental_MCPClient as MCPClient } from '@ai-sdk/mcp';
+import type { ActivityOptions } from '@temporalio/workflow';
+export type McpClientFactory = (args: unknown) => Promise<MCPClient>;
+export type McpClientFactories = {
+    [serviceName: string]: McpClientFactory;
+};
+/**
+ * Options for the Temporal MCP Client
+ *
+ * @experimental The AI SDK plugin is an experimental feature; APIs may change without notice.
+ */
+export interface TemporalMCPClientOptions {
+    readonly name: string;
+    readonly clientArgs?: any;
+    readonly activityOptions?: ActivityOptions;
+}
+/**
+ * A Temporal MCP Client which uses activities to execute list tools and call tools.
+ * This should match by name an MCPClientFactory registered with the AI SDK plugin to function.
+ *
+ * This is intentionally similar to the AI SDK's ``MCPClient``, but is lacking functions not used by the framework.
+ * It doesn't close either since the underlying clients are created in activities.
+ *
+ * @experimental The AI SDK plugin is an experimental feature; APIs may change without notice.
+ */
+export declare class TemporalMCPClient {
+    readonly options: TemporalMCPClientOptions;
+    constructor(options: TemporalMCPClientOptions);
+    tools(): Promise<ToolSet>;
+}

package/lib/mcp.js

@@ -0,0 +1,72 @@
+"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 (mod) {
+    if (mod && mod.__esModule) return mod;
+    var result = {};
+    if (mod != null) for (var k in mod) if (k !== "default" && Object.prototype.hasOwnProperty.call(mod, k)) __createBinding(result, mod, k);
+    __setModuleDefault(result, mod);
+    return result;
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.TemporalMCPClient = void 0;
+const workflow = __importStar(require("@temporalio/workflow"));
+/**
+ * A Temporal MCP Client which uses activities to execute list tools and call tools.
+ * This should match by name an MCPClientFactory registered with the AI SDK plugin to function.
+ *
+ * This is intentionally similar to the AI SDK's ``MCPClient``, but is lacking functions not used by the framework.
+ * It doesn't close either since the underlying clients are created in activities.
+ *
+ * @experimental The AI SDK plugin is an experimental feature; APIs may change without notice.
+ */
+class TemporalMCPClient {
+    options;
+    constructor(options) {
+        this.options = options;
+    }
+    async tools() {
+        workflow.log.info(`Options: ${this.options.activityOptions}`);
+        const activities = workflow.proxyActivities({ startToCloseTimeout: '10 minutes', ...this.options.activityOptions });
+        const listActivity = activities[this.options.name + '-listTools'];
+        const tools = await listActivity({ clientArgs: this.options.clientArgs });
+        return Object.fromEntries(Object.entries(tools).map(([toolName, toolResult]) => [
+            toolName,
+            {
+                execute: async (args, options) => {
+                    const activities = workflow.proxyActivities({
+                        summary: toolName,
+                        startToCloseTimeout: '10 minutes',
+                        ...this.options,
+                    });
+                    const callActivity = activities[this.options.name + '-callTool'];
+                    return await callActivity({ name: toolName, args, options, clientArgs: this.options.clientArgs });
+                },
+                inputSchema: {
+                    ...toolResult.inputSchema,
+                    _type: undefined,
+                    validate: undefined,
+                    [Symbol.for('vercel.ai.schema')]: true,
+                    [Symbol.for('vercel.ai.validator')]: true,
+                },
+                type: 'dynamic',
+            },
+        ]));
+    }
+}
+exports.TemporalMCPClient = TemporalMCPClient;
+//# sourceMappingURL=mcp.js.map

package/lib/mcp.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"mcp.js","sourceRoot":"","sources":["../src/mcp.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;AAEA,+DAAiD;AAmBjD;;;;;;;;GAQG;AACH,MAAa,iBAAiB;IACP;IAArB,YAAqB,OAAiC;QAAjC,YAAO,GAAP,OAAO,CAA0B;IAAG,CAAC;IAE1D,KAAK,CAAC,KAAK;QACT,QAAQ,CAAC,GAAG,CAAC,IAAI,CAAC,YAAY,IAAI,CAAC,OAAO,CAAC,eAAe,EAAE,CAAC,CAAC;QAC9D,MAAM,UAAU,GAAG,QAAQ,CAAC,eAAe,CAAC,EAAE,mBAAmB,EAAE,YAAY,EAAE,GAAG,IAAI,CAAC,OAAO,CAAC,eAAe,EAAE,CAAC,CAAC;QAEpH,MAAM,YAAY,GAAG,UAAU,CAAC,IAAI,CAAC,OAAO,CAAC,IAAI,GAAG,YAAY,CAAC,CAAC;QAClE,MAAM,KAAK,GAAmC,MAAM,YAAa,CAAC,EAAE,UAAU,EAAE,IAAI,CAAC,OAAO,CAAC,UAAU,EAAE,CAAC,CAAC;QAC3G,OAAO,MAAM,CAAC,WAAW,CACvB,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,QAAQ,EAAE,UAAU,CAAC,EAAE,EAAE,CAAC;YACpD,QAAQ;YACR;gBACE,OAAO,EAAE,KAAK,EAAE,IAAS,EAAE,OAAO,EAAE,EAAE;oBACpC,MAAM,UAAU,GAAG,QAAQ,CAAC,eAAe,CAAC;wBAC1C,OAAO,EAAE,QAAQ;wBACjB,mBAAmB,EAAE,YAAY;wBACjC,GAAG,IAAI,CAAC,OAAO;qBAChB,CAAC,CAAC;oBACH,MAAM,YAAY,GAAG,UAAU,CAAC,IAAI,CAAC,OAAO,CAAC,IAAI,GAAG,WAAW,CAAE,CAAC;oBAClE,OAAO,MAAM,YAAY,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,IAAI,EAAE,OAAO,EAAE,UAAU,EAAE,IAAI,CAAC,OAAO,CAAC,UAAU,EAAE,CAAC,CAAC;gBACpG,CAAC;gBACD,WAAW,EAAE;oBACX,GAAG,UAAU,CAAC,WAAW;oBACzB,KAAK,EAAE,SAAS;oBAChB,QAAQ,EAAE,SAAS;oBACnB,CAAC,MAAM,CAAC,GAAG,CAAC,kBAAkB,CAAC,CAAC,EAAE,IAAI;oBACtC,CAAC,MAAM,CAAC,GAAG,CAAC,qBAAqB,CAAC,CAAC,EAAE,IAAI;iBAC1C;gBACD,IAAI,EAAE,SAAS;aAChB;SACF,CAAC,CACH,CAAC;IACJ,CAAC;CACF;AAlCD,8CAkCC"}

package/lib/plugin.d.ts

@@ -0,0 +1,25 @@
+import type { ProviderV2 } from '@ai-sdk/provider';
+import { SimplePlugin } from '@temporalio/plugin';
+import type { McpClientFactories } from './mcp';
+/**
+ * Options for the AI SDK plugin
+ *
+ * @experimental The AI SDK plugin is an experimental feature; APIs may change without notice.
+ */
+export interface AiSdkPluginOptions {
+    modelProvider: ProviderV2;
+    /**
+     * This object contains a mapping of server names to functions which create MCP clients.
+     * Any TemporalMCPClient used in a workflow should have its associated servername listed in this object.
+     */
+    mcpClientFactories?: McpClientFactories;
+}
+/**
+ * A Temporal plugin that integrates AI SDK providers for use in workflows.
+ * This plugin creates activities that allow workflows to invoke AI models.
+ *
+ * @experimental The AI SDK plugin is an experimental feature; APIs may change without notice.
+ */
+export declare class AiSdkPlugin extends SimplePlugin {
+    constructor(options: AiSdkPluginOptions);
+}

package/lib/plugin.js

@@ -0,0 +1,21 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.AiSdkPlugin = void 0;
+const plugin_1 = require("@temporalio/plugin");
+const activities_1 = require("./activities");
+/**
+ * A Temporal plugin that integrates AI SDK providers for use in workflows.
+ * This plugin creates activities that allow workflows to invoke AI models.
+ *
+ * @experimental The AI SDK plugin is an experimental feature; APIs may change without notice.
+ */
+class AiSdkPlugin extends plugin_1.SimplePlugin {
+    constructor(options) {
+        super({
+            name: 'AiSDKPlugin',
+            activities: (0, activities_1.createActivities)(options.modelProvider, options.mcpClientFactories),
+        });
+    }
+}
+exports.AiSdkPlugin = AiSdkPlugin;
+//# sourceMappingURL=plugin.js.map

package/lib/plugin.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"plugin.js","sourceRoot":"","sources":["../src/plugin.ts"],"names":[],"mappings":";;;AACA,+CAAkD;AAClD,6CAAgD;AAkBhD;;;;;GAKG;AACH,MAAa,WAAY,SAAQ,qBAAY;IAC3C,YAAY,OAA2B;QACrC,KAAK,CAAC;YACJ,IAAI,EAAE,aAAa;YACnB,UAAU,EAAE,IAAA,6BAAgB,EAAC,OAAO,CAAC,aAAa,EAAE,OAAO,CAAC,kBAAkB,CAAC;SAChF,CAAC,CAAC;IACL,CAAC;CACF;AAPD,kCAOC"}

package/lib/provider.d.ts

@@ -0,0 +1,47 @@
+import type { EmbeddingModelV2, ImageModelV2, LanguageModelV2, LanguageModelV2CallOptions, ProviderV2, SharedV2Headers } from '@ai-sdk/provider';
+import { ActivityOptions } from '@temporalio/workflow';
+import { InvokeModelResult } from './activities';
+/**
+ * A language model implementation that delegates AI model calls to Temporal activities.
+ * This allows workflows to invoke AI models through the Temporal execution model.
+ *
+ * @experimental The AI SDK integration is an experimental feature; APIs may change without notice.
+ */
+export declare class TemporalLanguageModel implements LanguageModelV2 {
+    readonly modelId: string;
+    readonly options?: ActivityOptions | undefined;
+    readonly specificationVersion = "v2";
+    readonly provider = "temporal";
+    readonly supportedUrls: {};
+    constructor(modelId: string, options?: ActivityOptions | undefined);
+    doGenerate(options: LanguageModelV2CallOptions): Promise<InvokeModelResult>;
+    doStream(_options: LanguageModelV2CallOptions): PromiseLike<{
+        stream: any;
+        request?: {
+            body?: unknown;
+        };
+        response?: {
+            headers?: SharedV2Headers;
+        };
+    }>;
+}
+/**
+ * A Temporal-specific provider implementation that creates AI models which execute
+ * through Temporal activities. This provider integrates AI SDK models with Temporal's
+ * execution model to ensure reliable, durable AI model invocations.
+ *
+ * @experimental The AI SDK integration is an experimental feature; APIs may change without notice.
+ */
+export declare class TemporalProvider implements ProviderV2 {
+    readonly options?: ActivityOptions | undefined;
+    constructor(options?: ActivityOptions | undefined);
+    imageModel(_modelId: string): ImageModelV2;
+    languageModel(modelId: string): LanguageModelV2;
+    textEmbeddingModel(_modelId: string): EmbeddingModelV2<string>;
+}
+/**
+ * A singleton instance of TemporalProvider for convenient use in applications.
+ *
+ * @experimental The AI SDK integration is an experimental feature; APIs may change without notice.
+ */
+export declare const temporalProvider: TemporalProvider;

package/lib/provider.js

@@ -0,0 +1,90 @@
+"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 (mod) {
+    if (mod && mod.__esModule) return mod;
+    var result = {};
+    if (mod != null) for (var k in mod) if (k !== "default" && Object.prototype.hasOwnProperty.call(mod, k)) __createBinding(result, mod, k);
+    __setModuleDefault(result, mod);
+    return result;
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.temporalProvider = exports.TemporalProvider = exports.TemporalLanguageModel = void 0;
+const workflow = __importStar(require("@temporalio/workflow"));
+const common_1 = require("@temporalio/common");
+/**
+ * A language model implementation that delegates AI model calls to Temporal activities.
+ * This allows workflows to invoke AI models through the Temporal execution model.
+ *
+ * @experimental The AI SDK integration is an experimental feature; APIs may change without notice.
+ */
+class TemporalLanguageModel {
+    modelId;
+    options;
+    specificationVersion = 'v2';
+    provider = 'temporal';
+    supportedUrls = {};
+    constructor(modelId, options) {
+        this.modelId = modelId;
+        this.options = options;
+    }
+    async doGenerate(options) {
+        const activities = workflow.proxyActivities({ startToCloseTimeout: '10 minutes', ...this.options });
+        const result = await activities.invokeModel({ modelId: this.modelId, options });
+        if (result === undefined) {
+            throw common_1.ApplicationFailure.nonRetryable('Received undefined response from model activity.');
+        }
+        if (result.response !== undefined) {
+            result.response.timestamp = new Date(result.response.timestamp);
+        }
+        return result;
+    }
+    doStream(_options) {
+        throw common_1.ApplicationFailure.nonRetryable('Streaming not supported.');
+    }
+}
+exports.TemporalLanguageModel = TemporalLanguageModel;
+/**
+ * A Temporal-specific provider implementation that creates AI models which execute
+ * through Temporal activities. This provider integrates AI SDK models with Temporal's
+ * execution model to ensure reliable, durable AI model invocations.
+ *
+ * @experimental The AI SDK integration is an experimental feature; APIs may change without notice.
+ */
+class TemporalProvider {
+    options;
+    constructor(options) {
+        this.options = options;
+    }
+    imageModel(_modelId) {
+        throw new Error('Not implemented');
+    }
+    languageModel(modelId) {
+        return new TemporalLanguageModel(modelId, this.options);
+    }
+    textEmbeddingModel(_modelId) {
+        throw new Error('Not implemented');
+    }
+}
+exports.TemporalProvider = TemporalProvider;
+/**
+ * A singleton instance of TemporalProvider for convenient use in applications.
+ *
+ * @experimental The AI SDK integration is an experimental feature; APIs may change without notice.
+ */
+exports.temporalProvider = new TemporalProvider();
+//# sourceMappingURL=provider.js.map

package/lib/provider.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"provider.js","sourceRoot":"","sources":["../src/provider.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;AAQA,+DAAiD;AAEjD,+CAAwD;AAGxD;;;;;GAKG;AACH,MAAa,qBAAqB;IAMrB;IACA;IANF,oBAAoB,GAAG,IAAI,CAAC;IAC5B,QAAQ,GAAG,UAAU,CAAC;IACtB,aAAa,GAAG,EAAE,CAAC;IAE5B,YACW,OAAe,EACf,OAAyB;QADzB,YAAO,GAAP,OAAO,CAAQ;QACf,YAAO,GAAP,OAAO,CAAkB;IACjC,CAAC;IAEJ,KAAK,CAAC,UAAU,CAAC,OAAmC;QAClD,MAAM,UAAU,GAAG,QAAQ,CAAC,eAAe,CAAC,EAAE,mBAAmB,EAAE,YAAY,EAAE,GAAG,IAAI,CAAC,OAAO,EAAE,CAAC,CAAC;QACpG,MAAM,MAAM,GAAG,MAAM,UAAU,CAAC,WAAY,CAAC,EAAE,OAAO,EAAE,IAAI,CAAC,OAAO,EAAE,OAAO,EAAE,CAAC,CAAC;QACjF,IAAI,MAAM,KAAK,SAAS,EAAE,CAAC;YACzB,MAAM,2BAAkB,CAAC,YAAY,CAAC,kDAAkD,CAAC,CAAC;QAC5F,CAAC;QACD,IAAI,MAAM,CAAC,QAAQ,KAAK,SAAS,EAAE,CAAC;YAClC,MAAM,CAAC,QAAQ,CAAC,SAAS,GAAG,IAAI,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC;QAClE,CAAC;QACD,OAAO,MAAM,CAAC;IAChB,CAAC;IAED,QAAQ,CAAC,QAAoC;QAK3C,MAAM,2BAAkB,CAAC,YAAY,CAAC,0BAA0B,CAAC,CAAC;IACpE,CAAC;CACF;AA7BD,sDA6BC;AAED;;;;;;GAMG;AACH,MAAa,gBAAgB;IACN;IAArB,YAAqB,OAAyB;QAAzB,YAAO,GAAP,OAAO,CAAkB;IAAG,CAAC;IAElD,UAAU,CAAC,QAAgB;QACzB,MAAM,IAAI,KAAK,CAAC,iBAAiB,CAAC,CAAC;IACrC,CAAC;IAED,aAAa,CAAC,OAAe;QAC3B,OAAO,IAAI,qBAAqB,CAAC,OAAO,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC;IAC1D,CAAC;IAED,kBAAkB,CAAC,QAAgB;QACjC,MAAM,IAAI,KAAK,CAAC,iBAAiB,CAAC,CAAC;IACrC,CAAC;CACF;AAdD,4CAcC;AAED;;;;GAIG;AACU,QAAA,gBAAgB,GAAqB,IAAI,gBAAgB,EAAE,CAAC"}

package/package.json

@@ -0,0 +1,48 @@
+{
+  "name": "@temporalio/ai-sdk",
+  "version": "1.14.0",
+  "description": "Temporal AI SDK integration package",
+  "main": "lib/index.js",
+  "types": "./lib/index.d.ts",
+  "keywords": [
+    "temporal",
+    "workflow",
+    "ai",
+    "ai-sdk",
+    "llm"
+  ],
+  "author": "Temporal Technologies Inc. <sdk@temporal.io>",
+  "license": "MIT",
+  "dependencies": {
+    "@temporalio/plugin": "1.14.0",
+    "@temporalio/workflow": "1.14.0",
+    "@ungap/structured-clone": "^1.3.0",
+    "headers-polyfill": "^4.0.3",
+    "web-streams-polyfill": "^4.2.0"
+  },
+  "peerDependencies": {
+    "@ai-sdk/mcp": "^0.0.8",
+    "@ai-sdk/provider": "^2.0.0",
+    "ai": "^5.0.91"
+  },
+  "engines": {
+    "node": ">= 18.0.0"
+  },
+  "bugs": {
+    "url": "https://github.com/temporalio/sdk-typescript/issues"
+  },
+  "repository": {
+    "type": "git",
+    "url": "git+https://github.com/temporalio/sdk-typescript.git",
+    "directory": "packages/ai-sdk"
+  },
+  "homepage": "https://github.com/temporalio/sdk-typescript/tree/main/packages/ai-sdk",
+  "publishConfig": {
+    "access": "public"
+  },
+  "files": [
+    "src",
+    "lib"
+  ],
+  "gitHead": "eb428ac36568b48be67fdba662c38631867b932d"
+}

package/src/activities.ts

@@ -0,0 +1,112 @@
+import type {
+  LanguageModelV2CallOptions,
+  LanguageModelV2CallWarning,
+  LanguageModelV2Content,
+  LanguageModelV2FinishReason,
+  LanguageModelV2ResponseMetadata,
+  LanguageModelV2Usage,
+  ProviderV2,
+  SharedV2Headers,
+  SharedV2ProviderMetadata,
+} from '@ai-sdk/provider';
+import type { ToolCallOptions } from 'ai';
+import { ApplicationFailure } from '@temporalio/common';
+import type { McpClientFactories, McpClientFactory } from './mcp';
+
+export interface InvokeModelArgs {
+  modelId: string;
+  options: LanguageModelV2CallOptions;
+}
+
+export interface InvokeModelResult {
+  content: Array<LanguageModelV2Content>;
+  finishReason: LanguageModelV2FinishReason;
+  usage: LanguageModelV2Usage;
+  providerMetadata?: SharedV2ProviderMetadata;
+  request?: { body?: unknown };
+  response?: LanguageModelV2ResponseMetadata & { headers?: SharedV2Headers; body?: unknown };
+  warnings: Array<LanguageModelV2CallWarning>;
+}
+
+export interface ListToolResult {
+  description?: string;
+  inputSchema: any;
+}
+
+export interface ListToolArgs {
+  clientArgs?: any;
+}
+
+export interface CallToolArgs {
+  clientArgs?: any;
+  name: string;
+  args: any;
+  options: ToolCallOptions;
+}
+
+/**
+ * Creates Temporal activities for AI model invocation using the provided AI SDK provider.
+ * These activities allow workflows to call AI models while maintaining Temporal's
+ * execution guarantees and replay safety.
+ *
+ * @param provider The AI SDK provider to use for model invocations
+ * @param mcpClientFactories A mapping of server names to functions to create mcp clients
+ * @returns An object containing the activity functions
+ *
+ * @experimental The AI SDK integration is an experimental feature; APIs may change without notice.
+ */
+export function createActivities(provider: ProviderV2, mcpClientFactories?: McpClientFactories): object {
+  let activities = {
+    async invokeModel(args: InvokeModelArgs): Promise<InvokeModelResult> {
+      const model = provider.languageModel(args.modelId);
+      return await model.doGenerate(args.options);
+    },
+  };
+  if (mcpClientFactories !== undefined) {
+    Object.entries(mcpClientFactories).forEach(([name, func]) => {
+      activities = {
+        ...activities,
+        ...activitiesForName(name, func),
+      };
+    });
+  }
+  return activities;
+}
+
+function activitiesForName(name: string, mcpClientFactory: McpClientFactory): object {
+  async function listToolsActivity(args: ListToolArgs): Promise<Record<string, ListToolResult>> {
+    const mcpClient = await mcpClientFactory(args.clientArgs);
+    try {
+      const tools = await mcpClient.tools();
+
+      return Object.fromEntries(
+        Object.entries(tools).map(([k, v]) => [
+          k,
+          {
+            description: v.description,
+            inputSchema: v.inputSchema,
+          },
+        ])
+      );
+    } finally {
+      await mcpClient.close();
+    }
+  }
+  async function callToolActivity(args: CallToolArgs): Promise<any> {
+    const mcpClient = await mcpClientFactory(args.clientArgs);
+    try {
+      const tools = await mcpClient.tools();
+      const tool = tools[args.name];
+      if (tool === undefined) {
+        throw ApplicationFailure.retryable(`Tool ${args.name} not found.`);
+      }
+      return tool.execute(args.args, args.options);
+    } finally {
+      await mcpClient.close();
+    }
+  }
+  return {
+    [name + '-listTools']: listToolsActivity,
+    [name + '-callTool']: callToolActivity,
+  };
+}

package/src/index.ts

@@ -0,0 +1,6 @@
+// eslint-disable-next-line import/no-unassigned-import
+import './load-polyfills';
+
+export * from './mcp';
+export * from './plugin';
+export * from './provider';

package/src/load-polyfills.ts

@@ -0,0 +1,18 @@
+import { Headers } from 'headers-polyfill';
+import { inWorkflowContext } from '@temporalio/workflow';
+
+if (inWorkflowContext()) {
+  // Apply Headers polyfill
+  if (typeof globalThis.Headers === 'undefined') {
+    globalThis.Headers = Headers;
+  }
+
+  // eslint-disable-next-line @typescript-eslint/no-require-imports,import/no-unassigned-import
+  require('web-streams-polyfill/polyfill');
+  // Attach the polyfill as a Global function
+  if (!('structuredClone' in globalThis)) {
+    // eslint-disable-next-line @typescript-eslint/no-require-imports,import/no-unassigned-import
+    const structuredClone = require('@ungap/structured-clone');
+    globalThis.structuredClone = structuredClone.default;
+  }
+}

package/src/mcp.ts

@@ -0,0 +1,65 @@
+import type { ToolSet } from 'ai';
+import type { experimental_MCPClient as MCPClient } from '@ai-sdk/mcp';
+import * as workflow from '@temporalio/workflow';
+import type { ActivityOptions } from '@temporalio/workflow';
+import type { ListToolResult } from './activities';
+
+export type McpClientFactory = (args: unknown) => Promise<MCPClient>;
+export type McpClientFactories = { [serviceName: string]: McpClientFactory };
+
+/**
+ * Options for the Temporal MCP Client
+ *
+ * @experimental The AI SDK plugin is an experimental feature; APIs may change without notice.
+ */
+export interface TemporalMCPClientOptions {
+  readonly name: string;
+  // eslint-disable-next-line @typescript-eslint/explicit-module-boundary-types
+  readonly clientArgs?: any;
+  readonly activityOptions?: ActivityOptions;
+}
+
+/**
+ * A Temporal MCP Client which uses activities to execute list tools and call tools.
+ * This should match by name an MCPClientFactory registered with the AI SDK plugin to function.
+ *
+ * This is intentionally similar to the AI SDK's ``MCPClient``, but is lacking functions not used by the framework.
+ * It doesn't close either since the underlying clients are created in activities.
+ *
+ * @experimental The AI SDK plugin is an experimental feature; APIs may change without notice.
+ */
+export class TemporalMCPClient {
+  constructor(readonly options: TemporalMCPClientOptions) {}
+
+  async tools(): Promise<ToolSet> {
+    workflow.log.info(`Options: ${this.options.activityOptions}`);
+    const activities = workflow.proxyActivities({ startToCloseTimeout: '10 minutes', ...this.options.activityOptions });
+
+    const listActivity = activities[this.options.name + '-listTools'];
+    const tools: Record<string, ListToolResult> = await listActivity!({ clientArgs: this.options.clientArgs });
+    return Object.fromEntries(
+      Object.entries(tools).map(([toolName, toolResult]) => [
+        toolName,
+        {
+          execute: async (args: any, options) => {
+            const activities = workflow.proxyActivities({
+              summary: toolName,
+              startToCloseTimeout: '10 minutes',
+              ...this.options,
+            });
+            const callActivity = activities[this.options.name + '-callTool']!;
+            return await callActivity({ name: toolName, args, options, clientArgs: this.options.clientArgs });
+          },
+          inputSchema: {
+            ...toolResult.inputSchema,
+            _type: undefined,
+            validate: undefined,
+            [Symbol.for('vercel.ai.schema')]: true,
+            [Symbol.for('vercel.ai.validator')]: true,
+          },
+          type: 'dynamic',
+        },
+      ])
+    );
+  }
+}

package/src/plugin.ts

@@ -0,0 +1,34 @@
+import type { ProviderV2 } from '@ai-sdk/provider';
+import { SimplePlugin } from '@temporalio/plugin';
+import { createActivities } from './activities';
+import type { McpClientFactories } from './mcp';
+
+/**
+ * Options for the AI SDK plugin
+ *
+ * @experimental The AI SDK plugin is an experimental feature; APIs may change without notice.
+ */
+export interface AiSdkPluginOptions {
+  modelProvider: ProviderV2;
+
+  /**
+   * This object contains a mapping of server names to functions which create MCP clients.
+   * Any TemporalMCPClient used in a workflow should have its associated servername listed in this object.
+   */
+  mcpClientFactories?: McpClientFactories;
+}
+
+/**
+ * A Temporal plugin that integrates AI SDK providers for use in workflows.
+ * This plugin creates activities that allow workflows to invoke AI models.
+ *
+ * @experimental The AI SDK plugin is an experimental feature; APIs may change without notice.
+ */
+export class AiSdkPlugin extends SimplePlugin {
+  constructor(options: AiSdkPluginOptions) {
+    super({
+      name: 'AiSDKPlugin',
+      activities: createActivities(options.modelProvider, options.mcpClientFactories),
+    });
+  }
+}

package/src/provider.ts

@@ -0,0 +1,79 @@
+import type {
+  EmbeddingModelV2,
+  ImageModelV2,
+  LanguageModelV2,
+  LanguageModelV2CallOptions,
+  ProviderV2,
+  SharedV2Headers,
+} from '@ai-sdk/provider';
+import * as workflow from '@temporalio/workflow';
+import { ActivityOptions } from '@temporalio/workflow';
+import { ApplicationFailure } from '@temporalio/common';
+import { InvokeModelResult } from './activities';
+
+/**
+ * A language model implementation that delegates AI model calls to Temporal activities.
+ * This allows workflows to invoke AI models through the Temporal execution model.
+ *
+ * @experimental The AI SDK integration is an experimental feature; APIs may change without notice.
+ */
+export class TemporalLanguageModel implements LanguageModelV2 {
+  readonly specificationVersion = 'v2';
+  readonly provider = 'temporal';
+  readonly supportedUrls = {};
+
+  constructor(
+    readonly modelId: string,
+    readonly options?: ActivityOptions
+  ) {}
+
+  async doGenerate(options: LanguageModelV2CallOptions): Promise<InvokeModelResult> {
+    const activities = workflow.proxyActivities({ startToCloseTimeout: '10 minutes', ...this.options });
+    const result = await activities.invokeModel!({ modelId: this.modelId, options });
+    if (result === undefined) {
+      throw ApplicationFailure.nonRetryable('Received undefined response from model activity.');
+    }
+    if (result.response !== undefined) {
+      result.response.timestamp = new Date(result.response.timestamp);
+    }
+    return result;
+  }
+
+  doStream(_options: LanguageModelV2CallOptions): PromiseLike<{
+    stream: any;
+    request?: { body?: unknown };
+    response?: { headers?: SharedV2Headers };
+  }> {
+    throw ApplicationFailure.nonRetryable('Streaming not supported.');
+  }
+}
+
+/**
+ * A Temporal-specific provider implementation that creates AI models which execute
+ * through Temporal activities. This provider integrates AI SDK models with Temporal's
+ * execution model to ensure reliable, durable AI model invocations.
+ *
+ * @experimental The AI SDK integration is an experimental feature; APIs may change without notice.
+ */
+export class TemporalProvider implements ProviderV2 {
+  constructor(readonly options?: ActivityOptions) {}
+
+  imageModel(_modelId: string): ImageModelV2 {
+    throw new Error('Not implemented');
+  }
+
+  languageModel(modelId: string): LanguageModelV2 {
+    return new TemporalLanguageModel(modelId, this.options);
+  }
+
+  textEmbeddingModel(_modelId: string): EmbeddingModelV2<string> {
+    throw new Error('Not implemented');
+  }
+}
+
+/**
+ * A singleton instance of TemporalProvider for convenient use in applications.
+ *
+ * @experimental The AI SDK integration is an experimental feature; APIs may change without notice.
+ */
+export const temporalProvider: TemporalProvider = new TemporalProvider();