Struggling to setup postgresql in prod, pglite in dev

import { defineConfig } from "drizzle-kit";

export default defineConfig({
  out: "./drizzle",
  schema: "./lib/server/schema/index.ts",
  dialect: "postgresql",
  driver: "pglite",
  dbCredentials: {
    url: "./dev-db",
  },
});

import { defineConfig } from "drizzle-kit";
import { env } from "~/lib/env/server";

export default defineConfig({
   out: "./drizzle",
   schema: "./lib/server/schema/index.ts",
   breakpoints: true,
   dialect: "postgresql",
   dbCredentials: {
      url: env.DATABASE_URL,
   },
});

import { drizzle as drizzlePostgres } from "drizzle-orm/postgres-js";
import { drizzle as drizzlePGLite } from "drizzle-orm/pglite";
import postgres from "postgres";
import { env } from "../env/server";
import * as schema from "./schema";
import { migrate } from "drizzle-orm/pglite/migrator";

export const table = schema;

function initDb() {
   if (import.meta.env.PROD) {
      const driver = postgres(env.DATABASE_URL);
      const db = drizzlePostgres(driver, { schema });
      return db;
   } 
   return drizzlePGLite("./dev-db", { schema });
}

export const db = initDb();