import { z, ZodTypeAny, ZodRawShape } from 'zod';
import { CallToolResultSchema } from './CallToolResultSchema';
type Primitive = string | number | boolean | bigint | null | undefined;
type Flatten<T> = T extends Primitive ? T : T extends Array<infer U> ? Array<Flatten<U>> : T extends Set<infer U> ? Set<Flatten<U>> : T extends Map<infer K, infer V> ? Map<Flatten<K>, Flatten<V>> : T extends object ? {
    [K in keyof T]: Flatten<T[K]>;
} : T;
type Infer<Schema extends ZodTypeAny> = Flatten<z.infer<Schema>>;
export type CallToolResult = Infer<typeof CallToolResultSchema>;
export type ESTool = {
    name: string;
    params: {
        description: string;
        inputSchema: ZodRawShape;
        outputSchema?: ZodRawShape;
    };
    func: (args: any) => Promise<CallToolResult>;
};
/**
 * MCP工具类型
 */
export type MCPToolParams = {
    inputSchema: {
        [x: string]: unknown;
        type: "object";
        properties?: Record<string, object> | undefined;
        required?: string[] | undefined;
    };
    name: string;
    description?: string | undefined;
    outputSchema?: {
        [x: string]: unknown;
        type: "object";
        properties?: Record<string, object> | undefined;
        required?: string[] | undefined;
    } | undefined;
    annotations?: {
        title?: string | undefined;
        readOnlyHint?: boolean | undefined;
        destructiveHint?: boolean | undefined;
        idempotentHint?: boolean | undefined;
        openWorldHint?: boolean | undefined;
    } | undefined;
    execution?: {
        taskSupport?: "optional" | "required" | "forbidden" | undefined;
    } | undefined;
    _meta?: Record<string, unknown> | undefined;
    icons?: {
        src: string;
        mimeType?: string | undefined;
        sizes?: string[] | undefined;
        theme?: "light" | "dark" | undefined;
    }[] | undefined;
    title?: string | undefined;
};
export interface ToolCallParams {
    name: string;
    arguments: Record<string, any>;
}
export {};