modelfusion 0.100.0 → 0.102.0

package/model-provider/automatic1111/Automatic1111ImageGenerationModel.d.ts

@@ -6,6 +6,28 @@ import { PromptTemplate } from "../../model-function/PromptTemplate.js";
 import { ImageGenerationModel, ImageGenerationModelSettings } from "../../model-function/generate-image/ImageGenerationModel.js";
 import { PromptTemplateImageGenerationModel } from "../../model-function/generate-image/PromptTemplateImageGenerationModel.js";
 import { Automatic1111ImageGenerationPrompt } from "./Automatic1111ImageGenerationPrompt.js";
+export interface Automatic1111ImageGenerationSettings extends ImageGenerationModelSettings {
+    api?: ApiConfiguration;
+    /**
+     * Stable Diffusion checkpoint.
+     */
+    model: string;
+    height?: number;
+    width?: number;
+    /**
+     * Sampling method.
+     */
+    sampler?: string;
+    /**
+     * Sampling steps.
+     */
+    steps?: number;
+    /**
+     * CFG Scale.
+     */
+    cfgScale?: number;
+    seed?: number;
+}
 /**
  * Create an image generation model that calls the AUTOMATIC1111 Stable Diffusion Web UI API.
  *
@@ -17,26 +39,18 @@ export declare class Automatic1111ImageGenerationModel extends AbstractModel<Aut
     get modelName(): string;
     callAPI(input: Automatic1111ImageGenerationPrompt, options?: FunctionOptions): Promise<Automatic1111ImageGenerationResponse>;
     get settingsForEvent(): Partial<Automatic1111ImageGenerationSettings>;
-    doGenerateImage(prompt: Automatic1111ImageGenerationPrompt, options?: FunctionOptions): Promise<{
+    doGenerateImages(prompt: Automatic1111ImageGenerationPrompt, options?: FunctionOptions): Promise<{
         response: {
             images: string[];
             parameters: {};
             info: string;
         };
-        base64Image: string;
+        base64Images: string[];
     }>;
     withTextPrompt(): PromptTemplateImageGenerationModel<string, Automatic1111ImageGenerationPrompt, Automatic1111ImageGenerationSettings, this>;
     withPromptTemplate<INPUT_PROMPT>(promptTemplate: PromptTemplate<INPUT_PROMPT, Automatic1111ImageGenerationPrompt>): PromptTemplateImageGenerationModel<INPUT_PROMPT, Automatic1111ImageGenerationPrompt, Automatic1111ImageGenerationSettings, this>;
     withSettings(additionalSettings: Automatic1111ImageGenerationSettings): this;
 }
-export interface Automatic1111ImageGenerationSettings extends ImageGenerationModelSettings {
-    api?: ApiConfiguration;
-    model: string;
-    height?: number;
-    width?: number;
-    sampler?: string;
-    steps?: number;
-}
 declare const Automatic1111ImageGenerationResponseSchema: z.ZodObject<{
     images: z.ZodArray<z.ZodString, "many">;
     parameters: z.ZodObject<{}, "strip", z.ZodTypeAny, {}, {}>;

package/model-provider/automatic1111/Automatic1111ImageGenerationModel.js

@@ -25,13 +25,31 @@ export class Automatic1111ImageGenerationModel extends AbstractModel {
         return this.settings.model;
     }
     async callAPI(input, options) {
+        const api = this.settings.api ?? new Automatic1111ApiConfiguration();
+        const abortSignal = options?.run?.abortSignal;
         return callWithRetryAndThrottle({
-            retry: this.settings.api?.retry,
-            throttle: this.settings.api?.throttle,
-            call: async () => callAutomatic1111ImageGenerationAPI({
-                ...this.settings,
-                abortSignal: options?.run?.abortSignal,
-                prompt: input.prompt,
+            retry: api.retry,
+            throttle: api.throttle,
+            call: async () => postJsonToApi({
+                url: api.assembleUrl(`/txt2img`),
+                headers: api.headers,
+                body: {
+                    prompt: input.prompt,
+                    negative_prompt: input.negativePrompt,
+                    seed: this.settings.seed,
+                    batch_size: this.settings.numberOfGenerations,
+                    height: this.settings.height,
+                    width: this.settings.width,
+                    cfg_scale: this.settings.cfgScale,
+                    sampler_index: this.settings.sampler,
+                    steps: this.settings.steps,
+                    override_settings: {
+                        sd_model_checkpoint: this.settings.model,
+                    },
+                },
+                failedResponseHandler: failedAutomatic1111CallResponseHandler,
+                successfulResponseHandler: createJsonResponseHandler(Automatic1111ImageGenerationResponseSchema),
+                abortSignal,
             }),
         });
     }
@@ -43,11 +61,11 @@ export class Automatic1111ImageGenerationModel extends AbstractModel {
             steps: this.settings.steps,
         };
     }
-    async doGenerateImage(prompt, options) {
+    async doGenerateImages(prompt, options) {
         const response = await this.callAPI(prompt, options);
         return {
             response,
-            base64Image: response.images[0],
+            base64Images: response.images,
         };
     }
     withTextPrompt() {
@@ -68,24 +86,3 @@ const Automatic1111ImageGenerationResponseSchema = z.object({
     parameters: z.object({}),
     info: z.string(),
 });
-async function callAutomatic1111ImageGenerationAPI({ api = new Automatic1111ApiConfiguration(), abortSignal, height, width, prompt, negativePrompt, sampler, steps, seed, model, }) {
-    return postJsonToApi({
-        url: api.assembleUrl(`/txt2img`),
-        headers: api.headers,
-        body: {
-            height,
-            width,
-            prompt,
-            negative_prompt: negativePrompt,
-            sampler_index: sampler,
-            steps,
-            seed,
-            override_settings: {
-                sd_model_checkpoint: model,
-            },
-        },
-        failedResponseHandler: failedAutomatic1111CallResponseHandler,
-        successfulResponseHandler: createJsonResponseHandler(Automatic1111ImageGenerationResponseSchema),
-        abortSignal,
-    });
-}

package/model-provider/automatic1111/Automatic1111ImageGenerationPrompt.d.ts

@@ -2,7 +2,6 @@ import { PromptTemplate } from "../../model-function/PromptTemplate.js";
 export type Automatic1111ImageGenerationPrompt = {
     prompt: string;
     negativePrompt?: string;
-    seed?: number;
 };
 /**
  * Formats a basic text prompt as an Automatic1111 prompt.

package/model-provider/automatic1111/index.cjs

@@ -26,10 +26,8 @@ var __importStar = (this && this.__importStar) || function (mod) {
     return result;
 };
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.automatic1111 = exports.Automatic1111Error = void 0;
+exports.automatic1111 = void 0;
 __exportStar(require("./Automatic1111ApiConfiguration.cjs"), exports);
-var Automatic1111Error_js_1 = require("./Automatic1111Error.cjs");
-Object.defineProperty(exports, "Automatic1111Error", { enumerable: true, get: function () { return Automatic1111Error_js_1.Automatic1111Error; } });
 exports.automatic1111 = __importStar(require("./Automatic1111Facade.cjs"));
 __exportStar(require("./Automatic1111ImageGenerationModel.cjs"), exports);
 __exportStar(require("./Automatic1111ImageGenerationPrompt.cjs"), exports);

package/model-provider/automatic1111/index.d.ts

@@ -1,5 +1,5 @@
 export * from "./Automatic1111ApiConfiguration.js";
-export { Automatic1111Error, Automatic1111ErrorData, } from "./Automatic1111Error.js";
+export { Automatic1111ErrorData } from "./Automatic1111Error.js";
 export * as automatic1111 from "./Automatic1111Facade.js";
 export * from "./Automatic1111ImageGenerationModel.js";
 export * from "./Automatic1111ImageGenerationPrompt.js";

package/model-provider/automatic1111/index.js

@@ -1,5 +1,4 @@
 export * from "./Automatic1111ApiConfiguration.js";
-export { Automatic1111Error, } from "./Automatic1111Error.js";
 export * as automatic1111 from "./Automatic1111Facade.js";
 export * from "./Automatic1111ImageGenerationModel.js";
 export * from "./Automatic1111ImageGenerationPrompt.js";

package/model-provider/openai/OpenAIImageGenerationModel.cjs

@@ -60,7 +60,7 @@ const calculateOpenAIImageGenerationCostInMillicents = ({ model, settings, }) =>
     if (cost == null) {
         return null;
     }
-    return (settings.n ?? 1) * cost;
+    return (settings.numberOfGenerations ?? 1) * cost;
 };
 exports.calculateOpenAIImageGenerationCostInMillicents = calculateOpenAIImageGenerationCostInMillicents;
 /**
@@ -88,38 +88,48 @@ class OpenAIImageGenerationModel extends AbstractModel_js_1.AbstractModel {
         return this.settings.model;
     }
     async callAPI(prompt, options) {
-        const run = options?.run;
+        const api = this.settings.api ?? new OpenAIApiConfiguration_js_1.OpenAIApiConfiguration();
+        const abortSignal = options?.run?.abortSignal;
+        const userId = options?.run?.userId;
         const responseFormat = options?.responseFormat;
-        const callSettings = {
-            ...this.settings,
-            user: this.settings.isUserIdForwardingEnabled ? run?.userId : undefined,
-            abortSignal: run?.abortSignal,
-            responseFormat,
-            prompt,
-        };
         return (0, callWithRetryAndThrottle_js_1.callWithRetryAndThrottle)({
-            retry: callSettings.api?.retry,
-            throttle: callSettings.api?.throttle,
-            call: async () => callOpenAIImageGenerationAPI(callSettings),
+            retry: api.retry,
+            throttle: api.throttle,
+            call: async () => {
+                return (0, postToApi_js_1.postJsonToApi)({
+                    url: api.assembleUrl("/images/generations"),
+                    headers: api.headers,
+                    body: {
+                        prompt,
+                        n: this.settings.numberOfGenerations,
+                        size: this.settings.size,
+                        response_format: responseFormat.type,
+                        user: this.settings.isUserIdForwardingEnabled ? userId : undefined,
+                    },
+                    failedResponseHandler: OpenAIError_js_1.failedOpenAICallResponseHandler,
+                    successfulResponseHandler: responseFormat?.handler,
+                    abortSignal,
+                });
+            },
         });
     }
     get settingsForEvent() {
         const eventSettingProperties = [
-            "n",
+            "numberOfGenerations",
             "size",
             "quality",
             "style",
         ];
         return Object.fromEntries(Object.entries(this.settings).filter(([key]) => eventSettingProperties.includes(key)));
     }
-    async doGenerateImage(prompt, options) {
+    async doGenerateImages(prompt, options) {
         const response = await this.callAPI(prompt, {
             responseFormat: exports.OpenAIImageGenerationResponseFormat.base64Json,
             ...options,
         });
         return {
             response,
-            base64Image: response.data[0].b64_json,
+            base64Images: response.data.map((item) => item.b64_json),
         };
     }
     withPromptTemplate(promptTemplate) {
@@ -155,19 +165,3 @@ exports.OpenAIImageGenerationResponseFormat = {
         handler: (0, postToApi_js_1.createJsonResponseHandler)(openAIImageGenerationBase64JsonSchema),
     },
 };
-async function callOpenAIImageGenerationAPI({ api = new OpenAIApiConfiguration_js_1.OpenAIApiConfiguration(), abortSignal, prompt, n, size, responseFormat, user, }) {
-    return (0, postToApi_js_1.postJsonToApi)({
-        url: api.assembleUrl("/images/generations"),
-        headers: api.headers,
-        body: {
-            prompt,
-            n,
-            size,
-            response_format: responseFormat.type,
-            user,
-        },
-        failedResponseHandler: OpenAIError_js_1.failedOpenAICallResponseHandler,
-        successfulResponseHandler: responseFormat?.handler,
-        abortSignal,
-    });
-}

package/model-provider/openai/OpenAIImageGenerationModel.d.ts

@@ -24,7 +24,6 @@ export declare const calculateOpenAIImageGenerationCostInMillicents: ({ model, s
 export type OpenAIImageModelType = keyof typeof OPENAI_IMAGE_MODELS;
 export interface OpenAIImageGenerationCallSettings {
     model: OpenAIImageModelType;
-    n?: number;
     size?: "256x256" | "512x512" | "1024x1024" | "1792x1024" | "1024x1792";
     quality?: "standard" | "hd";
     style?: "vivid" | "natural";
@@ -52,14 +51,14 @@ export declare class OpenAIImageGenerationModel extends AbstractModel<OpenAIImag
         responseFormat: OpenAIImageGenerationResponseFormatType<RESULT>;
     } & FunctionOptions): Promise<RESULT>;
     get settingsForEvent(): Partial<OpenAIImageGenerationSettings>;
-    doGenerateImage(prompt: string, options?: FunctionOptions): Promise<{
+    doGenerateImages(prompt: string, options?: FunctionOptions): Promise<{
         response: {
             data: {
                 b64_json: string;
             }[];
             created: number;
         };
-        base64Image: string;
+        base64Images: string[];
     }>;
     withPromptTemplate<INPUT_PROMPT>(promptTemplate: PromptTemplate<INPUT_PROMPT, string>): PromptTemplateImageGenerationModel<INPUT_PROMPT, string, OpenAIImageGenerationSettings, this>;
     withSettings(additionalSettings: Partial<OpenAIImageGenerationSettings>): this;

package/model-provider/openai/OpenAIImageGenerationModel.js

@@ -57,7 +57,7 @@ export const calculateOpenAIImageGenerationCostInMillicents = ({ model, settings
     if (cost == null) {
         return null;
     }
-    return (settings.n ?? 1) * cost;
+    return (settings.numberOfGenerations ?? 1) * cost;
 };
 /**
  * Create an image generation model that calls the OpenAI AI image creation API.
@@ -84,38 +84,48 @@ export class OpenAIImageGenerationModel extends AbstractModel {
         return this.settings.model;
     }
     async callAPI(prompt, options) {
-        const run = options?.run;
+        const api = this.settings.api ?? new OpenAIApiConfiguration();
+        const abortSignal = options?.run?.abortSignal;
+        const userId = options?.run?.userId;
         const responseFormat = options?.responseFormat;
-        const callSettings = {
-            ...this.settings,
-            user: this.settings.isUserIdForwardingEnabled ? run?.userId : undefined,
-            abortSignal: run?.abortSignal,
-            responseFormat,
-            prompt,
-        };
         return callWithRetryAndThrottle({
-            retry: callSettings.api?.retry,
-            throttle: callSettings.api?.throttle,
-            call: async () => callOpenAIImageGenerationAPI(callSettings),
+            retry: api.retry,
+            throttle: api.throttle,
+            call: async () => {
+                return postJsonToApi({
+                    url: api.assembleUrl("/images/generations"),
+                    headers: api.headers,
+                    body: {
+                        prompt,
+                        n: this.settings.numberOfGenerations,
+                        size: this.settings.size,
+                        response_format: responseFormat.type,
+                        user: this.settings.isUserIdForwardingEnabled ? userId : undefined,
+                    },
+                    failedResponseHandler: failedOpenAICallResponseHandler,
+                    successfulResponseHandler: responseFormat?.handler,
+                    abortSignal,
+                });
+            },
         });
     }
     get settingsForEvent() {
         const eventSettingProperties = [
-            "n",
+            "numberOfGenerations",
             "size",
             "quality",
             "style",
         ];
         return Object.fromEntries(Object.entries(this.settings).filter(([key]) => eventSettingProperties.includes(key)));
     }
-    async doGenerateImage(prompt, options) {
+    async doGenerateImages(prompt, options) {
         const response = await this.callAPI(prompt, {
             responseFormat: OpenAIImageGenerationResponseFormat.base64Json,
             ...options,
         });
         return {
             response,
-            base64Image: response.data[0].b64_json,
+            base64Images: response.data.map((item) => item.b64_json),
         };
     }
     withPromptTemplate(promptTemplate) {
@@ -150,19 +160,3 @@ export const OpenAIImageGenerationResponseFormat = {
         handler: createJsonResponseHandler(openAIImageGenerationBase64JsonSchema),
     },
 };
-async function callOpenAIImageGenerationAPI({ api = new OpenAIApiConfiguration(), abortSignal, prompt, n, size, responseFormat, user, }) {
-    return postJsonToApi({
-        url: api.assembleUrl("/images/generations"),
-        headers: api.headers,
-        body: {
-            prompt,
-            n,
-            size,
-            response_format: responseFormat.type,
-            user,
-        },
-        failedResponseHandler: failedOpenAICallResponseHandler,
-        successfulResponseHandler: responseFormat?.handler,
-        abortSignal,
-    });
-}

package/model-provider/stability/StabilityApiConfiguration.cjs

@@ -1,13 +1,20 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.StabilityApiConfiguration = void 0;
-const BaseUrlApiConfiguration_js_1 = require("../../core/api/BaseUrlApiConfiguration.cjs");
+const BaseUrlPartsApiConfiguration_js_1 = require("../../core/api/BaseUrlPartsApiConfiguration.cjs");
 const loadApiKey_js_1 = require("../../core/api/loadApiKey.cjs");
-class StabilityApiConfiguration extends BaseUrlApiConfiguration_js_1.BaseUrlApiConfiguration {
-    constructor({ baseUrl = "https://api.stability.ai/v1", apiKey, retry, throttle, } = {}) {
+/**
+ * Creates an API configuration for the Stability AI API.
+ * It calls the API at https://api.stability.ai/v1 by default.
+ */
+class StabilityApiConfiguration extends BaseUrlPartsApiConfiguration_js_1.BaseUrlPartsApiConfiguration {
+    constructor({ protocol = "https", host = "api.stability.ai", port = "443", path = "/v1", apiKey, headers, retry, throttle, } = {}) {
         super({
-            baseUrl,
-            headers: {
+            protocol,
+            host,
+            port,
+            path,
+            headers: headers ?? {
                 Authorization: `Bearer ${(0, loadApiKey_js_1.loadApiKey)({
                     apiKey,
                     environmentVariableName: "STABILITY_API_KEY",

package/model-provider/stability/StabilityApiConfiguration.d.ts

@@ -1,11 +1,10 @@
-import { BaseUrlApiConfiguration } from "../../core/api/BaseUrlApiConfiguration.js";
-import { RetryFunction } from "../../core/api/RetryFunction.js";
-import { ThrottleFunction } from "../../core/api/ThrottleFunction.js";
-export declare class StabilityApiConfiguration extends BaseUrlApiConfiguration {
-    constructor({ baseUrl, apiKey, retry, throttle, }?: {
-        baseUrl?: string;
+import { BaseUrlPartsApiConfiguration, BaseUrlPartsApiConfigurationOptions } from "../../core/api/BaseUrlPartsApiConfiguration.js";
+/**
+ * Creates an API configuration for the Stability AI API.
+ * It calls the API at https://api.stability.ai/v1 by default.
+ */
+export declare class StabilityApiConfiguration extends BaseUrlPartsApiConfiguration {
+    constructor({ protocol, host, port, path, apiKey, headers, retry, throttle, }?: Partial<BaseUrlPartsApiConfigurationOptions> & {
         apiKey?: string;
-        retry?: RetryFunction;
-        throttle?: ThrottleFunction;
     });
 }

package/model-provider/stability/StabilityApiConfiguration.js

@@ -1,10 +1,17 @@
-import { BaseUrlApiConfiguration } from "../../core/api/BaseUrlApiConfiguration.js";
+import { BaseUrlPartsApiConfiguration, } from "../../core/api/BaseUrlPartsApiConfiguration.js";
 import { loadApiKey } from "../../core/api/loadApiKey.js";
-export class StabilityApiConfiguration extends BaseUrlApiConfiguration {
-    constructor({ baseUrl = "https://api.stability.ai/v1", apiKey, retry, throttle, } = {}) {
+/**
+ * Creates an API configuration for the Stability AI API.
+ * It calls the API at https://api.stability.ai/v1 by default.
+ */
+export class StabilityApiConfiguration extends BaseUrlPartsApiConfiguration {
+    constructor({ protocol = "https", host = "api.stability.ai", port = "443", path = "/v1", apiKey, headers, retry, throttle, } = {}) {
         super({
-            baseUrl,
-            headers: {
+            protocol,
+            host,
+            port,
+            path,
+            headers: headers ?? {
                 Authorization: `Bearer ${loadApiKey({
                     apiKey,
                     environmentVariableName: "STABILITY_API_KEY",

package/model-provider/stability/StabilityError.cjs

@@ -1,37 +1,13 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.failedStabilityCallResponseHandler = exports.StabilityError = exports.stabilityErrorDataSchema = void 0;
+exports.failedStabilityCallResponseHandler = void 0;
 const zod_1 = require("zod");
-const ApiCallError_js_1 = require("../../core/api/ApiCallError.cjs");
+const postToApi_js_1 = require("../../core/api/postToApi.cjs");
 const ZodSchema_js_1 = require("../../core/schema/ZodSchema.cjs");
-const parseJSON_js_1 = require("../../core/schema/parseJSON.cjs");
-exports.stabilityErrorDataSchema = new ZodSchema_js_1.ZodSchema(zod_1.z.object({
+const stabilityErrorDataSchema = new ZodSchema_js_1.ZodSchema(zod_1.z.object({
     message: zod_1.z.string(),
 }));
-class StabilityError extends ApiCallError_js_1.ApiCallError {
-    constructor({ data, statusCode, url, requestBodyValues, message = data.message, }) {
-        super({ message, statusCode, requestBodyValues, url });
-        Object.defineProperty(this, "data", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: void 0
-        });
-        this.data = data;
-    }
-}
-exports.StabilityError = StabilityError;
-const failedStabilityCallResponseHandler = async ({ response, url, requestBodyValues }) => {
-    const responseBody = await response.text();
-    const parsedError = (0, parseJSON_js_1.parseJSON)({
-        text: responseBody,
-        schema: exports.stabilityErrorDataSchema,
-    });
-    return new StabilityError({
-        url,
-        requestBodyValues,
-        statusCode: response.status,
-        data: parsedError,
-    });
-};
-exports.failedStabilityCallResponseHandler = failedStabilityCallResponseHandler;
+exports.failedStabilityCallResponseHandler = (0, postToApi_js_1.createJsonErrorResponseHandler)({
+    errorSchema: stabilityErrorDataSchema,
+    errorToMessage: (error) => error.message,
+});

package/model-provider/stability/StabilityError.d.ts

@@ -1,18 +1,9 @@
 import { ApiCallError } from "../../core/api/ApiCallError.js";
 import { ResponseHandler } from "../../core/api/postToApi.js";
 import { ZodSchema } from "../../core/schema/ZodSchema.js";
-export declare const stabilityErrorDataSchema: ZodSchema<{
+declare const stabilityErrorDataSchema: ZodSchema<{
     message: string;
 }>;
 export type StabilityErrorData = (typeof stabilityErrorDataSchema)["_type"];
-export declare class StabilityError extends ApiCallError {
-    readonly data: StabilityErrorData;
-    constructor({ data, statusCode, url, requestBodyValues, message, }: {
-        message?: string;
-        statusCode: number;
-        url: string;
-        requestBodyValues: unknown;
-        data: StabilityErrorData;
-    });
-}
 export declare const failedStabilityCallResponseHandler: ResponseHandler<ApiCallError>;
+export {};

package/model-provider/stability/StabilityError.js

@@ -1,32 +1,10 @@
 import { z } from "zod";
-import { ApiCallError } from "../../core/api/ApiCallError.js";
+import { createJsonErrorResponseHandler, } from "../../core/api/postToApi.js";
 import { ZodSchema } from "../../core/schema/ZodSchema.js";
-import { parseJSON } from "../../core/schema/parseJSON.js";
-export const stabilityErrorDataSchema = new ZodSchema(z.object({
+const stabilityErrorDataSchema = new ZodSchema(z.object({
     message: z.string(),
 }));
-export class StabilityError extends ApiCallError {
-    constructor({ data, statusCode, url, requestBodyValues, message = data.message, }) {
-        super({ message, statusCode, requestBodyValues, url });
-        Object.defineProperty(this, "data", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: void 0
-        });
-        this.data = data;
-    }
-}
-export const failedStabilityCallResponseHandler = async ({ response, url, requestBodyValues }) => {
-    const responseBody = await response.text();
-    const parsedError = parseJSON({
-        text: responseBody,
-        schema: stabilityErrorDataSchema,
-    });
-    return new StabilityError({
-        url,
-        requestBodyValues,
-        statusCode: response.status,
-        data: parsedError,
-    });
-};
+export const failedStabilityCallResponseHandler = createJsonErrorResponseHandler({
+    errorSchema: stabilityErrorDataSchema,
+    errorToMessage: (error) => error.message,
+});

package/model-provider/stability/StabilityFacade.cjs

@@ -1,6 +1,7 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.ImageGenerator = void 0;
+exports.Api = exports.ImageGenerator = void 0;
+const StabilityApiConfiguration_js_1 = require("./StabilityApiConfiguration.cjs");
 const StabilityImageGenerationModel_js_1 = require("./StabilityImageGenerationModel.cjs");
 /**
  * Create an image generation model that calls the Stability AI image generation API.
@@ -10,12 +11,11 @@ const StabilityImageGenerationModel_js_1 = require("./StabilityImageGenerationMo
  * @example
  * const image = await generateImage(
  *   stability.ImageGenerator({
- *     model: "stable-diffusion-512-v2-1",
+ *     model: "stable-diffusion-v1-6",
  *     cfgScale: 7,
  *     clipGuidancePreset: "FAST_BLUE",
  *     height: 512,
  *     width: 512,
- *     samples: 1,
  *     steps: 30,
  *   })
  *   [
@@ -30,3 +30,11 @@ function ImageGenerator(settings) {
     return new StabilityImageGenerationModel_js_1.StabilityImageGenerationModel(settings);
 }
 exports.ImageGenerator = ImageGenerator;
+/**
+ * Creates an API configuration for the Stability AI API.
+ * It calls the API at https://api.stability.ai/v1 by default.
+ */
+function Api(settings) {
+    return new StabilityApiConfiguration_js_1.StabilityApiConfiguration(settings);
+}
+exports.Api = Api;

package/model-provider/stability/StabilityFacade.d.ts

@@ -1,3 +1,5 @@
+import { BaseUrlPartsApiConfigurationOptions } from "../../core/api/BaseUrlPartsApiConfiguration.js";
+import { StabilityApiConfiguration } from "./StabilityApiConfiguration.js";
 import { StabilityImageGenerationModel, StabilityImageGenerationSettings } from "./StabilityImageGenerationModel.js";
 /**
  * Create an image generation model that calls the Stability AI image generation API.
@@ -7,12 +9,11 @@ import { StabilityImageGenerationModel, StabilityImageGenerationSettings } from
  * @example
  * const image = await generateImage(
  *   stability.ImageGenerator({
- *     model: "stable-diffusion-512-v2-1",
+ *     model: "stable-diffusion-v1-6",
  *     cfgScale: 7,
  *     clipGuidancePreset: "FAST_BLUE",
  *     height: 512,
  *     width: 512,
- *     samples: 1,
  *     steps: 30,
  *   })
  *   [
@@ -24,3 +25,10 @@ import { StabilityImageGenerationModel, StabilityImageGenerationSettings } from
  * @returns A new instance of {@link StabilityImageGenerationModel}.
  */
 export declare function ImageGenerator(settings: StabilityImageGenerationSettings): StabilityImageGenerationModel;
+/**
+ * Creates an API configuration for the Stability AI API.
+ * It calls the API at https://api.stability.ai/v1 by default.
+ */
+export declare function Api(settings: Partial<BaseUrlPartsApiConfigurationOptions> & {
+    apiKey?: string;
+}): StabilityApiConfiguration;

package/model-provider/stability/StabilityFacade.js

@@ -1,3 +1,4 @@
+import { StabilityApiConfiguration } from "./StabilityApiConfiguration.js";
 import { StabilityImageGenerationModel, } from "./StabilityImageGenerationModel.js";
 /**
  * Create an image generation model that calls the Stability AI image generation API.
@@ -7,12 +8,11 @@ import { StabilityImageGenerationModel, } from "./StabilityImageGenerationModel.
  * @example
  * const image = await generateImage(
  *   stability.ImageGenerator({
- *     model: "stable-diffusion-512-v2-1",
+ *     model: "stable-diffusion-v1-6",
  *     cfgScale: 7,
  *     clipGuidancePreset: "FAST_BLUE",
  *     height: 512,
  *     width: 512,
- *     samples: 1,
  *     steps: 30,
  *   })
  *   [
@@ -26,3 +26,10 @@ import { StabilityImageGenerationModel, } from "./StabilityImageGenerationModel.
 export function ImageGenerator(settings) {
     return new StabilityImageGenerationModel(settings);
 }
+/**
+ * Creates an API configuration for the Stability AI API.
+ * It calls the API at https://api.stability.ai/v1 by default.
+ */
+export function Api(settings) {
+    return new StabilityApiConfiguration(settings);
+}