import type { AgentEndEvent, ExtensionAPI, ExtensionContext, ExtensionFactory, TurnEndEvent } from '@earendil-works/pi-coding-agent';
import type { WaggleConfig, WaggleTurn } from '@openwaggle/waggle-core';
export type PiWaggleModel = NonNullable<ReturnType<ExtensionContext['modelRegistry']['find']>>;
export type PiWaggleCustomMessage = Parameters<ExtensionAPI['sendMessage']>[0];
export type PiWaggleSendMessageOptions = NonNullable<Parameters<ExtensionAPI['sendMessage']>[1]>;
export interface PiWaggleTurnMetadataInput {
    readonly turnNumber: number;
    readonly agentIndex: number;
}
export interface PiWaggleTurnCompleteInput<TMeta> {
    readonly turn: WaggleTurn;
    readonly meta: TMeta;
    readonly messages: AgentEndEvent['messages'];
}
export interface PiWaggleTurnDecision {
    readonly continue: boolean;
}
export interface PiWaggleTurnMessageInput<TMeta> {
    readonly model: PiWaggleModel;
    readonly turn: WaggleTurn;
    readonly meta: TMeta;
}
export interface PiWaggleStartNextTurnInput<TMeta> extends PiWaggleTurnMessageInput<TMeta> {
    readonly message: PiWaggleCustomMessage;
}
export interface PiWaggleResolveTurnModelInput<TMeta> {
    readonly ctx: PiWaggleExtensionContext;
    readonly turn: WaggleTurn;
    readonly meta: TMeta;
}
export interface PiWaggleExtensionInput<TMeta> {
    readonly config: WaggleConfig;
    readonly createTurnMetadata: (input: PiWaggleTurnMetadataInput) => TMeta;
    readonly onTurnComplete: (input: PiWaggleTurnCompleteInput<TMeta>) => PiWaggleTurnDecision | Promise<PiWaggleTurnDecision>;
    readonly buildTurnMessage: (input: PiWaggleTurnMessageInput<TMeta>) => PiWaggleCustomMessage;
    readonly resolveTurnModel?: (input: PiWaggleResolveTurnModelInput<TMeta>) => PiWaggleModel | Promise<PiWaggleModel>;
    readonly startNextTurn?: (input: PiWaggleStartNextTurnInput<TMeta>) => void | Promise<void>;
    readonly canStartNextTurn?: () => boolean;
    readonly onActiveTurnChange?: (meta: TMeta) => void;
    readonly onTurnStart?: (meta: TMeta) => void;
}
export interface PiWaggleLoopInput<TMeta> extends PiWaggleExtensionInput<TMeta> {
    readonly onComplete: () => void;
    readonly onError: (error: unknown) => void;
}
export type PiWaggleAgentEndHandler = (event: AgentEndEvent, ctx: PiWaggleExtensionContext) => Promise<void> | void;
export type PiWaggleTurnEndHandler = (event: TurnEndEvent, ctx: PiWaggleExtensionContext) => Promise<void> | void;
export interface PiWaggleExtensionContext {
    readonly modelRegistry: Pick<ExtensionContext['modelRegistry'], 'find'>;
}
export interface PiWaggleLoopApi {
    readonly sendMessage: ExtensionAPI['sendMessage'];
    readonly setModel: ExtensionAPI['setModel'];
}
export interface PiWaggleExtensionApi extends PiWaggleLoopApi {
    readonly onAgentEnd: (handler: PiWaggleAgentEndHandler) => void;
}
export interface PiWaggleExtensionController {
    readonly factory: ExtensionFactory;
    readonly done: Promise<void>;
}
export declare function createPiWaggleLoopHandler<TMeta>(input: PiWaggleLoopInput<TMeta>, api: PiWaggleLoopApi): PiWaggleAgentEndHandler;
export interface PiWaggleTurnCompletionHandlers {
    readonly onTurnEnd: PiWaggleTurnEndHandler;
    readonly onAgentEnd: PiWaggleAgentEndHandler;
}
export declare function createPiWaggleTurnCompletionHandlers<TMeta>(input: PiWaggleLoopInput<TMeta>, api: PiWaggleLoopApi): PiWaggleTurnCompletionHandlers;
export declare function createPiWaggleTurnEndHandler<TMeta>(input: PiWaggleLoopInput<TMeta>, api: PiWaggleLoopApi): PiWaggleTurnEndHandler;
export declare function registerPiWaggleLoop<TMeta>(input: PiWaggleLoopInput<TMeta>, api: PiWaggleExtensionApi): void;
export declare function createPiWaggleExtension<TMeta>(input: PiWaggleExtensionInput<TMeta>): PiWaggleExtensionController;