import { AxiosResponse } from 'axios';
import { AnyObject } from '../../types';
import { IGenerate, IGenerateGroupedResultMessages, IGenerateGroupedResultTask, IGenerateSingleResultMessages, IGenerateSingleResultPrompt } from '../filters/interfaces/Generate.interface';
import { GetQueryGenerativeGroupedResult, GetQueryGenerativeSingleResult, GetQueryResult } from '../Unbody';
import { QueryBuilder } from './QueryBuilder';
export interface GenerateQuery<T extends AnyObject, Q extends QueryBuilder<T, GetQueryResult<T>>> {
    (type: 'singleResult', prompt: Required<IGenerate<T>>['singleResult']['prompt']): Omit<Q, 'exec'> & {
        exec: <R = GetQueryGenerativeSingleResult<GetQueryResult<T>>>() => Promise<AxiosResponse<R>>;
    };
    (type: 'singleResult', args: IGenerateSingleResultPrompt): Omit<Q, 'exec'> & {
        exec: <R = GetQueryGenerativeSingleResult<GetQueryResult<T>>>() => Promise<AxiosResponse<R>>;
    };
    (type: 'singleResult', args: IGenerateSingleResultMessages): Omit<Q, 'exec'> & {
        exec: <R = GetQueryGenerativeSingleResult<GetQueryResult<T>>>() => Promise<AxiosResponse<R>>;
    };
    (type: 'groupedResult', prompt: IGenerateGroupedResultTask<T>): Omit<Q, 'exec'> & {
        exec: <R = GetQueryGenerativeGroupedResult<GetQueryResult<T>>>() => Promise<AxiosResponse<R>>;
    };
    (type: 'groupedResult', prompt: IGenerateGroupedResultMessages): Omit<Q, 'exec'> & {
        exec: <R = GetQueryGenerativeGroupedResult<GetQueryResult<T>>>() => Promise<AxiosResponse<R>>;
    };
    (type: 'groupedResult', prompt: Required<IGenerate<T>>['groupedResult']['task'], properties?: Required<IGenerate<T>>['groupedResult']['properties']): Omit<Q, 'exec'> & {
        exec: <R = GetQueryGenerativeGroupedResult<GetQueryResult<T>>>() => Promise<AxiosResponse<R>>;
    };
    (type: 'groupedResult' | 'singleResult', prompt: Required<IGenerate<T>>['groupedResult']['task'] | Required<IGenerate<T>>['singleResult']['prompt'] | IGenerateSingleResultPrompt | IGenerateSingleResultMessages | IGenerateGroupedResultTask<T> | IGenerateGroupedResultMessages, properties?: Required<IGenerate<T>>['groupedResult']['properties']): Omit<Q, 'exec'> & {
        exec: <R = GetQueryGenerativeSingleResult<GetQueryResult<T>> | GetQueryGenerativeGroupedResult<GetQueryResult<T>>>() => Promise<AxiosResponse<R>>;
    };
    fromOne(prompt: string): Omit<Q, 'exec'> & {
        exec: <R = GetQueryGenerativeSingleResult<GetQueryResult<T>>>() => Promise<AxiosResponse<R>>;
    };
    fromOne(prompt: IGenerateSingleResultPrompt): Omit<Q, 'exec'> & {
        exec: <R = GetQueryGenerativeSingleResult<GetQueryResult<T>>>() => Promise<AxiosResponse<R>>;
    };
    fromOne(prompt: IGenerateSingleResultMessages): Omit<Q, 'exec'> & {
        exec: <R = GetQueryGenerativeSingleResult<GetQueryResult<T>>>() => Promise<AxiosResponse<R>>;
    };
    fromOne(prompt: string | IGenerateSingleResultPrompt | IGenerateSingleResultMessages): Omit<Q, 'exec'> & {
        exec: <R = GetQueryGenerativeSingleResult<GetQueryResult<T>>>() => Promise<AxiosResponse<R>>;
    };
    fromMany(task: IGenerateGroupedResultTask<T>): Omit<Q, 'exec'> & {
        exec: <R = GetQueryGenerativeGroupedResult<GetQueryResult<T>>>() => Promise<AxiosResponse<R>>;
    };
    fromMany(task: IGenerateGroupedResultMessages): Omit<Q, 'exec'> & {
        exec: <R = GetQueryGenerativeGroupedResult<GetQueryResult<T>>>() => Promise<AxiosResponse<R>>;
    };
    fromMany(task: string, properties?: Required<IGenerate<T>>['groupedResult']['properties']): Omit<Q, 'exec'> & {
        exec: <R = GetQueryGenerativeGroupedResult<GetQueryResult<T>>>() => Promise<AxiosResponse<R>>;
    };
}
export declare const createGenerateQuery: <T extends AnyObject, Q extends QueryBuilder<T, any>>(queryBuilder: Q) => GenerateQuery<T, Q>;
//# sourceMappingURL=GenerateQuery.d.ts.map