@prisma-next/extension-pgvector 0.3.0-dev.10 → 0.3.0-dev.113

package/dist/operation-types.d.mts.map

@@ -0,0 +1 @@
+{"version":3,"file":"operation-types.d.mts","names":[],"sources":["../src/types/operation-types.ts"],"sourcesContent":[],"mappings":";;;;;;AAaA;AAsBA;;;;;;;KAtBY,cAAA;;;;;;;;;;;;;;;;;;;KAsBA,mBAAA,GAAsB"}

package/dist/operation-types.mjs

@@ -0,0 +1 @@
+export {  };

package/dist/pack.d.mts

@@ -0,0 +1,10 @@
+import { t as CodecTypes } from "./codec-types-BcHC4dXB.mjs";
+import { ExtensionPackRef } from "@prisma-next/contract/framework-components";
+
+//#region src/exports/pack.d.ts
+declare const _default: ExtensionPackRef<"sql", "postgres"> & {
+  readonly __codecTypes?: CodecTypes;
+};
+//#endregion
+export { _default as default };
+//# sourceMappingURL=pack.d.mts.map

package/dist/pack.d.mts.map

@@ -0,0 +1 @@
+{"version":3,"file":"pack.d.mts","names":[],"sources":["../src/exports/pack.ts"],"sourcesContent":[],"mappings":";;;;cAEuD,UAIxB;0BACL;AAL6B,CAAA"}

package/dist/pack.mjs

@@ -0,0 +1,9 @@
+import { n as pgvectorPackMeta } from "./descriptor-meta-DKfA8aF2.mjs";
+
+//#region src/exports/pack.ts
+const pgvectorPack = pgvectorPackMeta;
+var pack_default = pgvectorPack;
+
+//#endregion
+export { pack_default as default };
+//# sourceMappingURL=pack.mjs.map

package/dist/pack.mjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"pack.mjs","names":[],"sources":["../src/exports/pack.ts"],"sourcesContent":["import type { ExtensionPackRef } from '@prisma-next/contract/framework-components';\nimport { pgvectorPackMeta } from '../core/descriptor-meta';\nimport type { CodecTypes } from '../types/codec-types';\n\nconst pgvectorPack = pgvectorPackMeta;\n\nexport default pgvectorPack as ExtensionPackRef<'sql', 'postgres'> & {\n  readonly __codecTypes?: CodecTypes;\n};\n"],"mappings":";;;AAIA,MAAM,eAAe;AAErB,mBAAe"}

package/dist/runtime.d.mts

@@ -0,0 +1,7 @@
+import { SqlRuntimeExtensionDescriptor } from "@prisma-next/sql-runtime";
+
+//#region src/exports/runtime.d.ts
+declare const pgvectorRuntimeDescriptor: SqlRuntimeExtensionDescriptor<'postgres'>;
+//#endregion
+export { pgvectorRuntimeDescriptor as default };
+//# sourceMappingURL=runtime.d.mts.map

package/dist/runtime.d.mts.map

@@ -0,0 +1 @@
+{"version":3,"file":"runtime.d.mts","names":[],"sources":["../src/exports/runtime.ts"],"sourcesContent":[],"mappings":";;;cA4CM,2BAA2B"}

package/dist/runtime.mjs

