Mapped Types Not Inferring From Promise Correctly

export const FileRouter = UploadThingServerHelper({
  someKey: {
    middleware: async (req) => {
      const user = await auth(req);

      if (!user.userId) throw new Error("someProperty is required");

      return { userIds: user.userId };
    },
    onUpload: async (request) => {
      console.log("uploaded with the following metadata:", request.metadata); // THIS SHOULD INFER THE TYPE FROM THE RETURN
    },
  },
});

export const FileRouter = UploadThingServerHelper({
  someKey: {
    middleware: async (req) => {
      const user = await auth(req);

      if (!user.userId) throw new Error("someProperty is required");

      return { userIds: user.userId };
    },
    onUpload: async (request) => {
      console.log("uploaded with the following metadata:", request.metadata); // THIS SHOULD INFER THE TYPE FROM THE RETURN
    },
  },
});

type UploadableInputParams<TMiddlewareResponse> = {
  middleware: (request: Request) => Promise<TMiddlewareResponse>;
  onUpload: (response: {
    metadata: TMiddlewareResponse;
  }) => void;
};

export const UploadThingServerHelper = <TValidRoutes>(
  input: {
    readonly [K in keyof TValidRoutes]: UploadableInputParams<TValidRoutes[K]>;
  },
  config?: { callbackUrl?: string }
) => {...businessLogic};

type UploadableInputParams<TMiddlewareResponse> = {
  middleware: (request: Request) => Promise<TMiddlewareResponse>;
  onUpload: (response: {
    metadata: TMiddlewareResponse;
  }) => void;
};

export const UploadThingServerHelper = <TValidRoutes>(
  input: {
    readonly [K in keyof TValidRoutes]: UploadableInputParams<TValidRoutes[K]>;
  },
  config?: { callbackUrl?: string }
) => {...businessLogic};