Forward AI Gateway supported URLs

.changeset/forward-gateway-supported-urls.md

@@ -0,0 +1,5 @@
+---
+"ai-gateway-provider": patch
+---
+
+Forward supported URL metadata from wrapped provider models.

packages/ai-gateway-provider/src/index.ts

@@ -22,13 +22,13 @@ export class AiGatewayChatLanguageModel implements LanguageModelV3 {
 	readonly specificationVersion = "v3";
 	readonly defaultObjectGenerationMode = "json";
 
-	readonly supportedUrls: Record<string, RegExp[]> | PromiseLike<Record<string, RegExp[]>> = {
-		// No URLS are supported for this language model
-	};
-
 	readonly models: InternalLanguageModelV3[];
 	readonly config: AiGatewaySettings;
 
+	get supportedUrls(): Record<string, RegExp[]> | PromiseLike<Record<string, RegExp[]>> {
+		return this.models[0]?.supportedUrls ?? {};
+	}
+
 	get modelId(): string {
 		if (!this.models[0]) {
 			throw new Error("models cannot be empty array");

packages/ai-gateway-provider/test/supported-urls.test.ts

@@ -0,0 +1,26 @@
+import type { LanguageModelV3 } from "@ai-sdk/provider";
+import { describe, expect, it } from "vitest";
+import { createAiGateway } from "../src";
+
+describe("supportedUrls", () => {
+	it("forwards supported URLs from the wrapped model", async () => {
+		const supportedUrls = {
+			"application/pdf": [/^https:\/\/example\.com\/.*$/],
+			"image/*": [/^https:\/\//],
+		};
+		const wrappedModel = {
+			modelId: "test-model",
+			provider: "test-provider",
+			specificationVersion: "v3",
+			supportedUrls,
+		} as unknown as LanguageModelV3;
+
+		const aigateway = createAiGateway({
+			accountId: "test-account-id",
+			apiKey: "test-api-key",
+			gateway: "test-gateway",
+		});
+
+		await expect(await aigateway(wrappedModel).supportedUrls).toBe(supportedUrls);
+	});
+});