@ai-sdk/fireworks 2.0.57 → 2.0.58

package/CHANGELOG.md

@@ -1,5 +1,13 @@
 # @ai-sdk/fireworks
 
+## 2.0.58
+
+### Patch Changes
+
+- 647f4b3: Add the `promptCacheKey` provider option for Fireworks prompt cache affinity
+  and Kimi K2.6 model autocomplete, and remove the deprecated Kimi K2.5
+  serverless model from autocomplete.
+
 ## 2.0.57
 
 ### Patch Changes

package/dist/index.d.mts

@@ -2,8 +2,9 @@ import { z } from 'zod/v4';
 import { ImageModelV3, ProviderV3, LanguageModelV3, EmbeddingModelV3 } from '@ai-sdk/provider';
 import { FetchFunction } from '@ai-sdk/provider-utils';
 
-type FireworksChatModelId = 'accounts/fireworks/models/deepseek-v3' | 'accounts/fireworks/models/llama-v3p3-70b-instruct' | 'accounts/fireworks/models/llama-v3p2-3b-instruct' | 'accounts/fireworks/models/llama-v3p1-405b-instruct' | 'accounts/fireworks/models/llama-v3p1-8b-instruct' | 'accounts/fireworks/models/mixtral-8x7b-instruct' | 'accounts/fireworks/models/mixtral-8x22b-instruct' | 'accounts/fireworks/models/mixtral-8x7b-instruct-hf' | 'accounts/fireworks/models/qwen2p5-coder-32b-instruct' | 'accounts/fireworks/models/qwen2p5-72b-instruct' | 'accounts/fireworks/models/qwen-qwq-32b-preview' | 'accounts/fireworks/models/qwen2-vl-72b-instruct' | 'accounts/fireworks/models/llama-v3p2-11b-vision-instruct' | 'accounts/fireworks/models/qwq-32b' | 'accounts/fireworks/models/yi-large' | 'accounts/fireworks/models/kimi-k2-instruct' | 'accounts/fireworks/models/kimi-k2-thinking' | 'accounts/fireworks/models/kimi-k2p5' | 'accounts/fireworks/models/minimax-m2' | (string & {});
+type FireworksChatModelId = 'accounts/fireworks/models/deepseek-v3' | 'accounts/fireworks/models/llama-v3p3-70b-instruct' | 'accounts/fireworks/models/llama-v3p2-3b-instruct' | 'accounts/fireworks/models/llama-v3p1-405b-instruct' | 'accounts/fireworks/models/llama-v3p1-8b-instruct' | 'accounts/fireworks/models/mixtral-8x7b-instruct' | 'accounts/fireworks/models/mixtral-8x22b-instruct' | 'accounts/fireworks/models/mixtral-8x7b-instruct-hf' | 'accounts/fireworks/models/qwen2p5-coder-32b-instruct' | 'accounts/fireworks/models/qwen2p5-72b-instruct' | 'accounts/fireworks/models/qwen-qwq-32b-preview' | 'accounts/fireworks/models/qwen2-vl-72b-instruct' | 'accounts/fireworks/models/llama-v3p2-11b-vision-instruct' | 'accounts/fireworks/models/qwq-32b' | 'accounts/fireworks/models/yi-large' | 'accounts/fireworks/models/kimi-k2-instruct' | 'accounts/fireworks/models/kimi-k2-thinking' | 'accounts/fireworks/models/kimi-k2p6' | 'accounts/fireworks/models/minimax-m2' | (string & {});
 declare const fireworksLanguageModelOptions: z.ZodObject<{
+    promptCacheKey: z.ZodOptional<z.ZodString>;
     thinking: z.ZodOptional<z.ZodObject<{
         type: z.ZodOptional<z.ZodEnum<{
             enabled: "enabled";

package/dist/index.d.ts

@@ -2,8 +2,9 @@ import { z } from 'zod/v4';
 import { ImageModelV3, ProviderV3, LanguageModelV3, EmbeddingModelV3 } from '@ai-sdk/provider';
 import { FetchFunction } from '@ai-sdk/provider-utils';
 
-type FireworksChatModelId = 'accounts/fireworks/models/deepseek-v3' | 'accounts/fireworks/models/llama-v3p3-70b-instruct' | 'accounts/fireworks/models/llama-v3p2-3b-instruct' | 'accounts/fireworks/models/llama-v3p1-405b-instruct' | 'accounts/fireworks/models/llama-v3p1-8b-instruct' | 'accounts/fireworks/models/mixtral-8x7b-instruct' | 'accounts/fireworks/models/mixtral-8x22b-instruct' | 'accounts/fireworks/models/mixtral-8x7b-instruct-hf' | 'accounts/fireworks/models/qwen2p5-coder-32b-instruct' | 'accounts/fireworks/models/qwen2p5-72b-instruct' | 'accounts/fireworks/models/qwen-qwq-32b-preview' | 'accounts/fireworks/models/qwen2-vl-72b-instruct' | 'accounts/fireworks/models/llama-v3p2-11b-vision-instruct' | 'accounts/fireworks/models/qwq-32b' | 'accounts/fireworks/models/yi-large' | 'accounts/fireworks/models/kimi-k2-instruct' | 'accounts/fireworks/models/kimi-k2-thinking' | 'accounts/fireworks/models/kimi-k2p5' | 'accounts/fireworks/models/minimax-m2' | (string & {});
+type FireworksChatModelId = 'accounts/fireworks/models/deepseek-v3' | 'accounts/fireworks/models/llama-v3p3-70b-instruct' | 'accounts/fireworks/models/llama-v3p2-3b-instruct' | 'accounts/fireworks/models/llama-v3p1-405b-instruct' | 'accounts/fireworks/models/llama-v3p1-8b-instruct' | 'accounts/fireworks/models/mixtral-8x7b-instruct' | 'accounts/fireworks/models/mixtral-8x22b-instruct' | 'accounts/fireworks/models/mixtral-8x7b-instruct-hf' | 'accounts/fireworks/models/qwen2p5-coder-32b-instruct' | 'accounts/fireworks/models/qwen2p5-72b-instruct' | 'accounts/fireworks/models/qwen-qwq-32b-preview' | 'accounts/fireworks/models/qwen2-vl-72b-instruct' | 'accounts/fireworks/models/llama-v3p2-11b-vision-instruct' | 'accounts/fireworks/models/qwq-32b' | 'accounts/fireworks/models/yi-large' | 'accounts/fireworks/models/kimi-k2-instruct' | 'accounts/fireworks/models/kimi-k2-thinking' | 'accounts/fireworks/models/kimi-k2p6' | 'accounts/fireworks/models/minimax-m2' | (string & {});
 declare const fireworksLanguageModelOptions: z.ZodObject<{
+    promptCacheKey: z.ZodOptional<z.ZodString>;
     thinking: z.ZodOptional<z.ZodObject<{
         type: z.ZodOptional<z.ZodEnum<{
             enabled: "enabled";

package/dist/index.js

@@ -305,7 +305,7 @@ var import_provider_utils3 = require("@ai-sdk/provider-utils");
 var import_v42 = require("zod/v4");
 
 // src/version.ts
-var VERSION = true ? "2.0.57" : "0.0.0-test";
+var VERSION = true ? "2.0.58" : "0.0.0-test";
 
 // src/fireworks-provider.ts
 var fireworksErrorSchema = import_v42.z.object({
@@ -344,9 +344,18 @@ function createFireworks(options = {}) {
       transformRequestBody: (args) => {
         const thinking = args.thinking;
         const reasoningHistory = args.reasoningHistory;
-        const { thinking: _, reasoningHistory: __, ...rest } = args;
+        const promptCacheKey = args.promptCacheKey;
+        const {
+          thinking: _,
+          reasoningHistory: __,
+          promptCacheKey: ___,
+          ...rest
+        } = args;
         return {
           ...rest,
+          ...promptCacheKey !== void 0 && {
+            prompt_cache_key: promptCacheKey
+          },
           ...thinking && {
             thinking: {
               type: thinking.type,

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 type { ImageModelV3, SharedV3Warning } from '@ai-sdk/provider';\nimport {\n  combineHeaders,\n  convertImageModelFileToDataUri,\n  createBinaryResponseHandler,\n  createJsonResponseHandler,\n  createStatusCodeErrorResponseHandler,\n  delay,\n  getFromApi,\n  isSameOrigin,\n  postJsonToApi,\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 ImageModelV3 {\n  readonly specificationVersion = 'v3';\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<ImageModelV3['doGenerate']>[0]): Promise<\n    Awaited<ReturnType<ImageModelV3['doGenerate']>>\n  > {\n    const warnings: Array<SharedV3Warning> = [];\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<SharedV3Warning>;\n    currentDate: Date;\n  }): Promise<Awaited<ReturnType<ImageModelV3['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  EmbeddingModelV3,\n  ImageModelV3,\n  LanguageModelV3,\n  ProviderV3,\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 ProviderV3 {\n  /**\n   * Creates a model for text generation.\n   */\n  (modelId: FireworksChatModelId): LanguageModelV3;\n\n  /**\n   * Creates a chat model for text generation.\n   */\n  chatModel(modelId: FireworksChatModelId): LanguageModelV3;\n\n  /**\n   * Creates a completion model for text generation.\n   */\n  completionModel(modelId: FireworksCompletionModelId): LanguageModelV3;\n\n  /**\n   * Creates a chat model for text generation.\n   */\n  languageModel(modelId: FireworksChatModelId): LanguageModelV3;\n\n  /**\n   * Creates a text embedding model for text generation.\n   */\n  embeddingModel(modelId: FireworksEmbeddingModelId): EmbeddingModelV3;\n\n  /**\n   * @deprecated Use `embeddingModel` instead.\n   */\n  textEmbeddingModel(modelId: FireworksEmbeddingModelId): EmbeddingModelV3;\n\n  /**\n   * Creates a model for image generation.\n   */\n  image(modelId: FireworksImageModelId): ImageModelV3;\n\n  /**\n   * Creates a model for image generation.\n   */\n  imageModel(modelId: FireworksImageModelId): ImageModelV3;\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\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      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 = 'v3' 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,yBAWO;;;ACZP,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;;;ADNA,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;AAzIJ;AA0II,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;AAMD,UAAM,EAAE,OAAO,YAAY,gBAAgB,IAAI,UAAM,mCAAW;AAAA,MAC9D,KAAK;AAAA,MACL,aAAS,qCAAa,UAAU,KAAK,OAAO,OAAO,IAC/C,UACA;AAAA,MACJ;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;AArTtB;AAsTI,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;;;AEvWA,+BAKO;AAOP,IAAAC,yBAKO;AACP,IAAAC,aAAkB;;;AChBX,IAAM,UACX,OACI,WACA;;;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,cAAc;AAAA,MACd,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,cAAc;AAAA,IACd,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/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 type { ImageModelV3, SharedV3Warning } from '@ai-sdk/provider';\nimport {\n  combineHeaders,\n  convertImageModelFileToDataUri,\n  createBinaryResponseHandler,\n  createJsonResponseHandler,\n  createStatusCodeErrorResponseHandler,\n  delay,\n  getFromApi,\n  isSameOrigin,\n  postJsonToApi,\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 ImageModelV3 {\n  readonly specificationVersion = 'v3';\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<ImageModelV3['doGenerate']>[0]): Promise<\n    Awaited<ReturnType<ImageModelV3['doGenerate']>>\n  > {\n    const warnings: Array<SharedV3Warning> = [];\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<SharedV3Warning>;\n    currentDate: Date;\n  }): Promise<Awaited<ReturnType<ImageModelV3['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  EmbeddingModelV3,\n  ImageModelV3,\n  LanguageModelV3,\n  ProviderV3,\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 ProviderV3 {\n  /**\n   * Creates a model for text generation.\n   */\n  (modelId: FireworksChatModelId): LanguageModelV3;\n\n  /**\n   * Creates a chat model for text generation.\n   */\n  chatModel(modelId: FireworksChatModelId): LanguageModelV3;\n\n  /**\n   * Creates a completion model for text generation.\n   */\n  completionModel(modelId: FireworksCompletionModelId): LanguageModelV3;\n\n  /**\n   * Creates a chat model for text generation.\n   */\n  languageModel(modelId: FireworksChatModelId): LanguageModelV3;\n\n  /**\n   * Creates a text embedding model for text generation.\n   */\n  embeddingModel(modelId: FireworksEmbeddingModelId): EmbeddingModelV3;\n\n  /**\n   * @deprecated Use `embeddingModel` instead.\n   */\n  textEmbeddingModel(modelId: FireworksEmbeddingModelId): EmbeddingModelV3;\n\n  /**\n   * Creates a model for image generation.\n   */\n  image(modelId: FireworksImageModelId): ImageModelV3;\n\n  /**\n   * Creates a model for image generation.\n   */\n  imageModel(modelId: FireworksImageModelId): ImageModelV3;\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          ...rest\n        } = args;\n\n        return {\n          ...rest,\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 = 'v3' 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,yBAWO;;;ACZP,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;;;ADNA,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;AAzIJ;AA0II,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;AAMD,UAAM,EAAE,OAAO,YAAY,gBAAgB,IAAI,UAAM,mCAAW;AAAA,MAC9D,KAAK;AAAA,MACL,aAAS,qCAAa,UAAU,KAAK,OAAO,OAAO,IAC/C,UACA;AAAA,MACJ;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;AArTtB;AAsTI,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;;;AEvWA,+BAKO;AAOP,IAAAC,yBAKO;AACP,IAAAC,aAAkB;;;AChBX,IAAM,UACX,OACI,WACA;;;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,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,GAAG;AAAA,QACL,IAAI;AAEJ,eAAO;AAAA,UACL,GAAG;AAAA,UACH,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,iEAAwC,SAAS;AAAA,IACnD,GAAG,qBAAqB,YAAY;AAAA,IACpC,cAAc;AAAA,IACd,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"]}

package/dist/index.mjs

@@ -294,7 +294,7 @@ import {
 import { z as z2 } from "zod/v4";
 
 // src/version.ts
-var VERSION = true ? "2.0.57" : "0.0.0-test";
+var VERSION = true ? "2.0.58" : "0.0.0-test";
 
 // src/fireworks-provider.ts
 var fireworksErrorSchema = z2.object({
@@ -333,9 +333,18 @@ function createFireworks(options = {}) {
       transformRequestBody: (args) => {
         const thinking = args.thinking;
         const reasoningHistory = args.reasoningHistory;
-        const { thinking: _, reasoningHistory: __, ...rest } = args;
+        const promptCacheKey = args.promptCacheKey;
+        const {
+          thinking: _,
+          reasoningHistory: __,
+          promptCacheKey: ___,
+          ...rest
+        } = args;
         return {
           ...rest,
+          ...promptCacheKey !== void 0 && {
+            prompt_cache_key: promptCacheKey
+          },
           ...thinking && {
             thinking: {
               type: thinking.type,

package/dist/index.mjs.map

@@ -1 +1 @@
-{"version":3,"sources":["../src/fireworks-image-model.ts","../src/fireworks-image-api.ts","../src/fireworks-provider.ts","../src/version.ts"],"sourcesContent":["import type { ImageModelV3, SharedV3Warning } from '@ai-sdk/provider';\nimport {\n  combineHeaders,\n  convertImageModelFileToDataUri,\n  createBinaryResponseHandler,\n  createJsonResponseHandler,\n  createStatusCodeErrorResponseHandler,\n  delay,\n  getFromApi,\n  isSameOrigin,\n  postJsonToApi,\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 ImageModelV3 {\n  readonly specificationVersion = 'v3';\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<ImageModelV3['doGenerate']>[0]): Promise<\n    Awaited<ReturnType<ImageModelV3['doGenerate']>>\n  > {\n    const warnings: Array<SharedV3Warning> = [];\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<SharedV3Warning>;\n    currentDate: Date;\n  }): Promise<Awaited<ReturnType<ImageModelV3['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  EmbeddingModelV3,\n  ImageModelV3,\n  LanguageModelV3,\n  ProviderV3,\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 ProviderV3 {\n  /**\n   * Creates a model for text generation.\n   */\n  (modelId: FireworksChatModelId): LanguageModelV3;\n\n  /**\n   * Creates a chat model for text generation.\n   */\n  chatModel(modelId: FireworksChatModelId): LanguageModelV3;\n\n  /**\n   * Creates a completion model for text generation.\n   */\n  completionModel(modelId: FireworksCompletionModelId): LanguageModelV3;\n\n  /**\n   * Creates a chat model for text generation.\n   */\n  languageModel(modelId: FireworksChatModelId): LanguageModelV3;\n\n  /**\n   * Creates a text embedding model for text generation.\n   */\n  embeddingModel(modelId: FireworksEmbeddingModelId): EmbeddingModelV3;\n\n  /**\n   * @deprecated Use `embeddingModel` instead.\n   */\n  textEmbeddingModel(modelId: FireworksEmbeddingModelId): EmbeddingModelV3;\n\n  /**\n   * Creates a model for image generation.\n   */\n  image(modelId: FireworksImageModelId): ImageModelV3;\n\n  /**\n   * Creates a model for image generation.\n   */\n  imageModel(modelId: FireworksImageModelId): ImageModelV3;\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\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      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 = 'v3' 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,OAEK;;;ACZP,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;;;ADNA,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;AAzIJ;AA0II,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,eAAe,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,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;AArTtB;AAsTI,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;;;AEvWA;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,WACA;;;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;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,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"]}
+{"version":3,"sources":["../src/fireworks-image-model.ts","../src/fireworks-image-api.ts","../src/fireworks-provider.ts","../src/version.ts"],"sourcesContent":["import type { ImageModelV3, SharedV3Warning } from '@ai-sdk/provider';\nimport {\n  combineHeaders,\n  convertImageModelFileToDataUri,\n  createBinaryResponseHandler,\n  createJsonResponseHandler,\n  createStatusCodeErrorResponseHandler,\n  delay,\n  getFromApi,\n  isSameOrigin,\n  postJsonToApi,\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 ImageModelV3 {\n  readonly specificationVersion = 'v3';\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<ImageModelV3['doGenerate']>[0]): Promise<\n    Awaited<ReturnType<ImageModelV3['doGenerate']>>\n  > {\n    const warnings: Array<SharedV3Warning> = [];\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<SharedV3Warning>;\n    currentDate: Date;\n  }): Promise<Awaited<ReturnType<ImageModelV3['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  EmbeddingModelV3,\n  ImageModelV3,\n  LanguageModelV3,\n  ProviderV3,\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 ProviderV3 {\n  /**\n   * Creates a model for text generation.\n   */\n  (modelId: FireworksChatModelId): LanguageModelV3;\n\n  /**\n   * Creates a chat model for text generation.\n   */\n  chatModel(modelId: FireworksChatModelId): LanguageModelV3;\n\n  /**\n   * Creates a completion model for text generation.\n   */\n  completionModel(modelId: FireworksCompletionModelId): LanguageModelV3;\n\n  /**\n   * Creates a chat model for text generation.\n   */\n  languageModel(modelId: FireworksChatModelId): LanguageModelV3;\n\n  /**\n   * Creates a text embedding model for text generation.\n   */\n  embeddingModel(modelId: FireworksEmbeddingModelId): EmbeddingModelV3;\n\n  /**\n   * @deprecated Use `embeddingModel` instead.\n   */\n  textEmbeddingModel(modelId: FireworksEmbeddingModelId): EmbeddingModelV3;\n\n  /**\n   * Creates a model for image generation.\n   */\n  image(modelId: FireworksImageModelId): ImageModelV3;\n\n  /**\n   * Creates a model for image generation.\n   */\n  imageModel(modelId: FireworksImageModelId): ImageModelV3;\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          ...rest\n        } = args;\n\n        return {\n          ...rest,\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 = 'v3' 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,OAEK;;;ACZP,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;;;ADNA,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;AAzIJ;AA0II,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,eAAe,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,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;AArTtB;AAsTI,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;;;AEvWA;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,WACA;;;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,GAAG;AAAA,QACL,IAAI;AAEJ,eAAO;AAAA,UACL,GAAG;AAAA,UACH,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,9 +135,16 @@ const { text, reasoningText } = await generateText({
 
 The following optional provider options are available for Fireworks chat models:
 
+- **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.
+
 - **thinking** _object_
 
-  Configuration for thinking/reasoning models like Kimi K2.5.
+  Configuration for thinking/reasoning models like Kimi K2.6.
 
   - **type** _'enabled' | 'disabled'_
 
@@ -155,6 +162,90 @@ The following optional provider options are available for Fireworks chat models:
   - `'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 +279,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,6 +1,6 @@
 {
   "name": "@ai-sdk/fireworks",
-  "version": "2.0.57",
+  "version": "2.0.58",
   "license": "Apache-2.0",
   "sideEffects": false,
   "main": "./dist/index.js",
@@ -29,8 +29,8 @@
     }
   },
   "dependencies": {
-    "@ai-sdk/provider-utils": "4.0.30",
     "@ai-sdk/openai-compatible": "2.0.51",
+    "@ai-sdk/provider-utils": "4.0.30",
     "@ai-sdk/provider": "3.0.10"
   },
   "devDependencies": {

package/src/fireworks-chat-options.ts

@@ -20,11 +20,17 @@ export type FireworksChatModelId =
   | 'accounts/fireworks/models/yi-large'
   | 'accounts/fireworks/models/kimi-k2-instruct'
   | 'accounts/fireworks/models/kimi-k2-thinking'
-  | 'accounts/fireworks/models/kimi-k2p5'
+  | 'accounts/fireworks/models/kimi-k2p6'
   | 'accounts/fireworks/models/minimax-m2'
   | (string & {});
 
 export const fireworksLanguageModelOptions = z.object({
+  /**
+   * A stable key for routing requests with shared prompt prefixes to the same
+   * prompt cache.
+   */
+  promptCacheKey: z.string().optional(),
+
   thinking: z
     .object({
       type: z.enum(['enabled', 'disabled']).optional(),

package/src/fireworks-provider.ts

@@ -141,11 +141,20 @@ export function createFireworks(
           | { type?: string; budgetTokens?: number }
           | undefined;
         const reasoningHistory = args.reasoningHistory as string | undefined;
+        const promptCacheKey = args.promptCacheKey as string | undefined;
 
-        const { thinking: _, reasoningHistory: __, ...rest } = args;
+        const {
+          thinking: _,
+          reasoningHistory: __,
+          promptCacheKey: ___,
+          ...rest
+        } = args;
 
         return {
           ...rest,
+          ...(promptCacheKey !== undefined && {
+            prompt_cache_key: promptCacheKey,
+          }),
           ...(thinking && {
             thinking: {
               type: thinking.type,