[BUG]: Neon serverless driver accepts authToken as a promise, but the $withAuth does not

[pffigueiredo]

Report hasn't been filed before.

• I have verified that the bug I'm about to report hasn't been filed before.

What version of drizzle-orm are you using?

0.36.4

What version of drizzle-kit are you using?

0.27.1

Other packages

No response

Describe the Bug

Description

The @neondatabase/serverless driver allows to pass in authToken as functions, however, the recently introduced $withAuth only accepts string.

e.g.

Working example

 const db = drizzle(
        neon(process.env.NEXT_PUBLIC_DATABASE_AUTHENTICATED_URL!, {
              // returning a promise here works ✅
          authToken: () => Promise.resolve('my token'),
        }),
        { schema }
 );

Bugged example

      const db = drizzle(
        neon(process.env.NEXT_PUBLIC_DATABASE_AUTHENTICATED_URL!),
        { schema }
      );
      return db
      // only allows `string` ❌
        .$withAuth(() => Promise.resolve('my token'))
        .select()
        .from(schema.todos);

---

For this particular example, it doesn't seem too worrisome, but drizzle-orm should still use the same types as the Neon driver when possible, which seems like the case here 👀

[AndriiSherman]

Fixed in drizzle-orm@0.37.0