How can i assure to typescript the argument type of useMutation ?

export const useMutatePost = (type: "create" | "update" | "delete") => {
  const queryClient = useQueryClient();
  switch (type) {
    case "create":
      return useMutation((payload: CreatePost) => createPost(payload), {
        onSuccess: () => {
          queryClient.invalidateQueries(["profiles"]);
          queryClient.invalidateQueries(["user"]);
          queryClient.invalidateQueries(["posts"]);
        },
        onError: () => {
          throw new Error("Erro");
        },
      });

    case "update":
      return useMutation((payload: MutationPayload) => updatePost(payload), {
        onSuccess: (updatedPost: any) => {
          queryClient.setQueryData(["posts"], (prev: any) =>
            prev.map((post: Post) =>
              post.id === updatedPost.id ? updatedPost : post
            )
          );
        },
      });

    case "delete":
      return useMutation((payload: MutationPayload) => deletePost(payload), {
        onSuccess: (deletedPost: Post) => {
          queryClient.setQueryData(["posts"], (prev: any) =>
            prev.filter((post: Post) => post.id != deletedPost.id)
          );
        },
      });
  }
};