@@ -0,0 +1,76 @@
+import { n as pgvectorPackMeta, r as pgvectorQueryOperations, t as pgvectorOperationSignature } from "./descriptor-meta-DKfA8aF2.mjs";
+import { n as VECTOR_MAX_DIM, t as VECTOR_CODEC_ID } from "./constants-BmZpaev5.mjs";
+import { codec, createCodecRegistry, defineCodecs } from "@prisma-next/sql-relational-core/ast";
+import { type } from "arktype";
+
+//#region src/core/codecs.ts
+/**
+* Vector codec implementation for pgvector extension.
+*
+* Provides encoding/decoding for the `vector` PostgreSQL type.
+* Wire format is a string like `[1,2,3]` (PostgreSQL vector text format).
+*/
+const pgVectorCodec = codec({
+	typeId: "pg/vector@1",
+	targetTypes: ["vector"],
+	traits: ["equality"],
+	encode: (value) => {
+		if (!Array.isArray(value)) throw new Error("Vector value must be an array of numbers");
+		if (!value.every((v) => typeof v === "number")) throw new Error("Vector value must contain only numbers");
+		return `[${value.join(",")}]`;
+	},
+	decode: (wire) => {
+		if (typeof wire !== "string") throw new Error("Vector wire value must be a string");
+		if (!wire.startsWith("[") || !wire.endsWith("]")) throw new Error(`Invalid vector format: expected "[...]", got "${wire}"`);
+		const content = wire.slice(1, -1).trim();
+		if (content === "") return [];
+		return content.split(",").map((v) => {
+			const num = Number.parseFloat(v.trim());
+			if (Number.isNaN(num)) throw new Error(`Invalid vector value: "${v}" is not a number`);
+			return num;
+		});
+	},
+	meta: { db: { sql: { postgres: { nativeType: "vector" } } } }
+});
+const codecs = defineCodecs().add("vector", pgVectorCodec);
+const codecDefinitions = codecs.codecDefinitions;
+const dataTypes = codecs.dataTypes;
+
+//#endregion
+//#region src/exports/runtime.ts
+const parameterizedCodecDescriptors = [{
+	codecId: VECTOR_CODEC_ID,
+	paramsSchema: type({ length: "number" }).narrow((params, ctx) => {
+		const { length } = params;
+		if (!Number.isInteger(length)) return ctx.mustBe("an integer");
+		if (length < 1 || length > VECTOR_MAX_DIM) return ctx.mustBe(`in the range [1, ${VECTOR_MAX_DIM}]`);
+		return true;
+	})
+}];
+function createPgvectorCodecRegistry() {
+	const registry = createCodecRegistry();
+	for (const def of Object.values(codecDefinitions)) registry.register(def.codec);
+	return registry;
+}
+const pgvectorRuntimeDescriptor = {
+	kind: "extension",
+	id: pgvectorPackMeta.id,
+	version: pgvectorPackMeta.version,
+	familyId: "sql",
+	targetId: "postgres",
+	codecs: createPgvectorCodecRegistry,
+	operationSignatures: () => [pgvectorOperationSignature],
+	queryOperations: () => pgvectorQueryOperations,
+	parameterizedCodecs: () => parameterizedCodecDescriptors,
+	create() {
+		return {
+			familyId: "sql",
+			targetId: "postgres"
+		};
+	}
+};
+var runtime_default = pgvectorRuntimeDescriptor;
+
+//#endregion
+export { runtime_default as default };
+//# sourceMappingURL=runtime.mjs.map

