axios-inference-polling
Version:
A package for polling the inference status until it is completed.
29 lines (28 loc) • 1.11 kB
TypeScript
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;
}>;
}