@temporalio/ai-sdk 1.14.1 → 1.14.2

package/lib/activities.d.ts

@@ -1,46 +1,36 @@
-import type { LanguageModelV3CallOptions, LanguageModelV3GenerateResult, EmbeddingModelV3Result, SharedV3ProviderOptions, SharedV3Headers, ProviderV3 } from '@ai-sdk/provider';
-import type { FlexibleSchema, ToolExecutionOptions } from 'ai';
+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';
-/**
- * Arguments for invoking a language model activity.
- */
 export interface InvokeModelArgs {
     modelId: string;
-    options: LanguageModelV3CallOptions;
+    options: LanguageModelV2CallOptions;
 }
-/**
- * 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;
+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 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: FlexibleSchema<unknown>;
+    inputSchema: any;
 }
 export interface ListToolArgs {
-    clientArgs?: unknown;
+    clientArgs?: any;
 }
 export interface CallToolArgs {
-    clientArgs?: unknown;
+    clientArgs?: any;
     name: string;
-    input: unknown;
-    options: ToolExecutionOptions;
+    args: any;
+    options: ToolCallOptions;
 }
 /**
  * Creates Temporal activities for AI model invocation using the provided AI SDK provider.
@@ -53,4 +43,4 @@ export interface CallToolArgs {
  *
  * @experimental The AI SDK integration is an experimental feature; APIs may change without notice.
  */
-export declare function createActivities(provider: ProviderV3, mcpClientFactories?: McpClientFactories): object;
+export declare function createActivities(provider: ProviderV2, mcpClientFactories?: McpClientFactories): object;

package/lib/activities.js

