import type { AgentToolContext, ToolCallContext } from "@pk-nerdsaver-ai/pi-agent-core";
import type { CustomToolContext } from "../extensibility/custom-tools/types";
import type { ExtensionUIContext } from "../extensibility/extensions/types";
declare module "@pk-nerdsaver-ai/pi-agent-core" {
    interface AgentToolContext extends CustomToolContext {
        ui?: ExtensionUIContext;
        hasUI?: boolean;
        toolNames?: string[];
        toolCall?: ToolCallContext;
    }
}
export declare class ToolContextStore {
    #private;
    private readonly getBaseContext;
    constructor(getBaseContext: () => CustomToolContext);
    getContext(toolCall?: ToolCallContext): AgentToolContext;
    setUIContext(uiContext: ExtensionUIContext, hasUI: boolean): void;
    setToolNames(names: string[]): void;
}
