Interface LobeRuntimeAI

packages/model-runtime/src/core/BaseAI.ts:29–72

} from '../types';

export interface LobeRuntimeAI {
  baseURL?: string;
  chat?: (payload: ChatStreamPayload, options?: ChatMethodOptions) => Promise<Response>;
  createImage?: (
    payload: CreateImagePayload,
    options?: CreateImageMethodOptions,
  ) => Promise<CreateImageResponse>;

  createVideo?: (
    payload: CreateVideoPayload,
    options?: CreateVideoMethodOptions,
  ) => Promise<CreateVideoResponse>;

  embeddings?: (payload: EmbeddingsPayload, options?: EmbeddingsOptions) => Promise<Embeddings[]>;

  generateObject?: (
    payload: GenerateObjectPayload,
    options?: GenerateObjectOptions,
  ) => Promise<any>;

  handleCreateVideoWebhook?: (
    payload: HandleCreateVideoWebhookPayload,
  ) => Promise<HandleCreateVideoWebhookResult>;

  handlePollVideoStatus?: (
    inferenceId: string,
  ) => Promise<
    | { status: 'success'; videoUrl: string }
    | { status: 'failed'; error: string }
    | { status: 'pending' }
  >;

  models?: () => Promise<any>;

  // Model management related interface
  pullModel?: (params: PullModelParams, options?: ModelRequestOptions) => Promise<Response>;

  textToSpeech?: (
    payload: TextToSpeechPayload,
    options?: TextToSpeechOptions,
  ) => Promise<ArrayBuffer>;

  transcribe?: (payload: ASRPayload, options?: ASROptions) => Promise<ASRResponse>;
}
/* eslint-enabled */

export abstract class LobeOpenAICompatibleRuntime {