package/dist/runtime.mjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"runtime.mjs","names":["arktype","pgvectorRuntimeDescriptor: SqlRuntimeExtensionDescriptor<'postgres'>"],"sources":["../src/core/codecs.ts","../src/exports/runtime.ts"],"sourcesContent":["/**\n * Vector codec implementation for pgvector extension.\n *\n * Provides encoding/decoding for the `vector` PostgreSQL type.\n * Wire format is a string like `[1,2,3]` (PostgreSQL vector text format).\n */\n\nimport { codec, defineCodecs } from '@prisma-next/sql-relational-core/ast';\n\nconst pgVectorCodec = codec({\n  typeId: 'pg/vector@1',\n  targetTypes: ['vector'],\n  traits: ['equality'],\n  encode: (value: number[]): string => {\n    // Validate that value is an array of numbers\n    if (!Array.isArray(value)) {\n      throw new Error('Vector value must be an array of numbers');\n    }\n    if (!value.every((v) => typeof v === 'number')) {\n      throw new Error('Vector value must contain only numbers');\n    }\n    // Format as PostgreSQL vector text format: [1,2,3]\n    // PostgreSQL's pg library requires the vector format string\n    return `[${value.join(',')}]`;\n  },\n  decode: (wire: string): number[] => {\n    // Handle string format from PostgreSQL: [1,2,3]\n    if (typeof wire !== 'string') {\n      throw new Error('Vector wire value must be a string');\n    }\n    // Parse PostgreSQL vector format: [1,2,3]\n    if (!wire.startsWith('[') || !wire.endsWith(']')) {\n      throw new Error(`Invalid vector format: expected \"[...]\", got \"${wire}\"`);\n    }\n    const content = wire.slice(1, -1).trim();\n    if (content === '') {\n      return [];\n    }\n    const values = content.split(',').map((v) => {\n      const num = Number.parseFloat(v.trim());\n      if (Number.isNaN(num)) {\n        throw new Error(`Invalid vector value: \"${v}\" is not a number`);\n      }\n      return num;\n    });\n    return values;\n  },\n  meta: {\n    db: {\n      sql: {\n        postgres: {\n          nativeType: 'vector',\n        },\n      },\n    },\n  },\n});\n\n// Build codec definitions using the builder DSL\nconst codecs = defineCodecs().add('vector', pgVectorCodec);\n\n// Export derived structures directly from codecs builder\nexport const codecDefinitions = codecs.codecDefinitions;\nexport const dataTypes = codecs.dataTypes;\n\n// Export types derived from codecs builder\nexport type CodecTypes = typeof codecs.CodecTypes;\n","import { createCodecRegistry } from '@prisma-next/sql-relational-core/ast';\nimport type {\n  RuntimeParameterizedCodecDescriptor,\n  SqlRuntimeExtensionDescriptor,\n} from '@prisma-next/sql-runtime';\nimport { type as arktype } from 'arktype';\nimport { codecDefinitions } from '../core/codecs';\nimport { VECTOR_CODEC_ID, VECTOR_MAX_DIM } from '../core/constants';\nimport {\n  pgvectorOperationSignature,\n  pgvectorPackMeta,\n  pgvectorQueryOperations,\n} from '../core/descriptor-meta';\n\nconst vectorParamsSchema = arktype({\n  length: 'number',\n}).narrow((params, ctx) => {\n  const { length } = params;\n  if (!Number.isInteger(length)) {\n    return ctx.mustBe('an integer');\n  }\n  if (length < 1 || length > VECTOR_MAX_DIM) {\n    return ctx.mustBe(`in the range [1, ${VECTOR_MAX_DIM}]`);\n  }\n  return true;\n});\n\nconst parameterizedCodecDescriptors = [\n  {\n    codecId: VECTOR_CODEC_ID,\n    paramsSchema: vectorParamsSchema,\n  },\n] as const satisfies ReadonlyArray<\n  RuntimeParameterizedCodecDescriptor<{ readonly length: number }>\n>;\n\nfunction createPgvectorCodecRegistry() {\n  const registry = createCodecRegistry();\n  for (const def of Object.values(codecDefinitions)) {\n    registry.register(def.codec);\n  }\n  return registry;\n}\n\nconst pgvectorRuntimeDescriptor: SqlRuntimeExtensionDescriptor<'postgres'> = {\n  kind: 'extension' as const,\n  id: pgvectorPackMeta.id,\n  version: pgvectorPackMeta.version,\n  familyId: 'sql' as const,\n  targetId: 'postgres' as const,\n  codecs: createPgvectorCodecRegistry,\n  operationSignatures: () => [pgvectorOperationSignature],\n  queryOperations: () => pgvectorQueryOperations,\n  parameterizedCodecs: () => parameterizedCodecDescriptors,\n  create() {\n    return {\n      familyId: 'sql' as const,\n      targetId: 'postgres' as const,\n    };\n  },\n};\n\nexport default pgvectorRuntimeDescriptor;\n"],"mappings":";;;;;;;;;;;;AASA,MAAM,gBAAgB,MAAM;CAC1B,QAAQ;CACR,aAAa,CAAC,SAAS;CACvB,QAAQ,CAAC,WAAW;CACpB,SAAS,UAA4B;AAEnC,MAAI,CAAC,MAAM,QAAQ,MAAM,CACvB,OAAM,IAAI,MAAM,2CAA2C;AAE7D,MAAI,CAAC,MAAM,OAAO,MAAM,OAAO,MAAM,SAAS,CAC5C,OAAM,IAAI,MAAM,yCAAyC;AAI3D,SAAO,IAAI,MAAM,KAAK,IAAI,CAAC;;CAE7B,SAAS,SAA2B;AAElC,MAAI,OAAO,SAAS,SAClB,OAAM,IAAI,MAAM,qCAAqC;AAGvD,MAAI,CAAC,KAAK,WAAW,IAAI,IAAI,CAAC,KAAK,SAAS,IAAI,CAC9C,OAAM,IAAI,MAAM,iDAAiD,KAAK,GAAG;EAE3E,MAAM,UAAU,KAAK,MAAM,GAAG,GAAG,CAAC,MAAM;AACxC,MAAI,YAAY,GACd,QAAO,EAAE;AASX,SAPe,QAAQ,MAAM,IAAI,CAAC,KAAK,MAAM;GAC3C,MAAM,MAAM,OAAO,WAAW,EAAE,MAAM,CAAC;AACvC,OAAI,OAAO,MAAM,IAAI,CACnB,OAAM,IAAI,MAAM,0BAA0B,EAAE,mBAAmB;AAEjE,UAAO;IACP;;CAGJ,MAAM,EACJ,IAAI,EACF,KAAK,EACH,UAAU,EACR,YAAY,UACb,EACF,EACF,EACF;CACF,CAAC;AAGF,MAAM,SAAS,cAAc,CAAC,IAAI,UAAU,cAAc;AAG1D,MAAa,mBAAmB,OAAO;AACvC,MAAa,YAAY,OAAO;;;;ACpChC,MAAM,gCAAgC,CACpC;CACE,SAAS;CACT,cAhBuBA,KAAQ,EACjC,QAAQ,UACT,CAAC,CAAC,QAAQ,QAAQ,QAAQ;EACzB,MAAM,EAAE,WAAW;AACnB,MAAI,CAAC,OAAO,UAAU,OAAO,CAC3B,QAAO,IAAI,OAAO,aAAa;AAEjC,MAAI,SAAS,KAAK,SAAS,eACzB,QAAO,IAAI,OAAO,oBAAoB,eAAe,GAAG;AAE1D,SAAO;GACP;CAMC,CACF;AAID,SAAS,8BAA8B;CACrC,MAAM,WAAW,qBAAqB;AACtC,MAAK,MAAM,OAAO,OAAO,OAAO,iBAAiB,CAC/C,UAAS,SAAS,IAAI,MAAM;AAE9B,QAAO;;AAGT,MAAMC,4BAAuE;CAC3E,MAAM;CACN,IAAI,iBAAiB;CACrB,SAAS,iBAAiB;CAC1B,UAAU;CACV,UAAU;CACV,QAAQ;CACR,2BAA2B,CAAC,2BAA2B;CACvD,uBAAuB;CACvB,2BAA2B;CAC3B,SAAS;AACP,SAAO;GACL,UAAU;GACV,UAAU;GACX;;CAEJ;AAED,sBAAe"}

