How to create the type of the client with the schema

I want to create a type for the drizzle client with my schema. What I'm doing now is this:
const client = () => drizzle(neon(connectionString!), { schema });

type DrizzleClient = ReturnType<typeof client>

but I want to be able to do something like:
type DrizzleClient = typeof drizzle<schema>


Is there a way to do that?
Was this page helpful?