modelfusion 0.132.0 → 0.134.0

package/model-function/generate-transcription/generateTranscription.js

@@ -1,18 +0,0 @@
-import { executeStandardCall } from "../executeStandardCall.js";
-export async function generateTranscription({ model, audioData, mimeType, fullResponse, ...options }) {
-    const input = { mimeType, audioData };
-    const callResponse = await executeStandardCall({
-        functionType: "generate-transcription",
-        input,
-        model,
-        options,
-        generateResponse: async (options) => {
-            const result = await model.doTranscribe(input, options);
-            return {
-                rawResponse: result.rawResponse,
-                extractedValue: result.transcription,
-            };
-        },
-    });
-    return fullResponse ? callResponse : callResponse.value;
-}

package/model-function/index.cjs

@@ -1,38 +0,0 @@
-"use strict";
-var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
-    if (k2 === undefined) k2 = k;
-    var desc = Object.getOwnPropertyDescriptor(m, k);
-    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
-      desc = { enumerable: true, get: function() { return m[k]; } };
-    }
-    Object.defineProperty(o, k2, desc);
-}) : (function(o, m, k, k2) {
-    if (k2 === undefined) k2 = k;
-    o[k2] = m[k];
-}));
-var __exportStar = (this && this.__exportStar) || function(m, exports) {
-    for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
-};
-Object.defineProperty(exports, "__esModule", { value: true });
-__exportStar(require("./Delta.cjs"), exports);
-__exportStar(require("./Model.cjs"), exports);
-__exportStar(require("./ModelCallEvent.cjs"), exports);
-__exportStar(require("./ModelCallMetadata.cjs"), exports);
-__exportStar(require("./ModelInformation.cjs"), exports);
-__exportStar(require("./PromptTemplate.cjs"), exports);
-__exportStar(require("./classify/index.cjs"), exports);
-__exportStar(require("./embed/EmbeddingEvent.cjs"), exports);
-__exportStar(require("./embed/EmbeddingModel.cjs"), exports);
-__exportStar(require("./embed/embed.cjs"), exports);
-__exportStar(require("./generate-image/ImageGenerationEvent.cjs"), exports);
-__exportStar(require("./generate-image/ImageGenerationModel.cjs"), exports);
-__exportStar(require("./generate-image/PromptTemplateImageGenerationModel.cjs"), exports);
-__exportStar(require("./generate-image/generateImage.cjs"), exports);
-__exportStar(require("./generate-speech/index.cjs"), exports);
-__exportStar(require("./generate-object/index.cjs"), exports);
-__exportStar(require("./generate-text/index.cjs"), exports);
-__exportStar(require("./generate-transcription/TranscriptionEvent.cjs"), exports);
-__exportStar(require("./generate-transcription/TranscriptionModel.cjs"), exports);
-__exportStar(require("./generate-transcription/generateTranscription.cjs"), exports);
-__exportStar(require("./tokenize-text/Tokenizer.cjs"), exports);
-__exportStar(require("./tokenize-text/countTokens.cjs"), exports);

package/model-function/index.d.ts

@@ -1,22 +0,0 @@
-export * from "./Delta.js";
-export * from "./Model.js";
-export * from "./ModelCallEvent.js";
-export * from "./ModelCallMetadata.js";
-export * from "./ModelInformation.js";
-export * from "./PromptTemplate.js";
-export * from "./classify/index.js";
-export * from "./embed/EmbeddingEvent.js";
-export * from "./embed/EmbeddingModel.js";
-export * from "./embed/embed.js";
-export * from "./generate-image/ImageGenerationEvent.js";
-export * from "./generate-image/ImageGenerationModel.js";
-export * from "./generate-image/PromptTemplateImageGenerationModel.js";
-export * from "./generate-image/generateImage.js";
-export * from "./generate-speech/index.js";
-export * from "./generate-object/index.js";
-export * from "./generate-text/index.js";
-export * from "./generate-transcription/TranscriptionEvent.js";
-export * from "./generate-transcription/TranscriptionModel.js";
-export * from "./generate-transcription/generateTranscription.js";
-export * from "./tokenize-text/Tokenizer.js";
-export * from "./tokenize-text/countTokens.js";

