@ai-sdk/fireworks 0.0.8 → 0.0.10

package/CHANGELOG.md

@@ -1,5 +1,31 @@
 # @ai-sdk/fireworks
 
+## 0.0.10
+
+### Patch Changes
+
+- 90fb95a: chore (provider-utils): switch to unified test server
+- e6dfef4: feat (provider/fireworks): Support add'l image models.
+- Updated dependencies [90fb95a]
+- Updated dependencies [e6dfef4]
+- Updated dependencies [6636db6]
+  - @ai-sdk/provider-utils@2.0.7
+  - @ai-sdk/openai-compatible@0.0.15
+
+## 0.0.9
+
+### Patch Changes
+
+- 19a2ce7: feat (provider/fireworks): Add image model support.
+- 6337688: feat: change image generation errors to warnings
+- Updated dependencies [19a2ce7]
+- Updated dependencies [19a2ce7]
+- Updated dependencies [43b37f7]
+- Updated dependencies [6337688]
+  - @ai-sdk/provider@1.0.4
+  - @ai-sdk/provider-utils@2.0.6
+  - @ai-sdk/openai-compatible@0.0.14
+
 ## 0.0.8
 
 ### Patch Changes

package/README.md

@@ -1,34 +1,50 @@
 # AI SDK - Fireworks Provider
 
