import type { ClientExtensions } from '@aws-amplify/data-schema/runtime';
import type { getSchema } from '../types';
import type { AiClientState } from './shared';
export interface UseAIGenerationHookWrapper<Key extends keyof AIGenerationClient<Schema>['generations'], Schema extends Record<any, any>> {
    useAIGeneration: <U extends Key>(routeName: U) => [
        Awaited<AiClientState<Schema[U]['returnType']>>,
        (input: Schema[U]['args']) => void
    ];
}
export type UseAIGenerationHook<Key extends keyof AIGenerationClient<Schema>['generations'], Schema extends Record<any, any>> = (routeName: Key) => [
    Awaited<AiClientState<Schema[Key]['returnType']>>,
    (input: Schema[Key]['args']) => void
];
type AIGenerationClient<T extends Record<any, any>> = Pick<ClientExtensions<T>, 'generations'>;
export declare function createUseAIGeneration<Client extends Record<'generations' | 'conversations', Record<string, any>>, Schema extends getSchema<Client>>(client: Client): UseAIGenerationHook<keyof Client['generations'], Client>;
export {};