package/model-function/index.js

@@ -1,22 +0,0 @@
-export * from "./Delta.js";
-export * from "./Model.js";
-export * from "./ModelCallEvent.js";
-export * from "./ModelCallMetadata.js";
-export * from "./ModelInformation.js";
-export * from "./PromptTemplate.js";
-export * from "./classify/index.js";
-export * from "./embed/EmbeddingEvent.js";
-export * from "./embed/EmbeddingModel.js";
-export * from "./embed/embed.js";
-export * from "./generate-image/ImageGenerationEvent.js";
-export * from "./generate-image/ImageGenerationModel.js";
-export * from "./generate-image/PromptTemplateImageGenerationModel.js";
-export * from "./generate-image/generateImage.js";
-export * from "./generate-speech/index.js";
-export * from "./generate-object/index.js";
-export * from "./generate-text/index.js";
-export * from "./generate-transcription/TranscriptionEvent.js";
-export * from "./generate-transcription/TranscriptionModel.js";
-export * from "./generate-transcription/generateTranscription.js";
-export * from "./tokenize-text/Tokenizer.js";
-export * from "./tokenize-text/countTokens.js";

package/model-function/tokenize-text/Tokenizer.cjs

@@ -1,2 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });

package/model-function/tokenize-text/Tokenizer.d.ts

@@ -1,43 +0,0 @@
-/**
- * Interface for a basic tokenizer capable of converting text into tokens.
- *
- * This serves as the base for tokenization functionalities where the focus is on the transformation of input text into a series of numeric tokens.
- */
-export interface BasicTokenizer {
-    /**
-     * Asynchronously tokenize the given text into a sequence of numeric tokens.
-     *
-     * @param text - Input text string that needs to be tokenized.
-     * @returns A promise containing an array of numbers, where each number is a token representing a part or the whole of the input text.
-     */
-    tokenize: (text: string) => PromiseLike<Array<number>>;
-}
-/**
- * Interface for a comprehensive tokenizer that extends the basic tokenization capabilities.
- *
- * In addition to basic tokenization, this interface provides methods for detokenization and
- * retrieving the original text corresponding to each token, enabling a more informative and reversible transformation process.
- */
-export interface FullTokenizer extends BasicTokenizer {
-    /**
-     * Asynchronously tokenize the given text, providing both the numeric tokens and their corresponding text.
-     *
-     * @param text - Input text string to be tokenized.
-     * @returns A promise containing an object with two arrays:
-     *          1. `tokens` - An array of numbers where each number is a token.
-     *          2. `tokenTexts` - An array of strings where each string represents the original text corresponding to each token.
-     */
-    tokenizeWithTexts: (text: string) => PromiseLike<{
-        tokens: Array<number>;
-        tokenTexts: Array<string>;
-    }>;
-    /**
-     * Asynchronously revert a sequence of numeric tokens back into the original text.
-     * Detokenization is the process of transforming tokens back to a human-readable format, and it's essential in scenarios
-     * where the output needs to be interpretable or when the tokenization process has to be reversible.
-     *
-     * @param tokens - An array of numeric tokens to be converted back to text.
-     * @returns A promise containing a string that represents the original text corresponding to the sequence of input tokens.
-     */
-    detokenize: (tokens: Array<number>) => PromiseLike<string>;
-}

package/model-function/tokenize-text/Tokenizer.js

@@ -1 +0,0 @@
-export {};

package/model-function/tokenize-text/countTokens.cjs

