@ai-sdk/gateway 2.0.64 → 2.0.66

package/dist/index.js

@@ -34,7 +34,7 @@ __export(index_exports, {
 module.exports = __toCommonJS(index_exports);
 
 // src/gateway-provider.ts
-var import_provider_utils10 = require("@ai-sdk/provider-utils");
+var import_provider_utils11 = require("@ai-sdk/provider-utils");
 
 // src/errors/as-gateway-error.ts
 var import_provider = require("@ai-sdk/provider");
@@ -535,9 +535,111 @@ var gatewayCreditsResponseSchema = (0, import_provider_utils4.lazyValidator)(
   )
 );
 
-// src/gateway-language-model.ts
+// src/gateway-spend-report.ts
 var import_provider_utils5 = require("@ai-sdk/provider-utils");
 var import_v45 = require("zod/v4");
+var GatewaySpendReport = class {
+  constructor(config) {
+    this.config = config;
+  }
+  async getSpendReport(params) {
+    try {
+      const baseUrl = new URL(this.config.baseURL);
+      const searchParams = new URLSearchParams();
+      searchParams.set("start_date", params.startDate);
+      searchParams.set("end_date", params.endDate);
+      if (params.groupBy) {
+        searchParams.set("group_by", params.groupBy);
+      }
+      if (params.datePart) {
+        searchParams.set("date_part", params.datePart);
+      }
+      if (params.userId) {
+        searchParams.set("user_id", params.userId);
+      }
+      if (params.model) {
+        searchParams.set("model", params.model);
+      }
+      if (params.provider) {
+        searchParams.set("provider", params.provider);
+      }
+      if (params.credentialType) {
+        searchParams.set("credential_type", params.credentialType);
+      }
+      if (params.tags && params.tags.length > 0) {
+        searchParams.set("tags", params.tags.join(","));
+      }
+      const { value } = await (0, import_provider_utils5.getFromApi)({
+        url: `${baseUrl.origin}/v1/report?${searchParams.toString()}`,
+        headers: await (0, import_provider_utils5.resolve)(this.config.headers()),
+        successfulResponseHandler: (0, import_provider_utils5.createJsonResponseHandler)(
+          gatewaySpendReportResponseSchema
+        ),
+        failedResponseHandler: (0, import_provider_utils5.createJsonErrorResponseHandler)({
+          errorSchema: import_v45.z.any(),
+          errorToMessage: (data) => data
+        }),
+        fetch: this.config.fetch
+      });
+      return value;
+    } catch (error) {
+      throw await asGatewayError(error);
+    }
+  }
+};
+var gatewaySpendReportResponseSchema = (0, import_provider_utils5.lazySchema)(
+  () => (0, import_provider_utils5.zodSchema)(
+    import_v45.z.object({
+      results: import_v45.z.array(
+        import_v45.z.object({
+          day: import_v45.z.string().optional(),
+          hour: import_v45.z.string().optional(),
+          user: import_v45.z.string().optional(),
+          model: import_v45.z.string().optional(),
+          tag: import_v45.z.string().optional(),
+          provider: import_v45.z.string().optional(),
+          credential_type: import_v45.z.enum(["byok", "system"]).optional(),
+          total_cost: import_v45.z.number(),
+          market_cost: import_v45.z.number().optional(),
+          input_tokens: import_v45.z.number().optional(),
+          output_tokens: import_v45.z.number().optional(),
+          cached_input_tokens: import_v45.z.number().optional(),
+          cache_creation_input_tokens: import_v45.z.number().optional(),
+          reasoning_tokens: import_v45.z.number().optional(),
+          request_count: import_v45.z.number().optional()
+        }).transform(
+          ({
+            credential_type,
+            total_cost,
+            market_cost,
+            input_tokens,
+            output_tokens,
+            cached_input_tokens,
+            cache_creation_input_tokens,
+            reasoning_tokens,
+            request_count,
+            ...rest
+          }) => ({
+            ...rest,
+            ...credential_type !== void 0 ? { credentialType: credential_type } : {},
+            totalCost: total_cost,
+            ...market_cost !== void 0 ? { marketCost: market_cost } : {},
+            ...input_tokens !== void 0 ? { inputTokens: input_tokens } : {},
+            ...output_tokens !== void 0 ? { outputTokens: output_tokens } : {},
+            ...cached_input_tokens !== void 0 ? { cachedInputTokens: cached_input_tokens } : {},
+            ...cache_creation_input_tokens !== void 0 ? { cacheCreationInputTokens: cache_creation_input_tokens } : {},
+            ...reasoning_tokens !== void 0 ? { reasoningTokens: reasoning_tokens } : {},
+            ...request_count !== void 0 ? { requestCount: request_count } : {}
+          })
+        )
+      )
+    })
+  )
+);
+
+// src/gateway-language-model.ts
+var import_provider_utils6 = require("@ai-sdk/provider-utils");
+var import_v46 = require("zod/v4");
 var GatewayLanguageModel = class {
   constructor(modelId, config) {
     this.modelId = modelId;
@@ -558,24 +660,24 @@ var GatewayLanguageModel = class {
   async doGenerate(options) {
     const { args, warnings } = await this.getArgs(options);
     const { abortSignal } = options;
-    const resolvedHeaders = await (0, import_provider_utils5.resolve)(this.config.headers());
+    const resolvedHeaders = await (0, import_provider_utils6.resolve)(this.config.headers());
     try {
       const {
         responseHeaders,
         value: responseBody,
         rawValue: rawResponse
-      } = await (0, import_provider_utils5.postJsonToApi)({
+      } = await (0, import_provider_utils6.postJsonToApi)({
         url: this.getUrl(),
-        headers: (0, import_provider_utils5.combineHeaders)(
+        headers: (0, import_provider_utils6.combineHeaders)(
           resolvedHeaders,
           options.headers,
           this.getModelConfigHeaders(this.modelId, false),
-          await (0, import_provider_utils5.resolve)(this.config.o11yHeaders)
+          await (0, import_provider_utils6.resolve)(this.config.o11yHeaders)
         ),
         body: args,
-        successfulResponseHandler: (0, import_provider_utils5.createJsonResponseHandler)(import_v45.z.any()),
-        failedResponseHandler: (0, import_provider_utils5.createJsonErrorResponseHandler)({
-          errorSchema: import_v45.z.any(),
+        successfulResponseHandler: (0, import_provider_utils6.createJsonResponseHandler)(import_v46.z.any()),
+        failedResponseHandler: (0, import_provider_utils6.createJsonErrorResponseHandler)({
+          errorSchema: import_v46.z.any(),
           errorToMessage: (data) => data
         }),
         ...abortSignal && { abortSignal },
@@ -594,20 +696,20 @@ var GatewayLanguageModel = class {
   async doStream(options) {
     const { args, warnings } = await this.getArgs(options);
     const { abortSignal } = options;
-    const resolvedHeaders = await (0, import_provider_utils5.resolve)(this.config.headers());
+    const resolvedHeaders = await (0, import_provider_utils6.resolve)(this.config.headers());
     try {
-      const { value: response, responseHeaders } = await (0, import_provider_utils5.postJsonToApi)({
+      const { value: response, responseHeaders } = await (0, import_provider_utils6.postJsonToApi)({
         url: this.getUrl(),
-        headers: (0, import_provider_utils5.combineHeaders)(
+        headers: (0, import_provider_utils6.combineHeaders)(
           resolvedHeaders,
           options.headers,
           this.getModelConfigHeaders(this.modelId, true),
-          await (0, import_provider_utils5.resolve)(this.config.o11yHeaders)
+          await (0, import_provider_utils6.resolve)(this.config.o11yHeaders)
         ),
         body: args,
-        successfulResponseHandler: (0, import_provider_utils5.createEventSourceResponseHandler)(import_v45.z.any()),
-        failedResponseHandler: (0, import_provider_utils5.createJsonErrorResponseHandler)({
-          errorSchema: import_v45.z.any(),
+        successfulResponseHandler: (0, import_provider_utils6.createEventSourceResponseHandler)(import_v46.z.any()),
+        failedResponseHandler: (0, import_provider_utils6.createJsonErrorResponseHandler)({
+          errorSchema: import_v46.z.any(),
           errorToMessage: (data) => data
         }),
         ...abortSignal && { abortSignal },
@@ -685,8 +787,8 @@ var GatewayLanguageModel = class {
 };
 
 // src/gateway-embedding-model.ts
-var import_provider_utils6 = require("@ai-sdk/provider-utils");
-var import_v46 = require("zod/v4");
+var import_provider_utils7 = require("@ai-sdk/provider-utils");
+var import_v47 = require("zod/v4");
 var GatewayEmbeddingModel = class {
   constructor(modelId, config) {
     this.modelId = modelId;
@@ -705,29 +807,29 @@ var GatewayEmbeddingModel = class {
     providerOptions
   }) {
     var _a9;
-    const resolvedHeaders = await (0, import_provider_utils6.resolve)(this.config.headers());
+    const resolvedHeaders = await (0, import_provider_utils7.resolve)(this.config.headers());
     try {
       const {
         responseHeaders,
         value: responseBody,
         rawValue
-      } = await (0, import_provider_utils6.postJsonToApi)({
+      } = await (0, import_provider_utils7.postJsonToApi)({
         url: this.getUrl(),
-        headers: (0, import_provider_utils6.combineHeaders)(
+        headers: (0, import_provider_utils7.combineHeaders)(
           resolvedHeaders,
           headers != null ? headers : {},
           this.getModelConfigHeaders(),
-          await (0, import_provider_utils6.resolve)(this.config.o11yHeaders)
+          await (0, import_provider_utils7.resolve)(this.config.o11yHeaders)
         ),
         body: {
           input: values.length === 1 ? values[0] : values,
           ...providerOptions ? { providerOptions } : {}
         },
-        successfulResponseHandler: (0, import_provider_utils6.createJsonResponseHandler)(
+        successfulResponseHandler: (0, import_provider_utils7.createJsonResponseHandler)(
           gatewayEmbeddingResponseSchema
         ),
-        failedResponseHandler: (0, import_provider_utils6.createJsonErrorResponseHandler)({
-          errorSchema: import_v46.z.any(),
+        failedResponseHandler: (0, import_provider_utils7.createJsonErrorResponseHandler)({
+          errorSchema: import_v47.z.any(),
           errorToMessage: (data) => data
         }),
         ...abortSignal && { abortSignal },
@@ -753,19 +855,19 @@ var GatewayEmbeddingModel = class {
     };
   }
 };
-var gatewayEmbeddingResponseSchema = (0, import_provider_utils6.lazyValidator)(
-  () => (0, import_provider_utils6.zodSchema)(
-    import_v46.z.object({
-      embeddings: import_v46.z.array(import_v46.z.array(import_v46.z.number())),
-      usage: import_v46.z.object({ tokens: import_v46.z.number() }).nullish(),
-      providerMetadata: import_v46.z.record(import_v46.z.string(), import_v46.z.record(import_v46.z.string(), import_v46.z.unknown())).optional()
+var gatewayEmbeddingResponseSchema = (0, import_provider_utils7.lazyValidator)(
+  () => (0, import_provider_utils7.zodSchema)(
+    import_v47.z.object({
+      embeddings: import_v47.z.array(import_v47.z.array(import_v47.z.number())),
+      usage: import_v47.z.object({ tokens: import_v47.z.number() }).nullish(),
+      providerMetadata: import_v47.z.record(import_v47.z.string(), import_v47.z.record(import_v47.z.string(), import_v47.z.unknown())).optional()
     })
   )
 );
 
 // src/gateway-image-model.ts
-var import_provider_utils7 = require("@ai-sdk/provider-utils");
-var import_v47 = require("zod/v4");
+var import_provider_utils8 = require("@ai-sdk/provider-utils");
+var import_v48 = require("zod/v4");
 var GatewayImageModel = class {
   constructor(modelId, config) {
     this.modelId = modelId;
@@ -788,19 +890,19 @@ var GatewayImageModel = class {
     abortSignal
   }) {
     var _a9, _b9, _c, _d;
-    const resolvedHeaders = await (0, import_provider_utils7.resolve)(this.config.headers());
+    const resolvedHeaders = await (0, import_provider_utils8.resolve)(this.config.headers());
     try {
       const {
         responseHeaders,
         value: responseBody,
         rawValue
-      } = await (0, import_provider_utils7.postJsonToApi)({
+      } = await (0, import_provider_utils8.postJsonToApi)({
         url: this.getUrl(),
-        headers: (0, import_provider_utils7.combineHeaders)(
+        headers: (0, import_provider_utils8.combineHeaders)(
           resolvedHeaders,
           headers != null ? headers : {},
           this.getModelConfigHeaders(),
-          await (0, import_provider_utils7.resolve)(this.config.o11yHeaders)
+          await (0, import_provider_utils8.resolve)(this.config.o11yHeaders)
         ),
         body: {
           prompt,
@@ -810,11 +912,11 @@ var GatewayImageModel = class {
           ...seed && { seed },
           ...providerOptions && { providerOptions }
         },
-        successfulResponseHandler: (0, import_provider_utils7.createJsonResponseHandler)(
+        successfulResponseHandler: (0, import_provider_utils8.createJsonResponseHandler)(
           gatewayImageResponseSchema
         ),
-        failedResponseHandler: (0, import_provider_utils7.createJsonErrorResponseHandler)({
-          errorSchema: import_v47.z.any(),
+        failedResponseHandler: (0, import_provider_utils8.createJsonErrorResponseHandler)({
+          errorSchema: import_v48.z.any(),
           errorToMessage: (data) => data
         }),
         ...abortSignal && { abortSignal },
@@ -852,32 +954,32 @@ var GatewayImageModel = class {
     };
   }
 };
-var providerMetadataEntrySchema = import_v47.z.object({
-  images: import_v47.z.array(import_v47.z.unknown()).optional()
-}).catchall(import_v47.z.unknown());
-var gatewayImageUsageSchema = import_v47.z.object({
-  inputTokens: import_v47.z.number().nullish(),
-  outputTokens: import_v47.z.number().nullish(),
-  totalTokens: import_v47.z.number().nullish()
+var providerMetadataEntrySchema = import_v48.z.object({
+  images: import_v48.z.array(import_v48.z.unknown()).optional()
+}).catchall(import_v48.z.unknown());
+var gatewayImageUsageSchema = import_v48.z.object({
+  inputTokens: import_v48.z.number().nullish(),
+  outputTokens: import_v48.z.number().nullish(),
+  totalTokens: import_v48.z.number().nullish()
 });
-var gatewayImageResponseSchema = import_v47.z.object({
-  images: import_v47.z.array(import_v47.z.string()),
+var gatewayImageResponseSchema = import_v48.z.object({
+  images: import_v48.z.array(import_v48.z.string()),
   // Always base64 strings over the wire
-  warnings: import_v47.z.array(
-    import_v47.z.object({
-      type: import_v47.z.literal("other"),
-      message: import_v47.z.string()
+  warnings: import_v48.z.array(
+    import_v48.z.object({
+      type: import_v48.z.literal("other"),
+      message: import_v48.z.string()
     })
   ).optional(),
-  providerMetadata: import_v47.z.record(import_v47.z.string(), providerMetadataEntrySchema).optional(),
+  providerMetadata: import_v48.z.record(import_v48.z.string(), providerMetadataEntrySchema).optional(),
   usage: gatewayImageUsageSchema.optional()
 });
 
 // src/tool/parallel-search.ts
-var import_provider_utils8 = require("@ai-sdk/provider-utils");
+var import_provider_utils9 = require("@ai-sdk/provider-utils");
 var import_zod = require("zod");
-var parallelSearchInputSchema = (0, import_provider_utils8.lazySchema)(
-  () => (0, import_provider_utils8.zodSchema)(
+var parallelSearchInputSchema = (0, import_provider_utils9.lazySchema)(
+  () => (0, import_provider_utils9.zodSchema)(
     import_zod.z.object({
       objective: import_zod.z.string().describe(
         "Natural-language description of the web research goal, including source or freshness guidance and broader context from the task. Maximum 5000 characters."
@@ -912,8 +1014,8 @@ var parallelSearchInputSchema = (0, import_provider_utils8.lazySchema)(
     })
   )
 );
-var parallelSearchOutputSchema = (0, import_provider_utils8.lazySchema)(
-  () => (0, import_provider_utils8.zodSchema)(
+var parallelSearchOutputSchema = (0, import_provider_utils9.lazySchema)(
+  () => (0, import_provider_utils9.zodSchema)(
     import_zod.z.union([
       // Success response
       import_zod.z.object({
@@ -944,7 +1046,7 @@ var parallelSearchOutputSchema = (0, import_provider_utils8.lazySchema)(
     ])
   )
 );
-var parallelSearchToolFactory = (0, import_provider_utils8.createProviderDefinedToolFactoryWithOutputSchema)({
+var parallelSearchToolFactory = (0, import_provider_utils9.createProviderDefinedToolFactoryWithOutputSchema)({
   id: "gateway.parallel_search",
   name: "parallel_search",
   inputSchema: parallelSearchInputSchema,
@@ -953,10 +1055,10 @@ var parallelSearchToolFactory = (0, import_provider_utils8.createProviderDefined
 var parallelSearch = (config = {}) => parallelSearchToolFactory(config);
 
 // src/tool/perplexity-search.ts
-var import_provider_utils9 = require("@ai-sdk/provider-utils");
+var import_provider_utils10 = require("@ai-sdk/provider-utils");
 var import_zod2 = require("zod");
-var perplexitySearchInputSchema = (0, import_provider_utils9.lazySchema)(
-  () => (0, import_provider_utils9.zodSchema)(
+var perplexitySearchInputSchema = (0, import_provider_utils10.lazySchema)(
+  () => (0, import_provider_utils10.zodSchema)(
     import_zod2.z.object({
       query: import_zod2.z.union([import_zod2.z.string(), import_zod2.z.array(import_zod2.z.string())]).describe(
         "Search query (string) or multiple queries (array of up to 5 strings). Multi-query searches return combined results from all queries."
@@ -997,8 +1099,8 @@ var perplexitySearchInputSchema = (0, import_provider_utils9.lazySchema)(
     })
   )
 );
-var perplexitySearchOutputSchema = (0, import_provider_utils9.lazySchema)(
-  () => (0, import_provider_utils9.zodSchema)(
+var perplexitySearchOutputSchema = (0, import_provider_utils10.lazySchema)(
+  () => (0, import_provider_utils10.zodSchema)(
     import_zod2.z.union([
       // Success response
       import_zod2.z.object({
@@ -1028,7 +1130,7 @@ var perplexitySearchOutputSchema = (0, import_provider_utils9.lazySchema)(
     ])
   )
 );
-var perplexitySearchToolFactory = (0, import_provider_utils9.createProviderDefinedToolFactoryWithOutputSchema)({
+var perplexitySearchToolFactory = (0, import_provider_utils10.createProviderDefinedToolFactoryWithOutputSchema)({
   id: "gateway.perplexity_search",
   name: "perplexity_search",
   inputSchema: perplexitySearchInputSchema,
@@ -1066,10 +1168,10 @@ async function getVercelRequestId() {
 }
 
 // src/gateway-provider.ts
-var import_provider_utils11 = require("@ai-sdk/provider-utils");
+var import_provider_utils12 = require("@ai-sdk/provider-utils");
 
 // src/version.ts
-var VERSION = true ? "2.0.64" : "0.0.0-test";
+var VERSION = true ? "2.0.66" : "0.0.0-test";
 
 // src/gateway-provider.ts
 var AI_GATEWAY_PROTOCOL_VERSION = "0.0.1";
@@ -1079,11 +1181,11 @@ function createGatewayProvider(options = {}) {
   let metadataCache = null;
   const cacheRefreshMillis = (_a9 = options.metadataCacheRefreshMillis) != null ? _a9 : 1e3 * 60 * 5;
   let lastFetchTime = 0;
-  const baseURL = (_b9 = (0, import_provider_utils10.withoutTrailingSlash)(options.baseURL)) != null ? _b9 : "https://ai-gateway.vercel.sh/v1/ai";
+  const baseURL = (_b9 = (0, import_provider_utils11.withoutTrailingSlash)(options.baseURL)) != null ? _b9 : "https://ai-gateway.vercel.sh/v1/ai";
   const getHeaders = async () => {
     const auth = await getGatewayAuthToken(options);
     if (auth) {
-      return (0, import_provider_utils11.withUserAgentSuffix)(
+      return (0, import_provider_utils12.withUserAgentSuffix)(
         {
           Authorization: `Bearer ${auth.token}`,
           "ai-gateway-protocol-version": AI_GATEWAY_PROTOCOL_VERSION,
@@ -1100,19 +1202,19 @@ function createGatewayProvider(options = {}) {
     });
   };
   const createO11yHeaders = () => {
-    const deploymentId = (0, import_provider_utils10.loadOptionalSetting)({
+    const deploymentId = (0, import_provider_utils11.loadOptionalSetting)({
       settingValue: void 0,
       environmentVariableName: "VERCEL_DEPLOYMENT_ID"
     });
-    const environment = (0, import_provider_utils10.loadOptionalSetting)({
+    const environment = (0, import_provider_utils11.loadOptionalSetting)({
       settingValue: void 0,
       environmentVariableName: "VERCEL_ENV"
     });
-    const region = (0, import_provider_utils10.loadOptionalSetting)({
+    const region = (0, import_provider_utils11.loadOptionalSetting)({
       settingValue: void 0,
       environmentVariableName: "VERCEL_REGION"
     });
-    const projectId = (0, import_provider_utils10.loadOptionalSetting)({
+    const projectId = (0, import_provider_utils11.loadOptionalSetting)({
       settingValue: void 0,
       environmentVariableName: "VERCEL_PROJECT_ID"
     });
@@ -1169,6 +1271,18 @@ function createGatewayProvider(options = {}) {
       );
     });
   };
+  const getSpendReport = async (params) => {
+    return new GatewaySpendReport({
+      baseURL,
+      headers: getHeaders,
+      fetch: options.fetch
+    }).getSpendReport(params).catch(async (error) => {
+      throw await asGatewayError(
+        error,
+        await parseAuthMethod(await getHeaders())
+      );
+    });
+  };
   const provider = function(modelId) {
     if (new.target) {
       throw new Error(
@@ -1179,6 +1293,7 @@ function createGatewayProvider(options = {}) {
   };
   provider.getAvailableModels = getAvailableModels;
   provider.getCredits = getCredits;
+  provider.getSpendReport = getSpendReport;
   provider.imageModel = (modelId) => {
     return new GatewayImageModel(modelId, {
       provider: "gateway",
@@ -1203,7 +1318,7 @@ function createGatewayProvider(options = {}) {
 }
 var gateway = createGatewayProvider();
 async function getGatewayAuthToken(options) {
-  const apiKey = (0, import_provider_utils10.loadOptionalSetting)({
+  const apiKey = (0, import_provider_utils11.loadOptionalSetting)({
     settingValue: options.apiKey,
     environmentVariableName: "AI_GATEWAY_API_KEY"
   });