@ai-sdk/gateway 4.0.0-beta.39 → 4.0.0-beta.40

package/dist/index.mjs

@@ -1357,68 +1357,160 @@ var gatewayVideoEventSchema = z10.discriminatedUnion("type", [
   })
 ]);
 
-// src/tool/parallel-search.ts
+// src/gateway-reranking-model.ts
 import {
-  createProviderToolFactoryWithOutputSchema,
+  combineHeaders as combineHeaders5,
+  createJsonErrorResponseHandler as createJsonErrorResponseHandler8,
+  createJsonResponseHandler as createJsonResponseHandler7,
   lazySchema as lazySchema8,
+  postJsonToApi as postJsonToApi5,
+  resolve as resolve8,
   zodSchema as zodSchema8
 } from "@ai-sdk/provider-utils";
-import { z as z11 } from "zod";
-var parallelSearchInputSchema = lazySchema8(
+import { z as z11 } from "zod/v4";
+var GatewayRerankingModel = class {
+  constructor(modelId, config) {
+    this.modelId = modelId;
+    this.config = config;
+    this.specificationVersion = "v4";
+  }
+  get provider() {
+    return this.config.provider;
+  }
+  async doRerank({
+    documents,
+    query,
+    topN,
+    headers,
+    abortSignal,
+    providerOptions
+  }) {
+    const resolvedHeaders = await resolve8(this.config.headers());
+    try {
+      const {
+        responseHeaders,
+        value: responseBody,
+        rawValue
+      } = await postJsonToApi5({
+        url: this.getUrl(),
+        headers: combineHeaders5(
+          resolvedHeaders,
+          headers != null ? headers : {},
+          this.getModelConfigHeaders(),
+          await resolve8(this.config.o11yHeaders)
+        ),
+        body: {
+          documents,
+          query,
+          ...topN != null ? { topN } : {},
+          ...providerOptions ? { providerOptions } : {}
+        },
+        successfulResponseHandler: createJsonResponseHandler7(
+          gatewayRerankingResponseSchema
+        ),
+        failedResponseHandler: createJsonErrorResponseHandler8({
+          errorSchema: z11.any(),
+          errorToMessage: (data) => data
+        }),
+        ...abortSignal && { abortSignal },
+        fetch: this.config.fetch
+      });
+      return {
+        ranking: responseBody.ranking,
+        providerMetadata: responseBody.providerMetadata,
+        response: { headers: responseHeaders, body: rawValue },
+        warnings: []
+      };
+    } catch (error) {
+      throw await asGatewayError(error, await parseAuthMethod(resolvedHeaders));
+    }
+  }
+  getUrl() {
+    return `${this.config.baseURL}/reranking-model`;
+  }
+  getModelConfigHeaders() {
+    return {
+      "ai-reranking-model-specification-version": "4",
+      "ai-model-id": this.modelId
+    };
+  }
+};
+var gatewayRerankingResponseSchema = lazySchema8(
   () => zodSchema8(
     z11.object({
-      objective: z11.string().describe(
+      ranking: z11.array(
+        z11.object({
+          index: z11.number(),
+          relevanceScore: z11.number()
+        })
+      ),
+      providerMetadata: z11.record(z11.string(), z11.record(z11.string(), z11.unknown())).optional()
+    })
+  )
+);
+
+// src/tool/parallel-search.ts
+import {
+  createProviderToolFactoryWithOutputSchema,
+  lazySchema as lazySchema9,
+  zodSchema as zodSchema9
+} from "@ai-sdk/provider-utils";
+import { z as z12 } from "zod";
+var parallelSearchInputSchema = lazySchema9(
+  () => zodSchema9(
+    z12.object({
+      objective: z12.string().describe(
         "Natural-language description of the web research goal, including source or freshness guidance and broader context from the task. Maximum 5000 characters."
       ),
-      search_queries: z11.array(z11.string()).optional().describe(
+      search_queries: z12.array(z12.string()).optional().describe(
         "Optional search queries to supplement the objective. Maximum 200 characters per query."
       ),
-      mode: z11.enum(["one-shot", "agentic"]).optional().describe(
+      mode: z12.enum(["one-shot", "agentic"]).optional().describe(
         'Mode preset: "one-shot" for comprehensive results with longer excerpts (default), "agentic" for concise, token-efficient results for multi-step workflows.'
       ),
-      max_results: z11.number().optional().describe(
+      max_results: z12.number().optional().describe(
         "Maximum number of results to return (1-20). Defaults to 10 if not specified."
       ),
-      source_policy: z11.object({
-        include_domains: z11.array(z11.string()).optional().describe("List of domains to include in search results."),
-        exclude_domains: z11.array(z11.string()).optional().describe("List of domains to exclude from search results."),
-        after_date: z11.string().optional().describe(
+      source_policy: z12.object({
+        include_domains: z12.array(z12.string()).optional().describe("List of domains to include in search results."),
+        exclude_domains: z12.array(z12.string()).optional().describe("List of domains to exclude from search results."),
+        after_date: z12.string().optional().describe(
           "Only include results published after this date (ISO 8601 format)."
         )
       }).optional().describe(
         "Source policy for controlling which domains to include/exclude and freshness."
       ),
-      excerpts: z11.object({
-        max_chars_per_result: z11.number().optional().describe("Maximum characters per result."),
-        max_chars_total: z11.number().optional().describe("Maximum total characters across all results.")
+      excerpts: z12.object({
+        max_chars_per_result: z12.number().optional().describe("Maximum characters per result."),
+        max_chars_total: z12.number().optional().describe("Maximum total characters across all results.")
       }).optional().describe("Excerpt configuration for controlling result length."),
-      fetch_policy: z11.object({
-        max_age_seconds: z11.number().optional().describe(
+      fetch_policy: z12.object({
+        max_age_seconds: z12.number().optional().describe(
           "Maximum age in seconds for cached content. Set to 0 to always fetch fresh content."
         )
       }).optional().describe("Fetch policy for controlling content freshness.")
     })
   )
 );
-var parallelSearchOutputSchema = lazySchema8(
-  () => zodSchema8(
-    z11.union([
+var parallelSearchOutputSchema = lazySchema9(
+  () => zodSchema9(
+    z12.union([
       // Success response
-      z11.object({
-        searchId: z11.string(),
-        results: z11.array(
-          z11.object({
-            url: z11.string(),
-            title: z11.string(),
-            excerpt: z11.string(),
-            publishDate: z11.string().nullable().optional(),
-            relevanceScore: z11.number().optional()
+      z12.object({
+        searchId: z12.string(),
+        results: z12.array(
+          z12.object({
+            url: z12.string(),
+            title: z12.string(),
+            excerpt: z12.string(),
+            publishDate: z12.string().nullable().optional(),
+            relevanceScore: z12.number().optional()
           })
         )
       }),
       // Error response
-      z11.object({
-        error: z11.enum([
+      z12.object({
+        error: z12.enum([
           "api_error",
           "rate_limit",
           "timeout",
@@ -1426,8 +1518,8 @@ var parallelSearchOutputSchema = lazySchema8(
           "configuration_error",
           "unknown"
         ]),
-        statusCode: z11.number().optional(),
-        message: z11.string()
+        statusCode: z12.number().optional(),
+        message: z12.string()
       })
     ])
   )
@@ -1442,79 +1534,79 @@ var parallelSearch = (config = {}) => parallelSearchToolFactory(config);
 // src/tool/perplexity-search.ts
 import {
   createProviderToolFactoryWithOutputSchema as createProviderToolFactoryWithOutputSchema2,
-  lazySchema as lazySchema9,
-  zodSchema as zodSchema9
+  lazySchema as lazySchema10,
+  zodSchema as zodSchema10
 } from "@ai-sdk/provider-utils";
-import { z as z12 } from "zod";
-var perplexitySearchInputSchema = lazySchema9(
-  () => zodSchema9(
-    z12.object({
-      query: z12.union([z12.string(), z12.array(z12.string())]).describe(
+import { z as z13 } from "zod";
+var perplexitySearchInputSchema = lazySchema10(
+  () => zodSchema10(
+    z13.object({
+      query: z13.union([z13.string(), z13.array(z13.string())]).describe(
         "Search query (string) or multiple queries (array of up to 5 strings). Multi-query searches return combined results from all queries."
       ),
-      max_results: z12.number().optional().describe(
+      max_results: z13.number().optional().describe(
         "Maximum number of search results to return (1-20, default: 10)"
       ),
-      max_tokens_per_page: z12.number().optional().describe(
+      max_tokens_per_page: z13.number().optional().describe(
         "Maximum number of tokens to extract per search result page (256-2048, default: 2048)"
       ),
-      max_tokens: z12.number().optional().describe(
+      max_tokens: z13.number().optional().describe(
         "Maximum total tokens across all search results (default: 25000, max: 1000000)"
       ),
-      country: z12.string().optional().describe(
+      country: z13.string().optional().describe(
         "Two-letter ISO 3166-1 alpha-2 country code for regional search results (e.g., 'US', 'GB', 'FR')"
       ),
-      search_domain_filter: z12.array(z12.string()).optional().describe(
+      search_domain_filter: z13.array(z13.string()).optional().describe(
         "List of domains to include or exclude from search results (max 20). To include: ['nature.com', 'science.org']. To exclude: ['-example.com', '-spam.net']"
       ),
-      search_language_filter: z12.array(z12.string()).optional().describe(
+      search_language_filter: z13.array(z13.string()).optional().describe(
         "List of ISO 639-1 language codes to filter results (max 10, lowercase). Examples: ['en', 'fr', 'de']"
       ),
-      search_after_date: z12.string().optional().describe(
+      search_after_date: z13.string().optional().describe(
         "Include only results published after this date. Format: 'MM/DD/YYYY' (e.g., '3/1/2025'). Cannot be used with search_recency_filter."
       ),
-      search_before_date: z12.string().optional().describe(
+      search_before_date: z13.string().optional().describe(
         "Include only results published before this date. Format: 'MM/DD/YYYY' (e.g., '3/15/2025'). Cannot be used with search_recency_filter."
       ),
-      last_updated_after_filter: z12.string().optional().describe(
+      last_updated_after_filter: z13.string().optional().describe(
         "Include only results last updated after this date. Format: 'MM/DD/YYYY' (e.g., '3/1/2025'). Cannot be used with search_recency_filter."
       ),
-      last_updated_before_filter: z12.string().optional().describe(
+      last_updated_before_filter: z13.string().optional().describe(
         "Include only results last updated before this date. Format: 'MM/DD/YYYY' (e.g., '3/15/2025'). Cannot be used with search_recency_filter."
       ),
-      search_recency_filter: z12.enum(["day", "week", "month", "year"]).optional().describe(
+      search_recency_filter: z13.enum(["day", "week", "month", "year"]).optional().describe(
         "Filter results by relative time period. Cannot be used with search_after_date or search_before_date."
       )
     })
   )
 );
-var perplexitySearchOutputSchema = lazySchema9(
-  () => zodSchema9(
-    z12.union([
+var perplexitySearchOutputSchema = lazySchema10(
+  () => zodSchema10(
+    z13.union([
       // Success response
-      z12.object({
-        results: z12.array(
-          z12.object({
-            title: z12.string(),
-            url: z12.string(),
-            snippet: z12.string(),
-            date: z12.string().optional(),
-            lastUpdated: z12.string().optional()
+      z13.object({
+        results: z13.array(
+          z13.object({
+            title: z13.string(),
+            url: z13.string(),
+            snippet: z13.string(),
+            date: z13.string().optional(),
+            lastUpdated: z13.string().optional()
           })
         ),
-        id: z12.string()
+        id: z13.string()
       }),
       // Error response
-      z12.object({
-        error: z12.enum([
+      z13.object({
+        error: z13.enum([
           "api_error",
           "rate_limit",
           "timeout",
           "invalid_input",
           "unknown"
         ]),
-        statusCode: z12.number().optional(),
-        message: z12.string()
+        statusCode: z13.number().optional(),
+        message: z13.string()
       })
     ])
   )
@@ -1559,7 +1651,7 @@ async function getVercelRequestId() {
 import { withUserAgentSuffix } from "@ai-sdk/provider-utils";
 
 // src/version.ts
-var VERSION = true ? "4.0.0-beta.39" : "0.0.0-test";
+var VERSION = true ? "4.0.0-beta.40" : "0.0.0-test";
 
 // src/gateway-provider.ts
 var AI_GATEWAY_PROTOCOL_VERSION = "0.0.1";
@@ -1728,6 +1820,17 @@ function createGatewayProvider(options = {}) {
       o11yHeaders: createO11yHeaders()
     });
   };
+  const createRerankingModel = (modelId) => {
+    return new GatewayRerankingModel(modelId, {
+      provider: "gateway",
+      baseURL,
+      headers: getHeaders,
+      fetch: options.fetch,
+      o11yHeaders: createO11yHeaders()
+    });
+  };
+  provider.rerankingModel = createRerankingModel;
+  provider.reranking = createRerankingModel;
   provider.chat = provider.languageModel;
   provider.embedding = provider.embeddingModel;
   provider.image = provider.imageModel;