Is it possible to Typescript behavior with generic key that extends Union types? - arktype