import { PreviewConfig, Prompt, PromptElement, RenderOutput, SynchronousPreviewConfig, SynchronousPrompt } from './types';
import { StreamChatCompletionResponse } from './openai';
import { UsableTokenizer } from './tokenizer';
import { ChatCompletionResponseMessage, CreateChatCompletionResponse } from 'openai';
import { OutputCatcher } from './outputCatcher.ai';
export type PreviewManagerGetPromptQuery = {
    promptId: string;
    propsId: string;
    tokenLimit: number;
    tokenizer: UsableTokenizer;
    shouldBuildSourceMap: boolean;
};
export type PreviewManagerGetRemotePromptQuery = {
    promptId: string;
    promptDump: string;
    tokenLimit: number;
    tokenizer: UsableTokenizer;
};
export type PreviewManagerGetRemotePropsQuery = {
    promptId: string;
    promptDump: string;
};
export type PreviewManagerGetPromptOutputQuery = {
    promptId: string;
    propsId: string;
    tokenLimit: number;
    tokenizer: UsableTokenizer;
    completion: ChatCompletionResponseMessage | ChatCompletionResponseMessage[];
    stream: boolean;
    shouldBuildSourceMap: boolean;
};
export type PreviewManagerLiveModeQuery = {
    alreadySeenLiveModeId?: string;
};
export type PreviewManagerLiveModeResultQuery = {
    output: string;
};
export interface IPreviewManager {
    registerConfig<T>(config: PreviewConfig<T>): void;
    dump<T>(config: PreviewConfig<T>, props: T): void;
    getPreviews(): Record<string, {
        dumps: string[];
        saved: string[];
    }>;
    getPrompt(query: PreviewManagerGetPromptQuery): Promise<RenderOutput>;
    getPromptFromRemote(query: PreviewManagerGetRemotePromptQuery): Promise<RenderOutput>;
}
type LiveModeOutput = {
    liveModeId: string;
};
export declare function configFromPrompt<T, ReturnT = never>(prompt: Prompt<T, ReturnT>): PreviewConfig<T>;
export declare function configFromSynchronousPrompt<T, ReturnT = never>(prompt: SynchronousPrompt<T, ReturnT>, options?: {
    protoProps?: ProtoPropsType<T>;
}): SynchronousPreviewConfig<T, ReturnT>;
export declare function dumpProps<T, ReturnT = never>(config: PreviewConfig<T, ReturnT>, props: Omit<T, "onReturn">): string;
export declare function defaultYamlDump<T>(props: T): string;
export declare function defaultYamlLoad<T>(dump: string): T;
export type ProtoPropsType<T> = {
    fromJsonString(jsonString: string): T;
};
export type ProtoProps = {
    toJsonString(): string;
};
declare class PreviewManagerImpl implements IPreviewManager {
    private _shouldDump;
    get shouldDump(): boolean;
    set shouldDump(value: boolean);
    private readonly previews;
    getConfig(promptId: string): PreviewConfig<unknown>;
    getPreviews(): {
        [x: string]: {
            dumps: string[];
            saved: string[];
        } | {
            dumps: string[];
            saved: string[];
        };
    };
    getPrompt(query: PreviewManagerGetPromptQuery): Promise<RenderOutput>;
    getPromptOutput(query: PreviewManagerGetPromptOutputQuery): Promise<unknown>;
    getPromptFromRemote(query: PreviewManagerGetRemotePromptQuery): Promise<RenderOutput>;
    getPropsFromRemote(query: PreviewManagerGetRemotePropsQuery): Promise<unknown>;
    getPromptFunctionFromRemote(query: PreviewManagerGetRemotePropsQuery): Prompt<unknown, never>;
    getPromptFromRemoteElement(query: Omit<PreviewManagerGetRemotePromptQuery, "promptId" | "promptDump">, element: PromptElement): Promise<RenderOutput>;
    getRemoteElement(promptId: string, promptDump: string): PromptElement | Promise<PromptElement>;
    getElement(promptId: string, propsId: string, outputCatcher?: OutputCatcher<unknown>): PromptElement | Promise<PromptElement>;
    registerConfig<T, ReturnT = never>(config: PreviewConfig<T, ReturnT>): void;
    register<T, ReturnT = never>(prompt: Prompt<T, ReturnT>): void;
    registerProtoPrompt<T extends ProtoProps, ReturnT = never>(prompt: Prompt<T, ReturnT>, protoProps: ProtoPropsType<T>): void;
    configFromSynchronousPrompt<T, ReturnT = never>(prompt: SynchronousPrompt<T, ReturnT>): SynchronousPreviewConfig<T, ReturnT>;
    hydrate<T>(config: PreviewConfig<T>, dump: string): T;
    maybeDump<T, ReturnT = never>(prompt: Prompt<T, ReturnT>, props: Omit<T, "onReturn">): void;
    dump<T>(config: PreviewConfig<T>, props: T): void;
    private lastLiveModeData;
    private lastLiveModeOutputPromise;
    private resolveLastLiveModeOutputPromise;
    private liveModeResultPromise;
    private resolveLiveModeResult;
    constructor();
    getLiveModePromptCompletion(promptElement: PromptElement, options: {
        model: string;
        abortSignal?: AbortSignal;
    }): Promise<CreateChatCompletionResponse>;
    streamLiveModePromptCompletion(promptElement: PromptElement, options: {
        model: string;
        abortSignal?: AbortSignal;
    }): AsyncGenerator<StreamChatCompletionResponse>;
    liveMode(query: PreviewManagerLiveModeQuery, abortSignal?: AbortSignal): Promise<LiveModeOutput>;
    liveModeResult(query: PreviewManagerLiveModeResultQuery): void;
    private getDump;
}
export declare const PreviewManager: PreviewManagerImpl;
export declare function register(): <T, ReturnT = never>(target: any, propertyKey: string, descriptor: TypedPropertyDescriptor<Prompt<T, ReturnT>>) => void;
export {};
//# sourceMappingURL=preview.d.ts.map