How to type a function with column key of table - Drizzle Team