-The **[Fireworks provider](https://sdk.vercel.ai/providers/ai-sdk-providers/fireworks)** for the [AI SDK](https://sdk.vercel.ai/docs) contains language model support for the [Fireworks](https://fireworks.ai) platform.
+The **[Fireworks provider](https://sdk.vercel.ai/providers/ai-sdk-providers/fireworks)** for the [AI SDK](https://sdk.vercel.ai/docs) contains language model and image model support for the [Fireworks](https://fireworks.ai) platform.
 
 ## Setup
 
 The Fireworks provider is available in the `@ai-sdk/fireworks` module. You can install it with
 
-\```bash
+```bash
 npm i @ai-sdk/fireworks
-\```
+```
 
 ## Provider Instance
 
 You can import the default provider instance `fireworks` from `@ai-sdk/fireworks`:
 
-\```ts
+```ts
 import { fireworks } from '@ai-sdk/fireworks';
-\```
+```
 
-## Example
+## Language Model Example
 
-\```ts
+```ts
 import { fireworks } from '@ai-sdk/fireworks';
 import { generateText } from 'ai';
 
 const { text } = await generateText({
-model: fireworks('accounts/fireworks/models/llama-v2-13b-chat'),
-prompt: 'Write a JavaScript function that sorts a list:',
+  model: fireworks('accounts/fireworks/models/deepseek-v3'),
+  prompt: 'Write a JavaScript function that sorts a list:',
 });
-\```
+```
+
+## Image Model Examples
+
+```ts
+import { fireworks } from '@ai-sdk/fireworks';
+import { experimental_generateImage as generateImage } from 'ai';
+import fs from 'fs';
+
+const { image } = await generateImage({
+  model: fireworks.image('accounts/fireworks/models/flux-1-dev-fp8'),
+  prompt: 'A serene mountain landscape at sunset',
+});
+const filename = `image-${Date.now()}.png`;
+fs.writeFileSync(filename, image.uint8Array);
+console.log(`Image saved to ${filename}`);
+```
 
 ## Documentation
 

package/dist/index.d.mts

@@ -1,4 +1,4 @@
-import { LanguageModelV1, EmbeddingModelV1 } from '@ai-sdk/provider';
+import { LanguageModelV1, EmbeddingModelV1, ImageModelV1 } from '@ai-sdk/provider';
 import { FetchFunction } from '@ai-sdk/provider-utils';
 import { z } from 'zod';
 import { OpenAICompatibleChatSettings, OpenAICompatibleCompletionSettings, OpenAICompatibleEmbeddingSettings } from '@ai-sdk/openai-compatible';
@@ -15,6 +15,8 @@ type FireworksEmbeddingModelId = 'nomic-ai/nomic-embed-text-v1.5' | (string & {}
 interface FireworksEmbeddingSettings extends OpenAICompatibleEmbeddingSettings {
 }
 
+type FireworksImageModelId = 'accounts/fireworks/models/flux-1-dev-fp8' | 'accounts/fireworks/models/flux-1-schnell-fp8' | 'accounts/fireworks/models/playground-v2-5-1024px-aesthetic' | 'accounts/fireworks/models/japanese-stable-diffusion-xl' | 'accounts/fireworks/models/playground-v2-1024px-aesthetic' | 'accounts/fireworks/models/SSD-1B' | 'accounts/fireworks/models/stable-diffusion-xl-1024-v1-0' | (string & {});
+
 type FireworksErrorData = z.infer<typeof fireworksErrorSchema>;
 declare const fireworksErrorSchema: z.ZodObject<{
     error: z.ZodString;
@@ -25,7 +27,8 @@ declare const fireworksErrorSchema: z.ZodObject<{
 }>;
 interface FireworksProviderSettings {
     /**
-  Fireworks API key.
+  Fireworks API key. Default value is taken from the `FIREWORKS_API_KEY`
+  environment variable.
   */
     apiKey?: string;
     /**
@@ -59,6 +62,10 @@ interface FireworksProvider {
   Creates a text embedding model for text generation.
   */
     textEmbeddingModel(modelId: FireworksEmbeddingModelId, settings?: FireworksEmbeddingSettings): EmbeddingModelV1<string>;
+    /**
+  Creates a model for image generation.
+  */
+    image(modelId: FireworksImageModelId): ImageModelV1;
 }
 declare function createFireworks(options?: FireworksProviderSettings): FireworksProvider;
 declare const fireworks: FireworksProvider;

package/dist/index.d.ts

@@ -1,4 +1,4 @@
-import { LanguageModelV1, EmbeddingModelV1 } from '@ai-sdk/provider';
+import { LanguageModelV1, EmbeddingModelV1, ImageModelV1 } from '@ai-sdk/provider';
 import { FetchFunction } from '@ai-sdk/provider-utils';
 import { z } from 'zod';
 import { OpenAICompatibleChatSettings, OpenAICompatibleCompletionSettings, OpenAICompatibleEmbeddingSettings } from '@ai-sdk/openai-compatible';
@@ -15,6 +15,8 @@ type FireworksEmbeddingModelId = 'nomic-ai/nomic-embed-text-v1.5' | (string & {}
 interface FireworksEmbeddingSettings extends OpenAICompatibleEmbeddingSettings {
 }
 
+type FireworksImageModelId = 'accounts/fireworks/models/flux-1-dev-fp8' | 'accounts/fireworks/models/flux-1-schnell-fp8' | 'accounts/fireworks/models/playground-v2-5-1024px-aesthetic' | 'accounts/fireworks/models/japanese-stable-diffusion-xl' | 'accounts/fireworks/models/playground-v2-1024px-aesthetic' | 'accounts/fireworks/models/SSD-1B' | 'accounts/fireworks/models/stable-diffusion-xl-1024-v1-0' | (string & {});
+
 type FireworksErrorData = z.infer<typeof fireworksErrorSchema>;
 declare const fireworksErrorSchema: z.ZodObject<{
     error: z.ZodString;
@@ -25,7 +27,8 @@ declare const fireworksErrorSchema: z.ZodObject<{
 }>;
 interface FireworksProviderSettings {
     /**
-  Fireworks API key.
+  Fireworks API key. Default value is taken from the `FIREWORKS_API_KEY`
+  environment variable.
   */
     apiKey?: string;
     /**
@@ -59,6 +62,10 @@ interface FireworksProvider {
   Creates a text embedding model for text generation.
   */
     textEmbeddingModel(modelId: FireworksEmbeddingModelId, settings?: FireworksEmbeddingSettings): EmbeddingModelV1<string>;
+    /**
+  Creates a model for image generation.
+  */
+    image(modelId: FireworksImageModelId): ImageModelV1;
 }
 declare function createFireworks(options?: FireworksProviderSettings): FireworksProvider;
 declare const fireworks: FireworksProvider;

package/dist/index.js

@@ -27,8 +27,172 @@ module.exports = __toCommonJS(src_exports);
 
 // src/fireworks-provider.ts
 var import_openai_compatible = require("@ai-sdk/openai-compatible");
-var import_provider_utils = require("@ai-sdk/provider-utils");
+var import_provider_utils2 = require("@ai-sdk/provider-utils");
 var import_zod = require("zod");
+
+// src/fireworks-image-model.ts
+var import_provider = require("@ai-sdk/provider");
+var import_provider_utils = require("@ai-sdk/provider-utils");
+var modelToBackendConfig = {
+  "accounts/fireworks/models/flux-1-dev-fp8": {
+    urlFormat: "workflows"
+  },
+  "accounts/fireworks/models/flux-1-schnell-fp8": {
+    urlFormat: "workflows"
+  },
+  "accounts/fireworks/models/playground-v2-5-1024px-aesthetic": {
+    urlFormat: "image_generation",
+    supportsSize: true
+  },
+  "accounts/fireworks/models/japanese-stable-diffusion-xl": {
+    urlFormat: "image_generation",
+    supportsSize: true
+  },
+  "accounts/fireworks/models/playground-v2-1024px-aesthetic": {
+    urlFormat: "image_generation",
+    supportsSize: true
+  },
+  "accounts/fireworks/models/stable-diffusion-xl-1024-v1-0": {
+    urlFormat: "image_generation",
+    supportsSize: true
+  },
+  "accounts/fireworks/models/SSD-1B": {
+    urlFormat: "image_generation",
+    supportsSize: true
+  }
+};
+function getUrlForModel(baseUrl, modelId) {
+  var _a;
+  switch ((_a = modelToBackendConfig[modelId]) == null ? void 0 : _a.urlFormat) {
+    case "image_generation":
+      return `${baseUrl}/image_generation/${modelId}`;
+    case "workflows":
+    default:
+      return `${baseUrl}/workflows/${modelId}/text_to_image`;
+  }
+}
+var createBinaryResponseHandler = () => async ({ response, url, requestBodyValues }) => {
+  const responseHeaders = (0, import_provider_utils.extractResponseHeaders)(response);
+  if (!response.body) {
+    throw new import_provider.APICallError({
+      message: "Response body is empty",
+      url,
+      requestBodyValues,
+      statusCode: response.status,
+      responseHeaders,
+      responseBody: void 0
+    });
+  }
+  try {
+    const buffer = await response.arrayBuffer();
+    return {
+      responseHeaders,
+      value: buffer
+    };
+  } catch (error) {
+    throw new import_provider.APICallError({
+      message: "Failed to read response as array buffer",
+      url,
+      requestBodyValues,
+      statusCode: response.status,
+      responseHeaders,
+      responseBody: void 0,
+      cause: error
+    });
+  }
+};
+var statusCodeErrorResponseHandler = async ({
+  response,
+  url,
+  requestBodyValues
+}) => {
+  const responseHeaders = (0, import_provider_utils.extractResponseHeaders)(response);
+  const responseBody = await response.text();
+  return {
+    responseHeaders,
+    value: new import_provider.APICallError({
+      message: response.statusText,
+      url,
+      requestBodyValues,
+      statusCode: response.status,
+      responseHeaders,
+      responseBody
+    })
+  };
+};
+async function postImageToApi(params) {
+  var _a, _b;
+  const splitSize = (_a = params.size) == null ? void 0 : _a.split("x");
+  const { value: response } = await (0, import_provider_utils.postJsonToApi)({
+    url: getUrlForModel(params.baseUrl, params.modelId),
+    headers: params.headers,
+    body: {
+      prompt: params.prompt,
+      aspect_ratio: params.aspectRatio,
+      seed: params.seed,
+      ...splitSize && { width: splitSize[0], height: splitSize[1] },
+      ...(_b = params.providerOptions.fireworks) != null ? _b : {}
+    },
+    failedResponseHandler: statusCodeErrorResponseHandler,
+    successfulResponseHandler: createBinaryResponseHandler(),
+    abortSignal: params.abortSignal,
+    fetch: params.fetch
+  });
+  return response;
+}
+var FireworksImageModel = class {
+  constructor(modelId, config) {
+    this.modelId = modelId;
+    this.config = config;
+    this.specificationVersion = "v1";
+    this.maxImagesPerCall = 1;
+  }
+  get provider() {
+    return this.config.provider;
+  }
+  async doGenerate({
+    prompt,
+    n,
+    size,
+    aspectRatio,
+    seed,
+    providerOptions,
+    headers,
+    abortSignal
+  }) {
+    const warnings = [];
+    const backendConfig = modelToBackendConfig[this.modelId];
+    if (!(backendConfig == null ? void 0 : backendConfig.supportsSize) && size != null) {
+      warnings.push({
+        type: "unsupported-setting",
+        setting: "size",
+        details: "This model does not support the `size` option. Use `aspectRatio` instead."
+      });
+    }
+    if ((backendConfig == null ? void 0 : backendConfig.supportsSize) && aspectRatio != null) {
+      warnings.push({
+        type: "unsupported-setting",
+        setting: "aspectRatio",
+        details: "This model does not support the `aspectRatio` option."
+      });
+    }
+    const response = await postImageToApi({
+      baseUrl: this.config.baseURL,
+      prompt,
+      aspectRatio,
+      size,
+      seed,
+      modelId: this.modelId,
+      providerOptions,
+      headers: (0, import_provider_utils.combineHeaders)(this.config.headers(), headers),
+      abortSignal,
+      fetch: this.config.fetch
+    });
+    return { images: [new Uint8Array(response)], warnings };
+  }
+};
+
+// src/fireworks-provider.ts
 var fireworksErrorSchema = import_zod.z.object({
   error: import_zod.z.string()
 });
@@ -36,13 +200,12 @@ var fireworksErrorStructure = {
   errorSchema: fireworksErrorSchema,
   errorToMessage: (data) => data.error
 };
+var defaultBaseURL = "https://api.fireworks.ai/inference/v1";
 function createFireworks(options = {}) {
   var _a;
-  const baseURL = (0, import_provider_utils.withoutTrailingSlash)(
-    (_a = options.baseURL) != null ? _a : "https://api.fireworks.ai/inference/v1"
-  );
+  const baseURL = (0, import_provider_utils2.withoutTrailingSlash)((_a = options.baseURL) != null ? _a : defaultBaseURL);
   const getHeaders = () => ({
-    Authorization: `Bearer ${(0, import_provider_utils.loadApiKey)({
+    Authorization: `Bearer ${(0, import_provider_utils2.loadApiKey)({
       apiKey: options.apiKey,
       environmentVariableName: "FIREWORKS_API_KEY",
       description: "Fireworks API key"
@@ -53,29 +216,32 @@ function createFireworks(options = {}) {
     provider: `fireworks.${modelType}`,
     url: ({ path }) => `${baseURL}${path}`,
     headers: getHeaders,
-    fetch: options.fetch,
-    errorStructure: fireworksErrorStructure
+    fetch: options.fetch
   });
   const createChatModel = (modelId, settings = {}) => {
     return new import_openai_compatible.OpenAICompatibleChatLanguageModel(modelId, settings, {
       ...getCommonModelConfig("chat"),
+      errorStructure: fireworksErrorStructure,
       defaultObjectGenerationMode: "json"
     });
   };
-  const createCompletionModel = (modelId, settings = {}) => new import_openai_compatible.OpenAICompatibleCompletionLanguageModel(
-    modelId,
-    settings,
-    getCommonModelConfig("completion")
-  );
-  const createTextEmbeddingModel = (modelId, settings = {}) => new import_openai_compatible.OpenAICompatibleEmbeddingModel(
-    modelId,
-    settings,
-    getCommonModelConfig("embedding")
-  );
+  const createCompletionModel = (modelId, settings = {}) => new import_openai_compatible.OpenAICompatibleCompletionLanguageModel(modelId, settings, {
+    ...getCommonModelConfig("completion"),
+    errorStructure: fireworksErrorStructure
+  });
+  const createTextEmbeddingModel = (modelId, settings = {}) => new import_openai_compatible.OpenAICompatibleEmbeddingModel(modelId, settings, {
+    ...getCommonModelConfig("embedding"),
+    errorStructure: fireworksErrorStructure
+  });
+  const createImageModel = (modelId) => new FireworksImageModel(modelId, {
+    ...getCommonModelConfig("image"),
+    baseURL: baseURL != null ? baseURL : defaultBaseURL
+  });
   const provider = (modelId, settings) => createChatModel(modelId, settings);
   provider.completionModel = createCompletionModel;
   provider.chatModel = createChatModel;
   provider.textEmbeddingModel = createTextEmbeddingModel;
+  provider.image = createImageModel;
   return provider;
 }
 var fireworks = createFireworks();

package/dist/index.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../src/index.ts","../src/fireworks-provider.ts"],"sourcesContent":["export { createFireworks, fireworks } from './fireworks-provider';\nexport type {\n  FireworksProvider,\n  FireworksProviderSettings,\n  FireworksErrorData,\n} from './fireworks-provider';\n","import {\n  OpenAICompatibleChatLanguageModel,\n  OpenAICompatibleCompletionLanguageModel,\n  OpenAICompatibleEmbeddingModel,\n  ProviderErrorStructure,\n} from '@ai-sdk/openai-compatible';\nimport { EmbeddingModelV1, LanguageModelV1 } from '@ai-sdk/provider';\nimport {\n  FetchFunction,\n  loadApiKey,\n  withoutTrailingSlash,\n} from '@ai-sdk/provider-utils';\nimport { z } from 'zod';\nimport {\n  FireworksChatModelId,\n  FireworksChatSettings,\n} from './fireworks-chat-settings';\nimport {\n  FireworksCompletionModelId,\n  FireworksCompletionSettings,\n} from './fireworks-completion-settings';\nimport {\n  FireworksEmbeddingModelId,\n  FireworksEmbeddingSettings,\n} from './fireworks-embedding-settings';\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  /**\nFireworks API key.\n*/\n  apiKey?: string;\n  /**\nBase URL for the API calls.\n*/\n  baseURL?: string;\n  /**\nCustom headers to include in the requests.\n*/\n  headers?: Record<string, string>;\n  /**\nCustom fetch implementation. You can use it as a middleware to intercept requests,\nor to provide a custom fetch implementation for e.g. testing.\n*/\n  fetch?: FetchFunction;\n}\n\nexport interface FireworksProvider {\n  /**\nCreates a model for text generation.\n*/\n  (\n    modelId: FireworksChatModelId,\n    settings?: FireworksChatSettings,\n  ): LanguageModelV1;\n\n  /**\nCreates a chat model for text generation.\n*/\n  chatModel(\n    modelId: FireworksChatModelId,\n    settings?: FireworksChatSettings,\n  ): LanguageModelV1;\n\n  /**\nCreates a completion model for text generation.\n*/\n  completionModel(\n    modelId: FireworksCompletionModelId,\n    settings?: FireworksCompletionSettings,\n  ): LanguageModelV1;\n\n  /**\nCreates a text embedding model for text generation.\n*/\n  textEmbeddingModel(\n    modelId: FireworksEmbeddingModelId,\n    settings?: FireworksEmbeddingSettings,\n  ): EmbeddingModelV1<string>;\n}\n\nexport function createFireworks(\n  options: FireworksProviderSettings = {},\n): FireworksProvider {\n  const baseURL = withoutTrailingSlash(\n    options.baseURL ?? 'https://api.fireworks.ai/inference/v1',\n  );\n  const getHeaders = () => ({\n    Authorization: `Bearer ${loadApiKey({\n      apiKey: options.apiKey,\n      environmentVariableName: 'FIREWORKS_API_KEY',\n      description: 'Fireworks API key',\n    })}`,\n    ...options.headers,\n  });\n\n  interface CommonModelConfig {\n    provider: `fireworks.${string}`;\n    url: ({ path }: { path: string }) => string;\n    headers: () => Record<string, string>;\n    fetch?: FetchFunction;\n    errorStructure?: ProviderErrorStructure<FireworksErrorData>;\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    errorStructure: fireworksErrorStructure,\n  });\n\n  const createChatModel = (\n    modelId: FireworksChatModelId,\n    settings: FireworksChatSettings = {},\n  ) => {\n    return new OpenAICompatibleChatLanguageModel(modelId, settings, {\n      ...getCommonModelConfig('chat'),\n      defaultObjectGenerationMode: 'json',\n    });\n  };\n\n  const createCompletionModel = (\n    modelId: FireworksCompletionModelId,\n    settings: FireworksCompletionSettings = {},\n  ) =>\n    new OpenAICompatibleCompletionLanguageModel(\n      modelId,\n      settings,\n      getCommonModelConfig('completion'),\n    );\n\n  const createTextEmbeddingModel = (\n    modelId: FireworksEmbeddingModelId,\n    settings: FireworksEmbeddingSettings = {},\n  ) =>\n    new OpenAICompatibleEmbeddingModel(\n      modelId,\n      settings,\n      getCommonModelConfig('embedding'),\n    );\n\n  const provider = (\n    modelId: FireworksChatModelId,\n    settings?: FireworksChatSettings,\n  ) => createChatModel(modelId, settings);\n\n  provider.completionModel = createCompletionModel;\n  provider.chatModel = createChatModel;\n  provider.textEmbeddingModel = createTextEmbeddingModel;\n\n  return provider as FireworksProvider;\n}\n\nexport const fireworks = createFireworks();\n"],"mappings":";;;;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;;;ACAA,+BAKO;AAEP,4BAIO;AACP,iBAAkB;AAgBlB,IAAM,uBAAuB,aAAE,OAAO;AAAA,EACpC,OAAO,aAAE,OAAO;AAClB,CAAC;AAED,IAAM,0BAAsE;AAAA,EAC1E,aAAa;AAAA,EACb,gBAAgB,UAAQ,KAAK;AAC/B;AAwDO,SAAS,gBACd,UAAqC,CAAC,GACnB;AA7FrB;AA8FE,QAAM,cAAU;AAAA,KACd,aAAQ,YAAR,YAAmB;AAAA,EACrB;AACA,QAAM,aAAa,OAAO;AAAA,IACxB,eAAe,cAAU,kCAAW;AAAA,MAClC,QAAQ,QAAQ;AAAA,MAChB,yBAAyB;AAAA,MACzB,aAAa;AAAA,IACf,CAAC,CAAC;AAAA,IACF,GAAG,QAAQ;AAAA,EACb;AAUA,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,IACf,gBAAgB;AAAA,EAClB;AAEA,QAAM,kBAAkB,CACtB,SACA,WAAkC,CAAC,MAChC;AACH,WAAO,IAAI,2DAAkC,SAAS,UAAU;AAAA,MAC9D,GAAG,qBAAqB,MAAM;AAAA,MAC9B,6BAA6B;AAAA,IAC/B,CAAC;AAAA,EACH;AAEA,QAAM,wBAAwB,CAC5B,SACA,WAAwC,CAAC,MAEzC,IAAI;AAAA,IACF;AAAA,IACA;AAAA,IACA,qBAAqB,YAAY;AAAA,EACnC;AAEF,QAAM,2BAA2B,CAC/B,SACA,WAAuC,CAAC,MAExC,IAAI;AAAA,IACF;AAAA,IACA;AAAA,IACA,qBAAqB,WAAW;AAAA,EAClC;AAEF,QAAM,WAAW,CACf,SACA,aACG,gBAAgB,SAAS,QAAQ;AAEtC,WAAS,kBAAkB;AAC3B,WAAS,YAAY;AACrB,WAAS,qBAAqB;AAE9B,SAAO;AACT;AAEO,IAAM,YAAY,gBAAgB;","names":[]}
+{"version":3,"sources":["../src/index.ts","../src/fireworks-provider.ts","../src/fireworks-image-model.ts"],"sourcesContent":["export { createFireworks, fireworks } from './fireworks-provider';\nexport type {\n  FireworksProvider,\n  FireworksProviderSettings,\n  FireworksErrorData,\n} from './fireworks-provider';\n","import {\n  OpenAICompatibleChatLanguageModel,\n  OpenAICompatibleCompletionLanguageModel,\n  OpenAICompatibleEmbeddingModel,\n  ProviderErrorStructure,\n} from '@ai-sdk/openai-compatible';\nimport {\n  EmbeddingModelV1,\n  ImageModelV1,\n  LanguageModelV1,\n} from '@ai-sdk/provider';\nimport {\n  FetchFunction,\n  loadApiKey,\n  withoutTrailingSlash,\n} from '@ai-sdk/provider-utils';\nimport { z } from 'zod';\nimport {\n  FireworksChatModelId,\n  FireworksChatSettings,\n} from './fireworks-chat-settings';\nimport {\n  FireworksCompletionModelId,\n  FireworksCompletionSettings,\n} from './fireworks-completion-settings';\nimport {\n  FireworksEmbeddingModelId,\n  FireworksEmbeddingSettings,\n} from './fireworks-embedding-settings';\nimport {\n  FireworksImageModel,\n  FireworksImageModelId,\n} from './fireworks-image-model';\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  /**\nFireworks API key. Default value is taken from the `FIREWORKS_API_KEY`\nenvironment variable.\n*/\n  apiKey?: string;\n  /**\nBase URL for the API calls.\n*/\n  baseURL?: string;\n  /**\nCustom headers to include in the requests.\n*/\n  headers?: Record<string, string>;\n  /**\nCustom fetch implementation. You can use it as a middleware to intercept requests,\nor to provide a custom fetch implementation for e.g. testing.\n*/\n  fetch?: FetchFunction;\n}\n\nexport interface FireworksProvider {\n  /**\nCreates a model for text generation.\n*/\n  (\n    modelId: FireworksChatModelId,\n    settings?: FireworksChatSettings,\n  ): LanguageModelV1;\n\n  /**\nCreates a chat model for text generation.\n*/\n  chatModel(\n    modelId: FireworksChatModelId,\n    settings?: FireworksChatSettings,\n  ): LanguageModelV1;\n\n  /**\nCreates a completion model for text generation.\n*/\n  completionModel(\n    modelId: FireworksCompletionModelId,\n    settings?: FireworksCompletionSettings,\n  ): LanguageModelV1;\n\n  /**\nCreates a text embedding model for text generation.\n*/\n  textEmbeddingModel(\n    modelId: FireworksEmbeddingModelId,\n    settings?: FireworksEmbeddingSettings,\n  ): EmbeddingModelV1<string>;\n\n  /**\nCreates a model for image generation.\n*/\n  image(modelId: FireworksImageModelId): ImageModelV1;\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    Authorization: `Bearer ${loadApiKey({\n      apiKey: options.apiKey,\n      environmentVariableName: 'FIREWORKS_API_KEY',\n      description: 'Fireworks API key',\n    })}`,\n    ...options.headers,\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 = (\n    modelId: FireworksChatModelId,\n    settings: FireworksChatSettings = {},\n  ) => {\n    return new OpenAICompatibleChatLanguageModel(modelId, settings, {\n      ...getCommonModelConfig('chat'),\n      errorStructure: fireworksErrorStructure,\n      defaultObjectGenerationMode: 'json',\n    });\n  };\n\n  const createCompletionModel = (\n    modelId: FireworksCompletionModelId,\n    settings: FireworksCompletionSettings = {},\n  ) =>\n    new OpenAICompatibleCompletionLanguageModel(modelId, settings, {\n      ...getCommonModelConfig('completion'),\n      errorStructure: fireworksErrorStructure,\n    });\n\n  const createTextEmbeddingModel = (\n    modelId: FireworksEmbeddingModelId,\n    settings: FireworksEmbeddingSettings = {},\n  ) =>\n    new OpenAICompatibleEmbeddingModel(modelId, settings, {\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 = (\n    modelId: FireworksChatModelId,\n    settings?: FireworksChatSettings,\n  ) => createChatModel(modelId, settings);\n\n  provider.completionModel = createCompletionModel;\n  provider.chatModel = createChatModel;\n  provider.textEmbeddingModel = createTextEmbeddingModel;\n  provider.image = createImageModel;\n\n  return provider as FireworksProvider;\n}\n\nexport const fireworks = createFireworks();\n","import {\n  APICallError,\n  ImageModelV1,\n  ImageModelV1CallWarning,\n} from '@ai-sdk/provider';\nimport {\n  combineHeaders,\n  extractResponseHeaders,\n  FetchFunction,\n  postJsonToApi,\n  ResponseHandler,\n} from '@ai-sdk/provider-utils';\n\n// https://fireworks.ai/models?type=image\nexport type FireworksImageModelId =\n  | 'accounts/fireworks/models/flux-1-dev-fp8'\n  | 'accounts/fireworks/models/flux-1-schnell-fp8'\n  | 'accounts/fireworks/models/playground-v2-5-1024px-aesthetic'\n  | 'accounts/fireworks/models/japanese-stable-diffusion-xl'\n  | 'accounts/fireworks/models/playground-v2-1024px-aesthetic'\n  | 'accounts/fireworks/models/SSD-1B'\n  | 'accounts/fireworks/models/stable-diffusion-xl-1024-v1-0'\n  | (string & {});\n\ninterface FireworksImageModelBackendConfig {\n  urlFormat: 'workflows' | 'image_generation';\n  supportsSize?: 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/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  switch (modelToBackendConfig[modelId]?.urlFormat) {\n    case 'image_generation':\n      return `${baseUrl}/image_generation/${modelId}`;\n    case 'workflows':\n    default:\n      return `${baseUrl}/workflows/${modelId}/text_to_image`;\n  }\n}\n\ninterface FireworksImageModelConfig {\n  provider: string;\n  baseURL: string;\n  headers: () => Record<string, string>;\n  fetch?: FetchFunction;\n}\n\nconst createBinaryResponseHandler =\n  (): ResponseHandler<ArrayBuffer> =>\n  async ({ response, url, requestBodyValues }) => {\n    const responseHeaders = extractResponseHeaders(response);\n\n    if (!response.body) {\n      throw new APICallError({\n        message: 'Response body is empty',\n        url,\n        requestBodyValues,\n        statusCode: response.status,\n        responseHeaders,\n        responseBody: undefined,\n      });\n    }\n\n    try {\n      const buffer = await response.arrayBuffer();\n      return {\n        responseHeaders,\n        value: buffer,\n      };\n    } catch (error) {\n      throw new APICallError({\n        message: 'Failed to read response as array buffer',\n        url,\n        requestBodyValues,\n        statusCode: response.status,\n        responseHeaders,\n        responseBody: undefined,\n        cause: error,\n      });\n    }\n  };\n\nconst statusCodeErrorResponseHandler: ResponseHandler<APICallError> = async ({\n  response,\n  url,\n  requestBodyValues,\n}) => {\n  const responseHeaders = extractResponseHeaders(response);\n  const responseBody = await response.text();\n\n  return {\n    responseHeaders,\n    value: new APICallError({\n      message: response.statusText,\n      url,\n      requestBodyValues: requestBodyValues as Record<string, unknown>,\n      statusCode: response.status,\n      responseHeaders,\n      responseBody,\n    }),\n  };\n};\n\ninterface ImageRequestParams {\n  baseUrl: string;\n  modelId: FireworksImageModelId;\n  prompt: string;\n  aspectRatio?: string;\n  size?: string;\n  seed?: number;\n  providerOptions: Record<string, unknown>;\n  headers: Record<string, string | undefined>;\n  abortSignal?: AbortSignal;\n  fetch?: FetchFunction;\n}\n\nasync function postImageToApi(\n  params: ImageRequestParams,\n): Promise<ArrayBuffer> {\n  const splitSize = params.size?.split('x');\n  const { value: response } = await postJsonToApi({\n    url: getUrlForModel(params.baseUrl, params.modelId),\n    headers: params.headers,\n    body: {\n      prompt: params.prompt,\n      aspect_ratio: params.aspectRatio,\n      seed: params.seed,\n      ...(splitSize && { width: splitSize[0], height: splitSize[1] }),\n      ...(params.providerOptions.fireworks ?? {}),\n    },\n    failedResponseHandler: statusCodeErrorResponseHandler,\n    successfulResponseHandler: createBinaryResponseHandler(),\n    abortSignal: params.abortSignal,\n    fetch: params.fetch,\n  });\n\n  return response;\n}\n\nexport class FireworksImageModel implements ImageModelV1 {\n  readonly specificationVersion = 'v1';\n\n  get provider(): string {\n    return this.config.provider;\n  }\n\n  readonly maxImagesPerCall = 1;\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  }: Parameters<ImageModelV1['doGenerate']>[0]): Promise<\n    Awaited<ReturnType<ImageModelV1['doGenerate']>>\n  > {\n    const warnings: Array<ImageModelV1CallWarning> = [];\n\n    const backendConfig = modelToBackendConfig[this.modelId];\n    if (!backendConfig?.supportsSize && size != null) {\n      warnings.push({\n        type: 'unsupported-setting',\n        setting: '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-setting',\n        setting: 'aspectRatio',\n        details: 'This model does not support the `aspectRatio` option.',\n      });\n    }\n\n    const response = await postImageToApi({\n      baseUrl: this.config.baseURL,\n      prompt,\n      aspectRatio,\n      size,\n      seed,\n      modelId: this.modelId,\n      providerOptions,\n      headers: combineHeaders(this.config.headers(), headers),\n      abortSignal,\n      fetch: this.config.fetch,\n    });\n\n    return { images: [new Uint8Array(response)], warnings };\n  }\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;;;ACAA,+BAKO;AAMP,IAAAA,yBAIO;AACP,iBAAkB;;;AChBlB,sBAIO;AACP,4BAMO;AAkBP,IAAM,uBAEF;AAAA,EACF,4CAA4C;AAAA,IAC1C,WAAW;AAAA,EACb;AAAA,EACA,gDAAgD;AAAA,IAC9C,WAAW;AAAA,EACb;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;AA/DV;AAgEE,WAAQ,0BAAqB,OAAO,MAA5B,mBAA+B,WAAW;AAAA,IAChD,KAAK;AACH,aAAO,GAAG,OAAO,qBAAqB,OAAO;AAAA,IAC/C,KAAK;AAAA,IACL;AACE,aAAO,GAAG,OAAO,cAAc,OAAO;AAAA,EAC1C;AACF;AASA,IAAM,8BACJ,MACA,OAAO,EAAE,UAAU,KAAK,kBAAkB,MAAM;AAC9C,QAAM,sBAAkB,8CAAuB,QAAQ;AAEvD,MAAI,CAAC,SAAS,MAAM;AAClB,UAAM,IAAI,6BAAa;AAAA,MACrB,SAAS;AAAA,MACT;AAAA,MACA;AAAA,MACA,YAAY,SAAS;AAAA,MACrB;AAAA,MACA,cAAc;AAAA,IAChB,CAAC;AAAA,EACH;AAEA,MAAI;AACF,UAAM,SAAS,MAAM,SAAS,YAAY;AAC1C,WAAO;AAAA,MACL;AAAA,MACA,OAAO;AAAA,IACT;AAAA,EACF,SAAS,OAAO;AACd,UAAM,IAAI,6BAAa;AAAA,MACrB,SAAS;AAAA,MACT;AAAA,MACA;AAAA,MACA,YAAY,SAAS;AAAA,MACrB;AAAA,MACA,cAAc;AAAA,MACd,OAAO;AAAA,IACT,CAAC;AAAA,EACH;AACF;AAEF,IAAM,iCAAgE,OAAO;AAAA,EAC3E;AAAA,EACA;AAAA,EACA;AACF,MAAM;AACJ,QAAM,sBAAkB,8CAAuB,QAAQ;AACvD,QAAM,eAAe,MAAM,SAAS,KAAK;AAEzC,SAAO;AAAA,IACL;AAAA,IACA,OAAO,IAAI,6BAAa;AAAA,MACtB,SAAS,SAAS;AAAA,MAClB;AAAA,MACA;AAAA,MACA,YAAY,SAAS;AAAA,MACrB;AAAA,MACA;AAAA,IACF,CAAC;AAAA,EACH;AACF;AAeA,eAAe,eACb,QACsB;AAvJxB;AAwJE,QAAM,aAAY,YAAO,SAAP,mBAAa,MAAM;AACrC,QAAM,EAAE,OAAO,SAAS,IAAI,UAAM,qCAAc;AAAA,IAC9C,KAAK,eAAe,OAAO,SAAS,OAAO,OAAO;AAAA,IAClD,SAAS,OAAO;AAAA,IAChB,MAAM;AAAA,MACJ,QAAQ,OAAO;AAAA,MACf,cAAc,OAAO;AAAA,MACrB,MAAM,OAAO;AAAA,MACb,GAAI,aAAa,EAAE,OAAO,UAAU,CAAC,GAAG,QAAQ,UAAU,CAAC,EAAE;AAAA,MAC7D,IAAI,YAAO,gBAAgB,cAAvB,YAAoC,CAAC;AAAA,IAC3C;AAAA,IACA,uBAAuB;AAAA,IACvB,2BAA2B,4BAA4B;AAAA,IACvD,aAAa,OAAO;AAAA,IACpB,OAAO,OAAO;AAAA,EAChB,CAAC;AAED,SAAO;AACT;AAEO,IAAM,sBAAN,MAAkD;AAAA,EASvD,YACW,SACD,QACR;AAFS;AACD;AAVV,SAAS,uBAAuB;AAMhC,SAAS,mBAAmB;AAAA,EAKzB;AAAA,EATH,IAAI,WAAmB;AACrB,WAAO,KAAK,OAAO;AAAA,EACrB;AAAA,EASA,MAAM,WAAW;AAAA,IACf;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,EACF,GAEE;AACA,UAAM,WAA2C,CAAC;AAElD,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;AAEA,UAAM,WAAW,MAAM,eAAe;AAAA,MACpC,SAAS,KAAK,OAAO;AAAA,MACrB;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,MACA,SAAS,KAAK;AAAA,MACd;AAAA,MACA,aAAS,sCAAe,KAAK,OAAO,QAAQ,GAAG,OAAO;AAAA,MACtD;AAAA,MACA,OAAO,KAAK,OAAO;AAAA,IACrB,CAAC;AAED,WAAO,EAAE,QAAQ,CAAC,IAAI,WAAW,QAAQ,CAAC,GAAG,SAAS;AAAA,EACxD;AACF;;;ADvMA,IAAM,uBAAuB,aAAE,OAAO;AAAA,EACpC,OAAO,aAAE,OAAO;AAClB,CAAC;AAED,IAAM,0BAAsE;AAAA,EAC1E,aAAa;AAAA,EACb,gBAAgB,UAAQ,KAAK;AAC/B;AA8DA,IAAM,iBAAiB;AAEhB,SAAS,gBACd,UAAqC,CAAC,GACnB;AA7GrB;AA8GE,QAAM,cAAU,8CAAqB,aAAQ,YAAR,YAAmB,cAAc;AACtE,QAAM,aAAa,OAAO;AAAA,IACxB,eAAe,cAAU,mCAAW;AAAA,MAClC,QAAQ,QAAQ;AAAA,MAChB,yBAAyB;AAAA,MACzB,aAAa;AAAA,IACf,CAAC,CAAC;AAAA,IACF,GAAG,QAAQ;AAAA,EACb;AASA,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,CACtB,SACA,WAAkC,CAAC,MAChC;AACH,WAAO,IAAI,2DAAkC,SAAS,UAAU;AAAA,MAC9D,GAAG,qBAAqB,MAAM;AAAA,MAC9B,gBAAgB;AAAA,MAChB,6BAA6B;AAAA,IAC/B,CAAC;AAAA,EACH;AAEA,QAAM,wBAAwB,CAC5B,SACA,WAAwC,CAAC,MAEzC,IAAI,iEAAwC,SAAS,UAAU;AAAA,IAC7D,GAAG,qBAAqB,YAAY;AAAA,IACpC,gBAAgB;AAAA,EAClB,CAAC;AAEH,QAAM,2BAA2B,CAC/B,SACA,WAAuC,CAAC,MAExC,IAAI,wDAA+B,SAAS,UAAU;AAAA,IACpD,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,CACf,SACA,aACG,gBAAgB,SAAS,QAAQ;AAEtC,WAAS,kBAAkB;AAC3B,WAAS,YAAY;AACrB,WAAS,qBAAqB;AAC9B,WAAS,QAAQ;AAEjB,SAAO;AACT;AAEO,IAAM,YAAY,gBAAgB;","names":["import_provider_utils"]}

package/dist/index.mjs

@@ -9,6 +9,176 @@ import {
   withoutTrailingSlash
 } from "@ai-sdk/provider-utils";
 import { z } from "zod";
+
+// src/fireworks-image-model.ts
+import {
+  APICallError
+} from "@ai-sdk/provider";
+import {
+  combineHeaders,
+  extractResponseHeaders,
+  postJsonToApi
+} from "@ai-sdk/provider-utils";
+var modelToBackendConfig = {
+  "accounts/fireworks/models/flux-1-dev-fp8": {
+    urlFormat: "workflows"
+  },
+  "accounts/fireworks/models/flux-1-schnell-fp8": {
+    urlFormat: "workflows"
+  },
+  "accounts/fireworks/models/playground-v2-5-1024px-aesthetic": {
+    urlFormat: "image_generation",
+    supportsSize: true
+  },
+  "accounts/fireworks/models/japanese-stable-diffusion-xl": {
+    urlFormat: "image_generation",
+    supportsSize: true
+  },
+  "accounts/fireworks/models/playground-v2-1024px-aesthetic": {
+    urlFormat: "image_generation",
+    supportsSize: true
+  },
+  "accounts/fireworks/models/stable-diffusion-xl-1024-v1-0": {
+    urlFormat: "image_generation",
+    supportsSize: true
+  },
+  "accounts/fireworks/models/SSD-1B": {
+    urlFormat: "image_generation",
+    supportsSize: true
+  }
+};
+function getUrlForModel(baseUrl, modelId) {
+  var _a;
+  switch ((_a = modelToBackendConfig[modelId]) == null ? void 0 : _a.urlFormat) {
+    case "image_generation":
+      return `${baseUrl}/image_generation/${modelId}`;
+    case "workflows":
+    default:
+      return `${baseUrl}/workflows/${modelId}/text_to_image`;
+  }
+}
+var createBinaryResponseHandler = () => async ({ response, url, requestBodyValues }) => {
+  const responseHeaders = extractResponseHeaders(response);
+  if (!response.body) {
+    throw new APICallError({
+      message: "Response body is empty",
+      url,
+      requestBodyValues,
+      statusCode: response.status,
+      responseHeaders,
+      responseBody: void 0
+    });
+  }
+  try {
+    const buffer = await response.arrayBuffer();
+    return {
+      responseHeaders,
+      value: buffer
+    };
+  } catch (error) {
+    throw new APICallError({
+      message: "Failed to read response as array buffer",
+      url,
+      requestBodyValues,
+      statusCode: response.status,
+      responseHeaders,
+      responseBody: void 0,
+      cause: error
+    });
+  }
+};
+var statusCodeErrorResponseHandler = async ({
+  response,
+  url,
+  requestBodyValues
+}) => {
+  const responseHeaders = extractResponseHeaders(response);
+  const responseBody = await response.text();
+  return {
+    responseHeaders,
+    value: new APICallError({
+      message: response.statusText,
+      url,
+      requestBodyValues,
+      statusCode: response.status,
+      responseHeaders,
+      responseBody
+    })
+  };
+};
+async function postImageToApi(params) {
+  var _a, _b;
+  const splitSize = (_a = params.size) == null ? void 0 : _a.split("x");
+  const { value: response } = await postJsonToApi({
+    url: getUrlForModel(params.baseUrl, params.modelId),
+    headers: params.headers,
+    body: {
+      prompt: params.prompt,
+      aspect_ratio: params.aspectRatio,
+      seed: params.seed,
+      ...splitSize && { width: splitSize[0], height: splitSize[1] },
+      ...(_b = params.providerOptions.fireworks) != null ? _b : {}
+    },
+    failedResponseHandler: statusCodeErrorResponseHandler,
+    successfulResponseHandler: createBinaryResponseHandler(),
+    abortSignal: params.abortSignal,
+    fetch: params.fetch
+  });
+  return response;
+}
+var FireworksImageModel = class {
+  constructor(modelId, config) {
+    this.modelId = modelId;
+    this.config = config;
+    this.specificationVersion = "v1";
+    this.maxImagesPerCall = 1;
+  }
+  get provider() {
+    return this.config.provider;
+  }
+  async doGenerate({
+    prompt,
+    n,
+    size,
+    aspectRatio,
+    seed,
+    providerOptions,
+    headers,
+    abortSignal
+  }) {
+    const warnings = [];
+    const backendConfig = modelToBackendConfig[this.modelId];
+    if (!(backendConfig == null ? void 0 : backendConfig.supportsSize) && size != null) {
+      warnings.push({
+        type: "unsupported-setting",
+        setting: "size",
+        details: "This model does not support the `size` option. Use `aspectRatio` instead."
+      });
+    }
+    if ((backendConfig == null ? void 0 : backendConfig.supportsSize) && aspectRatio != null) {
+      warnings.push({
+        type: "unsupported-setting",
+        setting: "aspectRatio",
+        details: "This model does not support the `aspectRatio` option."
+      });
+    }
+    const response = await postImageToApi({
+      baseUrl: this.config.baseURL,
+      prompt,
+      aspectRatio,
+      size,
+      seed,
+      modelId: this.modelId,
+      providerOptions,
+      headers: combineHeaders(this.config.headers(), headers),
+      abortSignal,
+      fetch: this.config.fetch
+    });
+    return { images: [new Uint8Array(response)], warnings };
+  }
+};
+
+// src/fireworks-provider.ts
 var fireworksErrorSchema = z.object({
   error: z.string()
 });
@@ -16,11 +186,10 @@ var fireworksErrorStructure = {
   errorSchema: fireworksErrorSchema,
   errorToMessage: (data) => data.error
 };
+var defaultBaseURL = "https://api.fireworks.ai/inference/v1";
 function createFireworks(options = {}) {
   var _a;
-  const baseURL = withoutTrailingSlash(
-    (_a = options.baseURL) != null ? _a : "https://api.fireworks.ai/inference/v1"
-  );
+  const baseURL = withoutTrailingSlash((_a = options.baseURL) != null ? _a : defaultBaseURL);
   const getHeaders = () => ({
     Authorization: `Bearer ${loadApiKey({
       apiKey: options.apiKey,
@@ -33,29 +202,32 @@ function createFireworks(options = {}) {
     provider: `fireworks.${modelType}`,
     url: ({ path }) => `${baseURL}${path}`,
     headers: getHeaders,
-    fetch: options.fetch,
-    errorStructure: fireworksErrorStructure
+    fetch: options.fetch
   });
   const createChatModel = (modelId, settings = {}) => {
     return new OpenAICompatibleChatLanguageModel(modelId, settings, {
       ...getCommonModelConfig("chat"),
+      errorStructure: fireworksErrorStructure,
       defaultObjectGenerationMode: "json"
     });
   };
-  const createCompletionModel = (modelId, settings = {}) => new OpenAICompatibleCompletionLanguageModel(
-    modelId,
-    settings,
-    getCommonModelConfig("completion")
-  );
-  const createTextEmbeddingModel = (modelId, settings = {}) => new OpenAICompatibleEmbeddingModel(
-    modelId,
-    settings,
-    getCommonModelConfig("embedding")
-  );
+  const createCompletionModel = (modelId, settings = {}) => new OpenAICompatibleCompletionLanguageModel(modelId, settings, {
+    ...getCommonModelConfig("completion"),
+    errorStructure: fireworksErrorStructure
+  });
+  const createTextEmbeddingModel = (modelId, settings = {}) => new OpenAICompatibleEmbeddingModel(modelId, settings, {
+    ...getCommonModelConfig("embedding"),
+    errorStructure: fireworksErrorStructure
+  });
+  const createImageModel = (modelId) => new FireworksImageModel(modelId, {
+    ...getCommonModelConfig("image"),
+    baseURL: baseURL != null ? baseURL : defaultBaseURL
+  });
   const provider = (modelId, settings) => createChatModel(modelId, settings);
   provider.completionModel = createCompletionModel;
   provider.chatModel = createChatModel;
   provider.textEmbeddingModel = createTextEmbeddingModel;
+  provider.image = createImageModel;
   return provider;
 }
 var fireworks = createFireworks();

package/dist/index.mjs.map

@@ -1 +1 @@
-{"version":3,"sources":["../src/fireworks-provider.ts"],"sourcesContent":["import {\n  OpenAICompatibleChatLanguageModel,\n  OpenAICompatibleCompletionLanguageModel,\n  OpenAICompatibleEmbeddingModel,\n  ProviderErrorStructure,\n} from '@ai-sdk/openai-compatible';\nimport { EmbeddingModelV1, LanguageModelV1 } from '@ai-sdk/provider';\nimport {\n  FetchFunction,\n  loadApiKey,\n  withoutTrailingSlash,\n} from '@ai-sdk/provider-utils';\nimport { z } from 'zod';\nimport {\n  FireworksChatModelId,\n  FireworksChatSettings,\n} from './fireworks-chat-settings';\nimport {\n  FireworksCompletionModelId,\n  FireworksCompletionSettings,\n} from './fireworks-completion-settings';\nimport {\n  FireworksEmbeddingModelId,\n  FireworksEmbeddingSettings,\n} from './fireworks-embedding-settings';\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  /**\nFireworks API key.\n*/\n  apiKey?: string;\n  /**\nBase URL for the API calls.\n*/\n  baseURL?: string;\n  /**\nCustom headers to include in the requests.\n*/\n  headers?: Record<string, string>;\n  /**\nCustom fetch implementation. You can use it as a middleware to intercept requests,\nor to provide a custom fetch implementation for e.g. testing.\n*/\n  fetch?: FetchFunction;\n}\n\nexport interface FireworksProvider {\n  /**\nCreates a model for text generation.\n*/\n  (\n    modelId: FireworksChatModelId,\n    settings?: FireworksChatSettings,\n  ): LanguageModelV1;\n\n  /**\nCreates a chat model for text generation.\n*/\n  chatModel(\n    modelId: FireworksChatModelId,\n    settings?: FireworksChatSettings,\n  ): LanguageModelV1;\n\n  /**\nCreates a completion model for text generation.\n*/\n  completionModel(\n    modelId: FireworksCompletionModelId,\n    settings?: FireworksCompletionSettings,\n  ): LanguageModelV1;\n\n  /**\nCreates a text embedding model for text generation.\n*/\n  textEmbeddingModel(\n    modelId: FireworksEmbeddingModelId,\n    settings?: FireworksEmbeddingSettings,\n  ): EmbeddingModelV1<string>;\n}\n\nexport function createFireworks(\n  options: FireworksProviderSettings = {},\n): FireworksProvider {\n  const baseURL = withoutTrailingSlash(\n    options.baseURL ?? 'https://api.fireworks.ai/inference/v1',\n  );\n  const getHeaders = () => ({\n    Authorization: `Bearer ${loadApiKey({\n      apiKey: options.apiKey,\n      environmentVariableName: 'FIREWORKS_API_KEY',\n      description: 'Fireworks API key',\n    })}`,\n    ...options.headers,\n  });\n\n  interface CommonModelConfig {\n    provider: `fireworks.${string}`;\n    url: ({ path }: { path: string }) => string;\n    headers: () => Record<string, string>;\n    fetch?: FetchFunction;\n    errorStructure?: ProviderErrorStructure<FireworksErrorData>;\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    errorStructure: fireworksErrorStructure,\n  });\n\n  const createChatModel = (\n    modelId: FireworksChatModelId,\n    settings: FireworksChatSettings = {},\n  ) => {\n    return new OpenAICompatibleChatLanguageModel(modelId, settings, {\n      ...getCommonModelConfig('chat'),\n      defaultObjectGenerationMode: 'json',\n    });\n  };\n\n  const createCompletionModel = (\n    modelId: FireworksCompletionModelId,\n    settings: FireworksCompletionSettings = {},\n  ) =>\n    new OpenAICompatibleCompletionLanguageModel(\n      modelId,\n      settings,\n      getCommonModelConfig('completion'),\n    );\n\n  const createTextEmbeddingModel = (\n    modelId: FireworksEmbeddingModelId,\n    settings: FireworksEmbeddingSettings = {},\n  ) =>\n    new OpenAICompatibleEmbeddingModel(\n      modelId,\n      settings,\n      getCommonModelConfig('embedding'),\n    );\n\n  const provider = (\n    modelId: FireworksChatModelId,\n    settings?: FireworksChatSettings,\n  ) => createChatModel(modelId, settings);\n\n  provider.completionModel = createCompletionModel;\n  provider.chatModel = createChatModel;\n  provider.textEmbeddingModel = createTextEmbeddingModel;\n\n  return provider as FireworksProvider;\n}\n\nexport const fireworks = createFireworks();\n"],"mappings":";AAAA;AAAA,EACE;AAAA,EACA;AAAA,EACA;AAAA,OAEK;AAEP;AAAA,EAEE;AAAA,EACA;AAAA,OACK;AACP,SAAS,SAAS;AAgBlB,IAAM,uBAAuB,EAAE,OAAO;AAAA,EACpC,OAAO,EAAE,OAAO;AAClB,CAAC;AAED,IAAM,0BAAsE;AAAA,EAC1E,aAAa;AAAA,EACb,gBAAgB,UAAQ,KAAK;AAC/B;AAwDO,SAAS,gBACd,UAAqC,CAAC,GACnB;AA7FrB;AA8FE,QAAM,UAAU;AAAA,KACd,aAAQ,YAAR,YAAmB;AAAA,EACrB;AACA,QAAM,aAAa,OAAO;AAAA,IACxB,eAAe,UAAU,WAAW;AAAA,MAClC,QAAQ,QAAQ;AAAA,MAChB,yBAAyB;AAAA,MACzB,aAAa;AAAA,IACf,CAAC,CAAC;AAAA,IACF,GAAG,QAAQ;AAAA,EACb;AAUA,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,IACf,gBAAgB;AAAA,EAClB;AAEA,QAAM,kBAAkB,CACtB,SACA,WAAkC,CAAC,MAChC;AACH,WAAO,IAAI,kCAAkC,SAAS,UAAU;AAAA,MAC9D,GAAG,qBAAqB,MAAM;AAAA,MAC9B,6BAA6B;AAAA,IAC/B,CAAC;AAAA,EACH;AAEA,QAAM,wBAAwB,CAC5B,SACA,WAAwC,CAAC,MAEzC,IAAI;AAAA,IACF;AAAA,IACA;AAAA,IACA,qBAAqB,YAAY;AAAA,EACnC;AAEF,QAAM,2BAA2B,CAC/B,SACA,WAAuC,CAAC,MAExC,IAAI;AAAA,IACF;AAAA,IACA;AAAA,IACA,qBAAqB,WAAW;AAAA,EAClC;AAEF,QAAM,WAAW,CACf,SACA,aACG,gBAAgB,SAAS,QAAQ;AAEtC,WAAS,kBAAkB;AAC3B,WAAS,YAAY;AACrB,WAAS,qBAAqB;AAE9B,SAAO;AACT;AAEO,IAAM,YAAY,gBAAgB;","names":[]}
+{"version":3,"sources":["../src/fireworks-provider.ts","../src/fireworks-image-model.ts"],"sourcesContent":["import {\n  OpenAICompatibleChatLanguageModel,\n  OpenAICompatibleCompletionLanguageModel,\n  OpenAICompatibleEmbeddingModel,\n  ProviderErrorStructure,\n} from '@ai-sdk/openai-compatible';\nimport {\n  EmbeddingModelV1,\n  ImageModelV1,\n  LanguageModelV1,\n} from '@ai-sdk/provider';\nimport {\n  FetchFunction,\n  loadApiKey,\n  withoutTrailingSlash,\n} from '@ai-sdk/provider-utils';\nimport { z } from 'zod';\nimport {\n  FireworksChatModelId,\n  FireworksChatSettings,\n} from './fireworks-chat-settings';\nimport {\n  FireworksCompletionModelId,\n  FireworksCompletionSettings,\n} from './fireworks-completion-settings';\nimport {\n  FireworksEmbeddingModelId,\n  FireworksEmbeddingSettings,\n} from './fireworks-embedding-settings';\nimport {\n  FireworksImageModel,\n  FireworksImageModelId,\n} from './fireworks-image-model';\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  /**\nFireworks API key. Default value is taken from the `FIREWORKS_API_KEY`\nenvironment variable.\n*/\n  apiKey?: string;\n  /**\nBase URL for the API calls.\n*/\n  baseURL?: string;\n  /**\nCustom headers to include in the requests.\n*/\n  headers?: Record<string, string>;\n  /**\nCustom fetch implementation. You can use it as a middleware to intercept requests,\nor to provide a custom fetch implementation for e.g. testing.\n*/\n  fetch?: FetchFunction;\n}\n\nexport interface FireworksProvider {\n  /**\nCreates a model for text generation.\n*/\n  (\n    modelId: FireworksChatModelId,\n    settings?: FireworksChatSettings,\n  ): LanguageModelV1;\n\n  /**\nCreates a chat model for text generation.\n*/\n  chatModel(\n    modelId: FireworksChatModelId,\n    settings?: FireworksChatSettings,\n  ): LanguageModelV1;\n\n  /**\nCreates a completion model for text generation.\n*/\n  completionModel(\n    modelId: FireworksCompletionModelId,\n    settings?: FireworksCompletionSettings,\n  ): LanguageModelV1;\n\n  /**\nCreates a text embedding model for text generation.\n*/\n  textEmbeddingModel(\n    modelId: FireworksEmbeddingModelId,\n    settings?: FireworksEmbeddingSettings,\n  ): EmbeddingModelV1<string>;\n\n  /**\nCreates a model for image generation.\n*/\n  image(modelId: FireworksImageModelId): ImageModelV1;\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    Authorization: `Bearer ${loadApiKey({\n      apiKey: options.apiKey,\n      environmentVariableName: 'FIREWORKS_API_KEY',\n      description: 'Fireworks API key',\n    })}`,\n    ...options.headers,\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 = (\n    modelId: FireworksChatModelId,\n    settings: FireworksChatSettings = {},\n  ) => {\n    return new OpenAICompatibleChatLanguageModel(modelId, settings, {\n      ...getCommonModelConfig('chat'),\n      errorStructure: fireworksErrorStructure,\n      defaultObjectGenerationMode: 'json',\n    });\n  };\n\n  const createCompletionModel = (\n    modelId: FireworksCompletionModelId,\n    settings: FireworksCompletionSettings = {},\n  ) =>\n    new OpenAICompatibleCompletionLanguageModel(modelId, settings, {\n      ...getCommonModelConfig('completion'),\n      errorStructure: fireworksErrorStructure,\n    });\n\n  const createTextEmbeddingModel = (\n    modelId: FireworksEmbeddingModelId,\n    settings: FireworksEmbeddingSettings = {},\n  ) =>\n    new OpenAICompatibleEmbeddingModel(modelId, settings, {\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 = (\n    modelId: FireworksChatModelId,\n    settings?: FireworksChatSettings,\n  ) => createChatModel(modelId, settings);\n\n  provider.completionModel = createCompletionModel;\n  provider.chatModel = createChatModel;\n  provider.textEmbeddingModel = createTextEmbeddingModel;\n  provider.image = createImageModel;\n\n  return provider as FireworksProvider;\n}\n\nexport const fireworks = createFireworks();\n","import {\n  APICallError,\n  ImageModelV1,\n  ImageModelV1CallWarning,\n} from '@ai-sdk/provider';\nimport {\n  combineHeaders,\n  extractResponseHeaders,\n  FetchFunction,\n  postJsonToApi,\n  ResponseHandler,\n} from '@ai-sdk/provider-utils';\n\n// https://fireworks.ai/models?type=image\nexport type FireworksImageModelId =\n  | 'accounts/fireworks/models/flux-1-dev-fp8'\n  | 'accounts/fireworks/models/flux-1-schnell-fp8'\n  | 'accounts/fireworks/models/playground-v2-5-1024px-aesthetic'\n  | 'accounts/fireworks/models/japanese-stable-diffusion-xl'\n  | 'accounts/fireworks/models/playground-v2-1024px-aesthetic'\n  | 'accounts/fireworks/models/SSD-1B'\n  | 'accounts/fireworks/models/stable-diffusion-xl-1024-v1-0'\n  | (string & {});\n\ninterface FireworksImageModelBackendConfig {\n  urlFormat: 'workflows' | 'image_generation';\n  supportsSize?: 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/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  switch (modelToBackendConfig[modelId]?.urlFormat) {\n    case 'image_generation':\n      return `${baseUrl}/image_generation/${modelId}`;\n    case 'workflows':\n    default:\n      return `${baseUrl}/workflows/${modelId}/text_to_image`;\n  }\n}\n\ninterface FireworksImageModelConfig {\n  provider: string;\n  baseURL: string;\n  headers: () => Record<string, string>;\n  fetch?: FetchFunction;\n}\n\nconst createBinaryResponseHandler =\n  (): ResponseHandler<ArrayBuffer> =>\n  async ({ response, url, requestBodyValues }) => {\n    const responseHeaders = extractResponseHeaders(response);\n\n    if (!response.body) {\n      throw new APICallError({\n        message: 'Response body is empty',\n        url,\n        requestBodyValues,\n        statusCode: response.status,\n        responseHeaders,\n        responseBody: undefined,\n      });\n    }\n\n    try {\n      const buffer = await response.arrayBuffer();\n      return {\n        responseHeaders,\n        value: buffer,\n      };\n    } catch (error) {\n      throw new APICallError({\n        message: 'Failed to read response as array buffer',\n        url,\n        requestBodyValues,\n        statusCode: response.status,\n        responseHeaders,\n        responseBody: undefined,\n        cause: error,\n      });\n    }\n  };\n\nconst statusCodeErrorResponseHandler: ResponseHandler<APICallError> = async ({\n  response,\n  url,\n  requestBodyValues,\n}) => {\n  const responseHeaders = extractResponseHeaders(response);\n  const responseBody = await response.text();\n\n  return {\n    responseHeaders,\n    value: new APICallError({\n      message: response.statusText,\n      url,\n      requestBodyValues: requestBodyValues as Record<string, unknown>,\n      statusCode: response.status,\n      responseHeaders,\n      responseBody,\n    }),\n  };\n};\n\ninterface ImageRequestParams {\n  baseUrl: string;\n  modelId: FireworksImageModelId;\n  prompt: string;\n  aspectRatio?: string;\n  size?: string;\n  seed?: number;\n  providerOptions: Record<string, unknown>;\n  headers: Record<string, string | undefined>;\n  abortSignal?: AbortSignal;\n  fetch?: FetchFunction;\n}\n\nasync function postImageToApi(\n  params: ImageRequestParams,\n): Promise<ArrayBuffer> {\n  const splitSize = params.size?.split('x');\n  const { value: response } = await postJsonToApi({\n    url: getUrlForModel(params.baseUrl, params.modelId),\n    headers: params.headers,\n    body: {\n      prompt: params.prompt,\n      aspect_ratio: params.aspectRatio,\n      seed: params.seed,\n      ...(splitSize && { width: splitSize[0], height: splitSize[1] }),\n      ...(params.providerOptions.fireworks ?? {}),\n    },\n    failedResponseHandler: statusCodeErrorResponseHandler,\n    successfulResponseHandler: createBinaryResponseHandler(),\n    abortSignal: params.abortSignal,\n    fetch: params.fetch,\n  });\n\n  return response;\n}\n\nexport class FireworksImageModel implements ImageModelV1 {\n  readonly specificationVersion = 'v1';\n\n  get provider(): string {\n    return this.config.provider;\n  }\n\n  readonly maxImagesPerCall = 1;\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  }: Parameters<ImageModelV1['doGenerate']>[0]): Promise<\n    Awaited<ReturnType<ImageModelV1['doGenerate']>>\n  > {\n    const warnings: Array<ImageModelV1CallWarning> = [];\n\n    const backendConfig = modelToBackendConfig[this.modelId];\n    if (!backendConfig?.supportsSize && size != null) {\n      warnings.push({\n        type: 'unsupported-setting',\n        setting: '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-setting',\n        setting: 'aspectRatio',\n        details: 'This model does not support the `aspectRatio` option.',\n      });\n    }\n\n    const response = await postImageToApi({\n      baseUrl: this.config.baseURL,\n      prompt,\n      aspectRatio,\n      size,\n      seed,\n      modelId: this.modelId,\n      providerOptions,\n      headers: combineHeaders(this.config.headers(), headers),\n      abortSignal,\n      fetch: this.config.fetch,\n    });\n\n    return { images: [new Uint8Array(response)], warnings };\n  }\n}\n"],"mappings":";AAAA;AAAA,EACE;AAAA,EACA;AAAA,EACA;AAAA,OAEK;AAMP;AAAA,EAEE;AAAA,EACA;AAAA,OACK;AACP,SAAS,SAAS;;;AChBlB;AAAA,EACE;AAAA,OAGK;AACP;AAAA,EACE;AAAA,EACA;AAAA,EAEA;AAAA,OAEK;AAkBP,IAAM,uBAEF;AAAA,EACF,4CAA4C;AAAA,IAC1C,WAAW;AAAA,EACb;AAAA,EACA,gDAAgD;AAAA,IAC9C,WAAW;AAAA,EACb;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;AA/DV;AAgEE,WAAQ,0BAAqB,OAAO,MAA5B,mBAA+B,WAAW;AAAA,IAChD,KAAK;AACH,aAAO,GAAG,OAAO,qBAAqB,OAAO;AAAA,IAC/C,KAAK;AAAA,IACL;AACE,aAAO,GAAG,OAAO,cAAc,OAAO;AAAA,EAC1C;AACF;AASA,IAAM,8BACJ,MACA,OAAO,EAAE,UAAU,KAAK,kBAAkB,MAAM;AAC9C,QAAM,kBAAkB,uBAAuB,QAAQ;AAEvD,MAAI,CAAC,SAAS,MAAM;AAClB,UAAM,IAAI,aAAa;AAAA,MACrB,SAAS;AAAA,MACT;AAAA,MACA;AAAA,MACA,YAAY,SAAS;AAAA,MACrB;AAAA,MACA,cAAc;AAAA,IAChB,CAAC;AAAA,EACH;AAEA,MAAI;AACF,UAAM,SAAS,MAAM,SAAS,YAAY;AAC1C,WAAO;AAAA,MACL;AAAA,MACA,OAAO;AAAA,IACT;AAAA,EACF,SAAS,OAAO;AACd,UAAM,IAAI,aAAa;AAAA,MACrB,SAAS;AAAA,MACT;AAAA,MACA;AAAA,MACA,YAAY,SAAS;AAAA,MACrB;AAAA,MACA,cAAc;AAAA,MACd,OAAO;AAAA,IACT,CAAC;AAAA,EACH;AACF;AAEF,IAAM,iCAAgE,OAAO;AAAA,EAC3E;AAAA,EACA;AAAA,EACA;AACF,MAAM;AACJ,QAAM,kBAAkB,uBAAuB,QAAQ;AACvD,QAAM,eAAe,MAAM,SAAS,KAAK;AAEzC,SAAO;AAAA,IACL;AAAA,IACA,OAAO,IAAI,aAAa;AAAA,MACtB,SAAS,SAAS;AAAA,MAClB;AAAA,MACA;AAAA,MACA,YAAY,SAAS;AAAA,MACrB;AAAA,MACA;AAAA,IACF,CAAC;AAAA,EACH;AACF;AAeA,eAAe,eACb,QACsB;AAvJxB;AAwJE,QAAM,aAAY,YAAO,SAAP,mBAAa,MAAM;AACrC,QAAM,EAAE,OAAO,SAAS,IAAI,MAAM,cAAc;AAAA,IAC9C,KAAK,eAAe,OAAO,SAAS,OAAO,OAAO;AAAA,IAClD,SAAS,OAAO;AAAA,IAChB,MAAM;AAAA,MACJ,QAAQ,OAAO;AAAA,MACf,cAAc,OAAO;AAAA,MACrB,MAAM,OAAO;AAAA,MACb,GAAI,aAAa,EAAE,OAAO,UAAU,CAAC,GAAG,QAAQ,UAAU,CAAC,EAAE;AAAA,MAC7D,IAAI,YAAO,gBAAgB,cAAvB,YAAoC,CAAC;AAAA,IAC3C;AAAA,IACA,uBAAuB;AAAA,IACvB,2BAA2B,4BAA4B;AAAA,IACvD,aAAa,OAAO;AAAA,IACpB,OAAO,OAAO;AAAA,EAChB,CAAC;AAED,SAAO;AACT;AAEO,IAAM,sBAAN,MAAkD;AAAA,EASvD,YACW,SACD,QACR;AAFS;AACD;AAVV,SAAS,uBAAuB;AAMhC,SAAS,mBAAmB;AAAA,EAKzB;AAAA,EATH,IAAI,WAAmB;AACrB,WAAO,KAAK,OAAO;AAAA,EACrB;AAAA,EASA,MAAM,WAAW;AAAA,IACf;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,EACF,GAEE;AACA,UAAM,WAA2C,CAAC;AAElD,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;AAEA,UAAM,WAAW,MAAM,eAAe;AAAA,MACpC,SAAS,KAAK,OAAO;AAAA,MACrB;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,MACA,SAAS,KAAK;AAAA,MACd;AAAA,MACA,SAAS,eAAe,KAAK,OAAO,QAAQ,GAAG,OAAO;AAAA,MACtD;AAAA,MACA,OAAO,KAAK,OAAO;AAAA,IACrB,CAAC;AAED,WAAO,EAAE,QAAQ,CAAC,IAAI,WAAW,QAAQ,CAAC,GAAG,SAAS;AAAA,EACxD;AACF;;;ADvMA,IAAM,uBAAuB,EAAE,OAAO;AAAA,EACpC,OAAO,EAAE,OAAO;AAClB,CAAC;AAED,IAAM,0BAAsE;AAAA,EAC1E,aAAa;AAAA,EACb,gBAAgB,UAAQ,KAAK;AAC/B;AA8DA,IAAM,iBAAiB;AAEhB,SAAS,gBACd,UAAqC,CAAC,GACnB;AA7GrB;AA8GE,QAAM,UAAU,sBAAqB,aAAQ,YAAR,YAAmB,cAAc;AACtE,QAAM,aAAa,OAAO;AAAA,IACxB,eAAe,UAAU,WAAW;AAAA,MAClC,QAAQ,QAAQ;AAAA,MAChB,yBAAyB;AAAA,MACzB,aAAa;AAAA,IACf,CAAC,CAAC;AAAA,IACF,GAAG,QAAQ;AAAA,EACb;AASA,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,CACtB,SACA,WAAkC,CAAC,MAChC;AACH,WAAO,IAAI,kCAAkC,SAAS,UAAU;AAAA,MAC9D,GAAG,qBAAqB,MAAM;AAAA,MAC9B,gBAAgB;AAAA,MAChB,6BAA6B;AAAA,IAC/B,CAAC;AAAA,EACH;AAEA,QAAM,wBAAwB,CAC5B,SACA,WAAwC,CAAC,MAEzC,IAAI,wCAAwC,SAAS,UAAU;AAAA,IAC7D,GAAG,qBAAqB,YAAY;AAAA,IACpC,gBAAgB;AAAA,EAClB,CAAC;AAEH,QAAM,2BAA2B,CAC/B,SACA,WAAuC,CAAC,MAExC,IAAI,+BAA+B,SAAS,UAAU;AAAA,IACpD,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,CACf,SACA,aACG,gBAAgB,SAAS,QAAQ;AAEtC,WAAS,kBAAkB;AAC3B,WAAS,YAAY;AACrB,WAAS,qBAAqB;AAC9B,WAAS,QAAQ;AAEjB,SAAO;AACT;AAEO,IAAM,YAAY,gBAAgB;","names":[]}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@ai-sdk/fireworks",
-  "version": "0.0.8",
+  "version": "0.0.10",
   "license": "Apache-2.0",
   "sideEffects": false,
   "main": "./dist/index.js",
@@ -19,9 +19,9 @@
     }
   },
   "dependencies": {
-    "@ai-sdk/openai-compatible": "0.0.13",
-    "@ai-sdk/provider": "1.0.3",
-    "@ai-sdk/provider-utils": "2.0.5"
+    "@ai-sdk/openai-compatible": "0.0.15",
+    "@ai-sdk/provider": "1.0.4",
+    "@ai-sdk/provider-utils": "2.0.7"
   },
   "devDependencies": {
     "@types/node": "^18",