@@ -1,10 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.countTokens = void 0;
-/**
- * Count the number of tokens in the given text.
- */
-async function countTokens(tokenizer, text) {
-    return (await tokenizer.tokenize(text)).length;
-}
-exports.countTokens = countTokens;

package/model-function/tokenize-text/countTokens.d.ts

@@ -1,5 +0,0 @@
-import { BasicTokenizer } from "./Tokenizer.js";
-/**
- * Count the number of tokens in the given text.
- */
-export declare function countTokens(tokenizer: BasicTokenizer, text: string): Promise<number>;

package/model-function/tokenize-text/countTokens.js

@@ -1,6 +0,0 @@
-/**
- * Count the number of tokens in the given text.
- */
-export async function countTokens(tokenizer, text) {
-    return (await tokenizer.tokenize(text)).length;
-}

package/model-provider/automatic1111/Automatic1111ApiConfiguration.cjs

@@ -1,22 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.Automatic1111ApiConfiguration = void 0;
-const BaseUrlApiConfiguration_js_1 = require("../../core/api/BaseUrlApiConfiguration.cjs");
-/**
- * Creates an API configuration for the AUTOMATIC1111 Stable Diffusion Web UI API.
- * It calls the API at http://127.0.0.1:7860/sdapi/v1 by default.
- */
-class Automatic1111ApiConfiguration extends BaseUrlApiConfiguration_js_1.BaseUrlApiConfigurationWithDefaults {
-    constructor(settings = {}) {
-        super({
-            ...settings,
-            baseUrlDefaults: {
-                protocol: "http",
-                host: "127.0.0.1",
-                port: "7860",
-                path: "/sdapi/v1",
-            },
-        });
-    }
-}
-exports.Automatic1111ApiConfiguration = Automatic1111ApiConfiguration;

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

@@ -1,8 +0,0 @@
-import { BaseUrlApiConfigurationWithDefaults, PartialBaseUrlPartsApiConfigurationOptions } from "../../core/api/BaseUrlApiConfiguration.js";
-/**
- * Creates an API configuration for the AUTOMATIC1111 Stable Diffusion Web UI API.
- * It calls the API at http://127.0.0.1:7860/sdapi/v1 by default.
- */
-export declare class Automatic1111ApiConfiguration extends BaseUrlApiConfigurationWithDefaults {
-    constructor(settings?: PartialBaseUrlPartsApiConfigurationOptions);
-}

package/model-provider/automatic1111/Automatic1111ApiConfiguration.js

@@ -1,18 +0,0 @@
-import { BaseUrlApiConfigurationWithDefaults, } from "../../core/api/BaseUrlApiConfiguration.js";
-/**
- * Creates an API configuration for the AUTOMATIC1111 Stable Diffusion Web UI API.
- * It calls the API at http://127.0.0.1:7860/sdapi/v1 by default.
- */
-export class Automatic1111ApiConfiguration extends BaseUrlApiConfigurationWithDefaults {
-    constructor(settings = {}) {
-        super({
-            ...settings,
-            baseUrlDefaults: {
-                protocol: "http",
-                host: "127.0.0.1",
-                port: "7860",
-                path: "/sdapi/v1",
-            },
-        });
-    }
-}

package/model-provider/automatic1111/Automatic1111Error.cjs

@@ -1,16 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.failedAutomatic1111CallResponseHandler = void 0;
-const zod_1 = require("zod");
-const postToApi_js_1 = require("../../core/api/postToApi.cjs");
-const ZodSchema_js_1 = require("../../core/schema/ZodSchema.cjs");
-const automatic1111ErrorDataSchema = zod_1.z.object({
-    error: zod_1.z.string(),
-    detail: zod_1.z.string(),
-    body: zod_1.z.string(),
-    errors: zod_1.z.string(),
-});
-exports.failedAutomatic1111CallResponseHandler = (0, postToApi_js_1.createJsonErrorResponseHandler)({
-    errorSchema: (0, ZodSchema_js_1.zodSchema)(automatic1111ErrorDataSchema),
-    errorToMessage: (error) => error.detail,
-});

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

