import { Agent } from './agent';
import { RunContextWrapper } from './run-context';
import { Tool } from './tools';
/**
 * A class that receives callbacks on various lifecycle events in an agent run. Subclass and
 * override the methods you need.
 */
export declare class RunHooks<TContext> {
    /**
     * Called before the agent is invoked. Called each time the current agent changes.
     */
    onAgentStart(context: RunContextWrapper<TContext>, agent: Agent<TContext>): Promise<void>;
    /**
     * Called when the agent produces a final output.
     */
    onAgentEnd(context: RunContextWrapper<TContext>, agent: Agent<TContext>, output: any): Promise<void>;
    /**
     * Called when a handoff occurs.
     */
    onHandoff(context: RunContextWrapper<TContext>, fromAgent: Agent<TContext>, toAgent: Agent<TContext>): Promise<void>;
    /**
     * Called before a tool is invoked.
     */
    onToolStart(context: RunContextWrapper<TContext>, agent: Agent<TContext>, tool: Tool): Promise<void>;
    /**
     * Called after a tool is invoked.
     */
    onToolEnd(context: RunContextWrapper<TContext>, agent: Agent<TContext>, tool: Tool, result: string): Promise<void>;
}
/**
 * A class that receives callbacks on various lifecycle events for a specific agent. You can
 * set this on `agent.hooks` to receive events for that specific agent.
 *
 * Subclass and override the methods you need.
 */
export declare class AgentHooks<TContext> {
    /**
     * Called before the agent is invoked. Called each time the running agent is changed to this
     * agent.
     */
    onStart(context: RunContextWrapper<TContext>, agent: Agent<TContext>): Promise<void>;
    /**
     * Called when the agent produces a final output.
     */
    onEnd(context: RunContextWrapper<TContext>, agent: Agent<TContext>, output: any): Promise<void>;
    /**
     * Called when the agent is being handed off to. The `source` is the agent that is handing
     * off to this agent.
     */
    onHandoff(context: RunContextWrapper<TContext>, agent: Agent<TContext>, source: Agent<TContext>): Promise<void>;
    /**
     * Called before a tool is invoked.
     */
    onToolStart(context: RunContextWrapper<TContext>, agent: Agent<TContext>, tool: Tool): Promise<void>;
    /**
     * Called after a tool is invoked.
     */
    onToolEnd(context: RunContextWrapper<TContext>, agent: Agent<TContext>, tool: Tool, result: string): Promise<void>;
}