How can one make `jsonObjectFrom` respect nullability of underlying subquery?

export interface Account {
  id: string;
  name: string;
  ...
}

export interface User {
  id: string;
  account_id: string;
  ...
}

// part of select query for User
...
jsonObjectFrom(
  eb
  .selectFrom('account')
  .selectAll()
  .whereRef(
    'account.id',
    '=',
    'user.account_id'
   ),
).as('state'),

export interface Account {
  id: string;
  name: string;
  ...
}

export interface User {
  id: string;
  account_id: string;
  ...
}

// part of select query for User
...
jsonObjectFrom(
  eb
  .selectFrom('account')
  .selectAll()
  .whereRef(
    'account.id',
    '=',
    'user.account_id'
   ),
).as('state'),

function notNull<T>(expr: AliasableExpression<T>) {
  return expr.$castTo<Exclude<T, null>>()
}

notNull(
  jsonObjectFrom(
    eb
    .selectFrom('account')
    .selectAll()
    .whereRef(
      'account.id',
      '=',
      'user.account_id'
     ),
  )
).as('state'),

function notNull<T>(expr: AliasableExpression<T>) {
  return expr.$castTo<Exclude<T, null>>()
}

notNull(
  jsonObjectFrom(
    eb
    .selectFrom('account')
    .selectAll()
    .whereRef(
      'account.id',
      '=',
      'user.account_id'
     ),
  )
).as('state'),

function notNull<T>(expr: AliasableExpression<T>) {
  return expr as unknown as AliasableExpression<Exclude<T, null>>
}

function notNull<T>(expr: AliasableExpression<T>) {
  return expr as unknown as AliasableExpression<Exclude<T, null>>
}