ai-utils.js/model-provider/automatic1111/Automatic1111ImageGenerationModel.d.ts

Build AI applications, chatbots, and agents with JavaScript and TypeScript.

import { z } from "zod";
import { AbstractModel } from "../../model-function/AbstractModel.js";
import { FunctionOptions } from "../../model-function/FunctionOptions.js";
import { ImageGenerationModel, ImageGenerationModelSettings } from "../../model-function/generate-image/ImageGenerationModel.js";
import { RetryFunction } from "../../util/api/RetryFunction.js";
import { ThrottleFunction } from "../../util/api/ThrottleFunction.js";
/**
 * 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<Automatic1111ImageGenerationModelSettings> implements ImageGenerationModel<A111ImageGenerationPrompt, Automatic1111ImageGenerationResponse, Automatic1111ImageGenerationModelSettings> {
    constructor(settings: Automatic1111ImageGenerationModelSettings);
    readonly provider: "Automatic1111";
    get modelName(): string;
    callAPI(input: A111ImageGenerationPrompt, options?: FunctionOptions<Automatic1111ImageGenerationModelSettings>): Promise<Automatic1111ImageGenerationResponse>;
    generateImageResponse(prompt: A111ImageGenerationPrompt, options?: FunctionOptions<Automatic1111ImageGenerationModelSettings>): Promise<{
        parameters: {};
        images: string[];
        info: string;
    }>;
    extractBase64Image(response: Automatic1111ImageGenerationResponse): string;
    withSettings(additionalSettings: Automatic1111ImageGenerationModelSettings): this;
}
export interface Automatic1111ImageGenerationModelSettings extends ImageGenerationModelSettings {
    baseUrl?: string;
    retry?: RetryFunction;
    throttle?: ThrottleFunction;
    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, {}, {}>;
    info: z.ZodString;
}, "strip", z.ZodTypeAny, {
    parameters: {};
    images: string[];
    info: string;
}, {
    parameters: {};
    images: string[];
    info: string;
}>;
export type Automatic1111ImageGenerationResponse = z.infer<typeof Automatic1111ImageGenerationResponseSchema>;
export type A111ImageGenerationPrompt = {
    prompt: string;
    negativePrompt?: string;
    seed?: number;
};
export {};