How to pass typesafe react-hook-form form object

const form = useForm<FormData>({
    resolver: zodResolver(updateProfileSchema),
    defaultValues: {
      firstName: undefined,
      lastName: undefined,
      email: undefined,
      oldPassword: undefined,
      newPassword: undefined,
      imageUrl: undefined,
    },
    mode: 'onSubmit',
  });

<UploadButton
   form={form}
   onUploadComplete={(url: string) => setProfileImageUrl(url)}
/>

interface UploadButtonProps {
  form: UseFormReturn;
  onUploadComplete: (url: string) => void;
}

form.setValue('imageUrl', imageUrl);