type ToolExecuteFunction<TInput, TOutput, TContext> = undefined extends TContext ? (args, context?) => 
  | Promise<InferSchemaType<TOutput>>
  | InferSchemaType<TOutput> : (args, context) => 
  | Promise<InferSchemaType<TOutput>>
| InferSchemaType<TOutput>;

type ToolExecuteFunction<TInput, TOutput, TContext> = undefined extends TContext ? (args, context?) => 
  | Promise<InferSchemaType<TOutput>>
  | InferSchemaType<TOutput> : (args, context) => 
  | Promise<InferSchemaType<TOutput>>
| InferSchemaType<TOutput>;