import { z, ZodRawShape } from "zod";
import { Message } from "@langchain/langgraph-sdk";
import { ToolRenderData } from "./ToolUI.js";
import { HumanInTheLoopDecision, InterruptResponse as HumanInTheLoopResponse } from "../humanInTheLoop.js";
export interface UnionTool<Args extends ZodRawShape, Child extends Object = Object, ResponseType = any> {
    name: string;
    description: string;
    parameters: Args;
    /** 是否直接返回工具结果，而不是通过消息返回 */
    returnDirect?: boolean;
    execute?: ToolCallback<Args>;
    /** 工具执行成功后触发的附加消息 */
    callbackMessage?: (result: CallToolResult) => Message[];
    handler?: (args: z.infer<z.ZodObject<Args>>, context?: any) => ResponseType | Promise<ResponseType>;
    render?: (tool: ToolRenderData<z.infer<z.ZodObject<Args>>, ResponseType>) => Child;
    onlyRender?: boolean;
    /** 只允许指定的 agent 使用该工具，如果未指定，则所有 agent 都可以使用 */
    allowAgent?: string[];
    /** 只允许指定的 Graph 使用该工具 */
    allowGraph?: string[];
    /** 是否是纯净的 json schema 参数，而不是 zod 参数 */
    isPureParams?: boolean;
}
export type ToolCallback<Args extends ZodRawShape> = (args: z.infer<z.ZodObject<Args>>, context?: any) => CallToolResult | Promise<CallToolResult>;
export type CallToolResult = string | {
    type: "text";
    text: string;
}[] | HumanInTheLoopDecision | HumanInTheLoopResponse;
/** 用于格式校验 */
export declare const createTool: <Args extends ZodRawShape>(tool: UnionTool<Args>) => UnionTool<Args, Object, any>;
/**
 * create Type Safe Tool with zod and UI Render Feature
 */
export declare const createUITool: <Args extends ZodRawShape, Child extends Object = {}>(tool: UnionTool<Args, Child>) => UnionTool<Args, Child>;
export declare const createRenderUITool: <Args extends ZodRawShape, Child extends Object = {}>(tool: UnionTool<Args, Child>) => UnionTool<Args, Child>;
export declare const createInteractiveUITool: <Args extends ZodRawShape, Child extends Object = {}>(tool: UnionTool<Args, Child>) => UnionTool<Args, Child>;
export declare const createJSONDefineTool: <Args extends ZodRawShape>(tool: UnionTool<Args>) => {
    name: string;
    description: string;
    parameters: Args | ({
        anyOf: import("zod-to-json-schema").JsonSchema7DateType[];
    } & {
        title?: string;
        default?: any;
        description?: string;
        markdownDescription?: string;
    } & {
        $schema?: string | undefined;
        definitions?: {
            [key: string]: import("zod-to-json-schema").JsonSchema7Type;
        } | undefined;
    }) | ({
        type: "object" | "array";
    } & {
        title?: string;
        default?: any;
        description?: string;
        markdownDescription?: string;
    } & {
        $schema?: string | undefined;
        definitions?: {
            [key: string]: import("zod-to-json-schema").JsonSchema7Type;
        } | undefined;
    }) | ({
        type: ("string" | "number" | "boolean" | "null" | "integer") | ("string" | "number" | "boolean" | "null" | "integer")[];
    } & {
        title?: string;
        default?: any;
        description?: string;
        markdownDescription?: string;
    } & {
        $schema?: string | undefined;
        definitions?: {
            [key: string]: import("zod-to-json-schema").JsonSchema7Type;
        } | undefined;
    }) | ({
        anyOf: import("zod-to-json-schema").JsonSchema7Type[];
    } & {
        title?: string;
        default?: any;
        description?: string;
        markdownDescription?: string;
    } & {
        $schema?: string | undefined;
        definitions?: {
            [key: string]: import("zod-to-json-schema").JsonSchema7Type;
        } | undefined;
    }) | (import("zod-to-json-schema").JsonSchema7UndefinedType & {
        title?: string;
        default?: any;
        description?: string;
        markdownDescription?: string;
    } & {
        $schema?: string | undefined;
        definitions?: {
            [key: string]: import("zod-to-json-schema").JsonSchema7Type;
        } | undefined;
    }) | (import("zod-to-json-schema").JsonSchema7AnyType & {
        title?: string;
        default?: any;
        description?: string;
        markdownDescription?: string;
    } & {
        $schema?: string | undefined;
        definitions?: {
            [key: string]: import("zod-to-json-schema").JsonSchema7Type;
        } | undefined;
    }) | ({
        anyOf: [import("zod-to-json-schema").JsonSchema7Type, import("zod-to-json-schema").JsonSchema7NullType];
    } & {
        title?: string;
        default?: any;
        description?: string;
        markdownDescription?: string;
    } & {
        $schema?: string | undefined;
        definitions?: {
            [key: string]: import("zod-to-json-schema").JsonSchema7Type;
        } | undefined;
    }) | ({
        type: [string, "null"];
    } & {
        title?: string;
        default?: any;
        description?: string;
        markdownDescription?: string;
    } & {
        $schema?: string | undefined;
        definitions?: {
            [key: string]: import("zod-to-json-schema").JsonSchema7Type;
        } | undefined;
    }) | (import("zod-to-json-schema").JsonSchema7AllOfType & {
        title?: string;
        default?: any;
        description?: string;
        markdownDescription?: string;
    } & {
        $schema?: string | undefined;
        definitions?: {
            [key: string]: import("zod-to-json-schema").JsonSchema7Type;
        } | undefined;
    });
};