Is your feature request related to a problem? Please describe.

Looking for a way to define relationships between arguments in the tools inputSchema.

Describe the solution you'd like

Allow to use ZodDiscriminatedUnion as inputschema.

Describe alternatives you've considered

• Use of zod .refine method.
• Check the arguments relationship consistency in the tools handler.