@temporalio/ai-sdk 1.14.2 → 1.15.1

package/lib/activities.d.ts

@@ -1,36 +1,46 @@
-import type { LanguageModelV2CallOptions, LanguageModelV2CallWarning, LanguageModelV2Content, LanguageModelV2FinishReason, LanguageModelV2ResponseMetadata, LanguageModelV2Usage, ProviderV2, SharedV2Headers, SharedV2ProviderMetadata } from '@ai-sdk/provider';
-import type { ToolCallOptions } from 'ai';
+import type { LanguageModelV3CallOptions, LanguageModelV3GenerateResult, EmbeddingModelV3Result, SharedV3ProviderOptions, SharedV3Headers, ProviderV3 } from '@ai-sdk/provider';
+import { type Schema, type ToolExecutionOptions } from 'ai';
 import type { McpClientFactories } from './mcp';
+/**
+ * Arguments for invoking a language model activity.
+ */
 export interface InvokeModelArgs {
     modelId: string;
-    options: LanguageModelV2CallOptions;
+    options: LanguageModelV3CallOptions;
 }
-export interface InvokeModelResult {
-    content: Array<LanguageModelV2Content>;
-    finishReason: LanguageModelV2FinishReason;
-    usage: LanguageModelV2Usage;
-    providerMetadata?: SharedV2ProviderMetadata;
-    request?: {
-        body?: unknown;
-    };
-    response?: LanguageModelV2ResponseMetadata & {
-        headers?: SharedV2Headers;
-        body?: unknown;
-    };
-    warnings: Array<LanguageModelV2CallWarning>;
+/**
+ * Result from a language model invocation.
+ * This is an alias to the AI SDK's LanguageModelV3GenerateResult for type safety.
+ */
+export type InvokeModelResult = LanguageModelV3GenerateResult;
+/**
+ * Arguments for invoking an embedding model activity.
+ * Note: AbortSignal is not included as it cannot be serialized across activity boundaries.
+ * Temporal's workflow cancellation provides equivalent functionality.
+ */
+export interface InvokeEmbeddingModelArgs {
+    modelId: string;
+    values: string[];
+    providerOptions?: SharedV3ProviderOptions;
+    headers?: SharedV3Headers;
 }
+/**
+ * Result from an embedding model invocation.
+ * This is an alias to the AI SDK's EmbeddingModelV3Result for type safety.
+ */
+export type InvokeEmbeddingModelResult = EmbeddingModelV3Result;
 export interface ListToolResult {
     description?: string;
-    inputSchema: any;
+    inputSchema: Schema<unknown>;
 }
 export interface ListToolArgs {
-    clientArgs?: any;
+    clientArgs?: unknown;
 }
 export interface CallToolArgs {
-    clientArgs?: any;
+    clientArgs?: unknown;
     name: string;
-    args: any;
-    options: ToolCallOptions;
+    input: unknown;
+    options: ToolExecutionOptions;
 }
 /**
  * Creates Temporal activities for AI model invocation using the provided AI SDK provider.
@@ -43,4 +53,4 @@ export interface CallToolArgs {
  *
  * @experimental The AI SDK integration is an experimental feature; APIs may change without notice.
  */
-export declare function createActivities(provider: ProviderV2, mcpClientFactories?: McpClientFactories): object;
+export declare function createActivities(provider: ProviderV3, mcpClientFactories?: McpClientFactories): object;

package/lib/activities.js

@@ -1,6 +1,7 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.createActivities = createActivities;
+const ai_1 = require("ai");
 const common_1 = require("@temporalio/common");
 /**
  * Creates Temporal activities for AI model invocation using the provided AI SDK provider.
@@ -19,6 +20,14 @@ function createActivities(provider, mcpClientFactories) {
             const model = provider.languageModel(args.modelId);
             return await model.doGenerate(args.options);
         },
+        async invokeEmbeddingModel(args) {
+            const model = provider.embeddingModel(args.modelId);
+            return await model.doEmbed({
+                values: args.values,
+                providerOptions: args.providerOptions,
+                headers: args.headers,
+            });
+        },
     };
     if (mcpClientFactories !== undefined) {
         Object.entries(mcpClientFactories).forEach(([name, func]) => {
@@ -39,7 +48,8 @@ function activitiesForName(name, mcpClientFactory) {
                 k,
                 {
                     description: v.description,
-                    inputSchema: v.inputSchema,
+                    // Convert the FlexibleSchema to a Schema so that the shape is known outside the activity
+                    inputSchema: (0, ai_1.asSchema)(v.inputSchema),
                 },
             ]));
         }
@@ -55,7 +65,7 @@ function activitiesForName(name, mcpClientFactory) {
             if (tool === undefined) {
                 throw common_1.ApplicationFailure.retryable(`Tool ${args.name} not found.`);
             }
-            return tool.execute(args.args, args.options);
+            return await tool.execute(args.input, args.options);
         }
         finally {
             await mcpClient.close();

package/lib/activities.js.map

@@ -1 +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"}
+{"version":3,"file":"activities.js","sourceRoot":"","sources":["../src/activities.ts"],"names":[],"mappings":";;AAuEA,4CAwBC;AAvFD,2BAAsE;AACtE,+CAAwD;AAmDxD;;;;;;;;;;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;QACD,KAAK,CAAC,oBAAoB,CAAC,IAA8B;YACvD,MAAM,KAAK,GAAG,QAAQ,CAAC,cAAc,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;YACpD,OAAO,MAAM,KAAK,CAAC,OAAO,CAAC;gBACzB,MAAM,EAAE,IAAI,CAAC,MAAM;gBACnB,eAAe,EAAE,IAAI,CAAC,eAAe;gBACrC,OAAO,EAAE,IAAI,CAAC,OAAO;aACtB,CAAC,CAAC;QACL,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;YACtC,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,yFAAyF;oBACzF,WAAW,EAAE,IAAA,aAAQ,EAAC,CAAC,CAAC,WAAW,CAAC;iBACrC;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,MAAM,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,KAAK,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC;QACtD,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

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

package/lib/index.js

@@ -16,6 +16,7 @@ var __exportStar = (this && this.__exportStar) || function(m, exports) {
 Object.defineProperty(exports, "__esModule", { value: true });
 // eslint-disable-next-line import/no-unassigned-import
 require("./load-polyfills");
+__exportStar(require("./activities"), exports);
 __exportStar(require("./mcp"), exports);
 __exportStar(require("./plugin"), exports);
 __exportStar(require("./provider"), exports);

package/lib/index.js.map

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

package/lib/load-polyfills.js

@@ -11,7 +11,7 @@ if ((0, workflow_1.inWorkflowContext)()) {
     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
+        // eslint-disable-next-line @typescript-eslint/no-require-imports
         const structuredClone = require('@ungap/structured-clone');
         globalThis.structuredClone = structuredClone.default;
     }

package/lib/load-polyfills.js.map

@@ -1 +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"}
+{"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,iEAAiE;QACjE,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

@@ -12,7 +12,7 @@ export type McpClientFactories = {
  */
 export interface TemporalMCPClientOptions {
     readonly name: string;
-    readonly clientArgs?: any;
+    readonly clientArgs?: unknown;
     readonly activityOptions?: ActivityOptions;
 }
 /**

package/lib/mcp.js

@@ -40,22 +40,26 @@ class TemporalMCPClient {
         this.options = options;
     }
     async tools() {
-        workflow.log.info(`Options: ${this.options.activityOptions}`);
-        const activities = workflow.proxyActivities({ startToCloseTimeout: '10 minutes', ...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) => {
+                description: toolResult.description,
+                execute: async (input, 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 });
+                    return await callActivity({ name: toolName, input, options, clientArgs: this.options.clientArgs });
                 },
+                // Symbols and undefined values are lost on serialization, and need to be replaced to ensure the schema is used correctly
                 inputSchema: {
                     ...toolResult.inputSchema,
                     _type: undefined,

package/lib/mcp.js.map

@@ -1 +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"}
+{"version":3,"file":"mcp.js","sourceRoot":"","sources":["../src/mcp.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;AAEA,+DAAiD;AAkBjD;;;;;;;;GAQG;AACH,MAAa,iBAAiB;IACP;IAArB,YAAqB,OAAiC;QAAjC,YAAO,GAAP,OAAO,CAA0B;IAAG,CAAC;IAE1D,KAAK,CAAC,KAAK;QACT,MAAM,UAAU,GAAG,QAAQ,CAAC,eAAe,CAAC;YAC1C,mBAAmB,EAAE,YAAY;YACjC,GAAG,IAAI,CAAC,OAAO,CAAC,eAAe;SAChC,CAAC,CAAC;QAEH,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;QAE3G,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,WAAW,EAAE,UAAU,CAAC,WAAW;gBACnC,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,OAAO,EAAE,EAAE;oBAChC,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,KAAK,EAAE,OAAO,EAAE,UAAU,EAAE,IAAI,CAAC,OAAO,CAAC,UAAU,EAAE,CAAC,CAAC;gBACrG,CAAC;gBACD,yHAAyH;gBACzH,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;AAvCD,8CAuCC"}

package/lib/plugin.d.ts

@@ -1,4 +1,4 @@
-import type { ProviderV2 } from '@ai-sdk/provider';
+import type { ProviderV3 } from '@ai-sdk/provider';
 import { SimplePlugin } from '@temporalio/plugin';
 import type { McpClientFactories } from './mcp';
 /**
@@ -7,7 +7,7 @@ import type { McpClientFactories } from './mcp';
  * @experimental The AI SDK plugin is an experimental feature; APIs may change without notice.
  */
 export interface AiSdkPluginOptions {
-    modelProvider: ProviderV2;
+    modelProvider: ProviderV3;
     /**
      * 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.

package/lib/provider.d.ts

@@ -1,29 +1,63 @@
-import type { EmbeddingModelV2, ImageModelV2, LanguageModelV2, LanguageModelV2CallOptions, ProviderV2, SharedV2Headers } from '@ai-sdk/provider';
-import { ActivityOptions } from '@temporalio/workflow';
-import { InvokeModelResult } from './activities';
+import type { EmbeddingModelV3, EmbeddingModelV3CallOptions, EmbeddingModelV3Result, ImageModelV3, LanguageModelV3, LanguageModelV3CallOptions, LanguageModelV3GenerateResult, LanguageModelV3StreamResult, ProviderV3, TranscriptionModelV3 } from '@ai-sdk/provider';
+import type { ActivityOptions } from '@temporalio/workflow';
+/**
+ * Options for configuring the TemporalProvider with per-model activity settings.
+ */
+export interface TemporalProviderOptions {
+    /**
+     * Default activity options applied to all model types.
+     * These can be overridden by model-specific options.
+     */
+    default?: ActivityOptions;
+    /**
+     * Activity options specific to language model calls.
+     * Merged with default options, with these taking precedence.
+     */
+    languageModel?: ActivityOptions;
+    /**
+     * Activity options specific to embedding model calls.
+     * Merged with default options, with these taking precedence.
+     */
+    embeddingModel?: ActivityOptions;
+}
 /**
  * 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 {
+export declare class TemporalLanguageModel implements LanguageModelV3 {
     readonly modelId: string;
     readonly options?: ActivityOptions | undefined;
-    readonly specificationVersion = "v2";
+    readonly specificationVersion = "v3";
     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;
-        };
-    }>;
+    get supportedUrls(): Record<string, RegExp[]>;
+    doGenerate(options: LanguageModelV3CallOptions): Promise<LanguageModelV3GenerateResult>;
+    doStream(_options: LanguageModelV3CallOptions): PromiseLike<LanguageModelV3StreamResult>;
+}
+/**
+ * An embedding model implementation that delegates embedding generation to Temporal activities.
+ * This allows workflows to generate embeddings through the Temporal execution model.
+ *
+ * @experimental The AI SDK integration is an experimental feature; APIs may change without notice.
+ */
+export declare class TemporalEmbeddingModel implements EmbeddingModelV3 {
+    readonly modelId: string;
+    readonly options?: ActivityOptions | undefined;
+    readonly specificationVersion = "v3";
+    readonly provider = "temporal";
+    /**
+     * Undefined to let the underlying provider handle chunking, as it knows its own limits.
+     */
+    readonly maxEmbeddingsPerCall: undefined;
+    /**
+     * Indicates the underlying embedding model API can handle concurrent requests.
+     * Set to true since we delegate to the actual provider which manages its own concurrency.
+     */
+    readonly supportsParallelCalls = true;
+    constructor(modelId: string, options?: ActivityOptions | undefined);
+    doEmbed(options: EmbeddingModelV3CallOptions): Promise<EmbeddingModelV3Result>;
 }
 /**
  * A Temporal-specific provider implementation that creates AI models which execute
@@ -32,12 +66,14 @@ export declare class TemporalLanguageModel implements LanguageModelV2 {
  *
  * @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>;
+export declare class TemporalProvider implements ProviderV3 {
+    readonly options?: TemporalProviderOptions | undefined;
+    readonly specificationVersion = "v3";
+    constructor(options?: TemporalProviderOptions | undefined);
+    imageModel(_modelId: string): ImageModelV3;
+    languageModel(modelId: string): LanguageModelV3;
+    embeddingModel(modelId: string): EmbeddingModelV3;
+    transcriptionModel(_modelId: string): TranscriptionModelV3;
 }
 /**
  * A singleton instance of TemporalProvider for convenient use in applications.

package/lib/provider.js

@@ -23,7 +23,7 @@ var __importStar = (this && this.__importStar) || function (mod) {
     return result;
 };
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.temporalProvider = exports.TemporalProvider = exports.TemporalLanguageModel = void 0;
+exports.temporalProvider = exports.TemporalProvider = exports.TemporalEmbeddingModel = exports.TemporalLanguageModel = void 0;
 const workflow = __importStar(require("@temporalio/workflow"));
 const common_1 = require("@temporalio/common");
 /**
@@ -35,21 +35,30 @@ const common_1 = require("@temporalio/common");
 class TemporalLanguageModel {
     modelId;
     options;
-    specificationVersion = 'v2';
+    specificationVersion = 'v3';
     provider = 'temporal';
-    supportedUrls = {};
     constructor(modelId, options) {
         this.modelId = modelId;
         this.options = options;
     }
+    get supportedUrls() {
+        return {};
+    }
     async doGenerate(options) {
-        const activities = workflow.proxyActivities({ startToCloseTimeout: '10 minutes', ...this.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);
+        if (result.response !== undefined && result.response.timestamp) {
+            const timestamp = new Date(result.response.timestamp);
+            // Only set if it's a valid date
+            if (!isNaN(timestamp.getTime())) {
+                result.response.timestamp = timestamp;
+            }
         }
         return result;
     }
@@ -58,6 +67,49 @@ class TemporalLanguageModel {
     }
 }
 exports.TemporalLanguageModel = TemporalLanguageModel;
+/**
+ * An embedding model implementation that delegates embedding generation to Temporal activities.
+ * This allows workflows to generate embeddings through the Temporal execution model.
+ *
+ * @experimental The AI SDK integration is an experimental feature; APIs may change without notice.
+ */
+class TemporalEmbeddingModel {
+    modelId;
+    options;
+    specificationVersion = 'v3';
+    provider = 'temporal';
+    /**
+     * Undefined to let the underlying provider handle chunking, as it knows its own limits.
+     */
+    maxEmbeddingsPerCall = undefined;
+    /**
+     * Indicates the underlying embedding model API can handle concurrent requests.
+     * Set to true since we delegate to the actual provider which manages its own concurrency.
+     */
+    supportsParallelCalls = true;
+    constructor(modelId, options) {
+        this.modelId = modelId;
+        this.options = options;
+    }
+    async doEmbed(options) {
+        const activities = workflow.proxyActivities({
+            startToCloseTimeout: '10 minutes',
+            ...this.options,
+        });
+        const result = await activities.invokeEmbeddingModel({
+            modelId: this.modelId,
+            values: options.values,
+            providerOptions: options.providerOptions,
+            headers: options.headers,
+            // Note: abortSignal is not serializable, Temporal's cancellation handles this
+        });
+        if (result === undefined) {
+            throw common_1.ApplicationFailure.nonRetryable('Received undefined response from embedding model activity.');
+        }
+        return result;
+    }
+}
+exports.TemporalEmbeddingModel = TemporalEmbeddingModel;
 /**
  * A Temporal-specific provider implementation that creates AI models which execute
  * through Temporal activities. This provider integrates AI SDK models with Temporal's
@@ -67,6 +119,7 @@ exports.TemporalLanguageModel = TemporalLanguageModel;
  */
 class TemporalProvider {
     options;
+    specificationVersion = 'v3';
     constructor(options) {
         this.options = options;
     }
@@ -74,9 +127,18 @@ class TemporalProvider {
         throw new Error('Not implemented');
     }
     languageModel(modelId) {
-        return new TemporalLanguageModel(modelId, this.options);
+        return new TemporalLanguageModel(modelId, {
+            ...this.options?.default,
+            ...this.options?.languageModel,
+        });
+    }
+    embeddingModel(modelId) {
+        return new TemporalEmbeddingModel(modelId, {
+            ...this.options?.default,
+            ...this.options?.embeddingModel,
+        });
     }
-    textEmbeddingModel(_modelId) {
+    transcriptionModel(_modelId) {
         throw new Error('Not implemented');
     }
 }

package/lib/provider.js.map

@@ -1 +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"}
+{"version":3,"file":"provider.js","sourceRoot":"","sources":["../src/provider.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;AAYA,+DAAiD;AAEjD,+CAAwD;AAyBxD;;;;;GAKG;AACH,MAAa,qBAAqB;IAKrB;IACA;IALF,oBAAoB,GAAG,IAAI,CAAC;IAC5B,QAAQ,GAAG,UAAU,CAAC;IAE/B,YACW,OAAe,EACf,OAAyB;QADzB,YAAO,GAAP,OAAO,CAAQ;QACf,YAAO,GAAP,OAAO,CAAkB;IACjC,CAAC;IAEJ,IAAI,aAAa;QACf,OAAO,EAAE,CAAC;IACZ,CAAC;IAED,KAAK,CAAC,UAAU,CAAC,OAAmC;QAClD,MAAM,UAAU,GAAG,QAAQ,CAAC,eAAe,CAAC;YAC1C,mBAAmB,EAAE,YAAY;YACjC,GAAG,IAAI,CAAC,OAAO;SAChB,CAAC,CAAC;QACH,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,IAAI,MAAM,CAAC,QAAQ,CAAC,SAAS,EAAE,CAAC;YAC/D,MAAM,SAAS,GAAG,IAAI,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC;YACtD,gCAAgC;YAChC,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,OAAO,EAAE,CAAC,EAAE,CAAC;gBAChC,MAAM,CAAC,QAAQ,CAAC,SAAS,GAAG,SAAS,CAAC;YACxC,CAAC;QACH,CAAC;QACD,OAAO,MAAM,CAAC;IAChB,CAAC;IAED,QAAQ,CAAC,QAAoC;QAC3C,MAAM,2BAAkB,CAAC,YAAY,CAAC,0BAA0B,CAAC,CAAC;IACpE,CAAC;CACF;AAnCD,sDAmCC;AAED;;;;;GAKG;AACH,MAAa,sBAAsB;IActB;IACA;IAdF,oBAAoB,GAAG,IAAI,CAAC;IAC5B,QAAQ,GAAG,UAAU,CAAC;IAC/B;;OAEG;IACM,oBAAoB,GAAG,SAAS,CAAC;IAC1C;;;OAGG;IACM,qBAAqB,GAAG,IAAI,CAAC;IAEtC,YACW,OAAe,EACf,OAAyB;QADzB,YAAO,GAAP,OAAO,CAAQ;QACf,YAAO,GAAP,OAAO,CAAkB;IACjC,CAAC;IAEJ,KAAK,CAAC,OAAO,CAAC,OAAoC;QAChD,MAAM,UAAU,GAAG,QAAQ,CAAC,eAAe,CAAC;YAC1C,mBAAmB,EAAE,YAAY;YACjC,GAAG,IAAI,CAAC,OAAO;SAChB,CAAC,CAAC;QACH,MAAM,MAAM,GAAG,MAAM,UAAU,CAAC,oBAAqB,CAAC;YACpD,OAAO,EAAE,IAAI,CAAC,OAAO;YACrB,MAAM,EAAE,OAAO,CAAC,MAAM;YACtB,eAAe,EAAE,OAAO,CAAC,eAAe;YACxC,OAAO,EAAE,OAAO,CAAC,OAAO;YACxB,8EAA8E;SAC/E,CAAC,CAAC;QACH,IAAI,MAAM,KAAK,SAAS,EAAE,CAAC;YACzB,MAAM,2BAAkB,CAAC,YAAY,CAAC,4DAA4D,CAAC,CAAC;QACtG,CAAC;QACD,OAAO,MAAM,CAAC;IAChB,CAAC;CACF;AAnCD,wDAmCC;AAED;;;;;;GAMG;AACH,MAAa,gBAAgB;IAGN;IAFZ,oBAAoB,GAAG,IAAI,CAAC;IAErC,YAAqB,OAAiC;QAAjC,YAAO,GAAP,OAAO,CAA0B;IAAG,CAAC;IAE1D,UAAU,CAAC,QAAgB;QACzB,MAAM,IAAI,KAAK,CAAC,iBAAiB,CAAC,CAAC;IACrC,CAAC;IAED,aAAa,CAAC,OAAe;QAC3B,OAAO,IAAI,qBAAqB,CAAC,OAAO,EAAE;YACxC,GAAG,IAAI,CAAC,OAAO,EAAE,OAAO;YACxB,GAAG,IAAI,CAAC,OAAO,EAAE,aAAa;SAC/B,CAAC,CAAC;IACL,CAAC;IAED,cAAc,CAAC,OAAe;QAC5B,OAAO,IAAI,sBAAsB,CAAC,OAAO,EAAE;YACzC,GAAG,IAAI,CAAC,OAAO,EAAE,OAAO;YACxB,GAAG,IAAI,CAAC,OAAO,EAAE,cAAc;SAChC,CAAC,CAAC;IACL,CAAC;IAED,kBAAkB,CAAC,QAAgB;QACjC,MAAM,IAAI,KAAK,CAAC,iBAAiB,CAAC,CAAC;IACrC,CAAC;CACF;AA1BD,4CA0BC;AAED;;;;GAIG;AACU,QAAA,gBAAgB,GAAqB,IAAI,gBAAgB,EAAE,CAAC"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@temporalio/ai-sdk",
-  "version": "1.14.2",
+  "version": "1.15.1",
   "description": "Temporal AI SDK integration package",
   "main": "lib/index.js",
   "types": "./lib/index.d.ts",
@@ -9,7 +9,8 @@
     "workflow",
     "ai",
     "ai-sdk",
-    "llm"
+    "llm",
+    "embeddings"
   ],
   "author": "Temporal Technologies Inc. <sdk@temporal.io>",
   "license": "MIT",
@@ -17,16 +18,17 @@
     "@ungap/structured-clone": "^1.3.0",
     "headers-polyfill": "^4.0.3",
     "web-streams-polyfill": "^4.2.0",
-    "@temporalio/plugin": "1.14.2",
-    "@temporalio/workflow": "1.14.2"
+    "@temporalio/workflow": "1.15.1",
+    "@temporalio/common": "1.15.1",
+    "@temporalio/plugin": "1.15.1"
   },
   "peerDependencies": {
-    "@ai-sdk/mcp": "^0.0.8",
-    "@ai-sdk/provider": "^2.0.0",
-    "ai": "^5.0.91"
+    "@ai-sdk/mcp": "^1.0.0",
+    "@ai-sdk/provider": "^3.0.0",
+    "ai": "^6.0.0"
   },
   "engines": {
-    "node": ">= 18.0.0"
+    "node": ">= 20.0.0"
   },
   "bugs": {
     "url": "https://github.com/temporalio/sdk-typescript/issues"

package/src/activities.ts

@@ -1,47 +1,61 @@
 import type {
-  LanguageModelV2CallOptions,
-  LanguageModelV2CallWarning,
-  LanguageModelV2Content,
-  LanguageModelV2FinishReason,
-  LanguageModelV2ResponseMetadata,
-  LanguageModelV2Usage,
-  ProviderV2,
-  SharedV2Headers,
-  SharedV2ProviderMetadata,
+  LanguageModelV3CallOptions,
+  LanguageModelV3GenerateResult,
+  EmbeddingModelV3Result,
+  SharedV3ProviderOptions,
+  SharedV3Headers,
+  ProviderV3,
 } from '@ai-sdk/provider';
-import type { ToolCallOptions } from 'ai';
+import { asSchema, type Schema, type ToolExecutionOptions } from 'ai';
 import { ApplicationFailure } from '@temporalio/common';
 import type { McpClientFactories, McpClientFactory } from './mcp';
 
+/**
+ * Arguments for invoking a language model activity.
+ */
 export interface InvokeModelArgs {
   modelId: string;
-  options: LanguageModelV2CallOptions;
+  options: LanguageModelV3CallOptions;
 }
 
-export interface InvokeModelResult {
-  content: Array<LanguageModelV2Content>;
-  finishReason: LanguageModelV2FinishReason;
-  usage: LanguageModelV2Usage;
-  providerMetadata?: SharedV2ProviderMetadata;
-  request?: { body?: unknown };
-  response?: LanguageModelV2ResponseMetadata & { headers?: SharedV2Headers; body?: unknown };
-  warnings: Array<LanguageModelV2CallWarning>;
+/**
+ * Result from a language model invocation.
+ * This is an alias to the AI SDK's LanguageModelV3GenerateResult for type safety.
+ */
+export type InvokeModelResult = LanguageModelV3GenerateResult;
+
+/**
+ * Arguments for invoking an embedding model activity.
+ * Note: AbortSignal is not included as it cannot be serialized across activity boundaries.
+ * Temporal's workflow cancellation provides equivalent functionality.
+ */
+export interface InvokeEmbeddingModelArgs {
+  modelId: string;
+  values: string[];
+  providerOptions?: SharedV3ProviderOptions;
+  headers?: SharedV3Headers;
 }
 
+/**
+ * Result from an embedding model invocation.
+ * This is an alias to the AI SDK's EmbeddingModelV3Result for type safety.
+ */
+export type InvokeEmbeddingModelResult = EmbeddingModelV3Result;
+
 export interface ListToolResult {
   description?: string;
-  inputSchema: any;
+  inputSchema: Schema<unknown>;
 }
 
 export interface ListToolArgs {
-  clientArgs?: any;
+  clientArgs?: unknown;
 }
 
 export interface CallToolArgs {
-  clientArgs?: any;
+  clientArgs?: unknown;
   name: string;
-  args: any;
-  options: ToolCallOptions;
+  input: unknown;
+  options: ToolExecutionOptions;
 }
 
 /**
@@ -55,12 +69,20 @@ export interface CallToolArgs {
  *
  * @experimental The AI SDK integration is an experimental feature; APIs may change without notice.
  */
-export function createActivities(provider: ProviderV2, mcpClientFactories?: McpClientFactories): object {
+export function createActivities(provider: ProviderV3, mcpClientFactories?: McpClientFactories): object {
   let activities = {
     async invokeModel(args: InvokeModelArgs): Promise<InvokeModelResult> {
       const model = provider.languageModel(args.modelId);
       return await model.doGenerate(args.options);
     },
+    async invokeEmbeddingModel(args: InvokeEmbeddingModelArgs): Promise<InvokeEmbeddingModelResult> {
+      const model = provider.embeddingModel(args.modelId);
+      return await model.doEmbed({
+        values: args.values,
+        providerOptions: args.providerOptions,
+        headers: args.headers,
+      });
+    },
   };
   if (mcpClientFactories !== undefined) {
     Object.entries(mcpClientFactories).forEach(([name, func]) => {
@@ -78,13 +100,13 @@ function activitiesForName(name: string, mcpClientFactory: McpClientFactory): ob
     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,
+            // Convert the FlexibleSchema to a Schema so that the shape is known outside the activity
+            inputSchema: asSchema(v.inputSchema),
           },
         ])
       );
@@ -92,7 +114,7 @@ function activitiesForName(name: string, mcpClientFactory: McpClientFactory): ob
       await mcpClient.close();
     }
   }
-  async function callToolActivity(args: CallToolArgs): Promise<any> {
+  async function callToolActivity(args: CallToolArgs): Promise<unknown> {
     const mcpClient = await mcpClientFactory(args.clientArgs);
     try {
       const tools = await mcpClient.tools();
@@ -100,7 +122,7 @@ function activitiesForName(name: string, mcpClientFactory: McpClientFactory): ob
       if (tool === undefined) {
         throw ApplicationFailure.retryable(`Tool ${args.name} not found.`);
       }
-      return tool.execute(args.args, args.options);
+      return await tool.execute(args.input, args.options);
     } finally {
       await mcpClient.close();
     }

package/src/index.ts

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

package/src/load-polyfills.ts

@@ -11,7 +11,7 @@ if (inWorkflowContext()) {
   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
+    // eslint-disable-next-line @typescript-eslint/no-require-imports
     const structuredClone = require('@ungap/structured-clone');
     globalThis.structuredClone = structuredClone.default;
   }

package/src/mcp.ts

@@ -14,8 +14,7 @@ export type McpClientFactories = { [serviceName: string]: McpClientFactory };
  */
 export interface TemporalMCPClientOptions {
   readonly name: string;
-  // eslint-disable-next-line @typescript-eslint/explicit-module-boundary-types
-  readonly clientArgs?: any;
+  readonly clientArgs?: unknown;
   readonly activityOptions?: ActivityOptions;
 }
 
@@ -32,24 +31,29 @@ 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 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) => {
+          description: toolResult.description,
+          execute: async (input, 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 });
+            return await callActivity({ name: toolName, input, options, clientArgs: this.options.clientArgs });
           },
+          // Symbols and undefined values are lost on serialization, and need to be replaced to ensure the schema is used correctly
           inputSchema: {
             ...toolResult.inputSchema,
             _type: undefined,

package/src/plugin.ts

@@ -1,4 +1,4 @@
-import type { ProviderV2 } from '@ai-sdk/provider';
+import type { ProviderV3 } from '@ai-sdk/provider';
 import { SimplePlugin } from '@temporalio/plugin';
 import { createActivities } from './activities';
 import type { McpClientFactories } from './mcp';
@@ -9,7 +9,7 @@ import type { McpClientFactories } from './mcp';
  * @experimental The AI SDK plugin is an experimental feature; APIs may change without notice.
  */
 export interface AiSdkPluginOptions {
-  modelProvider: ProviderV2;
+  modelProvider: ProviderV3;
 
   /**
    * This object contains a mapping of server names to functions which create MCP clients.

package/src/provider.ts

@@ -1,15 +1,41 @@
 import type {
-  EmbeddingModelV2,
-  ImageModelV2,
-  LanguageModelV2,
-  LanguageModelV2CallOptions,
-  ProviderV2,
-  SharedV2Headers,
+  EmbeddingModelV3,
+  EmbeddingModelV3CallOptions,
+  EmbeddingModelV3Result,
+  ImageModelV3,
+  LanguageModelV3,
+  LanguageModelV3CallOptions,
+  LanguageModelV3GenerateResult,
+  LanguageModelV3StreamResult,
+  ProviderV3,
+  TranscriptionModelV3,
 } from '@ai-sdk/provider';
 import * as workflow from '@temporalio/workflow';
-import { ActivityOptions } from '@temporalio/workflow';
+import type { ActivityOptions } from '@temporalio/workflow';
 import { ApplicationFailure } from '@temporalio/common';
-import { InvokeModelResult } from './activities';
+
+/**
+ * Options for configuring the TemporalProvider with per-model activity settings.
+ */
+export interface TemporalProviderOptions {
+  /**
+   * Default activity options applied to all model types.
+   * These can be overridden by model-specific options.
+   */
+  default?: ActivityOptions;
+
+  /**
+   * Activity options specific to language model calls.
+   * Merged with default options, with these taking precedence.
+   */
+  languageModel?: ActivityOptions;
+
+  /**
+   * Activity options specific to embedding model calls.
+   * Merged with default options, with these taking precedence.
+   */
+  embeddingModel?: ActivityOptions;
+}
 
 /**
  * A language model implementation that delegates AI model calls to Temporal activities.
@@ -17,37 +43,86 @@ import { InvokeModelResult } from './activities';
  *
  * @experimental The AI SDK integration is an experimental feature; APIs may change without notice.
  */
-export class TemporalLanguageModel implements LanguageModelV2 {
-  readonly specificationVersion = 'v2';
+export class TemporalLanguageModel implements LanguageModelV3 {
+  readonly specificationVersion = 'v3';
   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 });
+  get supportedUrls(): Record<string, RegExp[]> {
+    return {};
+  }
+
+  async doGenerate(options: LanguageModelV3CallOptions): Promise<LanguageModelV3GenerateResult> {
+    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);
+    if (result.response !== undefined && result.response.timestamp) {
+      const timestamp = new Date(result.response.timestamp);
+      // Only set if it's a valid date
+      if (!isNaN(timestamp.getTime())) {
+        result.response.timestamp = timestamp;
+      }
     }
     return result;
   }
 
-  doStream(_options: LanguageModelV2CallOptions): PromiseLike<{
-    stream: any;
-    request?: { body?: unknown };
-    response?: { headers?: SharedV2Headers };
-  }> {
+  doStream(_options: LanguageModelV3CallOptions): PromiseLike<LanguageModelV3StreamResult> {
     throw ApplicationFailure.nonRetryable('Streaming not supported.');
   }
 }
 
+/**
+ * An embedding model implementation that delegates embedding generation to Temporal activities.
+ * This allows workflows to generate embeddings through the Temporal execution model.
+ *
+ * @experimental The AI SDK integration is an experimental feature; APIs may change without notice.
+ */
+export class TemporalEmbeddingModel implements EmbeddingModelV3 {
+  readonly specificationVersion = 'v3';
+  readonly provider = 'temporal';
+  /**
+   * Undefined to let the underlying provider handle chunking, as it knows its own limits.
+   */
+  readonly maxEmbeddingsPerCall = undefined;
+  /**
+   * Indicates the underlying embedding model API can handle concurrent requests.
+   * Set to true since we delegate to the actual provider which manages its own concurrency.
+   */
+  readonly supportsParallelCalls = true;
+
+  constructor(
+    readonly modelId: string,
+    readonly options?: ActivityOptions
+  ) {}
+
+  async doEmbed(options: EmbeddingModelV3CallOptions): Promise<EmbeddingModelV3Result> {
+    const activities = workflow.proxyActivities({
+      startToCloseTimeout: '10 minutes',
+      ...this.options,
+    });
+    const result = await activities.invokeEmbeddingModel!({
+      modelId: this.modelId,
+      values: options.values,
+      providerOptions: options.providerOptions,
+      headers: options.headers,
+      // Note: abortSignal is not serializable, Temporal's cancellation handles this
+    });
+    if (result === undefined) {
+      throw ApplicationFailure.nonRetryable('Received undefined response from embedding model activity.');
+    }
+    return result;
+  }
+}
+
 /**
  * A Temporal-specific provider implementation that creates AI models which execute
  * through Temporal activities. This provider integrates AI SDK models with Temporal's
@@ -55,18 +130,30 @@ export class TemporalLanguageModel implements LanguageModelV2 {
  *
  * @experimental The AI SDK integration is an experimental feature; APIs may change without notice.
  */
-export class TemporalProvider implements ProviderV2 {
-  constructor(readonly options?: ActivityOptions) {}
+export class TemporalProvider implements ProviderV3 {
+  readonly specificationVersion = 'v3';
 
-  imageModel(_modelId: string): ImageModelV2 {
+  constructor(readonly options?: TemporalProviderOptions) {}
+
+  imageModel(_modelId: string): ImageModelV3 {
     throw new Error('Not implemented');
   }
 
-  languageModel(modelId: string): LanguageModelV2 {
-    return new TemporalLanguageModel(modelId, this.options);
+  languageModel(modelId: string): LanguageModelV3 {
+    return new TemporalLanguageModel(modelId, {
+      ...this.options?.default,
+      ...this.options?.languageModel,
+    });
+  }
+
+  embeddingModel(modelId: string): EmbeddingModelV3 {
+    return new TemporalEmbeddingModel(modelId, {
+      ...this.options?.default,
+      ...this.options?.embeddingModel,
+    });
   }
 
-  textEmbeddingModel(_modelId: string): EmbeddingModelV2<string> {
+  transcriptionModel(_modelId: string): TranscriptionModelV3 {
     throw new Error('Not implemented');
   }
 }