Type Annotations for Parsing Specific Properties with Zod

[zaru]

I’m working on creating a generic function that takes a Zod schema as an argument and parses only a specified property. I came up with the following implementations, and both work as expected.

However, I noticed that there are two different ways to specify the type annotations. I’m unsure which one is better or if there’s an even more optimal way to write this. I would appreciate any advice you could provide.

function validate<TShape extends ZodRawShape, K extends keyof TShape>(
    schema: ZodObject<TShape>,
    name: K,
    value: unknown,
) {
    return schema.shape[name].safeParse(value);
}

function validate2<T extends AnyZodObject, K extends keyof T["shape"]>(
    schema: T,
    key: K,
    value: unknown,
) {
    return schema.shape[key].safeParse(value);
}

Which of these type annotations is preferable, or is there a better way to achieve this functionality?

Thank you in advance for your help!