import { ProviderV3, ImageModelV3 } from '@ai-sdk/provider';
import { FetchFunction } from '@ai-sdk/provider-utils';

type RunwareImageSize = {
    width: number;
    height: number;
};
type RunwareImageModelId = string;
interface RunwareImageSettings {
    /**
     Override the maximum number of images per call (default 1).
     */
    maxImagesPerCall?: number;
    /**
     Output format for the generated images.
     */
    outputFormat?: 'JPG' | 'PNG' | 'WEBP';
    /**
     Output quality for the generated images (20-99).
     */
    outputQuality?: number;
    /**
     Output type for the generated images.
     */
    outputType?: 'URL' | 'base64Data' | 'dataURI';
    /**
     Check if generated images contain NSFW content.
     */
    checkNSFW?: boolean;
    /**
     Include cost information in the response.
     */
    includeCost?: boolean;
    /**
     Negative prompt to guide what should not appear in the generated images.
     */
    negativePrompt?: string;
    /**
     Number of inference steps (1-100).
     */
    steps?: number;
    /**
     Sampler/scheduler to use for generation.
     */
    scheduler?: string;
    /**
     CFG Scale for the generation (0-50).
     */
    CFGScale?: number;
    /**
     Clip Skip value (0-2).
     */
    clipSkip?: number;
    /**
     Prompt weighting method.
     */
    promptWeighting?: '' | 'compel' | 'sdEmbeds';
}

interface RunwareProviderSettings {
    /**
    Runware API key. Default value is taken from the `RUNWARE_API_KEY` environment variable.
     */
    apiKey?: string;
    /**
    Base URL for the API calls.
    The default prefix is `https://api.runware.ai/v1`.
     */
    baseURL?: string;
    /**
    Custom headers to include in the requests.
     */
    headers?: Record<string, string>;
    /**
    Custom fetch implementation. You can use it as a middleware to intercept
    requests, or to provide a custom fetch implementation for e.g. testing.
     */
    fetch?: FetchFunction;
}
interface RunwareProvider extends ProviderV3 {
    /**
    Creates a model for image generation.
     */
    image(modelId: RunwareImageModelId, settings?: RunwareImageSettings): ImageModelV3;
    /**
    Creates a model for image generation.
     */
    imageModel(modelId: RunwareImageModelId, settings?: RunwareImageSettings): ImageModelV3;
}
/**
Create a Runware provider instance.
 */
declare function createRunware(options?: RunwareProviderSettings): RunwareProvider;
/**
Default Runware provider instance.
 */
declare const runware: RunwareProvider;

export { type RunwareImageModelId, type RunwareImageSettings, type RunwareImageSize, type RunwareProvider, type RunwareProviderSettings, createRunware, runware };