@@ -1,22 +0,0 @@
-import { z } from "zod";
-import { ApiCallError } from "../../core/api/ApiCallError.js";
-import { ResponseHandler } from "../../core/api/postToApi.js";
-declare const automatic1111ErrorDataSchema: z.ZodObject<{
-    error: z.ZodString;
-    detail: z.ZodString;
-    body: z.ZodString;
-    errors: z.ZodString;
-}, "strip", z.ZodTypeAny, {
-    error: string;
-    errors: string;
-    body: string;
-    detail: string;
-}, {
-    error: string;
-    errors: string;
-    body: string;
-    detail: string;
-}>;
-export type Automatic1111ErrorData = z.infer<typeof automatic1111ErrorDataSchema>;
-export declare const failedAutomatic1111CallResponseHandler: ResponseHandler<ApiCallError>;
-export {};

package/model-provider/automatic1111/Automatic1111Error.js

@@ -1,13 +0,0 @@
-import { z } from "zod";
-import { createJsonErrorResponseHandler, } from "../../core/api/postToApi.js";
-import { zodSchema } from "../../core/schema/ZodSchema.js";
-const automatic1111ErrorDataSchema = z.object({
-    error: z.string(),
-    detail: z.string(),
-    body: z.string(),
-    errors: z.string(),
-});
-export const failedAutomatic1111CallResponseHandler = createJsonErrorResponseHandler({
-    errorSchema: zodSchema(automatic1111ErrorDataSchema),
-    errorToMessage: (error) => error.detail,
-});

package/model-provider/automatic1111/Automatic1111Facade.cjs

@@ -1,24 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.ImageGenerator = exports.Api = void 0;
-const Automatic1111ApiConfiguration_js_1 = require("./Automatic1111ApiConfiguration.cjs");
-const Automatic1111ImageGenerationModel_js_1 = require("./Automatic1111ImageGenerationModel.cjs");
-/**
- * Creates an API configuration for the AUTOMATIC1111 Stable Diffusion Web UI API.
- * It calls the API at http://127.0.0.1:7860/sdapi/v1 by default.
- */
-function Api(settings) {
-    return new Automatic1111ApiConfiguration_js_1.Automatic1111ApiConfiguration(settings);
-}
-exports.Api = Api;
-/**
- * Create an image generation model that calls the AUTOMATIC1111 Stable Diffusion Web UI API.
- *
- * @see https://github.com/AUTOMATIC1111/stable-diffusion-webui
- *
- * @return A new instance of ${@link Automatic1111ImageGenerationModel}.
- */
-function ImageGenerator(settings) {
-    return new Automatic1111ImageGenerationModel_js_1.Automatic1111ImageGenerationModel(settings);
-}
-exports.ImageGenerator = ImageGenerator;

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

@@ -1,16 +0,0 @@
-import { PartialBaseUrlPartsApiConfigurationOptions } from "../../core/api/BaseUrlApiConfiguration.js";
-import { Automatic1111ApiConfiguration } from "./Automatic1111ApiConfiguration.js";
-import { Automatic1111ImageGenerationModel, Automatic1111ImageGenerationSettings } from "./Automatic1111ImageGenerationModel.js";
-/**
- * Creates an API configuration for the AUTOMATIC1111 Stable Diffusion Web UI API.
- * It calls the API at http://127.0.0.1:7860/sdapi/v1 by default.
- */
-export declare function Api(settings: PartialBaseUrlPartsApiConfigurationOptions): Automatic1111ApiConfiguration;
-/**
- * Create an image generation model that calls the AUTOMATIC1111 Stable Diffusion Web UI API.
- *
- * @see https://github.com/AUTOMATIC1111/stable-diffusion-webui
- *
- * @return A new instance of ${@link Automatic1111ImageGenerationModel}.
- */
-export declare function ImageGenerator(settings: Automatic1111ImageGenerationSettings): Automatic1111ImageGenerationModel;

