export enum ErnieModels {
    PRO = "completions_pro",
    TURBO = "ernie-4.0-turbo-8k",
    SPEED_8k = "ernie_speed",
    SPEED_128k = "ernie-speed-128k",
    SPEED_PRO_128K = "ernie-speed-pro-128k",
    SPEED_APP_BUILDER = "ai_apaas",
    SPEED_APP_BUILDER_0614 = "ai_apaas_lite",
    LITE_8K = "ernie-lite-8k",
    LITE_PRO_128K = "ernie-lite-pro-128k",
    TINY_8K = "ernie_tiny_8k",
    NOVEL_8K = "ernie-novel-8k",
    CHAR_8K = "ernie-char-8k",
    FUNC_8K = "ernie-func-8k"
}

export interface ErnieMessage {
    role: 'user' | 'assistant';
    content: string;
}

export interface ErnieRequestParams {
    messages: ErnieMessage[];
    stream?: boolean;
    temperature?: number;
    top_p?: number;
    penalty_score?: number;
    system?: string;
    stop?: string[];
    min_output_tokens?: number;
    max_output_tokens?: number;
    frequency_penalty?: number;
    presence_penalty?: number;
    response_format?: string,
    user_id?: string;
}

export interface ErnieChatCompletion {
    id: string;
    object: string;
    created: number;
    sentence_id?: number;
    is_end?: boolean;
    is_truncated: boolean;
    result: string;
    need_clear_history: boolean;
    usage: {
        prompt_tokens: number;
        completion_tokens: number;
        total_tokens: number;
    };
}