Returned type inference when using tanstack query

type GetSetGroupsArgs = Parameters<typeof db.query.setGroup.findMany>[0];
export const useSetGroups = (args?: GetSetGroupsArgs) => {
  return useQuery({
    queryKey: ['setGroups', args],
    queryFn: () => db.query.setGroup.findMany(args),
  });
};

  const { data, error, status } = useSetGroups({
    where: eq(setGroup.workoutId, id),
    with: { sets: { with: { exercise: true } } },
  });