feat(fireworks): add FireworksImageModelOptions type with providerOptions validation

.changeset/fireworks-image-model-options.md

@@ -0,0 +1,5 @@
+---
+'@ai-sdk/fireworks': patch
+---
+
+Add `FireworksImageModelOptions` type and validate image generation provider options using Zod schema with `parseProviderOptions`.

examples/ai-functions/src/generate-image/fireworks.ts

@@ -1,4 +1,5 @@
 import { fireworks } from '@ai-sdk/fireworks';
+import type { FireworksImageModelOptions } from '@ai-sdk/fireworks';
 import { generateImage } from 'ai';
 import { presentImages } from '../lib/present-image';
 import { run } from '../lib/run';
@@ -17,7 +18,7 @@ run(async () => {
         // https://fireworks.ai/models/fireworks/stable-diffusion-xl-1024-v1-0/playground
         cfg_scale: 10,
         steps: 30,
-      },
+      } satisfies FireworksImageModelOptions,
     },
   });
 

packages/fireworks/src/fireworks-image-model.test.ts

@@ -64,15 +64,15 @@ describe('FireworksImageModel', () => {
         size: undefined,
         aspectRatio: '16:9',
         seed: 42,
-        providerOptions: { fireworks: { additional_param: 'value' } },
+        providerOptions: { fireworks: { cfg_scale: 10 } },
       });
 
       expect(await server.calls[0].requestBodyJson).toStrictEqual({
         prompt,
         aspect_ratio: '16:9',
         seed: 42,
         samples: 1,
-        additional_param: 'value',
+        cfg_scale: 10,
       });
     });
 

packages/fireworks/src/fireworks-image-model.ts

@@ -5,9 +5,13 @@ import {
   createBinaryResponseHandler,
   createStatusCodeErrorResponseHandler,
   FetchFunction,
+  parseProviderOptions,
   postJsonToApi,
 } from '@ai-sdk/provider-utils';
-import { FireworksImageModelId } from './fireworks-image-options';
+import {
+  FireworksImageModelId,
+  fireworksImageModelOptions,
+} from './fireworks-image-options';
 
 interface FireworksImageModelBackendConfig {
   urlFormat: 'workflows' | 'workflows_edit' | 'image_generation';
@@ -163,20 +167,52 @@ export class FireworksImageModel implements ImageModelV3 {
       });
     }
 
+    const fireworksOptions = await parseProviderOptions({
+      provider: 'fireworks',
+      providerOptions,
+      schema: fireworksImageModelOptions,
+    });
+
     const splitSize = size?.split('x');
     const currentDate = this.config._internal?.currentDate?.() ?? new Date();
+
+    const body: Record<string, unknown> = {
+      prompt,
+      aspect_ratio: aspectRatio,
+      seed,
+      samples: n,
+      ...(inputImage && { input_image: inputImage }),
+      ...(splitSize && { width: splitSize[0], height: splitSize[1] }),
+    };
+    if (fireworksOptions?.cfg_scale != null) {
+      body.cfg_scale = fireworksOptions.cfg_scale;
+    }
+    if (fireworksOptions?.steps != null) {
+      body.steps = fireworksOptions.steps;
+    }
+    if (fireworksOptions?.negative_prompt != null) {
+      body.negative_prompt = fireworksOptions.negative_prompt;
+    }
+    if (fireworksOptions?.strength != null) {
+      body.strength = fireworksOptions.strength;
+    }
+    if (fireworksOptions?.scheduler != null) {
+      body.scheduler = fireworksOptions.scheduler;
+    }
+    if (fireworksOptions?.safety_checker != null) {
+      body.safety_checker = fireworksOptions.safety_checker;
+    }
+    if (fireworksOptions?.output_format != null) {
+      body.output_format = fireworksOptions.output_format;
+    }
+    if (fireworksOptions?.safety_tolerance != null) {
+      body.safety_tolerance = fireworksOptions.safety_tolerance;
+    }
+
     const { value: response, responseHeaders } = await postJsonToApi({
       url: getUrlForModel(this.config.baseURL, this.modelId, hasInputImage),
       headers: combineHeaders(this.config.headers(), headers),
-      body: {
-        prompt,
-        aspect_ratio: aspectRatio,
-        seed,
-        samples: n,
-        ...(inputImage && { input_image: inputImage }),
-        ...(splitSize && { width: splitSize[0], height: splitSize[1] }),
-        ...(providerOptions.fireworks ?? {}),
-      },
+      body,
       failedResponseHandler: createStatusCodeErrorResponseHandler(),
       successfulResponseHandler: createBinaryResponseHandler(),
       abortSignal,

packages/fireworks/src/fireworks-image-options.ts

@@ -1,3 +1,5 @@
+import { z } from 'zod/v4';
+
 // https://fireworks.ai/models?type=image
 export type FireworksImageModelId =
   | 'accounts/fireworks/models/flux-1-dev-fp8'
@@ -10,3 +12,19 @@ export type FireworksImageModelId =
   | 'accounts/fireworks/models/SSD-1B'
   | 'accounts/fireworks/models/stable-diffusion-xl-1024-v1-0'
   | (string & {});
+
+// https://docs.fireworks.ai/api-reference/post-imagegeneration
+export const fireworksImageModelOptions = z.object({
+  cfg_scale: z.number().optional(),
+  steps: z.number().optional(),
+  negative_prompt: z.string().optional(),
+  strength: z.number().optional(),
+  scheduler: z.string().optional(),
+  safety_checker: z.boolean().optional(),
+  output_format: z.string().optional(),
+  safety_tolerance: z.number().optional(),
+});
+
+export type FireworksImageModelOptions = z.infer<
+  typeof fireworksImageModelOptions
+>;

packages/fireworks/src/index.ts

@@ -10,7 +10,10 @@ export type {
   FireworksEmbeddingModelOptions as FireworksEmbeddingProviderOptions,
 } from './fireworks-embedding-options';
 export { FireworksImageModel } from './fireworks-image-model';
-export type { FireworksImageModelId } from './fireworks-image-options';
+export type {
+  FireworksImageModelId,
+  FireworksImageModelOptions,
+} from './fireworks-image-options';
 export { fireworks, createFireworks } from './fireworks-provider';
 export type {
   FireworksProvider,