@ai-sdk/gateway 2.0.80 → 2.0.82

package/dist/index.mjs

@@ -636,16 +636,110 @@ var gatewaySpendReportResponseSchema = lazySchema(
   )
 );
 
+// src/gateway-generation-info.ts
+import {
+  createJsonErrorResponseHandler as createJsonErrorResponseHandler3,
+  createJsonResponseHandler as createJsonResponseHandler3,
+  getFromApi as getFromApi3,
+  lazySchema as lazySchema2,
+  resolve as resolve3,
+  zodSchema as zodSchema6
+} from "@ai-sdk/provider-utils";
+import { z as z6 } from "zod/v4";
+var GatewayGenerationInfoFetcher = class {
+  constructor(config) {
+    this.config = config;
+  }
+  async getGenerationInfo(params) {
+    try {
+      const baseUrl = new URL(this.config.baseURL);
+      const { value } = await getFromApi3({
+        url: `${baseUrl.origin}/v1/generation?id=${encodeURIComponent(params.id)}`,
+        headers: await resolve3(this.config.headers()),
+        successfulResponseHandler: createJsonResponseHandler3(
+          gatewayGenerationInfoResponseSchema
+        ),
+        failedResponseHandler: createJsonErrorResponseHandler3({
+          errorSchema: z6.any(),
+          errorToMessage: (data) => data
+        }),
+        fetch: this.config.fetch
+      });
+      return value;
+    } catch (error) {
+      throw await asGatewayError(error);
+    }
+  }
+};
+var gatewayGenerationInfoResponseSchema = lazySchema2(
+  () => zodSchema6(
+    z6.object({
+      data: z6.object({
+        id: z6.string(),
+        total_cost: z6.number(),
+        upstream_inference_cost: z6.number(),
+        usage: z6.number(),
+        created_at: z6.string(),
+        model: z6.string(),
+        is_byok: z6.boolean(),
+        provider_name: z6.string(),
+        streamed: z6.boolean(),
+        finish_reason: z6.string(),
+        latency: z6.number(),
+        generation_time: z6.number(),
+        native_tokens_prompt: z6.number(),
+        native_tokens_completion: z6.number(),
+        native_tokens_reasoning: z6.number(),
+        native_tokens_cached: z6.number(),
+        native_tokens_cache_creation: z6.number(),
+        billable_web_search_calls: z6.number()
+      }).transform(
+        ({
+          total_cost,
+          upstream_inference_cost,
+          created_at,
+          is_byok,
+          provider_name,
+          finish_reason,
+          generation_time,
+          native_tokens_prompt,
+          native_tokens_completion,
+          native_tokens_reasoning,
+          native_tokens_cached,
+          native_tokens_cache_creation,
+          billable_web_search_calls,
+          ...rest
+        }) => ({
+          ...rest,
+          totalCost: total_cost,
+          upstreamInferenceCost: upstream_inference_cost,
+          createdAt: created_at,
+          isByok: is_byok,
+          providerName: provider_name,
+          finishReason: finish_reason,
+          generationTime: generation_time,
+          promptTokens: native_tokens_prompt,
+          completionTokens: native_tokens_completion,
+          reasoningTokens: native_tokens_reasoning,
+          cachedTokens: native_tokens_cached,
+          cacheCreationTokens: native_tokens_cache_creation,
+          billableWebSearchCalls: billable_web_search_calls
+        })
+      )
+    }).transform(({ data }) => data)
+  )
+);
+
 // src/gateway-language-model.ts
 import {
   combineHeaders,
   createEventSourceResponseHandler,
-  createJsonErrorResponseHandler as createJsonErrorResponseHandler3,
-  createJsonResponseHandler as createJsonResponseHandler3,
+  createJsonErrorResponseHandler as createJsonErrorResponseHandler4,
+  createJsonResponseHandler as createJsonResponseHandler4,
   postJsonToApi,
-  resolve as resolve3
+  resolve as resolve4
 } from "@ai-sdk/provider-utils";
