`ReadonlyRecord.get` type error after Effect upgrade

import { Option, ReadonlyArray, ReadonlyRecord } from "effect";
import { constant, flow } from "effect/Function";

export type SortBy = <SortKey extends string, A, B>(_: {
  keys: readonly SortKey[];
  getKey: (_: A) => SortKey;
  reduce: (_: A[]) => B;
}) => (self: readonly A[]) => Option.Option<B>[];

export const sortBy: SortBy = ({ keys, getKey, reduce }) =>
  keys.length === 0
    ? constant([])
    : flow(
        ReadonlyArray.groupBy(getKey),
        ReadonlyRecord.map(reduce),
        (selfByKey) =>
          // @ts-expect-error Type 'string' is not assignable to type 'NonLiteralKey<SortKey>'
          ReadonlyArray.map(keys, (key) => ReadonlyRecord.get(selfByKey, key))
      );

import { Option, ReadonlyArray, ReadonlyRecord } from "effect";
import { constant, flow } from "effect/Function";

export type SortBy = <SortKey extends string, A, B>(_: {
  keys: readonly SortKey[];
  getKey: (_: A) => SortKey;
  reduce: (_: A[]) => B;
}) => (self: readonly A[]) => Option.Option<B>[];

export const sortBy: SortBy = ({ keys, getKey, reduce }) =>
  keys.length === 0
    ? constant([])
    : flow(
        ReadonlyArray.groupBy(getKey),
        ReadonlyRecord.map(reduce),
        (selfByKey) =>
          // @ts-expect-error Type 'string' is not assignable to type 'NonLiteralKey<SortKey>'
          ReadonlyArray.map(keys, (key) => ReadonlyRecord.get(selfByKey, key))
      );