UNPKG

axios-inference-polling

Version:

A package for polling the inference status until it is completed.

29 lines (28 loc) 1.11 kB
import { AxiosRequestConfig } from "axios"; export type Status = "starting" | "processing" | "succeeded" | "failed" | "canceled"; export type InfereceEventType = "CREATE_PREDICTION" | "UPDATE_PREDICTION"; export declare class Inference { private axios; private subscribers; constructor(); subscribeToEvents(handler: (event: any, type: InfereceEventType) => any): void; unsubscribeToEvents(handler: (event: any, type: InfereceEventType) => any): void; private emitEvent; createInferenceRequest<T, Y>({ create_prediction_url, payload, config, onError, }: { create_prediction_url: string; payload: Y; config?: AxiosRequestConfig; onError?: (error: any) => any; }): Promise<T & { status: Status; }>; pollInferenceRequest<T, Y>({ create_prediction_url, prediction_status_url, payload, config, onError, }: { create_prediction_url: string; payload: Y; config?: AxiosRequestConfig; prediction_status_url: string; onError?: (error: any) => any; }): Promise<T & { status: Status; }>; }