-import { z as z6 } from "zod/v4";
+import { z as z7 } from "zod/v4";
 var GatewayLanguageModel = class {
   constructor(modelId, config) {
     this.modelId = modelId;
@@ -666,7 +760,7 @@ var GatewayLanguageModel = class {
   async doGenerate(options) {
     const { args, warnings } = await this.getArgs(options);
     const { abortSignal } = options;
-    const resolvedHeaders = await resolve3(this.config.headers());
+    const resolvedHeaders = await resolve4(this.config.headers());
     try {
       const {
         responseHeaders,
@@ -678,12 +772,12 @@ var GatewayLanguageModel = class {
           resolvedHeaders,
           options.headers,
           this.getModelConfigHeaders(this.modelId, false),
-          await resolve3(this.config.o11yHeaders)
+          await resolve4(this.config.o11yHeaders)
         ),
         body: args,
-        successfulResponseHandler: createJsonResponseHandler3(z6.any()),
-        failedResponseHandler: createJsonErrorResponseHandler3({
-          errorSchema: z6.any(),
+        successfulResponseHandler: createJsonResponseHandler4(z7.any()),
+        failedResponseHandler: createJsonErrorResponseHandler4({
+          errorSchema: z7.any(),
           errorToMessage: (data) => data
         }),
         ...abortSignal && { abortSignal },
@@ -702,7 +796,7 @@ var GatewayLanguageModel = class {
   async doStream(options) {
     const { args, warnings } = await this.getArgs(options);
     const { abortSignal } = options;
-    const resolvedHeaders = await resolve3(this.config.headers());
+    const resolvedHeaders = await resolve4(this.config.headers());
     try {
       const { value: response, responseHeaders } = await postJsonToApi({
         url: this.getUrl(),
@@ -710,12 +804,12 @@ var GatewayLanguageModel = class {
           resolvedHeaders,
           options.headers,
           this.getModelConfigHeaders(this.modelId, true),
-          await resolve3(this.config.o11yHeaders)
+          await resolve4(this.config.o11yHeaders)
         ),
         body: args,
-        successfulResponseHandler: createEventSourceResponseHandler(z6.any()),
-        failedResponseHandler: createJsonErrorResponseHandler3({
-          errorSchema: z6.any(),
+        successfulResponseHandler: createEventSourceResponseHandler(z7.any()),
+        failedResponseHandler: createJsonErrorResponseHandler4({
+          errorSchema: z7.any(),
           errorToMessage: (data) => data
         }),
         ...abortSignal && { abortSignal },
@@ -795,14 +889,14 @@ var GatewayLanguageModel = class {
 // src/gateway-embedding-model.ts
 import {
   combineHeaders as combineHeaders2,
-  createJsonErrorResponseHandler as createJsonErrorResponseHandler4,
-  createJsonResponseHandler as createJsonResponseHandler4,
+  createJsonErrorResponseHandler as createJsonErrorResponseHandler5,
+  createJsonResponseHandler as createJsonResponseHandler5,
   lazyValidator as lazyValidator5,
   postJsonToApi as postJsonToApi2,
-  resolve as resolve4,
-  zodSchema as zodSchema6
+  resolve as resolve5,
+  zodSchema as zodSchema7
 } from "@ai-sdk/provider-utils";
-import { z as z7 } from "zod/v4";
+import { z as z8 } from "zod/v4";
 var GatewayEmbeddingModel = class {
   constructor(modelId, config) {
     this.modelId = modelId;
@@ -821,7 +915,7 @@ var GatewayEmbeddingModel = class {
     providerOptions
   }) {
     var _a9;
-    const resolvedHeaders = await resolve4(this.config.headers());
+    const resolvedHeaders = await resolve5(this.config.headers());
     try {
       const {
         responseHeaders,
@@ -833,17 +927,17 @@ var GatewayEmbeddingModel = class {
           resolvedHeaders,
           headers != null ? headers : {},
           this.getModelConfigHeaders(),
-          await resolve4(this.config.o11yHeaders)
+          await resolve5(this.config.o11yHeaders)
         ),
         body: {
           input: values.length === 1 ? values[0] : values,
           ...providerOptions ? { providerOptions } : {}
         },
-        successfulResponseHandler: createJsonResponseHandler4(
+        successfulResponseHandler: createJsonResponseHandler5(
           gatewayEmbeddingResponseSchema
         ),
-        failedResponseHandler: createJsonErrorResponseHandler4({
-          errorSchema: z7.any(),
+        failedResponseHandler: createJsonErrorResponseHandler5({
+          errorSchema: z8.any(),
           errorToMessage: (data) => data
         }),
         ...abortSignal && { abortSignal },
@@ -870,11 +964,11 @@ var GatewayEmbeddingModel = class {
   }
 };
 var gatewayEmbeddingResponseSchema = lazyValidator5(
-  () => zodSchema6(
-    z7.object({
-      embeddings: z7.array(z7.array(z7.number())),
-      usage: z7.object({ tokens: z7.number() }).nullish(),
-      providerMetadata: z7.record(z7.string(), z7.record(z7.string(), z7.unknown())).optional()
+  () => zodSchema7(
+    z8.object({
+      embeddings: z8.array(z8.array(z8.number())),
+      usage: z8.object({ tokens: z8.number() }).nullish(),
+      providerMetadata: z8.record(z8.string(), z8.record(z8.string(), z8.unknown())).optional()
     })
   )
 );
@@ -882,12 +976,12 @@ var gatewayEmbeddingResponseSchema = lazyValidator5(
 // src/gateway-image-model.ts
 import {
   combineHeaders as combineHeaders3,
-  createJsonResponseHandler as createJsonResponseHandler5,
-  createJsonErrorResponseHandler as createJsonErrorResponseHandler5,
+  createJsonResponseHandler as createJsonResponseHandler6,
+  createJsonErrorResponseHandler as createJsonErrorResponseHandler6,
   postJsonToApi as postJsonToApi3,
-  resolve as resolve5
+  resolve as resolve6
 } from "@ai-sdk/provider-utils";
-import { z as z8 } from "zod/v4";
+import { z as z9 } from "zod/v4";
 var GatewayImageModel = class {
   constructor(modelId, config) {
     this.modelId = modelId;
@@ -910,7 +1004,7 @@ var GatewayImageModel = class {
     abortSignal
   }) {
     var _a9, _b9, _c, _d;
-    const resolvedHeaders = await resolve5(this.config.headers());
+    const resolvedHeaders = await resolve6(this.config.headers());
     try {
       const {
         responseHeaders,
@@ -922,7 +1016,7 @@ var GatewayImageModel = class {
           resolvedHeaders,
           headers != null ? headers : {},
           this.getModelConfigHeaders(),
-          await resolve5(this.config.o11yHeaders)
+          await resolve6(this.config.o11yHeaders)
         ),
         body: {
           prompt,
@@ -932,11 +1026,11 @@ var GatewayImageModel = class {
           ...seed && { seed },
           ...providerOptions && { providerOptions }
         },
-        successfulResponseHandler: createJsonResponseHandler5(
+        successfulResponseHandler: createJsonResponseHandler6(
           gatewayImageResponseSchema
         ),
-        failedResponseHandler: createJsonErrorResponseHandler5({
-          errorSchema: z8.any(),
+        failedResponseHandler: createJsonErrorResponseHandler6({
+          errorSchema: z9.any(),
           errorToMessage: (data) => data
         }),
         ...abortSignal && { abortSignal },
@@ -974,89 +1068,89 @@ var GatewayImageModel = class {
     };
   }
 };
-var providerMetadataEntrySchema = z8.object({
-  images: z8.array(z8.unknown()).optional()
-}).catchall(z8.unknown());
-var gatewayImageUsageSchema = z8.object({
-  inputTokens: z8.number().nullish(),
-  outputTokens: z8.number().nullish(),
-  totalTokens: z8.number().nullish()
+var providerMetadataEntrySchema = z9.object({
+  images: z9.array(z9.unknown()).optional()
+}).catchall(z9.unknown());
+var gatewayImageUsageSchema = z9.object({
+  inputTokens: z9.number().nullish(),
+  outputTokens: z9.number().nullish(),
+  totalTokens: z9.number().nullish()
 });
-var gatewayImageResponseSchema = z8.object({
-  images: z8.array(z8.string()),
+var gatewayImageResponseSchema = z9.object({
+  images: z9.array(z9.string()),
   // Always base64 strings over the wire
-  warnings: z8.array(
-    z8.object({
-      type: z8.literal("other"),
-      message: z8.string()
+  warnings: z9.array(
+    z9.object({
+      type: z9.literal("other"),
+      message: z9.string()
     })
   ).optional(),
-  providerMetadata: z8.record(z8.string(), providerMetadataEntrySchema).optional(),
+  providerMetadata: z9.record(z9.string(), providerMetadataEntrySchema).optional(),
   usage: gatewayImageUsageSchema.optional()
 });
 
 // src/tool/parallel-search.ts
 import {
   createProviderDefinedToolFactoryWithOutputSchema,
-  lazySchema as lazySchema2,
-  zodSchema as zodSchema7
+  lazySchema as lazySchema3,
+  zodSchema as zodSchema8
 } from "@ai-sdk/provider-utils";
-import { z as z9 } from "zod";
-var parallelSearchInputSchema = lazySchema2(
-  () => zodSchema7(
-    z9.object({
-      objective: z9.string().describe(
+import { z as z10 } from "zod";
+var parallelSearchInputSchema = lazySchema3(
+  () => zodSchema8(
+    z10.object({
+      objective: z10.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: z9.array(z9.string()).optional().describe(
+      search_queries: z10.array(z10.string()).optional().describe(
         "Optional search queries to supplement the objective. Maximum 200 characters per query."
       ),
-      mode: z9.enum(["one-shot", "agentic"]).optional().describe(
+      mode: z10.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: z9.number().optional().describe(
+      max_results: z10.number().optional().describe(
         "Maximum number of results to return (1-20). Defaults to 10 if not specified."
       ),
-      source_policy: z9.object({
-        include_domains: z9.array(z9.string()).optional().describe("List of domains to include in search results."),
-        exclude_domains: z9.array(z9.string()).optional().describe("List of domains to exclude from search results."),
-        after_date: z9.string().optional().describe(
+      source_policy: z10.object({
+        include_domains: z10.array(z10.string()).optional().describe("List of domains to include in search results."),
+        exclude_domains: z10.array(z10.string()).optional().describe("List of domains to exclude from search results."),
+        after_date: z10.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: z9.object({
-        max_chars_per_result: z9.number().optional().describe("Maximum characters per result."),
-        max_chars_total: z9.number().optional().describe("Maximum total characters across all results.")
+      excerpts: z10.object({
+        max_chars_per_result: z10.number().optional().describe("Maximum characters per result."),
+        max_chars_total: z10.number().optional().describe("Maximum total characters across all results.")
       }).optional().describe("Excerpt configuration for controlling result length."),
-      fetch_policy: z9.object({
-        max_age_seconds: z9.number().optional().describe(
+      fetch_policy: z10.object({
+        max_age_seconds: z10.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 = lazySchema2(
-  () => zodSchema7(
-    z9.union([
+var parallelSearchOutputSchema = lazySchema3(
+  () => zodSchema8(
+    z10.union([
       // Success response
-      z9.object({
-        searchId: z9.string(),
-        results: z9.array(
-          z9.object({
-            url: z9.string(),
-            title: z9.string(),
-            excerpt: z9.string(),
-            publishDate: z9.string().nullable().optional(),
-            relevanceScore: z9.number().optional()
+      z10.object({
+        searchId: z10.string(),
+        results: z10.array(
+          z10.object({
+            url: z10.string(),
+            title: z10.string(),
+            excerpt: z10.string(),
+            publishDate: z10.string().nullable().optional(),
+            relevanceScore: z10.number().optional()
           })
         )
       }),
       // Error response
-      z9.object({
-        error: z9.enum([
+      z10.object({
+        error: z10.enum([
           "api_error",
           "rate_limit",
           "timeout",
@@ -1064,8 +1158,8 @@ var parallelSearchOutputSchema = lazySchema2(
           "configuration_error",
           "unknown"
         ]),
-        statusCode: z9.number().optional(),
-        message: z9.string()
+        statusCode: z10.number().optional(),
+        message: z10.string()
       })
     ])
   )
@@ -1081,79 +1175,79 @@ var parallelSearch = (config = {}) => parallelSearchToolFactory(config);
 // src/tool/perplexity-search.ts
 import {
   createProviderDefinedToolFactoryWithOutputSchema as createProviderDefinedToolFactoryWithOutputSchema2,
-  lazySchema as lazySchema3,
-  zodSchema as zodSchema8
+  lazySchema as lazySchema4,
+  zodSchema as zodSchema9
 } from "@ai-sdk/provider-utils";
-import { z as z10 } from "zod";
-var perplexitySearchInputSchema = lazySchema3(
-  () => zodSchema8(
-    z10.object({
-      query: z10.union([z10.string(), z10.array(z10.string())]).describe(
+import { z as z11 } from "zod";
+var perplexitySearchInputSchema = lazySchema4(
+  () => zodSchema9(
+    z11.object({
+      query: z11.union([z11.string(), z11.array(z11.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: z10.number().optional().describe(
+      max_results: z11.number().optional().describe(
         "Maximum number of search results to return (1-20, default: 10)"
       ),
-      max_tokens_per_page: z10.number().optional().describe(
+      max_tokens_per_page: z11.number().optional().describe(
         "Maximum number of tokens to extract per search result page (256-2048, default: 2048)"
       ),
-      max_tokens: z10.number().optional().describe(
+      max_tokens: z11.number().optional().describe(
         "Maximum total tokens across all search results (default: 25000, max: 1000000)"
       ),
-      country: z10.string().optional().describe(
+      country: z11.string().optional().describe(
         "Two-letter ISO 3166-1 alpha-2 country code for regional search results (e.g., 'US', 'GB', 'FR')"
       ),
-      search_domain_filter: z10.array(z10.string()).optional().describe(
+      search_domain_filter: z11.array(z11.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: z10.array(z10.string()).optional().describe(
+      search_language_filter: z11.array(z11.string()).optional().describe(
         "List of ISO 639-1 language codes to filter results (max 10, lowercase). Examples: ['en', 'fr', 'de']"
       ),
-      search_after_date: z10.string().optional().describe(
+      search_after_date: z11.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: z10.string().optional().describe(
+      search_before_date: z11.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: z10.string().optional().describe(
+      last_updated_after_filter: z11.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: z10.string().optional().describe(
+      last_updated_before_filter: z11.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: z10.enum(["day", "week", "month", "year"]).optional().describe(
+      search_recency_filter: z11.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 = lazySchema3(
-  () => zodSchema8(
-    z10.union([
+var perplexitySearchOutputSchema = lazySchema4(
+  () => zodSchema9(
+    z11.union([
       // Success response
-      z10.object({
-        results: z10.array(
-          z10.object({
-            title: z10.string(),
-            url: z10.string(),
-            snippet: z10.string(),
-            date: z10.string().optional(),
-            lastUpdated: z10.string().optional()
+      z11.object({
+        results: z11.array(
+          z11.object({
+            title: z11.string(),
+            url: z11.string(),
+            snippet: z11.string(),
+            date: z11.string().optional(),
+            lastUpdated: z11.string().optional()
           })
         ),
-        id: z10.string()
+        id: z11.string()
       }),
       // Error response
-      z10.object({
-        error: z10.enum([
+      z11.object({
+        error: z11.enum([
           "api_error",
           "rate_limit",
           "timeout",
           "invalid_input",
           "unknown"
         ]),
-        statusCode: z10.number().optional(),
-        message: z10.string()
+        statusCode: z11.number().optional(),
+        message: z11.string()
       })
     ])
   )
@@ -1199,7 +1293,7 @@ async function getVercelRequestId() {
 import { withUserAgentSuffix } from "@ai-sdk/provider-utils";
 
 // src/version.ts
-var VERSION = true ? "2.0.80" : "0.0.0-test";
+var VERSION = true ? "2.0.82" : "0.0.0-test";
 
 // src/gateway-provider.ts
 var AI_GATEWAY_PROTOCOL_VERSION = "0.0.1";
@@ -1311,6 +1405,18 @@ function createGatewayProvider(options = {}) {
       );
     });
   };
+  const getGenerationInfo = async (params) => {
+    return new GatewayGenerationInfoFetcher({
+      baseURL,
+      headers: getHeaders,
+      fetch: options.fetch
+    }).getGenerationInfo(params).catch(async (error) => {
+      throw await asGatewayError(
+        error,
+        await parseAuthMethod(await getHeaders())
+      );
+    });
+  };
   const provider = function(modelId) {
     if (new.target) {
       throw new Error(
@@ -1322,6 +1428,7 @@ function createGatewayProvider(options = {}) {
   provider.getAvailableModels = getAvailableModels;
   provider.getCredits = getCredits;
   provider.getSpendReport = getSpendReport;
+  provider.getGenerationInfo = getGenerationInfo;
   provider.imageModel = (modelId) => {
     return new GatewayImageModel(modelId, {
       provider: "gateway",