import { default as OpenAI } from 'openai';
import { RequestOptions } from 'openai/core';
import { APIResource } from '../../resource';

export declare class Embeddings extends APIResource {
    protected endpoints: Record<EmbeddingModel, string>;
    /**
     * Creates an embedding vector representing the input text.
     *
     * See https://cloud.baidu.com/doc/WENXINWORKSHOP/s/alj562vvu
     */
    create(params: EmbeddingCreateParams, options?: RequestOptions): Promise<OpenAI.CreateEmbeddingResponse>;
    static fromResponse(model: EmbeddingModel, data: CreateEmbeddingResponse): OpenAI.CreateEmbeddingResponse;
    /**
     * 如果 code 不为 0，抛出 APIError
     *
     * @param code -
     * @param message -
     */
    static assert(resp: CreateEmbeddingResponse): void;
}
export type EmbeddingModel = 'ernie-text-embedding';
export interface EmbeddingCreateParams {
    /**
     * 输入文本
     */
    input: string | Array<string> | Array<number> | Array<Array<number>>;
    /**
     * 模型
     */
    model: EmbeddingModel;
    /**
     * 用户 ID
     */
    user?: string;
}
type CreateEmbeddingResponse = {
    errorCode: number;
    errorMsg: string;
    result: OpenAI.CreateEmbeddingResponse;
};
export {};