@ai-sdk/fireworks 3.0.0-beta.6 → 3.0.0-beta.60

package/dist/index.js

@@ -1,52 +1,36 @@
-"use strict";
-var __defProp = Object.defineProperty;
-var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
-var __getOwnPropNames = Object.getOwnPropertyNames;
-var __hasOwnProp = Object.prototype.hasOwnProperty;
-var __export = (target, all) => {
-  for (var name in all)
-    __defProp(target, name, { get: all[name], enumerable: true });
-};
-var __copyProps = (to, from, except, desc) => {
-  if (from && typeof from === "object" || typeof from === "function") {
-    for (let key of __getOwnPropNames(from))
-      if (!__hasOwnProp.call(to, key) && key !== except)
-        __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
-  }
-  return to;
-};
-var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
-
-// src/index.ts
-var index_exports = {};
-__export(index_exports, {
-  FireworksImageModel: () => FireworksImageModel,
-  VERSION: () => VERSION,
-  createFireworks: () => createFireworks,
-  fireworks: () => fireworks
-});
-module.exports = __toCommonJS(index_exports);
-
 // src/fireworks-image-model.ts
-var import_provider_utils2 = require("@ai-sdk/provider-utils");
+import {
+  combineHeaders,
+  convertImageModelFileToDataUri,
+  createBinaryResponseHandler,
+  createJsonResponseHandler,
+  createStatusCodeErrorResponseHandler,
+  delay,
+  getFromApi,
+  isSameOrigin,
+  postJsonToApi,
+  serializeModelOptions,
+  WORKFLOW_SERIALIZE,
+  WORKFLOW_DESERIALIZE
+} from "@ai-sdk/provider-utils";
 
 // src/fireworks-image-api.ts
