Updated Prisma examples and tests

Author: ankane

.github/workflows/build.yml

@@ -18,7 +18,10 @@ jobs:
 
       # Node.js
       - uses: actions/setup-node@v4
+        with:
+          node-version: 24
       - run: npm install
+      - run: npx prisma generate
       - run: npx prisma migrate dev
       - run: npm test
 

.gitignore

@@ -1,4 +1,5 @@
 node_modules/
+prisma/generated/
 package-lock.json
 pnpm-lock.yaml
 bun.lock

README.md

@@ -405,13 +405,12 @@ Add the extension to the schema
 
 ```prisma
 generator client {
-  provider        = "prisma-client-js"
+  provider        = "prisma-client"
   previewFeatures = ["postgresqlExtensions"]
 }
 
 datasource db {
   provider   = "postgresql"
-  url        = env("DATABASE_URL")
   extensions = [vector]
 }
 ```
@@ -798,6 +797,7 @@ git clone https://github.com/pgvector/pgvector-node.git
 cd pgvector-node
 npm install
 createdb pgvector_node_test
+npx prisma generate
 npx prisma migrate dev
 npm test
 ```

package.json

@@ -95,15 +95,16 @@
   "devDependencies": {
     "@mikro-orm/core": "^6.1.1",
     "@mikro-orm/postgresql": "^6.1.1",
-    "@prisma/client": "^6.2.1",
+    "@prisma/adapter-pg": "^7.1.0",
+    "@prisma/client": "^7.1.0",
     "drizzle-orm": "^0.45.0",
     "knex": "^3.1.0",
     "kysely": "^0.28.2",
     "objection": "^3.1.3",
     "pg": "^8.6.0",
     "pg-promise": "^12.3.0",
     "postgres": "^3.3.4",
-    "prisma": "^6.2.1",
+    "prisma": "^7.1.0",
     "sequelize": "^6.6.2",
     "slonik": "^48.8.7",
     "typeorm": "^0.3.17",

prisma.config.js

@@ -0,0 +1,11 @@
+import { defineConfig, env } from 'prisma/config';
+
+export default defineConfig({
+  schema: 'prisma/schema.prisma',
+  migrations: {
+    path: 'prisma/migrations',
+  },
+  datasource: {
+    url: 'postgresql://runner@localhost/pgvector_node_test',
+  },
+});

prisma/schema.prisma

@@ -2,13 +2,15 @@
 // learn more about it in the docs: https://pris.ly/d/prisma-schema
 
 generator client {
-  provider        = "prisma-client-js"
+  provider        = "prisma-client"
+  output          = "./generated"
+  generatedFileExtension = "mts"
+  importFileExtension = "mts"
   previewFeatures = ["postgresqlExtensions"]
 }
 
 datasource db {
   provider   = "postgresql"
-  url        = "postgresql://runner@localhost/pgvector_node_test"
   extensions = [vector]
 }
 

tests/prisma.test.mjs

@@ -2,10 +2,16 @@ import assert from 'node:assert';
 import test, { beforeEach } from 'node:test';
 import pgvector from 'pgvector';
 import { SparseVector } from 'pgvector';
-import { PrismaClient } from '@prisma/client';
+import { PrismaClient } from '../prisma/generated/client.mts';
+import { PrismaPg } from '@prisma/adapter-pg';
+
+function prismaClient() {
+  const adapter = new PrismaPg({connectionString: 'postgresql://runner@localhost/pgvector_node_test'});
+  return new PrismaClient({adapter: adapter});
+}
 
 test('vector', async () => {
-  const prisma = new PrismaClient();
+  const prisma = prismaClient();
 
   // TODO use create when possible (field is not available in the generated client)
   // https://www.prisma.io/docs/concepts/components/prisma-schema/features-without-psl-equivalent#unsupported-field-types
@@ -21,10 +27,12 @@ test('vector', async () => {
   assert.deepEqual(pgvector.fromSql(items[0].embedding), [1, 1, 1]);
   assert.deepEqual(pgvector.fromSql(items[1].embedding), [1, 1, 2]);
   assert.deepEqual(pgvector.fromSql(items[2].embedding), [2, 2, 2]);
+
+  await prisma.$disconnect();
 });
 
 test('halfvec', async () => {
-  const prisma = new PrismaClient();
+  const prisma = prismaClient();
 
   // TODO use create when possible (field is not available in the generated client)
   // https://www.prisma.io/docs/concepts/components/prisma-schema/features-without-psl-equivalent#unsupported-field-types
@@ -40,10 +48,12 @@ test('halfvec', async () => {
   assert.deepEqual(pgvector.fromSql(items[0].half_embedding), [1, 1, 1]);
   assert.deepEqual(pgvector.fromSql(items[1].half_embedding), [1, 1, 2]);
   assert.deepEqual(pgvector.fromSql(items[2].half_embedding), [2, 2, 2]);
+
+  await prisma.$disconnect();
 });
 
 test('bit', async () => {
-  const prisma = new PrismaClient();
+  const prisma = prismaClient();
 
   await prisma.item.createMany({
     data: [
@@ -60,10 +70,12 @@ test('bit', async () => {
   assert.equal(items[0].binary_embedding, '101');
   assert.equal(items[1].binary_embedding, '111');
   assert.equal(items[2].binary_embedding, '000');
+
+  await prisma.$disconnect();
 });
 
 test('sparsevec', async () => {
-  const prisma = new PrismaClient();
+  const prisma = prismaClient();
 
   // TODO use create when possible (field is not available in the generated client)
   // https://www.prisma.io/docs/concepts/components/prisma-schema/features-without-psl-equivalent#unsupported-field-types
@@ -79,9 +91,12 @@ test('sparsevec', async () => {
   assert.deepEqual(pgvector.fromSql(items[0].sparse_embedding).toArray(), [1, 1, 1]);
   assert.deepEqual(pgvector.fromSql(items[1].sparse_embedding).toArray(), [1, 1, 2]);
   assert.deepEqual(pgvector.fromSql(items[2].sparse_embedding).toArray(), [2, 2, 2]);
+
+  await prisma.$disconnect();
 });
 
 beforeEach(async () => {
-  const prisma = new PrismaClient();
+  const prisma = prismaClient();
   await prisma.item.deleteMany({});
+  await prisma.$disconnect();
 });