import * as m3o from '@m3o/m3o-node';
export declare class AiService {
    private client;
    constructor(token: string);
    chat(request: ChatRequest): Promise<ChatResponse>;
    complete(request: CompleteRequest): Promise<CompleteResponse>;
    edit(request: EditRequest): Promise<EditResponse>;
    generate(request: GenerateRequest): Promise<GenerateResponse>;
    moderate(request: ModerateRequest): Promise<ModerateResponse>;
    stream(request: StreamRequest): Promise<m3o.Stream<StreamRequest, StreamResponse>>;
}
export interface ChatRequest {
    context?: {
        [key: string]: any;
    }[];
    model?: string;
    prompt?: string;
    role?: string;
}
export interface ChatResponse {
    reply?: string;
}
export interface CompleteRequest {
    text?: string;
}
export interface CompleteResponse {
    text?: string;
}
export interface Context {
    prompt?: string;
    reply?: string;
}
export interface EditRequest {
    instruction?: string;
    text?: string;
}
export interface EditResponse {
    text?: string;
}
export interface GenerateRequest {
    limit?: number;
    size?: string;
    text?: string;
}
export interface GenerateResponse {
    images?: Image[];
}
export interface Image {
    base64?: string;
    url?: string;
}
export interface ModerateRequest {
    text?: string;
}
export interface ModerateResponse {
    categories?: {
        [key: string]: boolean;
    };
    flagged?: boolean;
    scores?: {
        [key: string]: number;
    };
}
export interface StreamRequest {
    model?: string;
    prompt?: string;
}
export interface StreamResponse {
    partial?: boolean;
    words?: string[];
}