Typesafe references?

export const a = pgTable('a', {
  id: uuid("id").primaryKey().defaultRandom()
});
export const b = pgTable('b', {
  id: uuid("id").primaryKey().defaultRandom(),
  aId: timestamp('a_id').references(() => a.id).notNull()
});