package/model-provider/automatic1111/Automatic1111Facade.js

@@ -1,19 +0,0 @@
-import { Automatic1111ApiConfiguration } from "./Automatic1111ApiConfiguration.js";
-import { Automatic1111ImageGenerationModel, } from "./Automatic1111ImageGenerationModel.js";
-/**
- * Creates an API configuration for the AUTOMATIC1111 Stable Diffusion Web UI API.
- * It calls the API at http://127.0.0.1:7860/sdapi/v1 by default.
- */
-export function Api(settings) {
-    return new Automatic1111ApiConfiguration(settings);
-}
-/**
- * Create an image generation model that calls the AUTOMATIC1111 Stable Diffusion Web UI API.
- *
- * @see https://github.com/AUTOMATIC1111/stable-diffusion-webui
- *
- * @return A new instance of ${@link Automatic1111ImageGenerationModel}.
- */
-export function ImageGenerator(settings) {
-    return new Automatic1111ImageGenerationModel(settings);
-}

package/model-provider/automatic1111/Automatic1111ImageGenerationModel.cjs

@@ -1,101 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.Automatic1111ImageGenerationModel = void 0;
-const zod_1 = require("zod");
-const callWithRetryAndThrottle_js_1 = require("../../core/api/callWithRetryAndThrottle.cjs");
-const postToApi_js_1 = require("../../core/api/postToApi.cjs");
-const ZodSchema_js_1 = require("../../core/schema/ZodSchema.cjs");
-const AbstractModel_js_1 = require("../../model-function/AbstractModel.cjs");
-const PromptTemplateImageGenerationModel_js_1 = require("../../model-function/generate-image/PromptTemplateImageGenerationModel.cjs");
-const Automatic1111ApiConfiguration_js_1 = require("./Automatic1111ApiConfiguration.cjs");
-const Automatic1111Error_js_1 = require("./Automatic1111Error.cjs");
-const Automatic1111ImageGenerationPrompt_js_1 = require("./Automatic1111ImageGenerationPrompt.cjs");
-/**
- * Create an image generation model that calls the AUTOMATIC1111 Stable Diffusion Web UI API.
- *
- * @see https://github.com/AUTOMATIC1111/stable-diffusion-webui
- */
-class Automatic1111ImageGenerationModel extends AbstractModel_js_1.AbstractModel {
-    constructor(settings) {
-        super({ settings });
-        Object.defineProperty(this, "provider", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: "Automatic1111"
-        });
-    }
-    get modelName() {
-        return this.settings.model;
-    }
-    async callAPI(input, callOptions) {
-        const api = this.settings.api ?? new Automatic1111ApiConfiguration_js_1.Automatic1111ApiConfiguration();
-        const abortSignal = callOptions.run?.abortSignal;
-        return (0, callWithRetryAndThrottle_js_1.callWithRetryAndThrottle)({
-            retry: api.retry,
-            throttle: api.throttle,
-            call: async () => (0, postToApi_js_1.postJsonToApi)({
-                url: api.assembleUrl(`/txt2img`),
-                headers: api.headers({
-                    functionType: callOptions.functionType,
-                    functionId: callOptions.functionId,
-                    run: callOptions.run,
-                    callId: callOptions.callId,
-                }),
-                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: Automatic1111Error_js_1.failedAutomatic1111CallResponseHandler,
-                successfulResponseHandler: (0, postToApi_js_1.createJsonResponseHandler)((0, ZodSchema_js_1.zodSchema)(Automatic1111ImageGenerationResponseSchema)),
-                abortSignal,
-            }),
-        });
-    }
-    get settingsForEvent() {
-        const eventSettingProperties = [
-            "height",
-            "width",
-            "sampler",
-            "steps",
-            "cfgScale",
-            "seed",
-        ];
-        return Object.fromEntries(Object.entries(this.settings).filter(([key]) => eventSettingProperties.includes(key)));
-    }
-    async doGenerateImages(prompt, options) {
-        const rawResponse = await this.callAPI(prompt, options);
-        return {
-            rawResponse,
-            base64Images: rawResponse.images,
-        };
-    }
-    withTextPrompt() {
-        return this.withPromptTemplate((0, Automatic1111ImageGenerationPrompt_js_1.mapBasicPromptToAutomatic1111Format)());
-    }
-    withPromptTemplate(promptTemplate) {
-        return new PromptTemplateImageGenerationModel_js_1.PromptTemplateImageGenerationModel({
-            model: this,
-            promptTemplate,
-        });
-    }
-    withSettings(additionalSettings) {
-        return new Automatic1111ImageGenerationModel(Object.assign({}, this.settings, additionalSettings));
-    }
-}
-exports.Automatic1111ImageGenerationModel = Automatic1111ImageGenerationModel;
-const Automatic1111ImageGenerationResponseSchema = zod_1.z.object({
-    images: zod_1.z.array(zod_1.z.string()),
-    parameters: zod_1.z.object({}),
-    info: zod_1.z.string(),
-});

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

