workers-ai-provider 2.0.2 → 3.0.1

package/package.json

@@ -2,7 +2,7 @@
   "name": "workers-ai-provider",
   "description": "Workers AI Provider for the vercel AI SDK",
   "type": "module",
-  "version": "2.0.2",
+  "version": "3.0.1",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",
   "repository": {
@@ -12,7 +12,7 @@
   "bugs": {
     "url": "https://github.com/cloudflare/ai/issues"
   },
-  "authors": "Sunil Pai <spai@cloudflare.com>",
+  "authors": "Cloudflare Inc.",
   "license": "MIT",
   "files": [
     "dist",
@@ -30,13 +30,18 @@
     "chat",
     "serverless"
   ],
-  "dependencies": {
-    "@ai-sdk/provider": "^2.0.0",
-    "@ai-sdk/provider-utils": "^3.0.19"
+  "peerDependencies": {
+    "@ai-sdk/provider": "^3.0.0",
+    "@ai-sdk/provider-utils": "^4.0.0",
+    "ai": "^6.0.0",
+    "zod": "^3.25.0 || ^4.0.0"
   },
   "devDependencies": {
+    "@ai-sdk/provider": "^3.0.0",
+    "@ai-sdk/provider-utils": "^4.0.0",
     "@cloudflare/workers-types": "^4.20251221.0",
-    "zod": "^3.25.76"
+    "ai": "^6.0.0",
+    "zod": "^4.1.8"
   },
   "scripts": {
     "build": "rm -rf dist && tsup src/index.ts --dts --sourcemap --format esm --target es2020",

package/src/autorag-chat-language-model.ts

@@ -1,8 +1,4 @@
-import type {
-	LanguageModelV2,
-	LanguageModelV2CallWarning,
-	LanguageModelV2StreamPart,
-} from "@ai-sdk/provider";
+import type { LanguageModelV3, SharedV3Warning, LanguageModelV3StreamPart } from "@ai-sdk/provider";
 
 import type { AutoRAGChatSettings } from "./autorag-chat-settings";
 import { convertToWorkersAIChatMessages } from "./convert-to-workersai-chat-messages";
@@ -17,8 +13,8 @@ type AutoRAGChatConfig = {
 	gateway?: GatewayOptions;
 };
 
-export class AutoRAGChatLanguageModel implements LanguageModelV2 {
-	readonly specificationVersion = "v2";
+export class AutoRAGChatLanguageModel implements LanguageModelV3 {
+	readonly specificationVersion = "v3";
 	readonly defaultObjectGenerationMode = "json";
 
 	readonly supportedUrls: Record<string, RegExp[]> | PromiseLike<Record<string, RegExp[]>> = {
@@ -51,20 +47,20 @@ export class AutoRAGChatLanguageModel implements LanguageModelV2 {
 		toolChoice,
 		frequencyPenalty,
 		presencePenalty,
-	}: Parameters<LanguageModelV2["doGenerate"]>[0]) {
-		const warnings: LanguageModelV2CallWarning[] = [];
+	}: Parameters<LanguageModelV3["doGenerate"]>[0]) {
+		const warnings: SharedV3Warning[] = [];
 
 		if (frequencyPenalty != null) {
 			warnings.push({
-				setting: "frequencyPenalty",
-				type: "unsupported-setting",
+				feature: "frequencyPenalty",
+				type: "unsupported",
 			});
 		}
 
 		if (presencePenalty != null) {
 			warnings.push({
-				setting: "presencePenalty",
-				type: "unsupported-setting",
+				feature: "presencePenalty",
+				type: "unsupported",
 			});
 		}
 
@@ -79,7 +75,10 @@ export class AutoRAGChatLanguageModel implements LanguageModelV2 {
 		switch (type) {
 			case "text": {
 				return {
-					args: { ...baseArgs, ...prepareToolsAndToolChoice(tools, toolChoice) },
+					args: {
+						...baseArgs,
+						...prepareToolsAndToolChoice(tools, toolChoice),
+					},
 					warnings,
 				};
 			}
@@ -106,8 +105,8 @@ export class AutoRAGChatLanguageModel implements LanguageModelV2 {
 	}
 
 	async doGenerate(
-		options: Parameters<LanguageModelV2["doGenerate"]>[0],
-	): Promise<Awaited<ReturnType<LanguageModelV2["doGenerate"]>>> {
+		options: Parameters<LanguageModelV3["doGenerate"]>[0],
+	): Promise<Awaited<ReturnType<LanguageModelV3["doGenerate"]>>> {
 		const { warnings } = this.getArgs(options);
 		const { messages } = convertToWorkersAIChatMessages(options.prompt);
 
@@ -116,7 +115,7 @@ export class AutoRAGChatLanguageModel implements LanguageModelV2 {
 		});
 
 		return {
-			finishReason: "stop",
+			finishReason: { unified: "stop", raw: "stop" },
 
 			content: [
 				...output.data.map(({ file_id, filename, score }) => ({
@@ -140,14 +139,14 @@ export class AutoRAGChatLanguageModel implements LanguageModelV2 {
 	}
 
 	async doStream(
-		options: Parameters<LanguageModelV2["doStream"]>[0],
-	): Promise<Awaited<ReturnType<LanguageModelV2["doStream"]>>> {
+		options: Parameters<LanguageModelV3["doStream"]>[0],
+	): Promise<Awaited<ReturnType<LanguageModelV3["doStream"]>>> {
 		const { args, warnings } = this.getArgs(options);
 		const { messages } = convertToWorkersAIChatMessages(options.prompt);
 
 		const query = messages.map(({ content, role }) => `${role}: ${content}`).join("\n\n");
 
-		// Get the underlying streaming response (assume this returns a ReadableStream<LanguageModelV2StreamPart>)
+		// Get the underlying streaming response (assume this returns a ReadableStream<LanguageModelV3StreamPart>)
 		const response = await this.config.binding.aiSearch({
 			query,
 			stream: true,
@@ -155,12 +154,12 @@ export class AutoRAGChatLanguageModel implements LanguageModelV2 {
 
 		// Create a new stream that first emits the stream-start part with warnings,
 		// then pipes through the rest of the response stream
-		const stream = new ReadableStream<LanguageModelV2StreamPart>({
+		const stream = new ReadableStream<LanguageModelV3StreamPart>({
 			start(controller) {
 				// Emit the stream-start part with warnings
 				controller.enqueue({
 					type: "stream-start",
-					warnings: warnings as LanguageModelV2CallWarning[],
+					warnings: warnings as SharedV3Warning[],
 				});
 
 				// Pipe the rest of the response stream

package/src/convert-to-workersai-chat-messages.ts

@@ -1,19 +1,19 @@
-import type { LanguageModelV2Prompt, SharedV2ProviderMetadata } from "@ai-sdk/provider";
+import type { LanguageModelV3Prompt, SharedV3ProviderOptions } from "@ai-sdk/provider";
 import type { WorkersAIChatPrompt } from "./workersai-chat-prompt";
 
-export function convertToWorkersAIChatMessages(prompt: LanguageModelV2Prompt): {
+export function convertToWorkersAIChatMessages(prompt: LanguageModelV3Prompt): {
 	messages: WorkersAIChatPrompt;
 	images: {
-		mimeType: string | undefined;
+		mediaType: string | undefined;
 		image: Uint8Array;
-		providerOptions: SharedV2ProviderMetadata | undefined;
+		providerOptions: SharedV3ProviderOptions | undefined;
 	}[];
 } {
 	const messages: WorkersAIChatPrompt = [];
 	const images: {
-		mimeType: string | undefined;
+		mediaType: string | undefined;
 		image: Uint8Array;
-		providerOptions: SharedV2ProviderMetadata | undefined;
+		providerOptions: SharedV3ProviderOptions | undefined;
 	}[] = [];
 
 	for (const { role, content } of prompt) {
@@ -38,7 +38,7 @@ export function convertToWorkersAIChatMessages(prompt: LanguageModelV2Prompt): {
 										// For Llama 3.2 Vision model, which needs array of integers
 										images.push({
 											image: part.data,
-											mimeType: part.mediaType,
+											mediaType: part.mediaType,
 											providerOptions: part.providerOptions,
 										});
 									}
@@ -74,6 +74,11 @@ export function convertToWorkersAIChatMessages(prompt: LanguageModelV2Prompt): {
 							break;
 						}
 
+						case "file": {
+							// Handle file parts in assistant messages (V3)
+							break;
+						}
+
 						case "tool-call": {
 							text = JSON.stringify({
 								name: part.toolName,
@@ -90,9 +95,17 @@ export function convertToWorkersAIChatMessages(prompt: LanguageModelV2Prompt): {
 							});
 							break;
 						}
+
+						case "tool-result": {
+							// Handle tool results in assistant messages (V3)
+							break;
+						}
+
 						default: {
-							const exhaustiveCheck = part;
-							throw new Error(`Unsupported part type: ${exhaustiveCheck.type}`);
+							const exhaustiveCheck = part satisfies never;
+							throw new Error(
+								`Unsupported part type: ${(exhaustiveCheck as { type: string }).type}`,
+							);
 						}
 					}
 				}
@@ -115,12 +128,15 @@ export function convertToWorkersAIChatMessages(prompt: LanguageModelV2Prompt): {
 
 			case "tool": {
 				for (const [index, toolResponse] of content.entries()) {
-					messages.push({
-						content: JSON.stringify(toolResponse.output),
-						name: toolResponse.toolName,
-						tool_call_id: `functions.${toolResponse.toolName}:${index}`,
-						role: "tool",
-					});
+					if (toolResponse.type === "tool-result") {
+						messages.push({
+							content: JSON.stringify(toolResponse.output),
+							name: toolResponse.toolName,
+							tool_call_id: `functions.${toolResponse.toolName}:${index}`,
+							role: "tool",
+						});
+					}
+					// Skip tool-approval-response parts as they're not supported by Workers AI
 				}
 				break;
 			}

package/src/map-workersai-finish-reason.ts

@@ -1,6 +1,6 @@
-import type { LanguageModelV2FinishReason } from "@ai-sdk/provider";
+import type { LanguageModelV3FinishReason } from "@ai-sdk/provider";
 
-export function mapWorkersAIFinishReason(finishReasonOrResponse: any): LanguageModelV2FinishReason {
+export function mapWorkersAIFinishReason(finishReasonOrResponse: any): LanguageModelV3FinishReason {
 	let finishReason: string | null | undefined;
 
 	// If it's a string/null/undefined, use it directly (original behavior)
@@ -26,22 +26,23 @@ export function mapWorkersAIFinishReason(finishReasonOrResponse: any): LanguageM
 		}
 	}
 
+	const raw = finishReason ?? "stop";
+
 	switch (finishReason) {
 		case "stop":
-			return "stop";
+			return { unified: "stop", raw };
 		case "length":
 		case "model_length":
-			return "length";
+			return { unified: "length", raw };
 		case "tool_calls":
-			return "tool-calls";
+			return { unified: "tool-calls", raw };
 		case "error":
-			return "error";
+			return { unified: "error", raw };
 		case "other":
-			return "other";
 		case "unknown":
-			return "unknown";
+			return { unified: "other", raw };
 		default:
 			// Default to `stop` for backwards compatibility
-			return "stop";
+			return { unified: "stop", raw };
 	}
 }

package/src/map-workersai-usage.ts

@@ -1,4 +1,8 @@
-export function mapWorkersAIUsage(output: AiTextGenerationOutput | AiTextToImageOutput) {
+import type { LanguageModelV3Usage } from "@ai-sdk/provider";
+
+export function mapWorkersAIUsage(
+	output: AiTextGenerationOutput | AiTextToImageOutput,
+): LanguageModelV3Usage {
 	const usage = (
 		output as {
 			usage: { prompt_tokens: number; completion_tokens: number };
@@ -9,8 +13,17 @@ export function mapWorkersAIUsage(output: AiTextGenerationOutput | AiTextToImage
 	};
 
 	return {
-		outputTokens: usage.completion_tokens,
-		inputTokens: usage.prompt_tokens,
-		totalTokens: usage.prompt_tokens + usage.completion_tokens,
+		outputTokens: {
+			total: usage.completion_tokens,
+			text: undefined,
+			reasoning: undefined,
+		},
+		inputTokens: {
+			total: usage.prompt_tokens,
+			noCache: undefined,
+			cacheRead: undefined,
+			cacheWrite: undefined,
+		},
+		raw: { total: usage.prompt_tokens + usage.completion_tokens },
 	};
 }

package/src/streaming.ts

@@ -1,4 +1,4 @@
-import type { LanguageModelV2StreamPart } from "@ai-sdk/provider";
+import type { LanguageModelV3StreamPart, LanguageModelV3Usage } from "@ai-sdk/provider";
 import { generateId } from "ai";
 import { events } from "fetch-event-stream";
 import { mapWorkersAIUsage } from "./map-workersai-usage";
@@ -6,14 +6,25 @@ import { processPartialToolCalls } from "./utils";
 
 export function getMappedStream(response: Response) {
 	const chunkEvent = events(response);
-	let usage = { outputTokens: 0, inputTokens: 0, totalTokens: 0 };
+	let usage: LanguageModelV3Usage = {
+		outputTokens: { total: 0, text: undefined, reasoning: undefined },
+		inputTokens: {
+			total: 0,
+			noCache: undefined,
+			cacheRead: undefined,
+			cacheWrite: undefined,
+		},
+		raw: {
+			totalTokens: 0,
+		},
+	};
 	const partialToolCalls: any[] = [];
 
 	// Track start/delta/end IDs per v5 streaming protocol
 	let textId: string | null = null;
 	let reasoningId: string | null = null;
 
-	return new ReadableStream<LanguageModelV2StreamPart>({
+	return new ReadableStream<LanguageModelV3StreamPart>({
 		async start(controller) {
 			for await (const event of chunkEvent) {
 				if (!event.data) {
@@ -90,7 +101,7 @@ export function getMappedStream(response: Response) {
 			}
 
 			controller.enqueue({
-				finishReason: "stop",
+				finishReason: { unified: "stop", raw: "stop" },
 				type: "finish",
 				usage: usage,
 			});

package/src/utils.ts

@@ -1,4 +1,4 @@
-import type { LanguageModelV2, LanguageModelV2ToolCall } from "@ai-sdk/provider";
+import type { LanguageModelV3, LanguageModelV3ToolCall } from "@ai-sdk/provider";
 import { generateId } from "ai";
 
 /**
@@ -67,7 +67,6 @@ export function createRun(config: CreateRunConfig): AiRun {
 		inputs: AiModels[Name]["inputs"],
 		options?: AiOptions & Record<string, StringLike>,
 	): Promise<Response | ReadableStream<Uint8Array> | AiModels[Name]["postProcessedOutputs"]> {
-		// biome-ignore lint/correctness/noUnusedVariables: they need to be destructured
 		const { gateway, prefix, extraHeaders, returnRawResponse, ...passthroughOptions } =
 			options || {};
 
@@ -128,8 +127,8 @@ export function createRun(config: CreateRunConfig): AiRun {
 }
 
 export function prepareToolsAndToolChoice(
-	tools: Parameters<LanguageModelV2["doGenerate"]>[0]["tools"],
-	toolChoice: Parameters<LanguageModelV2["doGenerate"]>[0]["toolChoice"],
+	tools: Parameters<LanguageModelV3["doGenerate"]>[0]["tools"],
+	toolChoice: Parameters<LanguageModelV3["doGenerate"]>[0]["toolChoice"],
 ) {
 	if (tools == null) {
 		return { tool_choice: undefined, tools: undefined };
@@ -213,7 +212,7 @@ function mergePartialToolCalls(partialCalls: any[]) {
 	return Object.values(mergedCallsByIndex);
 }
 
-function processToolCall(toolCall: any): LanguageModelV2ToolCall {
+function processToolCall(toolCall: any): LanguageModelV3ToolCall {
 	// Check for OpenAI format tool calls first
 	if (toolCall.function && toolCall.id) {
 		return {
@@ -237,7 +236,7 @@ function processToolCall(toolCall: any): LanguageModelV2ToolCall {
 	};
 }
 
-export function processToolCalls(output: any): LanguageModelV2ToolCall[] {
+export function processToolCalls(output: any): LanguageModelV3ToolCall[] {
 	if (output.tool_calls && Array.isArray(output.tool_calls)) {
 		return output.tool_calls.map((toolCall: any) => {
 			const processedToolCall = processToolCall(toolCall);

package/src/workers-ai-embedding-model.ts

@@ -1,4 +1,9 @@
-import { type EmbeddingModelV2, TooManyEmbeddingValuesForCallError } from "@ai-sdk/provider";
+import type {
+	EmbeddingModelV3,
+	EmbeddingModelV3CallOptions,
+	EmbeddingModelV3Result,
+} from "@ai-sdk/provider";
+import { TooManyEmbeddingValuesForCallError } from "@ai-sdk/provider";
 import type { StringLike } from "./utils";
 import type { EmbeddingModels } from "./workersai-models";
 
@@ -19,12 +24,12 @@ export type WorkersAIEmbeddingSettings = {
 	[key: string]: StringLike;
 };
 
-export class WorkersAIEmbeddingModel implements EmbeddingModelV2<string> {
+export class WorkersAIEmbeddingModel implements EmbeddingModelV3 {
 	/**
-	 * Semantic version of the {@link EmbeddingModelV1} specification implemented
+	 * Semantic version of the {@link EmbeddingModelV3} specification implemented
 	 * by this class. It never changes.
 	 */
-	readonly specificationVersion = "v2";
+	readonly specificationVersion = "v3";
 	readonly modelId: EmbeddingModels;
 	private readonly config: WorkersAIEmbeddingConfig;
 	private readonly settings: WorkersAIEmbeddingSettings;
@@ -56,11 +61,7 @@ export class WorkersAIEmbeddingModel implements EmbeddingModelV2<string> {
 		this.config = config;
 	}
 
-	async doEmbed({
-		values,
-	}: Parameters<EmbeddingModelV2<string>["doEmbed"]>[0]): Promise<
-		Awaited<ReturnType<EmbeddingModelV2<string>["doEmbed"]>>
-	> {
+	async doEmbed({ values }: EmbeddingModelV3CallOptions): Promise<EmbeddingModelV3Result> {
 		if (values.length > this.maxEmbeddingsPerCall) {
 			throw new TooManyEmbeddingValuesForCallError({
 				maxEmbeddingsPerCall: this.maxEmbeddingsPerCall,
@@ -85,6 +86,7 @@ export class WorkersAIEmbeddingModel implements EmbeddingModelV2<string> {
 
 		return {
 			embeddings: response.data,
+			warnings: [],
 		};
 	}
 }

package/src/workersai-chat-language-model.ts

@@ -1,8 +1,4 @@
-import type {
-	LanguageModelV2,
-	LanguageModelV2CallWarning,
-	LanguageModelV2StreamPart,
-} from "@ai-sdk/provider";
+import type { LanguageModelV3, SharedV3Warning, LanguageModelV3StreamPart } from "@ai-sdk/provider";
 import { generateId } from "ai";
 import { convertToWorkersAIChatMessages } from "./convert-to-workersai-chat-messages";
 import { mapWorkersAIFinishReason } from "./map-workersai-finish-reason";
@@ -23,8 +19,8 @@ type WorkersAIChatConfig = {
 	gateway?: GatewayOptions;
 };
 
-export class WorkersAIChatLanguageModel implements LanguageModelV2 {
-	readonly specificationVersion = "v2";
+export class WorkersAIChatLanguageModel implements LanguageModelV3 {
+	readonly specificationVersion = "v3";
 	readonly defaultObjectGenerationMode = "json";
 
 	readonly supportedUrls: Record<string, RegExp[]> | PromiseLike<Record<string, RegExp[]>> = {
@@ -60,22 +56,22 @@ export class WorkersAIChatLanguageModel implements LanguageModelV2 {
 		frequencyPenalty,
 		presencePenalty,
 		seed,
-	}: Parameters<LanguageModelV2["doGenerate"]>[0]) {
+	}: Parameters<LanguageModelV3["doGenerate"]>[0]) {
 		const type = responseFormat?.type ?? "text";
 
-		const warnings: LanguageModelV2CallWarning[] = [];
+		const warnings: SharedV3Warning[] = [];
 
 		if (frequencyPenalty != null) {
 			warnings.push({
-				setting: "frequencyPenalty",
-				type: "unsupported-setting",
+				feature: "frequencyPenalty",
+				type: "unsupported",
 			});
 		}
 
 		if (presencePenalty != null) {
 			warnings.push({
-				setting: "presencePenalty",
-				type: "unsupported-setting",
+				feature: "presencePenalty",
+				type: "unsupported",
 			});
 		}
 
@@ -125,11 +121,10 @@ export class WorkersAIChatLanguageModel implements LanguageModelV2 {
 	}
 
 	async doGenerate(
-		options: Parameters<LanguageModelV2["doGenerate"]>[0],
-	): Promise<Awaited<ReturnType<LanguageModelV2["doGenerate"]>>> {
+		options: Parameters<LanguageModelV3["doGenerate"]>[0],
+	): Promise<Awaited<ReturnType<LanguageModelV3["doGenerate"]>>> {
 		const { args, warnings } = this.getArgs(options);
 
-		// biome-ignore lint/correctness/noUnusedVariables: this needs to be destructured
 		const { gateway, safePrompt, ...passthroughOptions } = this.settings;
 
 		// Extract image from messages if present
@@ -198,8 +193,8 @@ export class WorkersAIChatLanguageModel implements LanguageModelV2 {
 	}
 
 	async doStream(
-		options: Parameters<LanguageModelV2["doStream"]>[0],
-	): Promise<Awaited<ReturnType<LanguageModelV2["doStream"]>>> {
+		options: Parameters<LanguageModelV3["doStream"]>[0],
+	): Promise<Awaited<ReturnType<LanguageModelV3["doStream"]>>> {
 		const { args, warnings } = this.getArgs(options);
 
 		// Extract image from messages if present
@@ -221,12 +216,12 @@ export class WorkersAIChatLanguageModel implements LanguageModelV2 {
 
 			return {
 				// rawCall: { rawPrompt: messages, rawSettings: args },
-				stream: new ReadableStream<LanguageModelV2StreamPart>({
+				stream: new ReadableStream<LanguageModelV3StreamPart>({
 					async start(controller) {
 						// Emit the stream-start part with warnings
 						controller.enqueue({
 							type: "stream-start",
-							warnings: warnings as LanguageModelV2CallWarning[],
+							warnings: warnings as SharedV3Warning[],
 						});
 
 						for (const contentPart of response.content) {
@@ -315,12 +310,12 @@ export class WorkersAIChatLanguageModel implements LanguageModelV2 {
 
 		// Create a new stream that first emits the stream-start part with warnings,
 		// then pipes through the rest of the response stream
-		const stream = new ReadableStream<LanguageModelV2StreamPart>({
+		const stream = new ReadableStream<LanguageModelV3StreamPart>({
 			start(controller) {
 				// Emit the stream-start part with warnings
 				controller.enqueue({
 					type: "stream-start",
-					warnings: warnings as LanguageModelV2CallWarning[],
+					warnings: warnings as SharedV3Warning[],
 				});
 
 				// Pipe the rest of the response stream

package/src/workersai-image-model.ts

@@ -1,10 +1,10 @@
-import type { ImageModelV2, ImageModelV2CallWarning } from "@ai-sdk/provider";
+import type { ImageModelV3, SharedV3Warning } from "@ai-sdk/provider";
 import type { WorkersAIImageConfig } from "./workersai-image-config";
 import type { WorkersAIImageSettings } from "./workersai-image-settings";
 import type { ImageGenerationModels } from "./workersai-models";
 
-export class WorkersAIImageModel implements ImageModelV2 {
-	readonly specificationVersion = "v2";
+export class WorkersAIImageModel implements ImageModelV3 {
+	readonly specificationVersion = "v3";
 
 	get maxImagesPerCall(): number {
 		return this.settings.maxImagesPerCall ?? 1;
@@ -25,20 +25,20 @@ export class WorkersAIImageModel implements ImageModelV2 {
 		size,
 		aspectRatio,
 		seed,
-		// headers,
-		// abortSignal,
-	}: Parameters<ImageModelV2["doGenerate"]>[0]): Promise<
-		Awaited<ReturnType<ImageModelV2["doGenerate"]>>
+	}: // headers,
+	// abortSignal,
+	Parameters<ImageModelV3["doGenerate"]>[0]): Promise<
+		Awaited<ReturnType<ImageModelV3["doGenerate"]>>
 	> {
 		const { width, height } = getDimensionsFromSizeString(size);
 
-		const warnings: Array<ImageModelV2CallWarning> = [];
+		const warnings: Array<SharedV3Warning> = [];
 
 		if (aspectRatio != null) {
 			warnings.push({
 				details: "This model does not support aspect ratio. Use `size` instead.",
-				setting: "aspectRatio",
-				type: "unsupported-setting",
+				feature: "aspectRatio",
+				type: "unsupported",
 			});
 		}
 
@@ -47,7 +47,7 @@ export class WorkersAIImageModel implements ImageModelV2 {
 				this.modelId,
 				{
 					height,
-					prompt,
+					prompt: prompt!,
 					seed,
 					width,
 				},