@ai-sdk/fireworks - npm Package Compare versions

Comparing version 0.0.11 to 0.0.12

CHANGELOG.md

 # @ai-sdk/fireworks
 
+## 0.0.12
+
+### Patch Changes
+
+- 218d001: feat (provider): Add maxImagesPerCall setting to all image providers.
+- 971be76: feat (provider/fireworks): Send generateImage n parameter to Fireworks API as 'samples'.
+
 ## 0.0.11
@@ -4,0 +11,0 @@

dist/index.d.ts

@@ -19,2 +19,8 @@ import { LanguageModelV1, EmbeddingModelV1, ImageModelV1 } from '@ai-sdk/provider';
 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 & {});
+interface FireworksImageSettings {
+    /**
+  Override the maximum number of images per call (default 1)
+     */
+    maxImagesPerCall?: number;
+}
 
@@ -69,3 +75,3 @@ type FireworksErrorData = z.infer<typeof fireworksErrorSchema>;
   */
-    image(modelId: FireworksImageModelId): ImageModelV1;
+    image(modelId: FireworksImageModelId, settings?: FireworksImageSettings): ImageModelV1;
 }
@@ -72,0 +78,0 @@ declare function createFireworks(options?: FireworksProviderSettings): FireworksProvider;

dist/index.js

@@ -123,28 +123,8 @@ "use strict";
 };
-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) {
+  constructor(modelId, settings, config) {
     this.modelId = modelId;
+    this.settings = settings;
     this.config = config;
     this.specificationVersion = "v1";
-    this.maxImagesPerCall = 1;
   }
@@ -154,2 +134,6 @@ get provider() {
   }
+  get maxImagesPerCall() {
+    var _a;
+    return (_a = this.settings.maxImagesPerCall) != null ? _a : 1;
+  }
   async doGenerate({
@@ -165,2 +149,3 @@ prompt,
   }) {
+    var _a;
     const warnings = [];
@@ -182,11 +167,16 @@ const backendConfig = modelToBackendConfig[this.modelId];
     }
-    const response = await postImageToApi({
-      baseUrl: this.config.baseURL,
-      prompt,
-      aspectRatio,
-      size,
-      seed,
-      modelId: this.modelId,
-      providerOptions,
+    const splitSize = size == null ? void 0 : size.split("x");
+    const { value: response } = await (0, import_provider_utils.postJsonToApi)({
+      url: getUrlForModel(this.config.baseURL, this.modelId),
       headers: (0, import_provider_utils.combineHeaders)(this.config.headers(), headers),
+      body: {
+        prompt,
+        aspect_ratio: aspectRatio,
+        seed,
+        samples: n,
+        ...splitSize && { width: splitSize[0], height: splitSize[1] },
+        ...(_a = providerOptions.fireworks) != null ? _a : {}
+      },
+      failedResponseHandler: statusCodeErrorResponseHandler,
+      successfulResponseHandler: createBinaryResponseHandler(),
       abortSignal,
@@ -240,3 +230,3 @@ fetch: this.config.fetch
   });
-  const createImageModel = (modelId) => new FireworksImageModel(modelId, {
+  const createImageModel = (modelId, settings = {}) => new FireworksImageModel(modelId, settings, {
     ...getCommonModelConfig("image"),
@@ -243,0 +233,0 @@ baseURL: baseURL != null ? baseURL : defaultBaseURL

package.json

 {
   "name": "@ai-sdk/fireworks",
-  "version": "0.0.11",
+  "version": "0.0.12",
   "license": "Apache-2.0",
@@ -5,0 +5,0 @@ "sideEffects": false,

No alert changes

Worsened metrics

Total package byte prevSize

58131

decreased by-3.13%

Lines of code

539

decreased by-2.53%

No dependency changes