import { GoogleCtodService } from './index.js';
/**
 *  if data:
 *  data: base64 string,
 *  mimeType: image/jpeg,
 */
type Part = {
    text: string;
} | {
    inlineData: {
        data: string;
        mimeType: string;
    };
} | {
    fileData: {
        fileUri: string;
    };
};
export type GoogleMessage = {
    role: 'model' | 'user';
    parts: Part[];
};
export type Config = {
    /**
     * @zh 選擇運行的模型。
     * @en What model to use.
     */
    model: string;
    maxTokens: number;
    temperature: number;
    enableGoogleSearch: boolean;
    thinkingConfig: {
        enabled: boolean;
        level: 'THINKING_LEVEL_UNSPECIFIED' | 'LOW' | 'HIGH';
    };
};
export declare class GoogleChat {
    google: GoogleCtodService;
    config: Config;
    constructor(google: GoogleCtodService);
    /**
     * @zh 改變對話的一些設定
     * @en Change some settings of the conversation
     */
    setConfig(options: Partial<Config>): void;
    static getThinkingConfig(config?: Config['thinkingConfig']): {
        includeThoughts: boolean;
        thinkingLevel: any;
    } | undefined;
    /**
     * @zh 進行對話
     * @en Talk to the AI
     */
    talk(params: {
        system?: string;
        messages: GoogleMessage[];
    }): Promise<{
        text: string | undefined;
        newMessages: (GoogleMessage | {
            role: string;
            parts: {
                text: string | undefined;
            }[];
        })[];
    }>;
    /**
     * @zh 進行對話，並且以串流的方式輸出
     * @en Talk to the AI and output in a streaming way
     */
    talkStream(params: {
        system?: string;
        messages: GoogleMessage[];
        onMessage: (_message: string) => void;
        onEnd: (_params: {
            isManualCancelled: boolean;
        }) => void;
        onThinking?: (_thinking: string) => void;
        onError: (_error: any) => void;
    }): {
        isManualCancelled: () => boolean;
        cancel: () => void;
    };
}
export type GoogleChatTalkResponse = Awaited<ReturnType<GoogleChat['talk']>>;
export {};