@@ -19,14 +19,6 @@ 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]) => {
@@ -63,7 +55,7 @@ function activitiesForName(name, mcpClientFactory) {
             if (tool === undefined) {
                 throw common_1.ApplicationFailure.retryable(`Tool ${args.name} not found.`);
             }
-            return tool.execute(args.input, args.options);
+            return tool.execute(args.args, 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":";;AAuEA,4CAwBC;AAtFD,+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;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,KAAK,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC;QAChD,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":";;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

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

package/lib/index.js

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

package/lib/mcp.d.ts

@@ -1,4 +1,4 @@
-import { type ToolSet } from 'ai';
+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>;
@@ -12,7 +12,7 @@ export type McpClientFactories = {
  */
 export interface TemporalMCPClientOptions {
     readonly name: string;
-    readonly clientArgs?: unknown;
+    readonly clientArgs?: any;
     readonly activityOptions?: ActivityOptions;
 }
 /**

package/lib/mcp.js

@@ -24,7 +24,6 @@ var __importStar = (this && this.__importStar) || function (mod) {
 };
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.TemporalMCPClient = void 0;
-const ai_1 = require("ai");
 const workflow = __importStar(require("@temporalio/workflow"));
 /**
  * A Temporal MCP Client which uses activities to execute list tools and call tools.
@@ -41,25 +40,29 @@ class TemporalMCPClient {
         this.options = options;
     }
     async tools() {
-        const activities = workflow.proxyActivities({
-            startToCloseTimeout: '10 minutes',
-            ...this.options.activityOptions,
-        });
+        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 (input, options) => {
+                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, input, options, clientArgs: this.options.clientArgs });
+                    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,
                 },
-                inputSchema: (0, ai_1.jsonSchema)(toolResult.inputSchema),
                 type: 'dynamic',
             },
         ]));

package/lib/mcp.js.map

@@ -1 +1 @@
-{"version":3,"file":"mcp.js","sourceRoot":"","sources":["../src/mcp.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;AAAA,2BAA8C;AAG9C,+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;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,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,WAAW,EAAE,IAAA,eAAU,EAAC,UAAU,CAAC,WAA0B,CAAC;gBAC9D,IAAI,EAAE,SAAS;aAChB;SACF,CAAC,CACH,CAAC;IACJ,CAAC;CACF;AA9BD,8CA8BC"}
+{"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

@@ -1,4 +1,4 @@
-import type { ProviderV3 } from '@ai-sdk/provider';
+import type { ProviderV2 } 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: ProviderV3;
+    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.

package/lib/provider.d.ts

@@ -1,63 +1,29 @@
-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;
-}
+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 LanguageModelV3 {
+export declare class TemporalLanguageModel implements LanguageModelV2 {
     readonly modelId: string;
     readonly options?: ActivityOptions | undefined;
-    readonly specificationVersion = "v3";
+    readonly specificationVersion = "v2";
     readonly provider = "temporal";
+    readonly supportedUrls: {};
     constructor(modelId: string, options?: ActivityOptions | undefined);
-    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>;
+    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
@@ -66,14 +32,12 @@ export declare class TemporalEmbeddingModel implements EmbeddingModelV3 {
  *
  * @experimental The AI SDK integration is an experimental feature; APIs may change without notice.
  */
-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;
+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.

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.TemporalEmbeddingModel = exports.TemporalLanguageModel = void 0;
+exports.temporalProvider = exports.TemporalProvider = exports.TemporalLanguageModel = void 0;
 const workflow = __importStar(require("@temporalio/workflow"));
 const common_1 = require("@temporalio/common");
 /**
@@ -35,30 +35,21 @@ const common_1 = require("@temporalio/common");
 class TemporalLanguageModel {
     modelId;
     options;
-    specificationVersion = 'v3';
+    specificationVersion = 'v2';
     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) {
-            const timestamp = new Date(result.response.timestamp);
-            // Only set if it's a valid date
-            if (!isNaN(timestamp.getTime())) {
-                result.response.timestamp = timestamp;
-            }
+        if (result.response !== undefined) {
+            result.response.timestamp = new Date(result.response.timestamp);
         }
         return result;
     }
@@ -67,49 +58,6 @@ 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
@@ -119,7 +67,6 @@ exports.TemporalEmbeddingModel = TemporalEmbeddingModel;
  */
 class TemporalProvider {
     options;
-    specificationVersion = 'v3';
     constructor(options) {
         this.options = options;
     }
@@ -127,18 +74,9 @@ class TemporalProvider {
         throw new Error('Not implemented');
     }
     languageModel(modelId) {
-        return new TemporalLanguageModel(modelId, {
-            ...this.options?.default,
-            ...this.options?.languageModel,
-        });
-    }
-    embeddingModel(modelId) {
-        return new TemporalEmbeddingModel(modelId, {
-            ...this.options?.default,
-            ...this.options?.embeddingModel,
-        });
+        return new TemporalLanguageModel(modelId, this.options);
     }
-    transcriptionModel(_modelId) {
+    textEmbeddingModel(_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":";;;;;;;;;;;;;;;;;;;;;;;;;;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"}
+{"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

@@ -1,6 +1,6 @@
 {
   "name": "@temporalio/ai-sdk",
-  "version": "1.14.1",
+  "version": "1.14.2",
   "description": "Temporal AI SDK integration package",
   "main": "lib/index.js",
   "types": "./lib/index.d.ts",
@@ -9,22 +9,21 @@
     "workflow",
     "ai",
     "ai-sdk",
-    "llm",
-    "embeddings"
+    "llm"
   ],
   "author": "Temporal Technologies Inc. <sdk@temporal.io>",
   "license": "MIT",
   "dependencies": {
-    "@temporalio/plugin": "1.14.1",
-    "@temporalio/workflow": "1.14.1",
     "@ungap/structured-clone": "^1.3.0",
     "headers-polyfill": "^4.0.3",
-    "web-streams-polyfill": "^4.2.0"
+    "web-streams-polyfill": "^4.2.0",
+    "@temporalio/plugin": "1.14.2",
+    "@temporalio/workflow": "1.14.2"
   },
   "peerDependencies": {
-    "@ai-sdk/mcp": "^1.0.0",
-    "@ai-sdk/provider": "^3.0.0",
-    "ai": "^6.0.0"
+    "@ai-sdk/mcp": "^0.0.8",
+    "@ai-sdk/provider": "^2.0.0",
+    "ai": "^5.0.91"
   },
   "engines": {
     "node": ">= 18.0.0"
@@ -44,6 +43,5 @@
   "files": [
     "src",
     "lib"
-  ],
-  "gitHead": "42bf6087946b7bbda61c7856bc0723c1616f9973"
-}
+  ]
+}

package/src/activities.ts

@@ -1,61 +1,47 @@
 import type {
-  LanguageModelV3CallOptions,
-  LanguageModelV3GenerateResult,
-  EmbeddingModelV3Result,
-  SharedV3ProviderOptions,
-  SharedV3Headers,
-  ProviderV3,
+  LanguageModelV2CallOptions,
+  LanguageModelV2CallWarning,
+  LanguageModelV2Content,
+  LanguageModelV2FinishReason,
+  LanguageModelV2ResponseMetadata,
+  LanguageModelV2Usage,
+  ProviderV2,
+  SharedV2Headers,
+  SharedV2ProviderMetadata,
 } from '@ai-sdk/provider';
-import type { FlexibleSchema, ToolExecutionOptions } from 'ai';
+import type { ToolCallOptions } 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: LanguageModelV3CallOptions;
+  options: LanguageModelV2CallOptions;
 }
 
-/**
- * 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;
+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 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: FlexibleSchema<unknown>;
+  inputSchema: any;
 }
 
 export interface ListToolArgs {
-  clientArgs?: unknown;
+  clientArgs?: any;
 }
 
 export interface CallToolArgs {
-  clientArgs?: unknown;
+  clientArgs?: any;
   name: string;
-  input: unknown;
-  options: ToolExecutionOptions;
+  args: any;
+  options: ToolCallOptions;
 }
 
 /**
@@ -69,20 +55,12 @@ export interface CallToolArgs {
  *
  * @experimental The AI SDK integration is an experimental feature; APIs may change without notice.
  */
-export function createActivities(provider: ProviderV3, mcpClientFactories?: McpClientFactories): object {
+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);
     },
-    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]) => {
@@ -114,7 +92,7 @@ function activitiesForName(name: string, mcpClientFactory: McpClientFactory): ob
       await mcpClient.close();
     }
   }
-  async function callToolActivity(args: CallToolArgs): Promise<unknown> {
+  async function callToolActivity(args: CallToolArgs): Promise<any> {
     const mcpClient = await mcpClientFactory(args.clientArgs);
     try {
       const tools = await mcpClient.tools();
@@ -122,7 +100,7 @@ function activitiesForName(name: string, mcpClientFactory: McpClientFactory): ob
       if (tool === undefined) {
         throw ApplicationFailure.retryable(`Tool ${args.name} not found.`);
       }
-      return tool.execute(args.input, args.options);
+      return tool.execute(args.args, args.options);
     } finally {
       await mcpClient.close();
     }

package/src/index.ts

@@ -1,7 +1,6 @@
 // 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/mcp.ts

@@ -1,5 +1,4 @@
-import { jsonSchema, type ToolSet } from 'ai';
-import type { JSONSchema7 } from '@ai-sdk/provider';
+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';
@@ -15,7 +14,8 @@ export type McpClientFactories = { [serviceName: string]: McpClientFactory };
  */
 export interface TemporalMCPClientOptions {
   readonly name: string;
-  readonly clientArgs?: unknown;
+  // eslint-disable-next-line @typescript-eslint/explicit-module-boundary-types
+  readonly clientArgs?: any;
   readonly activityOptions?: ActivityOptions;
 }
 
@@ -32,10 +32,8 @@ export class TemporalMCPClient {
   constructor(readonly options: TemporalMCPClientOptions) {}
 
   async tools(): Promise<ToolSet> {
-    const activities = workflow.proxyActivities({
-      startToCloseTimeout: '10 minutes',
-      ...this.options.activityOptions,
-    });
+    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 });
@@ -43,16 +41,22 @@ export class TemporalMCPClient {
       Object.entries(tools).map(([toolName, toolResult]) => [
         toolName,
         {
-          execute: async (input, options) => {
+          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, input, options, clientArgs: this.options.clientArgs });
+            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,
           },
-          inputSchema: jsonSchema(toolResult.inputSchema as JSONSchema7),
           type: 'dynamic',
         },
       ])

package/src/plugin.ts

@@ -1,4 +1,4 @@
-import type { ProviderV3 } from '@ai-sdk/provider';
+import type { ProviderV2 } 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: ProviderV3;
+  modelProvider: ProviderV2;
 
   /**
    * This object contains a mapping of server names to functions which create MCP clients.

package/src/provider.ts

@@ -1,41 +1,15 @@
 import type {
-  EmbeddingModelV3,
-  EmbeddingModelV3CallOptions,
-  EmbeddingModelV3Result,
-  ImageModelV3,
-  LanguageModelV3,
-  LanguageModelV3CallOptions,
-  LanguageModelV3GenerateResult,
-  LanguageModelV3StreamResult,
-  ProviderV3,
-  TranscriptionModelV3,
+  EmbeddingModelV2,
+  ImageModelV2,
+  LanguageModelV2,
+  LanguageModelV2CallOptions,
+  ProviderV2,
+  SharedV2Headers,
 } from '@ai-sdk/provider';
 import * as workflow from '@temporalio/workflow';
-import type { ActivityOptions } from '@temporalio/workflow';
+import { ActivityOptions } from '@temporalio/workflow';
 import { ApplicationFailure } from '@temporalio/common';
-
-/**
- * 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;
-}
+import { InvokeModelResult } from './activities';
 
 /**
  * A language model implementation that delegates AI model calls to Temporal activities.
@@ -43,86 +17,37 @@ export interface TemporalProviderOptions {
  *
  * @experimental The AI SDK integration is an experimental feature; APIs may change without notice.
  */
-export class TemporalLanguageModel implements LanguageModelV3 {
-  readonly specificationVersion = 'v3';
+export class TemporalLanguageModel implements LanguageModelV2 {
+  readonly specificationVersion = 'v2';
   readonly provider = 'temporal';
+  readonly supportedUrls = {};
 
   constructor(
     readonly modelId: string,
     readonly options?: ActivityOptions
   ) {}
 
-  get supportedUrls(): Record<string, RegExp[]> {
-    return {};
-  }
-
-  async doGenerate(options: LanguageModelV3CallOptions): Promise<LanguageModelV3GenerateResult> {
-    const activities = workflow.proxyActivities({
-      startToCloseTimeout: '10 minutes',
-      ...this.options,
-    });
+  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) {
-      const timestamp = new Date(result.response.timestamp);
-      // Only set if it's a valid date
-      if (!isNaN(timestamp.getTime())) {
-        result.response.timestamp = timestamp;
-      }
+    if (result.response !== undefined) {
+      result.response.timestamp = new Date(result.response.timestamp);
     }
     return result;
   }
 
-  doStream(_options: LanguageModelV3CallOptions): PromiseLike<LanguageModelV3StreamResult> {
+  doStream(_options: LanguageModelV2CallOptions): PromiseLike<{
+    stream: any;
+    request?: { body?: unknown };
+    response?: { headers?: SharedV2Headers };
+  }> {
     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
@@ -130,30 +55,18 @@ export class TemporalEmbeddingModel implements EmbeddingModelV3 {
  *
  * @experimental The AI SDK integration is an experimental feature; APIs may change without notice.
  */
-export class TemporalProvider implements ProviderV3 {
-  readonly specificationVersion = 'v3';
+export class TemporalProvider implements ProviderV2 {
+  constructor(readonly options?: ActivityOptions) {}
 
-  constructor(readonly options?: TemporalProviderOptions) {}
-
-  imageModel(_modelId: string): ImageModelV3 {
+  imageModel(_modelId: string): ImageModelV2 {
     throw new Error('Not implemented');
   }
 
-  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,
-    });
+  languageModel(modelId: string): LanguageModelV2 {
+    return new TemporalLanguageModel(modelId, this.options);
   }
 
-  transcriptionModel(_modelId: string): TranscriptionModelV3 {
+  textEmbeddingModel(_modelId: string): EmbeddingModelV2<string> {
     throw new Error('Not implemented');
   }
 }