-var import_provider_utils = require("@ai-sdk/provider-utils");
-var import_v4 = require("zod/v4");
-var asyncSubmitResponseSchema = (0, import_provider_utils.lazySchema)(
-  () => (0, import_provider_utils.zodSchema)(
-    import_v4.z.object({
-      request_id: import_v4.z.string()
+import { lazySchema, zodSchema } from "@ai-sdk/provider-utils";
+import { z } from "zod/v4";
+var asyncSubmitResponseSchema = lazySchema(
+  () => zodSchema(
+    z.object({
+      request_id: z.string()
     })
   )
 );
-var asyncPollResponseSchema = (0, import_provider_utils.lazySchema)(
-  () => (0, import_provider_utils.zodSchema)(
-    import_v4.z.object({
-      id: import_v4.z.string(),
-      status: import_v4.z.string(),
-      result: import_v4.z.object({
-        sample: import_v4.z.string().optional()
+var asyncPollResponseSchema = lazySchema(
+  () => zodSchema(
+    z.object({
+      id: z.string(),
+      status: z.string(),
+      result: z.object({
+        sample: z.string().optional()
       }).nullable()
     })
   )
@@ -106,7 +90,7 @@ function getUrlForModel(baseUrl, modelId) {
 function getPollUrlForModel(baseUrl, modelId) {
   return `${baseUrl}/workflows/${modelId}/get_result`;
 }
-var FireworksImageModel = class {
+var FireworksImageModel = class _FireworksImageModel {
   constructor(modelId, config) {
     this.modelId = modelId;
     this.config = config;
@@ -116,6 +100,15 @@ var FireworksImageModel = class {
   get provider() {
     return this.config.provider;
   }
+  static [WORKFLOW_SERIALIZE](model) {
+    return serializeModelOptions({
+      modelId: model.modelId,
+      config: model.config
+    });
+  }
+  static [WORKFLOW_DESERIALIZE](options) {
+    return new _FireworksImageModel(options.modelId, options.config);
+  }
   async doGenerate({
     prompt,
     n,
@@ -128,7 +121,7 @@ var FireworksImageModel = class {
     files,
     mask
   }) {
-    var _a, _b, _c, _d;
+    var _a, _b, _c, _d, _e, _f;
     const warnings = [];
     const backendConfig = modelToBackendConfig[this.modelId];
     if (!(backendConfig == null ? void 0 : backendConfig.supportsSize) && size != null) {
@@ -148,7 +141,7 @@ var FireworksImageModel = class {
     const hasInputImage = files != null && files.length > 0;
     let inputImage;
     if (hasInputImage) {
-      inputImage = (0, import_provider_utils2.convertImageModelFileToDataUri)(files[0]);
+      inputImage = convertImageModelFileToDataUri(files[0]);
       if (files.length > 1) {
         warnings.push({
           type: "other",
@@ -165,7 +158,7 @@ var FireworksImageModel = class {
     }
     const splitSize = size == null ? void 0 : size.split("x");
     const currentDate = (_c = (_b = (_a = this.config._internal) == null ? void 0 : _a.currentDate) == null ? void 0 : _b.call(_a)) != null ? _c : /* @__PURE__ */ new Date();
-    const combinedHeaders = (0, import_provider_utils2.combineHeaders)(this.config.headers(), headers);
+    const combinedHeaders = combineHeaders((_e = (_d = this.config).headers) == null ? void 0 : _e.call(_d), headers);
     const body = {
       prompt,
       aspect_ratio: aspectRatio,
@@ -173,7 +166,7 @@ var FireworksImageModel = class {
       samples: n,
       ...inputImage && { input_image: inputImage },
       ...splitSize && { width: splitSize[0], height: splitSize[1] },
-      ...(_d = providerOptions.fireworks) != null ? _d : {}
+      ...(_f = providerOptions.fireworks) != null ? _f : {}
     };
     if ((backendConfig == null ? void 0 : backendConfig.urlFormat) === "workflows_async") {
       return this.doGenerateAsync({
@@ -184,12 +177,12 @@ var FireworksImageModel = class {
         currentDate
       });
     }
-    const { value: response, responseHeaders } = await (0, import_provider_utils2.postJsonToApi)({
+    const { value: response, responseHeaders } = await postJsonToApi({
       url: getUrlForModel(this.config.baseURL, this.modelId),
       headers: combinedHeaders,
       body,
-      failedResponseHandler: (0, import_provider_utils2.createStatusCodeErrorResponseHandler)(),
-      successfulResponseHandler: (0, import_provider_utils2.createBinaryResponseHandler)(),
+      failedResponseHandler: createStatusCodeErrorResponseHandler(),
+      successfulResponseHandler: createBinaryResponseHandler(),
       abortSignal,
       fetch: this.config.fetch
     });
@@ -214,12 +207,12 @@ var FireworksImageModel = class {
     warnings,
     currentDate
   }) {
-    const { value: submitResponse } = await (0, import_provider_utils2.postJsonToApi)({
+    const { value: submitResponse } = await postJsonToApi({
       url: getUrlForModel(this.config.baseURL, this.modelId),
       headers,
       body,
-      failedResponseHandler: (0, import_provider_utils2.createStatusCodeErrorResponseHandler)(),
-      successfulResponseHandler: (0, import_provider_utils2.createJsonResponseHandler)(
+      failedResponseHandler: createStatusCodeErrorResponseHandler(),
+      successfulResponseHandler: createJsonResponseHandler(
         asyncSubmitResponseSchema
       ),
       abortSignal,
@@ -231,12 +224,12 @@ var FireworksImageModel = class {
       headers,
       abortSignal
     });
-    const { value: imageBytes, responseHeaders } = await (0, import_provider_utils2.getFromApi)({
+    const { value: imageBytes, responseHeaders } = await getFromApi({
       url: imageUrl,
-      headers,
+      headers: isSameOrigin(imageUrl, this.config.baseURL) ? headers : void 0,
       abortSignal,
-      failedResponseHandler: (0, import_provider_utils2.createStatusCodeErrorResponseHandler)(),
-      successfulResponseHandler: (0, import_provider_utils2.createBinaryResponseHandler)(),
+      failedResponseHandler: createStatusCodeErrorResponseHandler(),
+      successfulResponseHandler: createBinaryResponseHandler(),
       fetch: this.config.fetch
     });
     return {
@@ -265,12 +258,12 @@ var FireworksImageModel = class {
     );
     const pollUrl = getPollUrlForModel(this.config.baseURL, this.modelId);
     for (let i = 0; i < maxPollAttempts; i++) {
-      const { value: pollResponse } = await (0, import_provider_utils2.postJsonToApi)({
+      const { value: pollResponse } = await postJsonToApi({
         url: pollUrl,
         headers,
         body: { id: requestId },
-        failedResponseHandler: (0, import_provider_utils2.createStatusCodeErrorResponseHandler)(),
-        successfulResponseHandler: (0, import_provider_utils2.createJsonResponseHandler)(
+        failedResponseHandler: createStatusCodeErrorResponseHandler(),
+        successfulResponseHandler: createJsonResponseHandler(
           asyncPollResponseSchema
         ),
         abortSignal,
@@ -291,7 +284,7 @@ var FireworksImageModel = class {
           `Fireworks image generation failed with status: ${status}`
         );
       }
-      await (0, import_provider_utils2.delay)(pollIntervalMillis);
+      await delay(pollIntervalMillis);
     }
     throw new Error(
       `Fireworks image generation timed out after ${pollTimeoutMillis}ms`
@@ -300,16 +293,24 @@ var FireworksImageModel = class {
 };
 
 // src/fireworks-provider.ts
-var import_openai_compatible = require("@ai-sdk/openai-compatible");
-var import_provider_utils3 = require("@ai-sdk/provider-utils");
-var import_v42 = require("zod/v4");
+import {
+  OpenAICompatibleChatLanguageModel,
+  OpenAICompatibleCompletionLanguageModel,
+  OpenAICompatibleEmbeddingModel
+} from "@ai-sdk/openai-compatible";
+import {
+  loadApiKey,
+  withoutTrailingSlash,
+  withUserAgentSuffix
+} from "@ai-sdk/provider-utils";
+import { z as z2 } from "zod/v4";
 
 // src/version.ts
-var VERSION = true ? "3.0.0-beta.6" : "0.0.0-test";
+var VERSION = true ? "3.0.0-beta.60" : "0.0.0-test";
 
 // src/fireworks-provider.ts
-var fireworksErrorSchema = import_v42.z.object({
-  error: import_v42.z.string()
+var fireworksErrorSchema = z2.object({
+  error: z2.string()
 });
 var fireworksErrorStructure = {
   errorSchema: fireworksErrorSchema,
@@ -318,10 +319,10 @@ var fireworksErrorStructure = {
 var defaultBaseURL = "https://api.fireworks.ai/inference/v1";
 function createFireworks(options = {}) {
   var _a;
-  const baseURL = (0, import_provider_utils3.withoutTrailingSlash)((_a = options.baseURL) != null ? _a : defaultBaseURL);
-  const getHeaders = () => (0, import_provider_utils3.withUserAgentSuffix)(
+  const baseURL = withoutTrailingSlash((_a = options.baseURL) != null ? _a : defaultBaseURL);
+  const getHeaders = () => withUserAgentSuffix(
     {
-      Authorization: `Bearer ${(0, import_provider_utils3.loadApiKey)({
+      Authorization: `Bearer ${loadApiKey({
         apiKey: options.apiKey,
         environmentVariableName: "FIREWORKS_API_KEY",
         description: "Fireworks API key"
@@ -337,15 +338,30 @@ function createFireworks(options = {}) {
     fetch: options.fetch
   });
   const createChatModel = (modelId) => {
-    return new import_openai_compatible.OpenAICompatibleChatLanguageModel(modelId, {
+    return new OpenAICompatibleChatLanguageModel(modelId, {
       ...getCommonModelConfig("chat"),
+      includeUsage: true,
       errorStructure: fireworksErrorStructure,
       transformRequestBody: (args) => {
         const thinking = args.thinking;
         const reasoningHistory = args.reasoningHistory;
-        const { thinking: _, reasoningHistory: __, ...rest } = args;
+        const promptCacheKey = args.promptCacheKey;
+        const {
+          thinking: _,
+          reasoningHistory: __,
+          promptCacheKey: ___,
+          reasoning_effort,
+          ...rest
+        } = args;
         return {
           ...rest,
+          ...reasoning_effort != null && {
+            // Workaround since OpenAI spec allows for 5 reasoning levels, but Fireworks only supports 3 of them.
+            reasoning_effort: reasoning_effort === "minimal" ? "low" : reasoning_effort === "xhigh" ? "high" : reasoning_effort
+          },
+          ...promptCacheKey !== void 0 && {
+            prompt_cache_key: promptCacheKey
+          },
           ...thinking && {
             thinking: {
               type: thinking.type,
@@ -361,11 +377,12 @@ function createFireworks(options = {}) {
       }
     });
   };
-  const createCompletionModel = (modelId) => new import_openai_compatible.OpenAICompatibleCompletionLanguageModel(modelId, {
+  const createCompletionModel = (modelId) => new OpenAICompatibleCompletionLanguageModel(modelId, {
     ...getCommonModelConfig("completion"),
+    includeUsage: true,
     errorStructure: fireworksErrorStructure
   });
-  const createEmbeddingModel = (modelId) => new import_openai_compatible.OpenAICompatibleEmbeddingModel(modelId, {
+  const createEmbeddingModel = (modelId) => new OpenAICompatibleEmbeddingModel(modelId, {
     ...getCommonModelConfig("embedding"),
     errorStructure: fireworksErrorStructure
   });
@@ -385,11 +402,10 @@ function createFireworks(options = {}) {
   return provider;
 }
 var fireworks = createFireworks();
-// Annotate the CommonJS export names for ESM import in node:
-0 && (module.exports = {
+export {
   FireworksImageModel,
   VERSION,
   createFireworks,
   fireworks
-});
+};
 //# sourceMappingURL=index.js.map

package/dist/index.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../src/index.ts","../src/fireworks-image-model.ts","../src/fireworks-image-api.ts","../src/fireworks-provider.ts","../src/version.ts"],"sourcesContent":["export type {\n  FireworksLanguageModelOptions,\n  /** @deprecated Use `FireworksLanguageModelOptions` instead. */\n  FireworksLanguageModelOptions as FireworksProviderOptions,\n} from './fireworks-chat-options';\nexport type {\n  FireworksEmbeddingModelId,\n  FireworksEmbeddingModelOptions,\n  /** @deprecated Use `FireworksEmbeddingModelOptions` instead. */\n  FireworksEmbeddingModelOptions as FireworksEmbeddingProviderOptions,\n} from './fireworks-embedding-options';\nexport { FireworksImageModel } from './fireworks-image-model';\nexport type { FireworksImageModelId } from './fireworks-image-options';\nexport { fireworks, createFireworks } from './fireworks-provider';\nexport type {\n  FireworksProvider,\n  FireworksProviderSettings,\n  FireworksErrorData,\n} from './fireworks-provider';\nexport { VERSION } from './version';\n","import { ImageModelV4, SharedV4Warning } from '@ai-sdk/provider';\nimport {\n  combineHeaders,\n  convertImageModelFileToDataUri,\n  createBinaryResponseHandler,\n  createJsonResponseHandler,\n  createStatusCodeErrorResponseHandler,\n  delay,\n  FetchFunction,\n  getFromApi,\n  postJsonToApi,\n} from '@ai-sdk/provider-utils';\nimport {\n  asyncPollResponseSchema,\n  asyncSubmitResponseSchema,\n} from './fireworks-image-api';\nimport { FireworksImageModelId } from './fireworks-image-options';\n\nconst DEFAULT_POLL_INTERVAL_MILLIS = 500;\nconst DEFAULT_POLL_TIMEOUT_MILLIS = 120000; // 2 minutes for image generation\n\ninterface FireworksImageModelBackendConfig {\n  urlFormat: 'workflows' | 'workflows_async' | 'image_generation';\n  supportsSize?: boolean;\n  supportsEditing?: boolean;\n}\n\nconst modelToBackendConfig: Partial<\n  Record<FireworksImageModelId, FireworksImageModelBackendConfig>\n> = {\n  'accounts/fireworks/models/flux-1-dev-fp8': {\n    urlFormat: 'workflows',\n  },\n  'accounts/fireworks/models/flux-1-schnell-fp8': {\n    urlFormat: 'workflows',\n  },\n  'accounts/fireworks/models/flux-kontext-pro': {\n    urlFormat: 'workflows_async',\n    supportsEditing: true,\n  },\n  'accounts/fireworks/models/flux-kontext-max': {\n    urlFormat: 'workflows_async',\n    supportsEditing: true,\n  },\n  'accounts/fireworks/models/playground-v2-5-1024px-aesthetic': {\n    urlFormat: 'image_generation',\n    supportsSize: true,\n  },\n  'accounts/fireworks/models/japanese-stable-diffusion-xl': {\n    urlFormat: 'image_generation',\n    supportsSize: true,\n  },\n  'accounts/fireworks/models/playground-v2-1024px-aesthetic': {\n    urlFormat: 'image_generation',\n    supportsSize: true,\n  },\n  'accounts/fireworks/models/stable-diffusion-xl-1024-v1-0': {\n    urlFormat: 'image_generation',\n    supportsSize: true,\n  },\n  'accounts/fireworks/models/SSD-1B': {\n    urlFormat: 'image_generation',\n    supportsSize: true,\n  },\n};\n\nfunction getUrlForModel(\n  baseUrl: string,\n  modelId: FireworksImageModelId,\n): string {\n  const config = modelToBackendConfig[modelId];\n\n  switch (config?.urlFormat) {\n    case 'image_generation':\n      return `${baseUrl}/image_generation/${modelId}`;\n    case 'workflows_async':\n      return `${baseUrl}/workflows/${modelId}`;\n    case 'workflows':\n    default:\n      return `${baseUrl}/workflows/${modelId}/text_to_image`;\n  }\n}\n\nfunction getPollUrlForModel(\n  baseUrl: string,\n  modelId: FireworksImageModelId,\n): string {\n  return `${baseUrl}/workflows/${modelId}/get_result`;\n}\n\ninterface FireworksImageModelConfig {\n  provider: string;\n  baseURL: string;\n  headers: () => Record<string, string>;\n  fetch?: FetchFunction;\n  /**\n   * Poll interval in milliseconds between status checks for async models.\n   * Defaults to 500ms.\n   */\n  pollIntervalMillis?: number;\n  /**\n   * Overall timeout in milliseconds for polling before giving up.\n   * Defaults to 120000ms (2 minutes).\n   */\n  pollTimeoutMillis?: number;\n  _internal?: {\n    currentDate?: () => Date;\n  };\n}\n\nexport class FireworksImageModel implements ImageModelV4 {\n  readonly specificationVersion = 'v4';\n  readonly maxImagesPerCall = 1;\n\n  get provider(): string {\n    return this.config.provider;\n  }\n\n  constructor(\n    readonly modelId: FireworksImageModelId,\n    private config: FireworksImageModelConfig,\n  ) {}\n\n  async doGenerate({\n    prompt,\n    n,\n    size,\n    aspectRatio,\n    seed,\n    providerOptions,\n    headers,\n    abortSignal,\n    files,\n    mask,\n  }: Parameters<ImageModelV4['doGenerate']>[0]): Promise<\n    Awaited<ReturnType<ImageModelV4['doGenerate']>>\n  > {\n    const warnings: Array<SharedV4Warning> = [];\n\n    const backendConfig = modelToBackendConfig[this.modelId];\n    if (!backendConfig?.supportsSize && size != null) {\n      warnings.push({\n        type: 'unsupported',\n        feature: 'size',\n        details:\n          'This model does not support the `size` option. Use `aspectRatio` instead.',\n      });\n    }\n\n    // Use supportsSize as a proxy for whether the model does not support\n    // aspectRatio. This invariant holds for the current set of models.\n    if (backendConfig?.supportsSize && aspectRatio != null) {\n      warnings.push({\n        type: 'unsupported',\n        feature: 'aspectRatio',\n        details: 'This model does not support the `aspectRatio` option.',\n      });\n    }\n\n    // Handle files for image editing\n    const hasInputImage = files != null && files.length > 0;\n    let inputImage: string | undefined;\n\n    if (hasInputImage) {\n      inputImage = convertImageModelFileToDataUri(files[0]);\n\n      if (files.length > 1) {\n        warnings.push({\n          type: 'other',\n          message:\n            'Fireworks only supports a single input image. Additional images are ignored.',\n        });\n      }\n    }\n\n    // Warn about mask - Fireworks Kontext models don't support explicit masks\n    if (mask != null) {\n      warnings.push({\n        type: 'unsupported',\n        feature: 'mask',\n        details:\n          'Fireworks Kontext models do not support explicit masks. Use the prompt to describe the areas to edit.',\n      });\n    }\n\n    const splitSize = size?.split('x');\n    const currentDate = this.config._internal?.currentDate?.() ?? new Date();\n    const combinedHeaders = combineHeaders(this.config.headers(), headers);\n\n    const body = {\n      prompt,\n      aspect_ratio: aspectRatio,\n      seed,\n      samples: n,\n      ...(inputImage && { input_image: inputImage }),\n      ...(splitSize && { width: splitSize[0], height: splitSize[1] }),\n      ...(providerOptions.fireworks ?? {}),\n    };\n\n    // Handle async models that require polling (e.g., flux-kontext-*)\n    if (backendConfig?.urlFormat === 'workflows_async') {\n      return this.doGenerateAsync({\n        body,\n        headers: combinedHeaders,\n        abortSignal,\n        warnings,\n        currentDate,\n      });\n    }\n\n    // Handle sync models that return binary directly\n    const { value: response, responseHeaders } = await postJsonToApi({\n      url: getUrlForModel(this.config.baseURL, this.modelId),\n      headers: combinedHeaders,\n      body,\n      failedResponseHandler: createStatusCodeErrorResponseHandler(),\n      successfulResponseHandler: createBinaryResponseHandler(),\n      abortSignal,\n      fetch: this.config.fetch,\n    });\n\n    return {\n      images: [response],\n      warnings,\n      response: {\n        timestamp: currentDate,\n        modelId: this.modelId,\n        headers: responseHeaders,\n      },\n    };\n  }\n\n  /**\n   * Handles async image generation for models like flux-kontext-* that return\n   * a request_id and require polling for results.\n   */\n  private async doGenerateAsync({\n    body,\n    headers,\n    abortSignal,\n    warnings,\n    currentDate,\n  }: {\n    body: Record<string, unknown>;\n    headers: Record<string, string | undefined>;\n    abortSignal: AbortSignal | undefined;\n    warnings: Array<SharedV4Warning>;\n    currentDate: Date;\n  }): Promise<Awaited<ReturnType<ImageModelV4['doGenerate']>>> {\n    // Submit the generation request\n    const { value: submitResponse } = await postJsonToApi({\n      url: getUrlForModel(this.config.baseURL, this.modelId),\n      headers,\n      body,\n      failedResponseHandler: createStatusCodeErrorResponseHandler(),\n      successfulResponseHandler: createJsonResponseHandler(\n        asyncSubmitResponseSchema,\n      ),\n      abortSignal,\n      fetch: this.config.fetch,\n    });\n\n    const requestId = submitResponse.request_id;\n\n    // Poll for the result\n    const imageUrl = await this.pollForImageUrl({\n      requestId,\n      headers,\n      abortSignal,\n    });\n\n    // Download the image from the URL\n    const { value: imageBytes, responseHeaders } = await getFromApi({\n      url: imageUrl,\n      headers,\n      abortSignal,\n      failedResponseHandler: createStatusCodeErrorResponseHandler(),\n      successfulResponseHandler: createBinaryResponseHandler(),\n      fetch: this.config.fetch,\n    });\n\n    return {\n      images: [imageBytes],\n      warnings,\n      response: {\n        timestamp: currentDate,\n        modelId: this.modelId,\n        headers: responseHeaders,\n      },\n    };\n  }\n\n  /**\n   * Polls the get_result endpoint until the image is ready.\n   */\n  private async pollForImageUrl({\n    requestId,\n    headers,\n    abortSignal,\n  }: {\n    requestId: string;\n    headers: Record<string, string | undefined>;\n    abortSignal: AbortSignal | undefined;\n  }): Promise<string> {\n    const pollIntervalMillis =\n      this.config.pollIntervalMillis ?? DEFAULT_POLL_INTERVAL_MILLIS;\n    const pollTimeoutMillis =\n      this.config.pollTimeoutMillis ?? DEFAULT_POLL_TIMEOUT_MILLIS;\n    const maxPollAttempts = Math.ceil(\n      pollTimeoutMillis / Math.max(1, pollIntervalMillis),\n    );\n\n    const pollUrl = getPollUrlForModel(this.config.baseURL, this.modelId);\n\n    for (let i = 0; i < maxPollAttempts; i++) {\n      const { value: pollResponse } = await postJsonToApi({\n        url: pollUrl,\n        headers,\n        body: { id: requestId },\n        failedResponseHandler: createStatusCodeErrorResponseHandler(),\n        successfulResponseHandler: createJsonResponseHandler(\n          asyncPollResponseSchema,\n        ),\n        abortSignal,\n        fetch: this.config.fetch,\n      });\n\n      const status = pollResponse.status;\n\n      if (status === 'Ready') {\n        const imageUrl = pollResponse.result?.sample;\n        if (typeof imageUrl === 'string') {\n          return imageUrl;\n        }\n        throw new Error(\n          'Fireworks poll response is Ready but missing result.sample',\n        );\n      }\n\n      if (status === 'Error' || status === 'Failed') {\n        throw new Error(\n          `Fireworks image generation failed with status: ${status}`,\n        );\n      }\n\n      // Wait before next poll attempt\n      await delay(pollIntervalMillis);\n    }\n\n    throw new Error(\n      `Fireworks image generation timed out after ${pollTimeoutMillis}ms`,\n    );\n  }\n}\n","import { lazySchema, zodSchema } from '@ai-sdk/provider-utils';\nimport { z } from 'zod/v4';\n\n// Schema for async submit response\nexport const asyncSubmitResponseSchema = lazySchema(() =>\n  zodSchema(\n    z.object({\n      request_id: z.string(),\n    }),\n  ),\n);\n\n// Schema for async poll response\nexport const asyncPollResponseSchema = lazySchema(() =>\n  zodSchema(\n    z.object({\n      id: z.string(),\n      status: z.string(),\n      result: z\n        .object({\n          sample: z.string().optional(),\n        })\n        .nullable(),\n    }),\n  ),\n);\n","import {\n  OpenAICompatibleChatLanguageModel,\n  OpenAICompatibleCompletionLanguageModel,\n  OpenAICompatibleEmbeddingModel,\n  ProviderErrorStructure,\n} from '@ai-sdk/openai-compatible';\nimport {\n  EmbeddingModelV4,\n  ImageModelV4,\n  LanguageModelV4,\n  ProviderV4,\n} from '@ai-sdk/provider';\nimport {\n  FetchFunction,\n  loadApiKey,\n  withoutTrailingSlash,\n  withUserAgentSuffix,\n} from '@ai-sdk/provider-utils';\nimport { z } from 'zod/v4';\nimport { FireworksChatModelId } from './fireworks-chat-options';\nimport { FireworksCompletionModelId } from './fireworks-completion-options';\nimport { FireworksEmbeddingModelId } from './fireworks-embedding-options';\nimport { FireworksImageModel } from './fireworks-image-model';\nimport { FireworksImageModelId } from './fireworks-image-options';\nimport { VERSION } from './version';\n\nexport type FireworksErrorData = z.infer<typeof fireworksErrorSchema>;\n\nconst fireworksErrorSchema = z.object({\n  error: z.string(),\n});\n\nconst fireworksErrorStructure: ProviderErrorStructure<FireworksErrorData> = {\n  errorSchema: fireworksErrorSchema,\n  errorToMessage: data => data.error,\n};\n\nexport interface FireworksProviderSettings {\n  /**\n   * Fireworks API key. Default value is taken from the `FIREWORKS_API_KEY`\n   * environment variable.\n   */\n  apiKey?: string;\n  /**\n   * Base URL for the API calls.\n   */\n  baseURL?: string;\n  /**\n   * Custom headers to include in the requests.\n   */\n  headers?: Record<string, string>;\n  /**\n   * Custom fetch implementation. You can use it as a middleware to intercept requests,\n   * or to provide a custom fetch implementation for e.g. testing.\n   */\n  fetch?: FetchFunction;\n}\n\nexport interface FireworksProvider extends ProviderV4 {\n  /**\n   * Creates a model for text generation.\n   */\n  (modelId: FireworksChatModelId): LanguageModelV4;\n\n  /**\n   * Creates a chat model for text generation.\n   */\n  chatModel(modelId: FireworksChatModelId): LanguageModelV4;\n\n  /**\n   * Creates a completion model for text generation.\n   */\n  completionModel(modelId: FireworksCompletionModelId): LanguageModelV4;\n\n  /**\n   * Creates a chat model for text generation.\n   */\n  languageModel(modelId: FireworksChatModelId): LanguageModelV4;\n\n  /**\n   * Creates a text embedding model for text generation.\n   */\n  embeddingModel(modelId: FireworksEmbeddingModelId): EmbeddingModelV4;\n\n  /**\n   * @deprecated Use `embeddingModel` instead.\n   */\n  textEmbeddingModel(modelId: FireworksEmbeddingModelId): EmbeddingModelV4;\n\n  /**\n   * Creates a model for image generation.\n   */\n  image(modelId: FireworksImageModelId): ImageModelV4;\n\n  /**\n   * Creates a model for image generation.\n   */\n  imageModel(modelId: FireworksImageModelId): ImageModelV4;\n}\n\nconst defaultBaseURL = 'https://api.fireworks.ai/inference/v1';\n\nexport function createFireworks(\n  options: FireworksProviderSettings = {},\n): FireworksProvider {\n  const baseURL = withoutTrailingSlash(options.baseURL ?? defaultBaseURL);\n  const getHeaders = () =>\n    withUserAgentSuffix(\n      {\n        Authorization: `Bearer ${loadApiKey({\n          apiKey: options.apiKey,\n          environmentVariableName: 'FIREWORKS_API_KEY',\n          description: 'Fireworks API key',\n        })}`,\n        ...options.headers,\n      },\n      `ai-sdk/fireworks/${VERSION}`,\n    );\n\n  interface CommonModelConfig {\n    provider: string;\n    url: ({ path }: { path: string }) => string;\n    headers: () => Record<string, string>;\n    fetch?: FetchFunction;\n  }\n\n  const getCommonModelConfig = (modelType: string): CommonModelConfig => ({\n    provider: `fireworks.${modelType}`,\n    url: ({ path }) => `${baseURL}${path}`,\n    headers: getHeaders,\n    fetch: options.fetch,\n  });\n\n  const createChatModel = (modelId: FireworksChatModelId) => {\n    return new OpenAICompatibleChatLanguageModel(modelId, {\n      ...getCommonModelConfig('chat'),\n      errorStructure: fireworksErrorStructure,\n      transformRequestBody: args => {\n        const thinking = args.thinking as\n          | { type?: string; budgetTokens?: number }\n          | undefined;\n        const reasoningHistory = args.reasoningHistory as string | undefined;\n\n        const { thinking: _, reasoningHistory: __, ...rest } = args;\n\n        return {\n          ...rest,\n          ...(thinking && {\n            thinking: {\n              type: thinking.type,\n              ...(thinking.budgetTokens !== undefined && {\n                budget_tokens: thinking.budgetTokens,\n              }),\n            },\n          }),\n          ...(reasoningHistory && {\n            reasoning_history: reasoningHistory,\n          }),\n        };\n      },\n    });\n  };\n\n  const createCompletionModel = (modelId: FireworksCompletionModelId) =>\n    new OpenAICompatibleCompletionLanguageModel(modelId, {\n      ...getCommonModelConfig('completion'),\n      errorStructure: fireworksErrorStructure,\n    });\n\n  const createEmbeddingModel = (modelId: FireworksEmbeddingModelId) =>\n    new OpenAICompatibleEmbeddingModel(modelId, {\n      ...getCommonModelConfig('embedding'),\n      errorStructure: fireworksErrorStructure,\n    });\n\n  const createImageModel = (modelId: FireworksImageModelId) =>\n    new FireworksImageModel(modelId, {\n      ...getCommonModelConfig('image'),\n      baseURL: baseURL ?? defaultBaseURL,\n    });\n\n  const provider = (modelId: FireworksChatModelId) => createChatModel(modelId);\n\n  provider.specificationVersion = 'v4' as const;\n  provider.completionModel = createCompletionModel;\n  provider.chatModel = createChatModel;\n  provider.languageModel = createChatModel;\n  provider.embeddingModel = createEmbeddingModel;\n  provider.textEmbeddingModel = createEmbeddingModel;\n  provider.image = createImageModel;\n  provider.imageModel = createImageModel;\n  return provider;\n}\n\nexport const fireworks = createFireworks();\n","// Version string of this package injected at build time.\ndeclare const __PACKAGE_VERSION__: string | undefined;\nexport const VERSION: string =\n  typeof __PACKAGE_VERSION__ !== 'undefined'\n    ? __PACKAGE_VERSION__\n    : '0.0.0-test';\n"],"mappings":";;;;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;;;ACCA,IAAAA,yBAUO;;;ACXP,4BAAsC;AACtC,gBAAkB;AAGX,IAAM,gCAA4B;AAAA,EAAW,UAClD;AAAA,IACE,YAAE,OAAO;AAAA,MACP,YAAY,YAAE,OAAO;AAAA,IACvB,CAAC;AAAA,EACH;AACF;AAGO,IAAM,8BAA0B;AAAA,EAAW,UAChD;AAAA,IACE,YAAE,OAAO;AAAA,MACP,IAAI,YAAE,OAAO;AAAA,MACb,QAAQ,YAAE,OAAO;AAAA,MACjB,QAAQ,YACL,OAAO;AAAA,QACN,QAAQ,YAAE,OAAO,EAAE,SAAS;AAAA,MAC9B,CAAC,EACA,SAAS;AAAA,IACd,CAAC;AAAA,EACH;AACF;;;ADPA,IAAM,+BAA+B;AACrC,IAAM,8BAA8B;AAQpC,IAAM,uBAEF;AAAA,EACF,4CAA4C;AAAA,IAC1C,WAAW;AAAA,EACb;AAAA,EACA,gDAAgD;AAAA,IAC9C,WAAW;AAAA,EACb;AAAA,EACA,8CAA8C;AAAA,IAC5C,WAAW;AAAA,IACX,iBAAiB;AAAA,EACnB;AAAA,EACA,8CAA8C;AAAA,IAC5C,WAAW;AAAA,IACX,iBAAiB;AAAA,EACnB;AAAA,EACA,8DAA8D;AAAA,IAC5D,WAAW;AAAA,IACX,cAAc;AAAA,EAChB;AAAA,EACA,0DAA0D;AAAA,IACxD,WAAW;AAAA,IACX,cAAc;AAAA,EAChB;AAAA,EACA,4DAA4D;AAAA,IAC1D,WAAW;AAAA,IACX,cAAc;AAAA,EAChB;AAAA,EACA,2DAA2D;AAAA,IACzD,WAAW;AAAA,IACX,cAAc;AAAA,EAChB;AAAA,EACA,oCAAoC;AAAA,IAClC,WAAW;AAAA,IACX,cAAc;AAAA,EAChB;AACF;AAEA,SAAS,eACP,SACA,SACQ;AACR,QAAM,SAAS,qBAAqB,OAAO;AAE3C,UAAQ,iCAAQ,WAAW;AAAA,IACzB,KAAK;AACH,aAAO,GAAG,OAAO,qBAAqB,OAAO;AAAA,IAC/C,KAAK;AACH,aAAO,GAAG,OAAO,cAAc,OAAO;AAAA,IACxC,KAAK;AAAA,IACL;AACE,aAAO,GAAG,OAAO,cAAc,OAAO;AAAA,EAC1C;AACF;AAEA,SAAS,mBACP,SACA,SACQ;AACR,SAAO,GAAG,OAAO,cAAc,OAAO;AACxC;AAsBO,IAAM,sBAAN,MAAkD;AAAA,EAQvD,YACW,SACD,QACR;AAFS;AACD;AATV,SAAS,uBAAuB;AAChC,SAAS,mBAAmB;AAAA,EASzB;AAAA,EAPH,IAAI,WAAmB;AACrB,WAAO,KAAK,OAAO;AAAA,EACrB;AAAA,EAOA,MAAM,WAAW;AAAA,IACf;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,EACF,GAEE;AAxIJ;AAyII,UAAM,WAAmC,CAAC;AAE1C,UAAM,gBAAgB,qBAAqB,KAAK,OAAO;AACvD,QAAI,EAAC,+CAAe,iBAAgB,QAAQ,MAAM;AAChD,eAAS,KAAK;AAAA,QACZ,MAAM;AAAA,QACN,SAAS;AAAA,QACT,SACE;AAAA,MACJ,CAAC;AAAA,IACH;AAIA,SAAI,+CAAe,iBAAgB,eAAe,MAAM;AACtD,eAAS,KAAK;AAAA,QACZ,MAAM;AAAA,QACN,SAAS;AAAA,QACT,SAAS;AAAA,MACX,CAAC;AAAA,IACH;AAGA,UAAM,gBAAgB,SAAS,QAAQ,MAAM,SAAS;AACtD,QAAI;AAEJ,QAAI,eAAe;AACjB,uBAAa,uDAA+B,MAAM,CAAC,CAAC;AAEpD,UAAI,MAAM,SAAS,GAAG;AACpB,iBAAS,KAAK;AAAA,UACZ,MAAM;AAAA,UACN,SACE;AAAA,QACJ,CAAC;AAAA,MACH;AAAA,IACF;AAGA,QAAI,QAAQ,MAAM;AAChB,eAAS,KAAK;AAAA,QACZ,MAAM;AAAA,QACN,SAAS;AAAA,QACT,SACE;AAAA,MACJ,CAAC;AAAA,IACH;AAEA,UAAM,YAAY,6BAAM,MAAM;AAC9B,UAAM,eAAc,sBAAK,OAAO,cAAZ,mBAAuB,gBAAvB,4CAA0C,oBAAI,KAAK;AACvE,UAAM,sBAAkB,uCAAe,KAAK,OAAO,QAAQ,GAAG,OAAO;AAErE,UAAM,OAAO;AAAA,MACX;AAAA,MACA,cAAc;AAAA,MACd;AAAA,MACA,SAAS;AAAA,MACT,GAAI,cAAc,EAAE,aAAa,WAAW;AAAA,MAC5C,GAAI,aAAa,EAAE,OAAO,UAAU,CAAC,GAAG,QAAQ,UAAU,CAAC,EAAE;AAAA,MAC7D,IAAI,qBAAgB,cAAhB,YAA6B,CAAC;AAAA,IACpC;AAGA,SAAI,+CAAe,eAAc,mBAAmB;AAClD,aAAO,KAAK,gBAAgB;AAAA,QAC1B;AAAA,QACA,SAAS;AAAA,QACT;AAAA,QACA;AAAA,QACA;AAAA,MACF,CAAC;AAAA,IACH;AAGA,UAAM,EAAE,OAAO,UAAU,gBAAgB,IAAI,UAAM,sCAAc;AAAA,MAC/D,KAAK,eAAe,KAAK,OAAO,SAAS,KAAK,OAAO;AAAA,MACrD,SAAS;AAAA,MACT;AAAA,MACA,2BAAuB,6DAAqC;AAAA,MAC5D,+BAA2B,oDAA4B;AAAA,MACvD;AAAA,MACA,OAAO,KAAK,OAAO;AAAA,IACrB,CAAC;AAED,WAAO;AAAA,MACL,QAAQ,CAAC,QAAQ;AAAA,MACjB;AAAA,MACA,UAAU;AAAA,QACR,WAAW;AAAA,QACX,SAAS,KAAK;AAAA,QACd,SAAS;AAAA,MACX;AAAA,IACF;AAAA,EACF;AAAA;AAAA;AAAA;AAAA;AAAA,EAMA,MAAc,gBAAgB;AAAA,IAC5B;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,EACF,GAM6D;AAE3D,UAAM,EAAE,OAAO,eAAe,IAAI,UAAM,sCAAc;AAAA,MACpD,KAAK,eAAe,KAAK,OAAO,SAAS,KAAK,OAAO;AAAA,MACrD;AAAA,MACA;AAAA,MACA,2BAAuB,6DAAqC;AAAA,MAC5D,+BAA2B;AAAA,QACzB;AAAA,MACF;AAAA,MACA;AAAA,MACA,OAAO,KAAK,OAAO;AAAA,IACrB,CAAC;AAED,UAAM,YAAY,eAAe;AAGjC,UAAM,WAAW,MAAM,KAAK,gBAAgB;AAAA,MAC1C;AAAA,MACA;AAAA,MACA;AAAA,IACF,CAAC;AAGD,UAAM,EAAE,OAAO,YAAY,gBAAgB,IAAI,UAAM,mCAAW;AAAA,MAC9D,KAAK;AAAA,MACL;AAAA,MACA;AAAA,MACA,2BAAuB,6DAAqC;AAAA,MAC5D,+BAA2B,oDAA4B;AAAA,MACvD,OAAO,KAAK,OAAO;AAAA,IACrB,CAAC;AAED,WAAO;AAAA,MACL,QAAQ,CAAC,UAAU;AAAA,MACnB;AAAA,MACA,UAAU;AAAA,QACR,WAAW;AAAA,QACX,SAAS,KAAK;AAAA,QACd,SAAS;AAAA,MACX;AAAA,IACF;AAAA,EACF;AAAA;AAAA;AAAA;AAAA,EAKA,MAAc,gBAAgB;AAAA,IAC5B;AAAA,IACA;AAAA,IACA;AAAA,EACF,GAIoB;AA/StB;AAgTI,UAAM,sBACJ,UAAK,OAAO,uBAAZ,YAAkC;AACpC,UAAM,qBACJ,UAAK,OAAO,sBAAZ,YAAiC;AACnC,UAAM,kBAAkB,KAAK;AAAA,MAC3B,oBAAoB,KAAK,IAAI,GAAG,kBAAkB;AAAA,IACpD;AAEA,UAAM,UAAU,mBAAmB,KAAK,OAAO,SAAS,KAAK,OAAO;AAEpE,aAAS,IAAI,GAAG,IAAI,iBAAiB,KAAK;AACxC,YAAM,EAAE,OAAO,aAAa,IAAI,UAAM,sCAAc;AAAA,QAClD,KAAK;AAAA,QACL;AAAA,QACA,MAAM,EAAE,IAAI,UAAU;AAAA,QACtB,2BAAuB,6DAAqC;AAAA,QAC5D,+BAA2B;AAAA,UACzB;AAAA,QACF;AAAA,QACA;AAAA,QACA,OAAO,KAAK,OAAO;AAAA,MACrB,CAAC;AAED,YAAM,SAAS,aAAa;AAE5B,UAAI,WAAW,SAAS;AACtB,cAAM,YAAW,kBAAa,WAAb,mBAAqB;AACtC,YAAI,OAAO,aAAa,UAAU;AAChC,iBAAO;AAAA,QACT;AACA,cAAM,IAAI;AAAA,UACR;AAAA,QACF;AAAA,MACF;AAEA,UAAI,WAAW,WAAW,WAAW,UAAU;AAC7C,cAAM,IAAI;AAAA,UACR,kDAAkD,MAAM;AAAA,QAC1D;AAAA,MACF;AAGA,gBAAM,8BAAM,kBAAkB;AAAA,IAChC;AAEA,UAAM,IAAI;AAAA,MACR,8CAA8C,iBAAiB;AAAA,IACjE;AAAA,EACF;AACF;;;AEjWA,+BAKO;AAOP,IAAAC,yBAKO;AACP,IAAAC,aAAkB;;;AChBX,IAAM,UACX,OACI,iBACA;;;ADuBN,IAAM,uBAAuB,aAAE,OAAO;AAAA,EACpC,OAAO,aAAE,OAAO;AAClB,CAAC;AAED,IAAM,0BAAsE;AAAA,EAC1E,aAAa;AAAA,EACb,gBAAgB,UAAQ,KAAK;AAC/B;AAiEA,IAAM,iBAAiB;AAEhB,SAAS,gBACd,UAAqC,CAAC,GACnB;AAxGrB;AAyGE,QAAM,cAAU,8CAAqB,aAAQ,YAAR,YAAmB,cAAc;AACtE,QAAM,aAAa,UACjB;AAAA,IACE;AAAA,MACE,eAAe,cAAU,mCAAW;AAAA,QAClC,QAAQ,QAAQ;AAAA,QAChB,yBAAyB;AAAA,QACzB,aAAa;AAAA,MACf,CAAC,CAAC;AAAA,MACF,GAAG,QAAQ;AAAA,IACb;AAAA,IACA,oBAAoB,OAAO;AAAA,EAC7B;AASF,QAAM,uBAAuB,CAAC,eAA0C;AAAA,IACtE,UAAU,aAAa,SAAS;AAAA,IAChC,KAAK,CAAC,EAAE,KAAK,MAAM,GAAG,OAAO,GAAG,IAAI;AAAA,IACpC,SAAS;AAAA,IACT,OAAO,QAAQ;AAAA,EACjB;AAEA,QAAM,kBAAkB,CAAC,YAAkC;AACzD,WAAO,IAAI,2DAAkC,SAAS;AAAA,MACpD,GAAG,qBAAqB,MAAM;AAAA,MAC9B,gBAAgB;AAAA,MAChB,sBAAsB,UAAQ;AAC5B,cAAM,WAAW,KAAK;AAGtB,cAAM,mBAAmB,KAAK;AAE9B,cAAM,EAAE,UAAU,GAAG,kBAAkB,IAAI,GAAG,KAAK,IAAI;AAEvD,eAAO;AAAA,UACL,GAAG;AAAA,UACH,GAAI,YAAY;AAAA,YACd,UAAU;AAAA,cACR,MAAM,SAAS;AAAA,cACf,GAAI,SAAS,iBAAiB,UAAa;AAAA,gBACzC,eAAe,SAAS;AAAA,cAC1B;AAAA,YACF;AAAA,UACF;AAAA,UACA,GAAI,oBAAoB;AAAA,YACtB,mBAAmB;AAAA,UACrB;AAAA,QACF;AAAA,MACF;AAAA,IACF,CAAC;AAAA,EACH;AAEA,QAAM,wBAAwB,CAAC,YAC7B,IAAI,iEAAwC,SAAS;AAAA,IACnD,GAAG,qBAAqB,YAAY;AAAA,IACpC,gBAAgB;AAAA,EAClB,CAAC;AAEH,QAAM,uBAAuB,CAAC,YAC5B,IAAI,wDAA+B,SAAS;AAAA,IAC1C,GAAG,qBAAqB,WAAW;AAAA,IACnC,gBAAgB;AAAA,EAClB,CAAC;AAEH,QAAM,mBAAmB,CAAC,YACxB,IAAI,oBAAoB,SAAS;AAAA,IAC/B,GAAG,qBAAqB,OAAO;AAAA,IAC/B,SAAS,4BAAW;AAAA,EACtB,CAAC;AAEH,QAAM,WAAW,CAAC,YAAkC,gBAAgB,OAAO;AAE3E,WAAS,uBAAuB;AAChC,WAAS,kBAAkB;AAC3B,WAAS,YAAY;AACrB,WAAS,gBAAgB;AACzB,WAAS,iBAAiB;AAC1B,WAAS,qBAAqB;AAC9B,WAAS,QAAQ;AACjB,WAAS,aAAa;AACtB,SAAO;AACT;AAEO,IAAM,YAAY,gBAAgB;","names":["import_provider_utils","import_provider_utils","import_v4"]}
+{"version":3,"sources":["../src/fireworks-image-model.ts","../src/fireworks-image-api.ts","../src/fireworks-provider.ts","../src/version.ts"],"sourcesContent":["import type { ImageModelV4, SharedV4Warning } from '@ai-sdk/provider';\nimport {\n  combineHeaders,\n  convertImageModelFileToDataUri,\n  createBinaryResponseHandler,\n  createJsonResponseHandler,\n  createStatusCodeErrorResponseHandler,\n  delay,\n  getFromApi,\n  isSameOrigin,\n  postJsonToApi,\n  serializeModelOptions,\n  WORKFLOW_SERIALIZE,\n  WORKFLOW_DESERIALIZE,\n  type FetchFunction,\n} from '@ai-sdk/provider-utils';\nimport {\n  asyncPollResponseSchema,\n  asyncSubmitResponseSchema,\n} from './fireworks-image-api';\nimport type { FireworksImageModelId } from './fireworks-image-options';\n\nconst DEFAULT_POLL_INTERVAL_MILLIS = 500;\nconst DEFAULT_POLL_TIMEOUT_MILLIS = 120000; // 2 minutes for image generation\n\ninterface FireworksImageModelBackendConfig {\n  urlFormat: 'workflows' | 'workflows_async' | 'image_generation';\n  supportsSize?: boolean;\n  supportsEditing?: boolean;\n}\n\nconst modelToBackendConfig: Partial<\n  Record<FireworksImageModelId, FireworksImageModelBackendConfig>\n> = {\n  'accounts/fireworks/models/flux-1-dev-fp8': {\n    urlFormat: 'workflows',\n  },\n  'accounts/fireworks/models/flux-1-schnell-fp8': {\n    urlFormat: 'workflows',\n  },\n  'accounts/fireworks/models/flux-kontext-pro': {\n    urlFormat: 'workflows_async',\n    supportsEditing: true,\n  },\n  'accounts/fireworks/models/flux-kontext-max': {\n    urlFormat: 'workflows_async',\n    supportsEditing: true,\n  },\n  'accounts/fireworks/models/playground-v2-5-1024px-aesthetic': {\n    urlFormat: 'image_generation',\n    supportsSize: true,\n  },\n  'accounts/fireworks/models/japanese-stable-diffusion-xl': {\n    urlFormat: 'image_generation',\n    supportsSize: true,\n  },\n  'accounts/fireworks/models/playground-v2-1024px-aesthetic': {\n    urlFormat: 'image_generation',\n    supportsSize: true,\n  },\n  'accounts/fireworks/models/stable-diffusion-xl-1024-v1-0': {\n    urlFormat: 'image_generation',\n    supportsSize: true,\n  },\n  'accounts/fireworks/models/SSD-1B': {\n    urlFormat: 'image_generation',\n    supportsSize: true,\n  },\n};\n\nfunction getUrlForModel(\n  baseUrl: string,\n  modelId: FireworksImageModelId,\n): string {\n  const config = modelToBackendConfig[modelId];\n\n  switch (config?.urlFormat) {\n    case 'image_generation':\n      return `${baseUrl}/image_generation/${modelId}`;\n    case 'workflows_async':\n      return `${baseUrl}/workflows/${modelId}`;\n    case 'workflows':\n    default:\n      return `${baseUrl}/workflows/${modelId}/text_to_image`;\n  }\n}\n\nfunction getPollUrlForModel(\n  baseUrl: string,\n  modelId: FireworksImageModelId,\n): string {\n  return `${baseUrl}/workflows/${modelId}/get_result`;\n}\n\ninterface FireworksImageModelConfig {\n  provider: string;\n  baseURL: string;\n  headers?: () => Record<string, string>;\n  fetch?: FetchFunction;\n  /**\n   * Poll interval in milliseconds between status checks for async models.\n   * Defaults to 500ms.\n   */\n  pollIntervalMillis?: number;\n  /**\n   * Overall timeout in milliseconds for polling before giving up.\n   * Defaults to 120000ms (2 minutes).\n   */\n  pollTimeoutMillis?: number;\n  _internal?: {\n    currentDate?: () => Date;\n  };\n}\n\nexport class FireworksImageModel implements ImageModelV4 {\n  readonly specificationVersion = 'v4';\n  readonly maxImagesPerCall = 1;\n\n  get provider(): string {\n    return this.config.provider;\n  }\n\n  static [WORKFLOW_SERIALIZE](model: FireworksImageModel) {\n    return serializeModelOptions({\n      modelId: model.modelId,\n      config: model.config,\n    });\n  }\n\n  static [WORKFLOW_DESERIALIZE](options: {\n    modelId: FireworksImageModelId;\n    config: FireworksImageModelConfig;\n  }) {\n    return new FireworksImageModel(options.modelId, options.config);\n  }\n\n  constructor(\n    readonly modelId: FireworksImageModelId,\n    private config: FireworksImageModelConfig,\n  ) {}\n\n  async doGenerate({\n    prompt,\n    n,\n    size,\n    aspectRatio,\n    seed,\n    providerOptions,\n    headers,\n    abortSignal,\n    files,\n    mask,\n  }: Parameters<ImageModelV4['doGenerate']>[0]): Promise<\n    Awaited<ReturnType<ImageModelV4['doGenerate']>>\n  > {\n    const warnings: Array<SharedV4Warning> = [];\n\n    const backendConfig = modelToBackendConfig[this.modelId];\n    if (!backendConfig?.supportsSize && size != null) {\n      warnings.push({\n        type: 'unsupported',\n        feature: 'size',\n        details:\n          'This model does not support the `size` option. Use `aspectRatio` instead.',\n      });\n    }\n\n    // Use supportsSize as a proxy for whether the model does not support\n    // aspectRatio. This invariant holds for the current set of models.\n    if (backendConfig?.supportsSize && aspectRatio != null) {\n      warnings.push({\n        type: 'unsupported',\n        feature: 'aspectRatio',\n        details: 'This model does not support the `aspectRatio` option.',\n      });\n    }\n\n    // Handle files for image editing\n    const hasInputImage = files != null && files.length > 0;\n    let inputImage: string | undefined;\n\n    if (hasInputImage) {\n      inputImage = convertImageModelFileToDataUri(files[0]);\n\n      if (files.length > 1) {\n        warnings.push({\n          type: 'other',\n          message:\n            'Fireworks only supports a single input image. Additional images are ignored.',\n        });\n      }\n    }\n\n    // Warn about mask - Fireworks Kontext models don't support explicit masks\n    if (mask != null) {\n      warnings.push({\n        type: 'unsupported',\n        feature: 'mask',\n        details:\n          'Fireworks Kontext models do not support explicit masks. Use the prompt to describe the areas to edit.',\n      });\n    }\n\n    const splitSize = size?.split('x');\n    const currentDate = this.config._internal?.currentDate?.() ?? new Date();\n    const combinedHeaders = combineHeaders(this.config.headers?.(), headers);\n\n    const body = {\n      prompt,\n      aspect_ratio: aspectRatio,\n      seed,\n      samples: n,\n      ...(inputImage && { input_image: inputImage }),\n      ...(splitSize && { width: splitSize[0], height: splitSize[1] }),\n      ...(providerOptions.fireworks ?? {}),\n    };\n\n    // Handle async models that require polling (e.g., flux-kontext-*)\n    if (backendConfig?.urlFormat === 'workflows_async') {\n      return this.doGenerateAsync({\n        body,\n        headers: combinedHeaders,\n        abortSignal,\n        warnings,\n        currentDate,\n      });\n    }\n\n    // Handle sync models that return binary directly\n    const { value: response, responseHeaders } = await postJsonToApi({\n      url: getUrlForModel(this.config.baseURL, this.modelId),\n      headers: combinedHeaders,\n      body,\n      failedResponseHandler: createStatusCodeErrorResponseHandler(),\n      successfulResponseHandler: createBinaryResponseHandler(),\n      abortSignal,\n      fetch: this.config.fetch,\n    });\n\n    return {\n      images: [response],\n      warnings,\n      response: {\n        timestamp: currentDate,\n        modelId: this.modelId,\n        headers: responseHeaders,\n      },\n    };\n  }\n\n  /**\n   * Handles async image generation for models like flux-kontext-* that return\n   * a request_id and require polling for results.\n   */\n  private async doGenerateAsync({\n    body,\n    headers,\n    abortSignal,\n    warnings,\n    currentDate,\n  }: {\n    body: Record<string, unknown>;\n    headers: Record<string, string | undefined>;\n    abortSignal: AbortSignal | undefined;\n    warnings: Array<SharedV4Warning>;\n    currentDate: Date;\n  }): Promise<Awaited<ReturnType<ImageModelV4['doGenerate']>>> {\n    // Submit the generation request\n    const { value: submitResponse } = await postJsonToApi({\n      url: getUrlForModel(this.config.baseURL, this.modelId),\n      headers,\n      body,\n      failedResponseHandler: createStatusCodeErrorResponseHandler(),\n      successfulResponseHandler: createJsonResponseHandler(\n        asyncSubmitResponseSchema,\n      ),\n      abortSignal,\n      fetch: this.config.fetch,\n    });\n\n    const requestId = submitResponse.request_id;\n\n    // Poll for the result\n    const imageUrl = await this.pollForImageUrl({\n      requestId,\n      headers,\n      abortSignal,\n    });\n\n    // Download the image from the URL. The URL comes from the provider\n    // response and typically points at a CDN, so only send credentials when it\n    // stays on the provider's own origin (never leak the API key to a CDN or an\n    // attacker-named host).\n    const { value: imageBytes, responseHeaders } = await getFromApi({\n      url: imageUrl,\n      headers: isSameOrigin(imageUrl, this.config.baseURL)\n        ? headers\n        : undefined,\n      abortSignal,\n      failedResponseHandler: createStatusCodeErrorResponseHandler(),\n      successfulResponseHandler: createBinaryResponseHandler(),\n      fetch: this.config.fetch,\n    });\n\n    return {\n      images: [imageBytes],\n      warnings,\n      response: {\n        timestamp: currentDate,\n        modelId: this.modelId,\n        headers: responseHeaders,\n      },\n    };\n  }\n\n  /**\n   * Polls the get_result endpoint until the image is ready.\n   */\n  private async pollForImageUrl({\n    requestId,\n    headers,\n    abortSignal,\n  }: {\n    requestId: string;\n    headers: Record<string, string | undefined>;\n    abortSignal: AbortSignal | undefined;\n  }): Promise<string> {\n    const pollIntervalMillis =\n      this.config.pollIntervalMillis ?? DEFAULT_POLL_INTERVAL_MILLIS;\n    const pollTimeoutMillis =\n      this.config.pollTimeoutMillis ?? DEFAULT_POLL_TIMEOUT_MILLIS;\n    const maxPollAttempts = Math.ceil(\n      pollTimeoutMillis / Math.max(1, pollIntervalMillis),\n    );\n\n    const pollUrl = getPollUrlForModel(this.config.baseURL, this.modelId);\n\n    for (let i = 0; i < maxPollAttempts; i++) {\n      const { value: pollResponse } = await postJsonToApi({\n        url: pollUrl,\n        headers,\n        body: { id: requestId },\n        failedResponseHandler: createStatusCodeErrorResponseHandler(),\n        successfulResponseHandler: createJsonResponseHandler(\n          asyncPollResponseSchema,\n        ),\n        abortSignal,\n        fetch: this.config.fetch,\n      });\n\n      const status = pollResponse.status;\n\n      if (status === 'Ready') {\n        const imageUrl = pollResponse.result?.sample;\n        if (typeof imageUrl === 'string') {\n          return imageUrl;\n        }\n        throw new Error(\n          'Fireworks poll response is Ready but missing result.sample',\n        );\n      }\n\n      if (status === 'Error' || status === 'Failed') {\n        throw new Error(\n          `Fireworks image generation failed with status: ${status}`,\n        );\n      }\n\n      // Wait before next poll attempt\n      await delay(pollIntervalMillis);\n    }\n\n    throw new Error(\n      `Fireworks image generation timed out after ${pollTimeoutMillis}ms`,\n    );\n  }\n}\n","import { lazySchema, zodSchema } from '@ai-sdk/provider-utils';\nimport { z } from 'zod/v4';\n\n// Schema for async submit response\nexport const asyncSubmitResponseSchema = lazySchema(() =>\n  zodSchema(\n    z.object({\n      request_id: z.string(),\n    }),\n  ),\n);\n\n// Schema for async poll response\nexport const asyncPollResponseSchema = lazySchema(() =>\n  zodSchema(\n    z.object({\n      id: z.string(),\n      status: z.string(),\n      result: z\n        .object({\n          sample: z.string().optional(),\n        })\n        .nullable(),\n    }),\n  ),\n);\n","import {\n  OpenAICompatibleChatLanguageModel,\n  OpenAICompatibleCompletionLanguageModel,\n  OpenAICompatibleEmbeddingModel,\n  type ProviderErrorStructure,\n} from '@ai-sdk/openai-compatible';\nimport type {\n  EmbeddingModelV4,\n  ImageModelV4,\n  LanguageModelV4,\n  ProviderV4,\n} from '@ai-sdk/provider';\nimport {\n  loadApiKey,\n  withoutTrailingSlash,\n  withUserAgentSuffix,\n  type FetchFunction,\n} from '@ai-sdk/provider-utils';\nimport { z } from 'zod/v4';\nimport type { FireworksChatModelId } from './fireworks-chat-options';\nimport type { FireworksCompletionModelId } from './fireworks-completion-options';\nimport type { FireworksEmbeddingModelId } from './fireworks-embedding-options';\nimport { FireworksImageModel } from './fireworks-image-model';\nimport type { FireworksImageModelId } from './fireworks-image-options';\nimport { VERSION } from './version';\n\nexport type FireworksErrorData = z.infer<typeof fireworksErrorSchema>;\n\nconst fireworksErrorSchema = z.object({\n  error: z.string(),\n});\n\nconst fireworksErrorStructure: ProviderErrorStructure<FireworksErrorData> = {\n  errorSchema: fireworksErrorSchema,\n  errorToMessage: data => data.error,\n};\n\nexport interface FireworksProviderSettings {\n  /**\n   * Fireworks API key. Default value is taken from the `FIREWORKS_API_KEY`\n   * environment variable.\n   */\n  apiKey?: string;\n  /**\n   * Base URL for the API calls.\n   */\n  baseURL?: string;\n  /**\n   * Custom headers to include in the requests.\n   */\n  headers?: Record<string, string>;\n  /**\n   * Custom fetch implementation. You can use it as a middleware to intercept requests,\n   * or to provide a custom fetch implementation for e.g. testing.\n   */\n  fetch?: FetchFunction;\n}\n\nexport interface FireworksProvider extends ProviderV4 {\n  /**\n   * Creates a model for text generation.\n   */\n  (modelId: FireworksChatModelId): LanguageModelV4;\n\n  /**\n   * Creates a chat model for text generation.\n   */\n  chatModel(modelId: FireworksChatModelId): LanguageModelV4;\n\n  /**\n   * Creates a completion model for text generation.\n   */\n  completionModel(modelId: FireworksCompletionModelId): LanguageModelV4;\n\n  /**\n   * Creates a chat model for text generation.\n   */\n  languageModel(modelId: FireworksChatModelId): LanguageModelV4;\n\n  /**\n   * Creates a text embedding model for text generation.\n   */\n  embeddingModel(modelId: FireworksEmbeddingModelId): EmbeddingModelV4;\n\n  /**\n   * @deprecated Use `embeddingModel` instead.\n   */\n  textEmbeddingModel(modelId: FireworksEmbeddingModelId): EmbeddingModelV4;\n\n  /**\n   * Creates a model for image generation.\n   */\n  image(modelId: FireworksImageModelId): ImageModelV4;\n\n  /**\n   * Creates a model for image generation.\n   */\n  imageModel(modelId: FireworksImageModelId): ImageModelV4;\n}\n\nconst defaultBaseURL = 'https://api.fireworks.ai/inference/v1';\n\nexport function createFireworks(\n  options: FireworksProviderSettings = {},\n): FireworksProvider {\n  const baseURL = withoutTrailingSlash(options.baseURL ?? defaultBaseURL);\n  const getHeaders = () =>\n    withUserAgentSuffix(\n      {\n        Authorization: `Bearer ${loadApiKey({\n          apiKey: options.apiKey,\n          environmentVariableName: 'FIREWORKS_API_KEY',\n          description: 'Fireworks API key',\n        })}`,\n        ...options.headers,\n      },\n      `ai-sdk/fireworks/${VERSION}`,\n    );\n\n  interface CommonModelConfig {\n    provider: string;\n    url: ({ path }: { path: string }) => string;\n    headers: () => Record<string, string>;\n    fetch?: FetchFunction;\n  }\n\n  const getCommonModelConfig = (modelType: string): CommonModelConfig => ({\n    provider: `fireworks.${modelType}`,\n    url: ({ path }) => `${baseURL}${path}`,\n    headers: getHeaders,\n    fetch: options.fetch,\n  });\n\n  const createChatModel = (modelId: FireworksChatModelId) => {\n    return new OpenAICompatibleChatLanguageModel(modelId, {\n      ...getCommonModelConfig('chat'),\n      includeUsage: true,\n      errorStructure: fireworksErrorStructure,\n      transformRequestBody: args => {\n        const thinking = args.thinking as\n          | { type?: string; budgetTokens?: number }\n          | undefined;\n        const reasoningHistory = args.reasoningHistory as string | undefined;\n        const promptCacheKey = args.promptCacheKey as string | undefined;\n\n        const {\n          thinking: _,\n          reasoningHistory: __,\n          promptCacheKey: ___,\n          reasoning_effort,\n          ...rest\n        } = args;\n\n        return {\n          ...rest,\n          ...(reasoning_effort != null && {\n            // Workaround since OpenAI spec allows for 5 reasoning levels, but Fireworks only supports 3 of them.\n            reasoning_effort:\n              reasoning_effort === 'minimal'\n                ? 'low'\n                : reasoning_effort === 'xhigh'\n                  ? 'high'\n                  : reasoning_effort,\n          }),\n          ...(promptCacheKey !== undefined && {\n            prompt_cache_key: promptCacheKey,\n          }),\n          ...(thinking && {\n            thinking: {\n              type: thinking.type,\n              ...(thinking.budgetTokens !== undefined && {\n                budget_tokens: thinking.budgetTokens,\n              }),\n            },\n          }),\n          ...(reasoningHistory && {\n            reasoning_history: reasoningHistory,\n          }),\n        };\n      },\n    });\n  };\n\n  const createCompletionModel = (modelId: FireworksCompletionModelId) =>\n    new OpenAICompatibleCompletionLanguageModel(modelId, {\n      ...getCommonModelConfig('completion'),\n      includeUsage: true,\n      errorStructure: fireworksErrorStructure,\n    });\n\n  const createEmbeddingModel = (modelId: FireworksEmbeddingModelId) =>\n    new OpenAICompatibleEmbeddingModel(modelId, {\n      ...getCommonModelConfig('embedding'),\n      errorStructure: fireworksErrorStructure,\n    });\n\n  const createImageModel = (modelId: FireworksImageModelId) =>\n    new FireworksImageModel(modelId, {\n      ...getCommonModelConfig('image'),\n      baseURL: baseURL ?? defaultBaseURL,\n    });\n\n  const provider = (modelId: FireworksChatModelId) => createChatModel(modelId);\n\n  provider.specificationVersion = 'v4' as const;\n  provider.completionModel = createCompletionModel;\n  provider.chatModel = createChatModel;\n  provider.languageModel = createChatModel;\n  provider.embeddingModel = createEmbeddingModel;\n  provider.textEmbeddingModel = createEmbeddingModel;\n  provider.image = createImageModel;\n  provider.imageModel = createImageModel;\n  return provider;\n}\n\nexport const fireworks = createFireworks();\n","// Version string of this package injected at build time.\ndeclare const __PACKAGE_VERSION__: string | undefined;\nexport const VERSION: string =\n  typeof __PACKAGE_VERSION__ !== 'undefined'\n    ? __PACKAGE_VERSION__\n    : '0.0.0-test';\n"],"mappings":";AACA;AAAA,EACE;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,OAEK;;;ACfP,SAAS,YAAY,iBAAiB;AACtC,SAAS,SAAS;AAGX,IAAM,4BAA4B;AAAA,EAAW,MAClD;AAAA,IACE,EAAE,OAAO;AAAA,MACP,YAAY,EAAE,OAAO;AAAA,IACvB,CAAC;AAAA,EACH;AACF;AAGO,IAAM,0BAA0B;AAAA,EAAW,MAChD;AAAA,IACE,EAAE,OAAO;AAAA,MACP,IAAI,EAAE,OAAO;AAAA,MACb,QAAQ,EAAE,OAAO;AAAA,MACjB,QAAQ,EACL,OAAO;AAAA,QACN,QAAQ,EAAE,OAAO,EAAE,SAAS;AAAA,MAC9B,CAAC,EACA,SAAS;AAAA,IACd,CAAC;AAAA,EACH;AACF;;;ADHA,IAAM,+BAA+B;AACrC,IAAM,8BAA8B;AAQpC,IAAM,uBAEF;AAAA,EACF,4CAA4C;AAAA,IAC1C,WAAW;AAAA,EACb;AAAA,EACA,gDAAgD;AAAA,IAC9C,WAAW;AAAA,EACb;AAAA,EACA,8CAA8C;AAAA,IAC5C,WAAW;AAAA,IACX,iBAAiB;AAAA,EACnB;AAAA,EACA,8CAA8C;AAAA,IAC5C,WAAW;AAAA,IACX,iBAAiB;AAAA,EACnB;AAAA,EACA,8DAA8D;AAAA,IAC5D,WAAW;AAAA,IACX,cAAc;AAAA,EAChB;AAAA,EACA,0DAA0D;AAAA,IACxD,WAAW;AAAA,IACX,cAAc;AAAA,EAChB;AAAA,EACA,4DAA4D;AAAA,IAC1D,WAAW;AAAA,IACX,cAAc;AAAA,EAChB;AAAA,EACA,2DAA2D;AAAA,IACzD,WAAW;AAAA,IACX,cAAc;AAAA,EAChB;AAAA,EACA,oCAAoC;AAAA,IAClC,WAAW;AAAA,IACX,cAAc;AAAA,EAChB;AACF;AAEA,SAAS,eACP,SACA,SACQ;AACR,QAAM,SAAS,qBAAqB,OAAO;AAE3C,UAAQ,iCAAQ,WAAW;AAAA,IACzB,KAAK;AACH,aAAO,GAAG,OAAO,qBAAqB,OAAO;AAAA,IAC/C,KAAK;AACH,aAAO,GAAG,OAAO,cAAc,OAAO;AAAA,IACxC,KAAK;AAAA,IACL;AACE,aAAO,GAAG,OAAO,cAAc,OAAO;AAAA,EAC1C;AACF;AAEA,SAAS,mBACP,SACA,SACQ;AACR,SAAO,GAAG,OAAO,cAAc,OAAO;AACxC;AAsBO,IAAM,sBAAN,MAAM,qBAA4C;AAAA,EAsBvD,YACW,SACD,QACR;AAFS;AACD;AAvBV,SAAS,uBAAuB;AAChC,SAAS,mBAAmB;AAAA,EAuBzB;AAAA,EArBH,IAAI,WAAmB;AACrB,WAAO,KAAK,OAAO;AAAA,EACrB;AAAA,EAEA,QAAQ,kBAAkB,EAAE,OAA4B;AACtD,WAAO,sBAAsB;AAAA,MAC3B,SAAS,MAAM;AAAA,MACf,QAAQ,MAAM;AAAA,IAChB,CAAC;AAAA,EACH;AAAA,EAEA,QAAQ,oBAAoB,EAAE,SAG3B;AACD,WAAO,IAAI,qBAAoB,QAAQ,SAAS,QAAQ,MAAM;AAAA,EAChE;AAAA,EAOA,MAAM,WAAW;AAAA,IACf;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,EACF,GAEE;AA1JJ;AA2JI,UAAM,WAAmC,CAAC;AAE1C,UAAM,gBAAgB,qBAAqB,KAAK,OAAO;AACvD,QAAI,EAAC,+CAAe,iBAAgB,QAAQ,MAAM;AAChD,eAAS,KAAK;AAAA,QACZ,MAAM;AAAA,QACN,SAAS;AAAA,QACT,SACE;AAAA,MACJ,CAAC;AAAA,IACH;AAIA,SAAI,+CAAe,iBAAgB,eAAe,MAAM;AACtD,eAAS,KAAK;AAAA,QACZ,MAAM;AAAA,QACN,SAAS;AAAA,QACT,SAAS;AAAA,MACX,CAAC;AAAA,IACH;AAGA,UAAM,gBAAgB,SAAS,QAAQ,MAAM,SAAS;AACtD,QAAI;AAEJ,QAAI,eAAe;AACjB,mBAAa,+BAA+B,MAAM,CAAC,CAAC;AAEpD,UAAI,MAAM,SAAS,GAAG;AACpB,iBAAS,KAAK;AAAA,UACZ,MAAM;AAAA,UACN,SACE;AAAA,QACJ,CAAC;AAAA,MACH;AAAA,IACF;AAGA,QAAI,QAAQ,MAAM;AAChB,eAAS,KAAK;AAAA,QACZ,MAAM;AAAA,QACN,SAAS;AAAA,QACT,SACE;AAAA,MACJ,CAAC;AAAA,IACH;AAEA,UAAM,YAAY,6BAAM,MAAM;AAC9B,UAAM,eAAc,sBAAK,OAAO,cAAZ,mBAAuB,gBAAvB,4CAA0C,oBAAI,KAAK;AACvE,UAAM,kBAAkB,gBAAe,gBAAK,QAAO,YAAZ,6BAAyB,OAAO;AAEvE,UAAM,OAAO;AAAA,MACX;AAAA,MACA,cAAc;AAAA,MACd;AAAA,MACA,SAAS;AAAA,MACT,GAAI,cAAc,EAAE,aAAa,WAAW;AAAA,MAC5C,GAAI,aAAa,EAAE,OAAO,UAAU,CAAC,GAAG,QAAQ,UAAU,CAAC,EAAE;AAAA,MAC7D,IAAI,qBAAgB,cAAhB,YAA6B,CAAC;AAAA,IACpC;AAGA,SAAI,+CAAe,eAAc,mBAAmB;AAClD,aAAO,KAAK,gBAAgB;AAAA,QAC1B;AAAA,QACA,SAAS;AAAA,QACT;AAAA,QACA;AAAA,QACA;AAAA,MACF,CAAC;AAAA,IACH;AAGA,UAAM,EAAE,OAAO,UAAU,gBAAgB,IAAI,MAAM,cAAc;AAAA,MAC/D,KAAK,eAAe,KAAK,OAAO,SAAS,KAAK,OAAO;AAAA,MACrD,SAAS;AAAA,MACT;AAAA,MACA,uBAAuB,qCAAqC;AAAA,MAC5D,2BAA2B,4BAA4B;AAAA,MACvD;AAAA,MACA,OAAO,KAAK,OAAO;AAAA,IACrB,CAAC;AAED,WAAO;AAAA,MACL,QAAQ,CAAC,QAAQ;AAAA,MACjB;AAAA,MACA,UAAU;AAAA,QACR,WAAW;AAAA,QACX,SAAS,KAAK;AAAA,QACd,SAAS;AAAA,MACX;AAAA,IACF;AAAA,EACF;AAAA;AAAA;AAAA;AAAA;AAAA,EAMA,MAAc,gBAAgB;AAAA,IAC5B;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,EACF,GAM6D;AAE3D,UAAM,EAAE,OAAO,eAAe,IAAI,MAAM,cAAc;AAAA,MACpD,KAAK,eAAe,KAAK,OAAO,SAAS,KAAK,OAAO;AAAA,MACrD;AAAA,MACA;AAAA,MACA,uBAAuB,qCAAqC;AAAA,MAC5D,2BAA2B;AAAA,QACzB;AAAA,MACF;AAAA,MACA;AAAA,MACA,OAAO,KAAK,OAAO;AAAA,IACrB,CAAC;AAED,UAAM,YAAY,eAAe;AAGjC,UAAM,WAAW,MAAM,KAAK,gBAAgB;AAAA,MAC1C;AAAA,MACA;AAAA,MACA;AAAA,IACF,CAAC;AAMD,UAAM,EAAE,OAAO,YAAY,gBAAgB,IAAI,MAAM,WAAW;AAAA,MAC9D,KAAK;AAAA,MACL,SAAS,aAAa,UAAU,KAAK,OAAO,OAAO,IAC/C,UACA;AAAA,MACJ;AAAA,MACA,uBAAuB,qCAAqC;AAAA,MAC5D,2BAA2B,4BAA4B;AAAA,MACvD,OAAO,KAAK,OAAO;AAAA,IACrB,CAAC;AAED,WAAO;AAAA,MACL,QAAQ,CAAC,UAAU;AAAA,MACnB;AAAA,MACA,UAAU;AAAA,QACR,WAAW;AAAA,QACX,SAAS,KAAK;AAAA,QACd,SAAS;AAAA,MACX;AAAA,IACF;AAAA,EACF;AAAA;AAAA;AAAA;AAAA,EAKA,MAAc,gBAAgB;AAAA,IAC5B;AAAA,IACA;AAAA,IACA;AAAA,EACF,GAIoB;AAtUtB;AAuUI,UAAM,sBACJ,UAAK,OAAO,uBAAZ,YAAkC;AACpC,UAAM,qBACJ,UAAK,OAAO,sBAAZ,YAAiC;AACnC,UAAM,kBAAkB,KAAK;AAAA,MAC3B,oBAAoB,KAAK,IAAI,GAAG,kBAAkB;AAAA,IACpD;AAEA,UAAM,UAAU,mBAAmB,KAAK,OAAO,SAAS,KAAK,OAAO;AAEpE,aAAS,IAAI,GAAG,IAAI,iBAAiB,KAAK;AACxC,YAAM,EAAE,OAAO,aAAa,IAAI,MAAM,cAAc;AAAA,QAClD,KAAK;AAAA,QACL;AAAA,QACA,MAAM,EAAE,IAAI,UAAU;AAAA,QACtB,uBAAuB,qCAAqC;AAAA,QAC5D,2BAA2B;AAAA,UACzB;AAAA,QACF;AAAA,QACA;AAAA,QACA,OAAO,KAAK,OAAO;AAAA,MACrB,CAAC;AAED,YAAM,SAAS,aAAa;AAE5B,UAAI,WAAW,SAAS;AACtB,cAAM,YAAW,kBAAa,WAAb,mBAAqB;AACtC,YAAI,OAAO,aAAa,UAAU;AAChC,iBAAO;AAAA,QACT;AACA,cAAM,IAAI;AAAA,UACR;AAAA,QACF;AAAA,MACF;AAEA,UAAI,WAAW,WAAW,WAAW,UAAU;AAC7C,cAAM,IAAI;AAAA,UACR,kDAAkD,MAAM;AAAA,QAC1D;AAAA,MACF;AAGA,YAAM,MAAM,kBAAkB;AAAA,IAChC;AAEA,UAAM,IAAI;AAAA,MACR,8CAA8C,iBAAiB;AAAA,IACjE;AAAA,EACF;AACF;;;AExXA;AAAA,EACE;AAAA,EACA;AAAA,EACA;AAAA,OAEK;AAOP;AAAA,EACE;AAAA,EACA;AAAA,EACA;AAAA,OAEK;AACP,SAAS,KAAAA,UAAS;;;AChBX,IAAM,UACX,OACI,kBACA;;;ADuBN,IAAM,uBAAuBC,GAAE,OAAO;AAAA,EACpC,OAAOA,GAAE,OAAO;AAClB,CAAC;AAED,IAAM,0BAAsE;AAAA,EAC1E,aAAa;AAAA,EACb,gBAAgB,UAAQ,KAAK;AAC/B;AAiEA,IAAM,iBAAiB;AAEhB,SAAS,gBACd,UAAqC,CAAC,GACnB;AAxGrB;AAyGE,QAAM,UAAU,sBAAqB,aAAQ,YAAR,YAAmB,cAAc;AACtE,QAAM,aAAa,MACjB;AAAA,IACE;AAAA,MACE,eAAe,UAAU,WAAW;AAAA,QAClC,QAAQ,QAAQ;AAAA,QAChB,yBAAyB;AAAA,QACzB,aAAa;AAAA,MACf,CAAC,CAAC;AAAA,MACF,GAAG,QAAQ;AAAA,IACb;AAAA,IACA,oBAAoB,OAAO;AAAA,EAC7B;AASF,QAAM,uBAAuB,CAAC,eAA0C;AAAA,IACtE,UAAU,aAAa,SAAS;AAAA,IAChC,KAAK,CAAC,EAAE,KAAK,MAAM,GAAG,OAAO,GAAG,IAAI;AAAA,IACpC,SAAS;AAAA,IACT,OAAO,QAAQ;AAAA,EACjB;AAEA,QAAM,kBAAkB,CAAC,YAAkC;AACzD,WAAO,IAAI,kCAAkC,SAAS;AAAA,MACpD,GAAG,qBAAqB,MAAM;AAAA,MAC9B,cAAc;AAAA,MACd,gBAAgB;AAAA,MAChB,sBAAsB,UAAQ;AAC5B,cAAM,WAAW,KAAK;AAGtB,cAAM,mBAAmB,KAAK;AAC9B,cAAM,iBAAiB,KAAK;AAE5B,cAAM;AAAA,UACJ,UAAU;AAAA,UACV,kBAAkB;AAAA,UAClB,gBAAgB;AAAA,UAChB;AAAA,UACA,GAAG;AAAA,QACL,IAAI;AAEJ,eAAO;AAAA,UACL,GAAG;AAAA,UACH,GAAI,oBAAoB,QAAQ;AAAA;AAAA,YAE9B,kBACE,qBAAqB,YACjB,QACA,qBAAqB,UACnB,SACA;AAAA,UACV;AAAA,UACA,GAAI,mBAAmB,UAAa;AAAA,YAClC,kBAAkB;AAAA,UACpB;AAAA,UACA,GAAI,YAAY;AAAA,YACd,UAAU;AAAA,cACR,MAAM,SAAS;AAAA,cACf,GAAI,SAAS,iBAAiB,UAAa;AAAA,gBACzC,eAAe,SAAS;AAAA,cAC1B;AAAA,YACF;AAAA,UACF;AAAA,UACA,GAAI,oBAAoB;AAAA,YACtB,mBAAmB;AAAA,UACrB;AAAA,QACF;AAAA,MACF;AAAA,IACF,CAAC;AAAA,EACH;AAEA,QAAM,wBAAwB,CAAC,YAC7B,IAAI,wCAAwC,SAAS;AAAA,IACnD,GAAG,qBAAqB,YAAY;AAAA,IACpC,cAAc;AAAA,IACd,gBAAgB;AAAA,EAClB,CAAC;AAEH,QAAM,uBAAuB,CAAC,YAC5B,IAAI,+BAA+B,SAAS;AAAA,IAC1C,GAAG,qBAAqB,WAAW;AAAA,IACnC,gBAAgB;AAAA,EAClB,CAAC;AAEH,QAAM,mBAAmB,CAAC,YACxB,IAAI,oBAAoB,SAAS;AAAA,IAC/B,GAAG,qBAAqB,OAAO;AAAA,IAC/B,SAAS,4BAAW;AAAA,EACtB,CAAC;AAEH,QAAM,WAAW,CAAC,YAAkC,gBAAgB,OAAO;AAE3E,WAAS,uBAAuB;AAChC,WAAS,kBAAkB;AAC3B,WAAS,YAAY;AACrB,WAAS,gBAAgB;AACzB,WAAS,iBAAiB;AAC1B,WAAS,qBAAqB;AAC9B,WAAS,QAAQ;AACjB,WAAS,aAAa;AACtB,SAAO;AACT;AAEO,IAAM,YAAY,gBAAgB;","names":["z","z"]}

package/docs/26-fireworks.mdx

@@ -122,7 +122,7 @@ import {
 import { generateText } from 'ai';
 
 const { text, reasoningText } = await generateText({
-  model: fireworks('accounts/fireworks/models/kimi-k2p5'),
+  model: fireworks('accounts/fireworks/models/kimi-k2p6'),
   providerOptions: {
     fireworks: {
       thinking: { type: 'enabled', budgetTokens: 4096 },
@@ -135,10 +135,16 @@ const { text, reasoningText } = await generateText({
 
 The following optional provider options are available for Fireworks chat models:
 
-- **thinking** _object_
+- **promptCacheKey** _string_
+
+  A stable, opaque key that improves prompt cache hit rates by routing requests
+  with shared prompt prefixes to the same Fireworks replica. Reuse the same key
+  for all steps and calls in one conversation, and use different keys for
+  unrelated conversations.
 
-  Configuration for thinking/reasoning models like Kimi K2.5.
+- **thinking** _object_
 
+  Configuration for thinking/reasoning models like Kimi K2.6.
   - **type** _'enabled' | 'disabled'_
 
     Whether to enable thinking mode.
@@ -150,11 +156,94 @@ The following optional provider options are available for Fireworks chat models:
 - **reasoningHistory** _'disabled' | 'interleaved' | 'preserved'_
 
   Controls how reasoning history is handled in multi-turn conversations:
-
   - `'disabled'`: Remove reasoning from history
   - `'interleaved'`: Include reasoning between tool calls within a single turn
   - `'preserved'`: Keep all reasoning in history
 
+### Prompt Cache Affinity
+
+[Fireworks prompt caching](https://docs.fireworks.ai/guides/prompt-caching)
+is automatic, but cached prefixes are local to a replica. Set
+`promptCacheKey` to an opaque session or conversation identifier to improve
+cache affinity. The provider sends it as Fireworks'
+[`prompt_cache_key`](https://docs.fireworks.ai/api-reference/post-chatcompletions)
+request field.
+
+The AI SDK reuses the same provider options for every model step in a
+multi-step `generateText` or `streamText` call, so one key covers the complete
+tool loop:
+
+```ts
+import {
+  fireworks,
+  type FireworksLanguageModelOptions,
+} from '@ai-sdk/fireworks';
+import { generateText, isStepCount, tool } from 'ai';
+import { z } from 'zod';
+
+const sessionId = 'conversation-123';
+
+const result = await generateText({
+  model: fireworks('accounts/fireworks/models/kimi-k2p6'),
+  providerOptions: {
+    fireworks: {
+      promptCacheKey: sessionId,
+    } satisfies FireworksLanguageModelOptions,
+  },
+  tools: {
+    weather: tool({
+      description: 'Get the weather for a city.',
+      inputSchema: z.object({ city: z.string() }),
+      execute: async ({ city }) => `It is sunny in ${city}.`,
+    }),
+  },
+  stopWhen: isStepCount(5),
+  prompt: 'What is the weather in San Francisco?',
+});
+
+console.log(result.usage.inputTokenDetails.cacheReadTokens);
+```
+
+For a reusable `ToolLoopAgent`, use call options to supply a key for each
+conversation. Pass the same key again for later agent calls that continue that
+conversation:
+
+```ts
+import {
+  fireworks,
+  type FireworksLanguageModelOptions,
+} from '@ai-sdk/fireworks';
+import { ToolLoopAgent } from 'ai';
+import { z } from 'zod';
+import { weatherTool } from './weather-tool';
+
+const agent = new ToolLoopAgent({
+  model: fireworks('accounts/fireworks/models/kimi-k2p6'),
+  callOptionsSchema: z.object({
+    sessionId: z.string(),
+  }),
+  prepareCall: ({ options, ...settings }) => ({
+    ...settings,
+    providerOptions: {
+      fireworks: {
+        promptCacheKey: options.sessionId,
+      } satisfies FireworksLanguageModelOptions,
+    },
+  }),
+  tools: { weather: weatherTool },
+});
+
+const result = await agent.generate({
+  prompt: 'What is the weather in San Francisco?',
+  options: {
+    sessionId: 'conversation-123',
+  },
+});
+```
+
+Use non-identifying values for cache keys rather than email addresses or other
+personal information.
+
 ### Completion Models
 
 You can create models that call the Fireworks completions API using the `.completionModel()` factory method:
@@ -188,7 +277,7 @@ const model = fireworks.completionModel(
 | `accounts/fireworks/models/yi-large`                       | <Cross size={18} /> | <Check size={18} /> | <Check size={18} /> | <Cross size={18} /> |
 | `accounts/fireworks/models/kimi-k2-instruct`               | <Cross size={18} /> | <Check size={18} /> | <Check size={18} /> | <Cross size={18} /> |
 | `accounts/fireworks/models/kimi-k2-thinking`               | <Cross size={18} /> | <Check size={18} /> | <Check size={18} /> | <Cross size={18} /> |
-| `accounts/fireworks/models/kimi-k2p5`                      | <Check size={18} /> | <Check size={18} /> | <Check size={18} /> | <Cross size={18} /> |
+| `accounts/fireworks/models/kimi-k2p6`                      | <Check size={18} /> | <Check size={18} /> | <Check size={18} /> | <Cross size={18} /> |
 | `accounts/fireworks/models/minimax-m2`                     | <Cross size={18} /> | <Check size={18} /> | <Check size={18} /> | <Cross size={18} /> |
 
 <Note>

package/package.json

@@ -1,10 +1,10 @@
 {
   "name": "@ai-sdk/fireworks",
-  "version": "3.0.0-beta.6",
+  "version": "3.0.0-beta.60",
+  "type": "module",
   "license": "Apache-2.0",
   "sideEffects": false,
   "main": "./dist/index.js",
-  "module": "./dist/index.mjs",
   "types": "./dist/index.d.ts",
   "files": [
     "dist/**/*",
@@ -24,36 +24,38 @@
     "./package.json": "./package.json",
     ".": {
       "types": "./dist/index.d.ts",
-      "import": "./dist/index.mjs",
-      "require": "./dist/index.js"
+      "import": "./dist/index.js",
+      "default": "./dist/index.js"
     }
   },
   "dependencies": {
-    "@ai-sdk/openai-compatible": "3.0.0-beta.6",
-    "@ai-sdk/provider": "4.0.0-beta.2",
-    "@ai-sdk/provider-utils": "5.0.0-beta.4"
+    "@ai-sdk/openai-compatible": "3.0.0-beta.58",
+    "@ai-sdk/provider": "4.0.0-beta.20",
+    "@ai-sdk/provider-utils": "5.0.0-beta.50"
   },
   "devDependencies": {
-    "@types/node": "20.17.24",
-    "tsup": "^8",
+    "@types/node": "22.19.19",
+    "tsup": "^8.5.1",
     "typescript": "5.8.3",
     "zod": "3.25.76",
-    "@ai-sdk/test-server": "2.0.0-beta.0",
+    "@ai-sdk/test-server": "2.0.0-beta.7",
     "@vercel/ai-tsconfig": "0.0.0"
   },
   "peerDependencies": {
     "zod": "^3.25.76 || ^4.1.8"
   },
   "engines": {
-    "node": ">=18"
+    "node": ">=22"
   },
   "publishConfig": {
-    "access": "public"
+    "access": "public",
+    "provenance": true
   },
   "homepage": "https://ai-sdk.dev/docs",
   "repository": {
     "type": "git",
-    "url": "git+https://github.com/vercel/ai.git"
+    "url": "https://github.com/vercel/ai",
+    "directory": "packages/fireworks"
   },
   "bugs": {
     "url": "https://github.com/vercel/ai/issues"
@@ -65,9 +67,7 @@
     "build": "pnpm clean && tsup --tsconfig tsconfig.build.json",
     "build:watch": "pnpm clean && tsup --watch",
     "clean": "del-cli dist docs *.tsbuildinfo",
-    "lint": "eslint \"./**/*.ts*\"",
     "type-check": "tsc --build",
-    "prettier-check": "prettier --check \"./**/*.ts*\"",
     "test": "pnpm test:node && pnpm test:edge",
     "test:update": "pnpm test:node -u",
     "test:watch": "vitest --config vitest.node.config.js",