import { type IChatModel } from "@empiricalrun/llm/chat";
import type { CanonicalMessage, ChatState, ChatStateGeneric, MessageCheckpoint, PendingToolCall, ProcessToolResultsResult, ReporterFunction, StreamingMessageReporterFunc, SupportedChatModels, TextMessagePart, TextMessageWithAttachmentsPart, ToolResult, ToolsForLLM } from "@empiricalrun/shared-types/chat-agent";
import type { TraceClient } from "@empiricalrun/shared-types/common";
import type { FileInfo } from "@empiricalrun/shared-types/test-gen";
import type { IToolExecutor } from "../../tools/executor/types";
type RunLoopParams<T> = {
    signal?: AbortSignal;
    reporter: ReporterFunction<T>;
    streamingMessageReporter?: StreamingMessageReporterFunc;
    onLLMResponse?: (response: T, selectedModel: SupportedChatModels) => Promise<void>;
};
type AgentParamsWithoutPrompt<T> = {
    trace?: TraceClient;
    chatState?: ChatStateGeneric<T>;
    workerEnv?: Record<string, string>;
    featureFlags?: string[];
    toolExecutor?: IToolExecutor;
    systemPrompt?: string;
    selectedModel: SupportedChatModels;
    repoInfoBuilder?: () => Promise<FileInfo>;
};
export type AgentParams<T> = AgentParamsWithoutPrompt<T> & ({
    systemPrompt: string;
} | {
    repoInfoBuilder: () => Promise<FileInfo>;
});
export type OnToolResultsProcessedCallback = (newChatState: ChatState) => Promise<void>;
export type OnToolCallCallback = (toolCalls: PendingToolCall[]) => Promise<void>;
export type OnLLMResponseCallback<T> = (response: T, selectedModel: SupportedChatModels) => Promise<void>;
export declare abstract class BaseAgent<T> {
    featureFlags: string[];
    tools: ToolsForLLM;
    selectedModel: SupportedChatModels;
    chatModel: IChatModel<T>;
    workerEnv?: Record<string, string>;
    toolExecutor?: IToolExecutor;
    trace?: TraceClient;
    systemPrompt: string | undefined;
    repoInfoBuilder: (() => Promise<FileInfo>) | undefined;
    onToolResultsProcessed?: OnToolResultsProcessedCallback;
    onToolCall?: OnToolCallCallback;
    onLLMResponse?: OnLLMResponseCallback<T>;
    constructor({ featureFlags, selectedModel, workerEnv, chatState, toolExecutor, trace, systemPrompt, repoInfoBuilder, }: AgentParams<T>);
    protected abstract getTools(): ToolsForLLM;
    protected abstract buildSystemPrompt(repoInfoBuilder: () => Promise<FileInfo>): Promise<string>;
    get messages(): CanonicalMessage[];
    get askUserForInput(): boolean;
    get chatState(): ChatStateGeneric<T>;
    pushUserMessage(parts: Array<TextMessagePart | TextMessageWithAttachmentsPart>): ChatStateGeneric<T>;
    setChatState(chatState: ChatStateGeneric<T>): void;
    changeModel(newModel: SupportedChatModels): void;
    stop(): ChatStateGeneric<T>;
    processToolResults(toolCalls: PendingToolCall[], toolResults: ToolResult[], checkpoint: MessageCheckpoint | null): ProcessToolResultsResult;
    toolCallHandler(toolCalls: PendingToolCall[]): Promise<void>;
    runLoop(params: RunLoopParams<any>): Promise<void>;
    getHumanReadableLatestMessage(): {
        role: string;
        textMessage: string;
    } | undefined;
}
export {};
//# sourceMappingURL=index.d.ts.map