@@ -1,68 +0,0 @@
-import { z } from "zod";
-import { FunctionCallOptions } from "../../core/FunctionOptions.js";
-import { ApiConfiguration } from "../../core/api/ApiConfiguration.js";
-import { AbstractModel } from "../../model-function/AbstractModel.js";
-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.
- *
- * @see https://github.com/AUTOMATIC1111/stable-diffusion-webui
- */
-export declare class Automatic1111ImageGenerationModel extends AbstractModel<Automatic1111ImageGenerationSettings> implements ImageGenerationModel<Automatic1111ImageGenerationPrompt, Automatic1111ImageGenerationSettings> {
-    constructor(settings: Automatic1111ImageGenerationSettings);
-    readonly provider: "Automatic1111";
-    get modelName(): string;
-    callAPI(input: Automatic1111ImageGenerationPrompt, callOptions: FunctionCallOptions): Promise<Automatic1111ImageGenerationResponse>;
-    get settingsForEvent(): Partial<Automatic1111ImageGenerationSettings>;
-    doGenerateImages(prompt: Automatic1111ImageGenerationPrompt, options: FunctionCallOptions): Promise<{
-        rawResponse: {
-            images: string[];
-            parameters: {};
-            info: 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: Partial<Automatic1111ImageGenerationSettings>): this;
-}
-declare const Automatic1111ImageGenerationResponseSchema: z.ZodObject<{
-    images: z.ZodArray<z.ZodString, "many">;
-    parameters: z.ZodObject<{}, "strip", z.ZodTypeAny, {}, {}>;
-    info: z.ZodString;
-}, "strip", z.ZodTypeAny, {
-    images: string[];
-    parameters: {};
-    info: string;
-}, {
-    images: string[];
-    parameters: {};
-    info: string;
-}>;
-export type Automatic1111ImageGenerationResponse = z.infer<typeof Automatic1111ImageGenerationResponseSchema>;
-export {};

package/model-provider/automatic1111/Automatic1111ImageGenerationModel.js

@@ -1,97 +0,0 @@
-import { z } from "zod";
-import { callWithRetryAndThrottle } from "../../core/api/callWithRetryAndThrottle.js";
-import { createJsonResponseHandler, postJsonToApi, } from "../../core/api/postToApi.js";
-import { zodSchema } from "../../core/schema/ZodSchema.js";
-import { AbstractModel } from "../../model-function/AbstractModel.js";
-import { PromptTemplateImageGenerationModel } from "../../model-function/generate-image/PromptTemplateImageGenerationModel.js";
-import { Automatic1111ApiConfiguration } from "./Automatic1111ApiConfiguration.js";
-import { failedAutomatic1111CallResponseHandler } from "./Automatic1111Error.js";
-import { mapBasicPromptToAutomatic1111Format, } from "./Automatic1111ImageGenerationPrompt.js";
-/**
- * Create an image generation model that calls the AUTOMATIC1111 Stable Diffusion Web UI API.
- *
- * @see https://github.com/AUTOMATIC1111/stable-diffusion-webui
- */
-export class Automatic1111ImageGenerationModel extends AbstractModel {
-    constructor(settings) {
-        super({ settings });
-        Object.defineProperty(this, "provider", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: "Automatic1111"
-        });
-    }
-    get modelName() {
-        return this.settings.model;
-    }
-    async callAPI(input, callOptions) {
-        const api = this.settings.api ?? new Automatic1111ApiConfiguration();
-        const abortSignal = callOptions.run?.abortSignal;
-        return callWithRetryAndThrottle({
-            retry: api.retry,
-            throttle: api.throttle,
-            call: async () => postJsonToApi({
-                url: api.assembleUrl(`/txt2img`),
-                headers: api.headers({
-                    functionType: callOptions.functionType,
-                    functionId: callOptions.functionId,
-                    run: callOptions.run,
-                    callId: callOptions.callId,
-                }),
-                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(zodSchema(Automatic1111ImageGenerationResponseSchema)),
-                abortSignal,
-            }),
-        });
-    }
-    get settingsForEvent() {
-        const eventSettingProperties = [
-            "height",
-            "width",
-            "sampler",
-            "steps",
-            "cfgScale",
-            "seed",
-        ];
-        return Object.fromEntries(Object.entries(this.settings).filter(([key]) => eventSettingProperties.includes(key)));
-    }
-    async doGenerateImages(prompt, options) {
-        const rawResponse = await this.callAPI(prompt, options);
-        return {
-            rawResponse,
-            base64Images: rawResponse.images,
-        };
-    }
-    withTextPrompt() {
-        return this.withPromptTemplate(mapBasicPromptToAutomatic1111Format());
-    }
-    withPromptTemplate(promptTemplate) {
-        return new PromptTemplateImageGenerationModel({
-            model: this,
-            promptTemplate,
-        });
-    }
-    withSettings(additionalSettings) {
-        return new Automatic1111ImageGenerationModel(Object.assign({}, this.settings, additionalSettings));
-    }
-}
-const Automatic1111ImageGenerationResponseSchema = z.object({
-    images: z.array(z.string()),
-    parameters: z.object({}),
-    info: z.string(),
-});