package/package.json

@@ -1,71 +1,59 @@
 {
   "name": "@prisma-next/extension-pgvector",
-  "version": "0.3.0-dev.10",
+  "version": "0.3.0-dev.113",
   "type": "module",
   "sideEffects": false,
   "dependencies": {
     "arktype": "^2.0.0",
-    "@prisma-next/cli": "0.3.0-dev.10",
-    "@prisma-next/contract": "0.3.0-dev.10",
-    "@prisma-next/contract-authoring": "0.3.0-dev.10",
-    "@prisma-next/core-control-plane": "0.3.0-dev.10",
-    "@prisma-next/family-sql": "0.3.0-dev.10",
-    "@prisma-next/sql-relational-core": "0.3.0-dev.10",
-    "@prisma-next/sql-runtime": "0.3.0-dev.10",
-    "@prisma-next/sql-operations": "0.3.0-dev.10",
-    "@prisma-next/sql-schema-ir": "0.3.0-dev.10"
+    "@prisma-next/contract": "0.3.0-dev.113",
+    "@prisma-next/cli": "0.3.0-dev.113",
+    "@prisma-next/contract-authoring": "0.3.0-dev.113",
+    "@prisma-next/core-control-plane": "0.3.0-dev.113",
+    "@prisma-next/sql-operations": "0.3.0-dev.113",
+    "@prisma-next/family-sql": "0.3.0-dev.113",
+    "@prisma-next/sql-relational-core": "0.3.0-dev.113",
+    "@prisma-next/sql-schema-ir": "0.3.0-dev.113",
+    "@prisma-next/sql-runtime": "0.3.0-dev.113"
   },
   "devDependencies": {
-    "@vitest/coverage-v8": "4.0.16",
-    "tsup": "8.5.1",
+    "tsdown": "0.18.4",
     "typescript": "5.9.3",
-    "vitest": "4.0.16",
-    "@prisma-next/adapter-postgres": "0.3.0-dev.10",
-    "@prisma-next/operations": "0.3.0-dev.10",
-    "@prisma-next/sql-contract": "0.3.0-dev.10",
-    "@prisma-next/sql-contract-ts": "0.3.0-dev.10",
-    "@prisma-next/sql-lane": "0.3.0-dev.10",
-    "@prisma-next/test-utils": "0.0.1"
+    "vitest": "4.0.17",
+    "@prisma-next/adapter-postgres": "0.3.0-dev.113",
+    "@prisma-next/operations": "0.3.0-dev.113",
+    "@prisma-next/sql-contract": "0.3.0-dev.113",
+    "@prisma-next/sql-contract-ts": "0.3.0-dev.113",
+    "@prisma-next/sql-lane": "0.3.0-dev.113",
+    "@prisma-next/test-utils": "0.0.1",
+    "@prisma-next/tsconfig": "0.0.0",
+    "@prisma-next/tsdown": "0.0.0"
   },
   "files": [
     "dist",
     "src"
   ],
   "exports": {
-    "./package.json": "./package.json",
-    "./control": {
-      "types": "./dist/exports/control.d.ts",
-      "import": "./dist/exports/control.js"
-    },
-    "./runtime": {
-      "types": "./dist/exports/runtime.d.ts",
-      "import": "./dist/exports/runtime.js"
-    },
-    "./pack": {
-      "types": "./dist/exports/pack.d.ts",
-      "import": "./dist/exports/pack.js"
-    },
-    "./codec-types": {
-      "types": "./dist/exports/codec-types.d.ts",
-      "import": "./dist/exports/codec-types.js"
-    },
-    "./column-types": {
-      "types": "./dist/exports/column-types.d.ts",
-      "import": "./dist/exports/column-types.js"
-    },
-    "./operation-types": {
-      "types": "./dist/exports/operation-types.d.ts",
-      "import": "./dist/exports/operation-types.js"
-    }
+    "./codec-types": "./dist/codec-types.mjs",
+    "./column-types": "./dist/column-types.mjs",
+    "./control": "./dist/control.mjs",
+    "./operation-types": "./dist/operation-types.mjs",
+    "./pack": "./dist/pack.mjs",
+    "./runtime": "./dist/runtime.mjs",
+    "./package.json": "./package.json"
+  },
+  "repository": {
+    "type": "git",
+    "url": "https://github.com/prisma/prisma-next.git",
+    "directory": "packages/3-extensions/pgvector"
   },
   "scripts": {
-    "build": "tsup --config tsup.config.ts && tsc --project tsconfig.build.json",
+    "build": "tsdown",
     "test": "vitest run",
     "test:coverage": "vitest run --coverage",
     "typecheck": "tsc --project tsconfig.json --noEmit",
-    "lint": "biome check . --config-path ../../../biome.json --error-on-warnings",
-    "lint:fix": "biome check --write . --config-path ../../../biome.json",
-    "lint:fix:unsafe": "biome check --write --unsafe . --config-path ../../../biome.json",
-    "clean": "node ../../../scripts/clean.mjs"
+    "lint": "biome check . --error-on-warnings",
+    "lint:fix": "biome check --write .",
+    "lint:fix:unsafe": "biome check --write --unsafe .",
+    "clean": "rm -rf dist dist-tsc dist-tsc-prod coverage .tmp-output"
   }
 }

