import { AxiosResponse } from 'axios';
import { HttpClient } from '../utils';
export type IGenerateTextOptions = {
    model?: string;
    topP?: number;
    maxTokens?: number;
    temperature?: number;
    presencePenalty?: number;
    frequencyPenalty?: number;
};
export type IGenerateMessageCommon = {
    name?: string;
    role?: 'user' | 'system' | 'assistant';
};
export type IGenerateTextMessage = {
    type?: 'text';
    content: string;
} & IGenerateMessageCommon;
export type IGenerateImageMessage = {
    type: 'image';
    content: {
        url: string;
    };
};
export type IGenerateMessage = IGenerateTextMessage | IGenerateImageMessage;
export type IGenerateTextResPayload = {
    content: string;
    metadata: {
        finishReason: string;
        usage: {
            inputTokens: number;
            outputTokens: number;
            totalTokens: number;
        };
    };
};
type ChatCompletionApiRes = {
    statusCode: number;
    message: string;
    data: {
        content: string;
        usageMetadata: {
            inputTokens: number;
            outputTokens: number;
            totalTokens: number;
        };
        finishReason: string;
    };
};
export type IGenerateTextRes = AxiosResponse<{
    data: ChatCompletionApiRes;
    payload: IGenerateTextResPayload;
}>;
export declare class Generative {
    httpClient: HttpClient;
    constructor(httpClient: HttpClient);
    text(prompt: string, options?: IGenerateTextOptions): Promise<IGenerateTextRes>;
    text(messages: IGenerateMessage[], options?: IGenerateTextOptions): Promise<IGenerateTextRes>;
}
export {};
//# sourceMappingURL=Generative.d.ts.map