index.d.ts

declare module "api2d"

declare class Api2d {
  constructor(
    key?: null | string, 
    apiBaseUrl?: null | string,
    timeout?: number,
  );

  setKey(key: string): void;

  setApiBaseUrl(apiBaseUrl: string): void;

  setTimeout(timeout: number): void;

  abort(): void;

  completion(options: {
    onMessage?: (message: string) => void;
    onEnd?: (message: string) => void;
    model?: string;
    stream?: boolean;
    noCache?: boolean;
    [key: string]: any;
  }): Promise<string> | Object;

  embeddings(options: {
      model?: string;
      [key: string]: any;
  }): Promise<any>;

  billing(): Promise<any>;

  vectorSave(options:{
    text: string,
    embedding: number[], 
    uuid?: string, 
    meta?: string
  }
  ): Promise<any>;

  vectorSearch(options:{
    searchable_id: string,
    embedding: number[], // 1536维的向量
    topk?: number
  }
  ): Promise<any>;

  vectorDelete(options:{
    uuid: string
  }): Promise<any>;
  vectorDeleteAll(): Promise<any>;

  textToSpeech(options: {
    text: string;
    voiceName: string;
    responseType: 'file' | 'stream';
    output: string | any;
  }): Promise<boolean> | Promise<void>;

  
  speechToText(options: {
    file: string;
    language: string;
    moderation?: boolean;
    moderation_stop?: boolean;
  }): Promise<string>;

  imageGenerate(options: {
    model?: string;
    prompt?: string;
    n?: number;
    size?: string;
    response_format?: string;
  }): Promise<any>;

}

export default Api2d