UNPKG

@huggingface/inference

Version:

Typescript client for the Hugging Face Inference Providers and Inference Endpoints

46 lines 2 kB
import type { TextToImageArgs } from "../tasks/cv/textToImage.js"; import type { ImageToImageArgs } from "../tasks/cv/imageToImage.js"; import type { TextToVideoArgs } from "../tasks/cv/textToVideo.js"; import type { ImageToVideoArgs } from "../tasks/cv/imageToVideo.js"; import type { BodyParams, RequestArgs, UrlParams } from "../types.js"; import type { TextToImageTaskHelper, TextToVideoTaskHelper, ImageToImageTaskHelper, ImageToVideoTaskHelper } from "./providerHelper.js"; import { TaskProviderHelper } from "./providerHelper.js"; /** * Response structure for initial task submission */ interface WaveSpeedAISubmitResponse { id: string; urls: { get: string; }; } /** * Response structure for WaveSpeed AI API with submit response data */ interface WaveSpeedAISubmitTaskResponse { code: number; message: string; data: WaveSpeedAISubmitResponse; } declare abstract class WavespeedAITask extends TaskProviderHelper { constructor(url?: string); makeRoute(params: UrlParams): string; preparePayload(params: BodyParams<ImageToImageArgs | TextToImageArgs | TextToVideoArgs | ImageToVideoArgs>): Record<string, unknown>; getResponse(response: WaveSpeedAISubmitTaskResponse, url?: string, headers?: Record<string, string>): Promise<Blob>; } export declare class WavespeedAITextToImageTask extends WavespeedAITask implements TextToImageTaskHelper { constructor(); } export declare class WavespeedAITextToVideoTask extends WavespeedAITask implements TextToVideoTaskHelper { constructor(); } export declare class WavespeedAIImageToImageTask extends WavespeedAITask implements ImageToImageTaskHelper { constructor(); preparePayloadAsync(args: ImageToImageArgs): Promise<RequestArgs>; } export declare class WavespeedAIImageToVideoTask extends WavespeedAITask implements ImageToVideoTaskHelper { constructor(); preparePayloadAsync(args: ImageToVideoArgs): Promise<RequestArgs>; } export {}; //# sourceMappingURL=wavespeed.d.ts.map