package/src/core/codecs.ts

@@ -7,9 +7,10 @@
 
 import { codec, defineCodecs } from '@prisma-next/sql-relational-core/ast';
 
-const pgVectorCodec = codec<'pg/vector@1', string, number[]>({
+const pgVectorCodec = codec({
   typeId: 'pg/vector@1',
   targetTypes: ['vector'],
+  traits: ['equality'],
   encode: (value: number[]): string => {
     // Validate that value is an array of numbers
     if (!Array.isArray(value)) {

package/src/core/constants.ts

@@ -0,0 +1,9 @@
+/**
+ * Codec ID for pgvector's vector type.
+ */
+export const VECTOR_CODEC_ID = 'pg/vector@1' as const;
+
+/**
+ * Maximum dimension for pgvector vectors (VECTOR_MAX_DIM from pgvector).
+ */
+export const VECTOR_MAX_DIM = 16000;

package/src/core/descriptor-meta.ts

@@ -1,5 +1,5 @@
-import type { ExtensionPackRef } from '@prisma-next/contract/framework-components';
 import type { SqlOperationSignature } from '@prisma-next/sql-operations';
+import type { QueryOperationDescriptor } from '@prisma-next/sql-relational-core/query-operations';
 
 const pgvectorTypeId = 'pg/vector@1' as const;
 
@@ -9,10 +9,6 @@ const cosineLowering = {
   template: '1 - ({{self}} <=> {{arg0}})',
 } as const;
 
-/**
- * Shared operation definition used by both pack metadata and runtime descriptor.
- * Frozen to prevent accidental mutation.
- */
 const cosineDistanceOperation = Object.freeze({
   method: 'cosineDistance',
   args: [{ kind: 'param' }],
@@ -20,6 +16,27 @@ const cosineDistanceOperation = Object.freeze({
   lowering: cosineLowering,
 } as const);
 
+export const pgvectorOperationSignature: SqlOperationSignature = {
+  forTypeId: pgvectorTypeId,
+  ...cosineDistanceOperation,
+};
+
+export const pgvectorQueryOperations: readonly QueryOperationDescriptor[] = [
+  {
+    method: 'cosineDistance',
+    args: [
+      { codecId: pgvectorTypeId, nullable: false },
+      { codecId: pgvectorTypeId, nullable: false },
+    ],
+    returns: { codecId: 'pg/float8@1', nullable: false },
+    lowering: {
+      targetFamily: 'sql',
+      strategy: 'function',
+      template: '1 - ({{self}} <=> {{arg0}})',
+    },
+  },
+];
+
 export const pgvectorPackMeta = {
   kind: 'extension',
   id: 'pgvector',
@@ -38,6 +55,16 @@ export const pgvectorPackMeta = {
         named: 'CodecTypes',
         alias: 'PgVectorTypes',
       },
+      typeImports: [
+        {
+          package: '@prisma-next/extension-pgvector/codec-types',
+          named: 'Vector',
+          alias: 'Vector',
+        },
+      ],
+      parameterized: {
+        [pgvectorTypeId]: 'Vector<{{length}}>',
+      },
     },
     operationTypes: {
       import: {
@@ -46,19 +73,15 @@ export const pgvectorPackMeta = {
         alias: 'PgVectorOperationTypes',
       },
     },
+    queryOperationTypes: {
+      import: {
+        package: '@prisma-next/extension-pgvector/operation-types',
+        named: 'QueryOperationTypes',
+        alias: 'PgVectorQueryOperationTypes',
+      },
+    },
     storage: [
       { typeId: pgvectorTypeId, familyId: 'sql', targetId: 'postgres', nativeType: 'vector' },
     ],
   },
-  operations: [
-    {
-      for: pgvectorTypeId,
-      ...cosineDistanceOperation,
-    },
-  ],
-} as const satisfies ExtensionPackRef<'sql', 'postgres'>;
-
-export const pgvectorRuntimeOperation: SqlOperationSignature = {
-  forTypeId: pgvectorTypeId,
-  ...cosineDistanceOperation,
-};
+} as const;

package/src/exports/codec-types.ts

@@ -4,4 +4,4 @@
  * Re-export from types module for public API.
  */
 
-export type { CodecTypes } from '../types/codec-types';
+export type { CodecTypes, Vector } from '../types/codec-types';

package/src/exports/column-types.ts

@@ -6,8 +6,41 @@
  */
 
 import type { ColumnTypeDescriptor } from '@prisma-next/contract-authoring';
+import { VECTOR_CODEC_ID, VECTOR_MAX_DIM } from '../core/constants';
 
+/**
+ * Static vector column descriptor without dimension.
+ * Use `vector(N)` for dimensioned vectors that produce `vector(N)` DDL.
+ */
 export const vectorColumn = {
-  codecId: 'pg/vector@1',
+  codecId: VECTOR_CODEC_ID,
   nativeType: 'vector',
 } as const satisfies ColumnTypeDescriptor;
+
+/**
+ * Factory for creating dimensioned vector column descriptors.
+ *
+ * @example
+ * ```typescript
+ * .column('embedding', { type: vector(1536), nullable: false })
+ * // Produces: nativeType: 'vector', typeParams: { length: 1536 }
+ * ```
+ *
+ * @param length - The dimension of the vector (e.g., 1536 for OpenAI embeddings)
+ * @returns A column type descriptor with `typeParams.length` set
+ * @throws {RangeError} If length is not an integer in the range [1, VECTOR_MAX_DIM]
+ */
+export function vector<N extends number>(
+  length: N,
+): ColumnTypeDescriptor & { readonly typeParams: { readonly length: N } } {
+  if (!Number.isInteger(length) || length < 1 || length > VECTOR_MAX_DIM) {
+    throw new RangeError(
+      `pgvector: dimension must be an integer in [1, ${VECTOR_MAX_DIM}], got ${length}`,
+    );
+  }
+  return {
+    codecId: VECTOR_CODEC_ID,
+    nativeType: 'vector',
+    typeParams: { length },
+  } as const;
+}

package/src/exports/control.ts

@@ -1,32 +1,33 @@
-import type { SchemaIssue } from '@prisma-next/core-control-plane/types';
 import type {
+  CodecControlHooks,
   ComponentDatabaseDependencies,
   SqlControlExtensionDescriptor,
 } from '@prisma-next/family-sql/control';
-import type { SqlSchemaIR } from '@prisma-next/sql-schema-ir/types';
-import { pgvectorPackMeta } from '../core/descriptor-meta';
+import { pgvectorOperationSignature, pgvectorPackMeta } from '../core/descriptor-meta';
 
-/**
- * Pure verification hook: checks whether the 'vector' extension is installed
- * based on the in-memory schema IR (no DB I/O).
- */
-function verifyVectorExtensionInstalled(schema: SqlSchemaIR): readonly SchemaIssue[] {
-  if (!schema.extensions.includes('vector')) {
-    return [
-      {
-        kind: 'extension_missing',
-        table: '',
-        message: 'Extension "vector" is missing from database (required by pgvector)',
-      },
-    ];
+const PGVECTOR_CODEC_ID = 'pg/vector@1' as const;
+
+function buildVectorIdentityValue(typeParams: Record<string, unknown> | undefined): string | null {
+  const length = typeParams?.['length'];
+  if (typeof length !== 'number' || !Number.isInteger(length) || length <= 0) {
+    return null;
   }
-  return [];
+
+  const zeroVector = `[${new Array(length).fill('0').join(',')}]`;
+  return `'${zeroVector}'::vector`;
 }
 
-/**
- * Database dependencies for the pgvector extension.
- * Declares the 'vector' Postgres extension as a required dependency.
- */
+const vectorControlPlaneHooks: CodecControlHooks = {
+  expandNativeType: ({ nativeType, typeParams }) => {
+    const length = typeParams?.['length'];
+    if (typeof length === 'number' && Number.isInteger(length) && length > 0) {
+      return `${nativeType}(${length})`;
+    }
+    return nativeType;
+  },
+  resolveIdentityValue: ({ typeParams }) => buildVectorIdentityValue(typeParams),
+};
+
 const pgvectorDatabaseDependencies: ComponentDatabaseDependencies<unknown> = {
   init: [
     {
@@ -59,17 +60,22 @@ const pgvectorDatabaseDependencies: ComponentDatabaseDependencies<unknown> = {
           ],
         },
       ],
-      verifyDatabaseDependencyInstalled: verifyVectorExtensionInstalled,
     },
   ],
 };
 
-/**
- * pgvector extension descriptor for CLI config.
- * Declares database dependencies for the 'vector' Postgres extension.
- */
 const pgvectorExtensionDescriptor: SqlControlExtensionDescriptor<'postgres'> = {
   ...pgvectorPackMeta,
+  types: {
+    ...pgvectorPackMeta.types,
+    codecTypes: {
+      ...pgvectorPackMeta.types.codecTypes,
+      controlPlaneHooks: {
+        [PGVECTOR_CODEC_ID]: vectorControlPlaneHooks,
+      },
+    },
+  },
+  operationSignatures: () => [pgvectorOperationSignature],
   databaseDependencies: pgvectorDatabaseDependencies,
   create: () => ({
     familyId: 'sql' as const,

package/src/exports/operation-types.ts

@@ -4,4 +4,4 @@
  * Re-export from types module for public API.
  */
 
-export type { OperationTypes } from '../types/operation-types';
+export type { OperationTypes, QueryOperationTypes } from '../types/operation-types';

package/src/exports/pack.ts

@@ -1,5 +1,9 @@
+import type { ExtensionPackRef } from '@prisma-next/contract/framework-components';
 import { pgvectorPackMeta } from '../core/descriptor-meta';
+import type { CodecTypes } from '../types/codec-types';
 
 const pgvectorPack = pgvectorPackMeta;
 
-export default pgvectorPack;
+export default pgvectorPack as ExtensionPackRef<'sql', 'postgres'> & {
+  readonly __codecTypes?: CodecTypes;
+};

package/src/exports/runtime.ts

@@ -1,47 +1,62 @@
-import type { SqlOperationSignature } from '@prisma-next/sql-operations';
-import type { CodecRegistry } from '@prisma-next/sql-relational-core/ast';
 import { createCodecRegistry } from '@prisma-next/sql-relational-core/ast';
 import type {
+  RuntimeParameterizedCodecDescriptor,
   SqlRuntimeExtensionDescriptor,
-  SqlRuntimeExtensionInstance,
 } from '@prisma-next/sql-runtime';
+import { type as arktype } from 'arktype';
 import { codecDefinitions } from '../core/codecs';
-import { pgvectorPackMeta, pgvectorRuntimeOperation } from '../core/descriptor-meta';
+import { VECTOR_CODEC_ID, VECTOR_MAX_DIM } from '../core/constants';
+import {
+  pgvectorOperationSignature,
+  pgvectorPackMeta,
+  pgvectorQueryOperations,
+} from '../core/descriptor-meta';
 
-/**
- * pgvector SQL runtime extension instance.
- * Provides codecs and operations for vector data type and similarity operations.
- */
-class PgVectorRuntimeExtensionInstance implements SqlRuntimeExtensionInstance<'postgres'> {
-  readonly familyId = 'sql' as const;
-  readonly targetId = 'postgres' as const;
-
-  codecs(): CodecRegistry {
-    const registry = createCodecRegistry();
-    // Register all codecs from codecDefinitions
-    for (const def of Object.values(codecDefinitions)) {
-      registry.register(def.codec);
-    }
-    return registry;
+const vectorParamsSchema = arktype({
+  length: 'number',
+}).narrow((params, ctx) => {
+  const { length } = params;
+  if (!Number.isInteger(length)) {
+    return ctx.mustBe('an integer');
+  }
+  if (length < 1 || length > VECTOR_MAX_DIM) {
+    return ctx.mustBe(`in the range [1, ${VECTOR_MAX_DIM}]`);
   }
+  return true;
+});
+
+const parameterizedCodecDescriptors = [
+  {
+    codecId: VECTOR_CODEC_ID,
+    paramsSchema: vectorParamsSchema,
+  },
+] as const satisfies ReadonlyArray<
+  RuntimeParameterizedCodecDescriptor<{ readonly length: number }>
+>;
 
-  operations(): ReadonlyArray<SqlOperationSignature> {
-    return [pgvectorRuntimeOperation];
+function createPgvectorCodecRegistry() {
+  const registry = createCodecRegistry();
+  for (const def of Object.values(codecDefinitions)) {
+    registry.register(def.codec);
   }
+  return registry;
 }
 
-/**
- * pgvector SQL runtime extension descriptor.
- * Provides metadata and factory for creating runtime extension instances.
- */
 const pgvectorRuntimeDescriptor: SqlRuntimeExtensionDescriptor<'postgres'> = {
   kind: 'extension' as const,
   id: pgvectorPackMeta.id,
   version: pgvectorPackMeta.version,
   familyId: 'sql' as const,
   targetId: 'postgres' as const,
-  create(): SqlRuntimeExtensionInstance<'postgres'> {
-    return new PgVectorRuntimeExtensionInstance();
+  codecs: createPgvectorCodecRegistry,
+  operationSignatures: () => [pgvectorOperationSignature],
+  queryOperations: () => pgvectorQueryOperations,
+  parameterizedCodecs: () => parameterizedCodecDescriptors,
+  create() {
+    return {
+      familyId: 'sql' as const,
+      targetId: 'postgres' as const,
+    };
   },
 };
 

package/src/types/codec-types.ts

@@ -7,4 +7,26 @@
  * Runtime codec implementations are provided by the extension's codec registry.
  */
 
-export type { CodecTypes } from '../core/codecs';
+import type { CodecTypes as CoreCodecTypes } from '../core/codecs';
+
+/**
+ * Type-level branded vector.
+ *
+ * The runtime values are plain number arrays, but parameterized column typing can
+ * carry the dimension at the type level (e.g. Vector<1536>).
+ */
+export type Vector<N extends number = number> = number[] & { readonly __vectorLength?: N };
+
+/**
+ * Codec types for pgvector.
+ *
+ * - Scalar output remains `number[]` (runtime representation).
+ * - `parameterizedOutput` enables lane typing to compute `Vector<N>` from column `typeParams`.
+ */
+export type CodecTypes = CoreCodecTypes & {
+  readonly 'pg/vector@1': CoreCodecTypes['pg/vector@1'] & {
+    readonly parameterizedOutput: <P extends { readonly length: number }>(
+      params: P,
+    ) => P extends { readonly length: infer N extends number } ? Vector<N> : Vector<number>;
+  };
+};

package/src/types/operation-types.ts

@@ -1,3 +1,5 @@
+import type { SqlQueryOperationTypes } from '@prisma-next/sql-contract/types';
+
 /**
  * Operation type definitions for pgvector extension.
  *
@@ -29,3 +31,14 @@ export type OperationTypes = {
     };
   };
 };
+
+/** Flat operation signatures for the query builder. */
+export type QueryOperationTypes = SqlQueryOperationTypes<{
+  readonly cosineDistance: {
+    readonly args: readonly [
+      { readonly codecId: 'pg/vector@1'; readonly nullable: boolean },
+      { readonly codecId: 'pg/vector@1'; readonly nullable: boolean },
+    ];
+    readonly returns: { readonly codecId: 'pg/float8@1'; readonly nullable: false };
+  };
+}>;