package/model-provider/automatic1111/Automatic1111ImageGenerationPrompt.cjs

@@ -1,12 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.mapBasicPromptToAutomatic1111Format = void 0;
-/**
- * Formats a basic text prompt as an Automatic1111 prompt.
- */
-function mapBasicPromptToAutomatic1111Format() {
-    return {
-        format: (description) => ({ prompt: description }),
-    };
-}
-exports.mapBasicPromptToAutomatic1111Format = mapBasicPromptToAutomatic1111Format;

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

@@ -1,9 +0,0 @@
-import { PromptTemplate } from "../../model-function/PromptTemplate.js";
-export type Automatic1111ImageGenerationPrompt = {
-    prompt: string;
-    negativePrompt?: string;
-};
-/**
- * Formats a basic text prompt as an Automatic1111 prompt.
- */
-export declare function mapBasicPromptToAutomatic1111Format(): PromptTemplate<string, Automatic1111ImageGenerationPrompt>;

package/model-provider/automatic1111/Automatic1111ImageGenerationPrompt.js

@@ -1,8 +0,0 @@
-/**
- * Formats a basic text prompt as an Automatic1111 prompt.
- */
-export function mapBasicPromptToAutomatic1111Format() {
-    return {
-        format: (description) => ({ prompt: description }),
-    };
-}