import { AsyncLocalStorage } from "node:async_hooks";
import { ContextGetter } from "@auth0/ai/authorizers";
import { ToolWrapper } from "../types";
type LlamaIndexContext = {
    threadID: string;
};
export declare const aiContext: AsyncLocalStorage<LlamaIndexContext>;
/**
 *
 * Set the context for the current async execution.
 *
 * This function should be called in the API route handler.
 *
 * @param params - The context parameters to set.
 * @param params.threadID - The thread ID to set in the context.
 */
export declare const setAIContext: (params: Pick<LlamaIndexContext, "threadID">) => void;
type TProtectFunc = (getContext: ContextGetter<[any, object | undefined]>, execute: (args_0: any, args_1: object | undefined) => any) => (args_0: any, args_1: object | undefined) => any;
export declare const createToolWrapper: (protect: TProtectFunc) => ToolWrapper;
export {};
//# sourceMappingURL=index.d.ts.map