@prisma-next/extension-pgvector 0.3.0-dev.12 → 0.3.0-dev.123

package/dist/descriptor-meta-IXkObAr0.mjs

@@ -0,0 +1,117 @@
+//#region src/core/descriptor-meta.ts
+const pgvectorTypeId = "pg/vector@1";
+const cosineDistanceLowering = {
+	targetFamily: "sql",
+	strategy: "function",
+	template: "{{self}} <=> {{arg0}}"
+};
+const cosineSimilarityLowering = {
+	targetFamily: "sql",
+	strategy: "function",
+	template: "1 - ({{self}} <=> {{arg0}})"
+};
+const cosineDistanceOperation = Object.freeze({
+	method: "cosineDistance",
+	args: [{ kind: "param" }],
+	returns: {
+		kind: "builtin",
+		type: "number"
+	},
+	lowering: cosineDistanceLowering
+});
+const cosineSimilarityOperation = Object.freeze({
+	method: "cosineSimilarity",
+	args: [{ kind: "param" }],
+	returns: {
+		kind: "builtin",
+		type: "number"
+	},
+	lowering: cosineSimilarityLowering
+});
+const pgvectorOperationSignatures = [{
+	forTypeId: pgvectorTypeId,
+	...cosineDistanceOperation
+}, {
+	forTypeId: pgvectorTypeId,
+	...cosineSimilarityOperation
+}];
+const pgvectorQueryOperations = [{
+	method: "cosineDistance",
+	args: [{
+		codecId: pgvectorTypeId,
+		nullable: false
+	}, {
+		codecId: pgvectorTypeId,
+		nullable: false
+	}],
+	returns: {
+		codecId: "pg/float8@1",
+		nullable: false
+	},
+	lowering: {
+		targetFamily: "sql",
+		strategy: "function",
+		template: "{{self}} <=> {{arg0}}"
+	}
+}, {
+	method: "cosineSimilarity",
+	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}})"
+	}
+}];
+const pgvectorPackMeta = {
+	kind: "extension",
+	id: "pgvector",
+	familyId: "sql",
+	targetId: "postgres",
+	version: "0.0.1",
+	capabilities: { postgres: { "pgvector/cosine": true } },
+	types: {
+		codecTypes: {
+			import: {
+				package: "@prisma-next/extension-pgvector/codec-types",
+				named: "CodecTypes",
+				alias: "PgVectorTypes"
+			},
+			typeImports: [{
+				package: "@prisma-next/extension-pgvector/codec-types",
+				named: "Vector",
+				alias: "Vector"
+			}],
+			parameterized: { [pgvectorTypeId]: "Vector<{{length}}>" }
+		},
+		operationTypes: { import: {
+			package: "@prisma-next/extension-pgvector/operation-types",
+			named: "OperationTypes",
+			alias: "PgVectorOperationTypes"
+		} },
+		queryOperationTypes: { import: {
+			package: "@prisma-next/extension-pgvector/operation-types",
+			named: "QueryOperationTypes",
+			alias: "PgVectorQueryOperationTypes"
+		} },
+		storage: [{
+			typeId: pgvectorTypeId,
+			familyId: "sql",
+			targetId: "postgres",
+			nativeType: "vector"
+		}]
+	}
+};
+
+//#endregion
+export { pgvectorPackMeta as n, pgvectorQueryOperations as r, pgvectorOperationSignatures as t };
+//# sourceMappingURL=descriptor-meta-IXkObAr0.mjs.map

package/dist/descriptor-meta-IXkObAr0.mjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"descriptor-meta-IXkObAr0.mjs","names":["pgvectorOperationSignatures: readonly SqlOperationSignature[]","pgvectorQueryOperations: readonly QueryOperationDescriptor[]"],"sources":["../src/core/descriptor-meta.ts"],"sourcesContent":["import type { SqlOperationSignature } from '@prisma-next/sql-operations';\nimport type { QueryOperationDescriptor } from '@prisma-next/sql-relational-core/query-operations';\n\nconst pgvectorTypeId = 'pg/vector@1' as const;\n\nconst cosineDistanceLowering = {\n  targetFamily: 'sql',\n  strategy: 'function',\n  template: '{{self}} <=> {{arg0}}',\n} as const;\n\nconst cosineSimilarityLowering = {\n  targetFamily: 'sql',\n  strategy: 'function',\n  template: '1 - ({{self}} <=> {{arg0}})',\n} as const;\n\nconst cosineDistanceOperation = Object.freeze({\n  method: 'cosineDistance',\n  args: [{ kind: 'param' }],\n  returns: { kind: 'builtin', type: 'number' },\n  lowering: cosineDistanceLowering,\n} as const);\n\nconst cosineSimilarityOperation = Object.freeze({\n  method: 'cosineSimilarity',\n  args: [{ kind: 'param' }],\n  returns: { kind: 'builtin', type: 'number' },\n  lowering: cosineSimilarityLowering,\n} as const);\n\nexport const pgvectorOperationSignatures: readonly SqlOperationSignature[] = [\n  { forTypeId: pgvectorTypeId, ...cosineDistanceOperation },\n  { forTypeId: pgvectorTypeId, ...cosineSimilarityOperation },\n];\n\nexport const pgvectorQueryOperations: readonly QueryOperationDescriptor[] = [\n  {\n    method: 'cosineDistance',\n    args: [\n      { codecId: pgvectorTypeId, nullable: false },\n      { codecId: pgvectorTypeId, nullable: false },\n    ],\n    returns: { codecId: 'pg/float8@1', nullable: false },\n    lowering: {\n      targetFamily: 'sql',\n      strategy: 'function',\n      template: '{{self}} <=> {{arg0}}',\n    },\n  },\n  {\n    method: 'cosineSimilarity',\n    args: [\n      { codecId: pgvectorTypeId, nullable: false },\n      { codecId: pgvectorTypeId, nullable: false },\n    ],\n    returns: { codecId: 'pg/float8@1', nullable: false },\n    lowering: {\n      targetFamily: 'sql',\n      strategy: 'function',\n      template: '1 - ({{self}} <=> {{arg0}})',\n    },\n  },\n];\n\nexport const pgvectorPackMeta = {\n  kind: 'extension',\n  id: 'pgvector',\n  familyId: 'sql',\n  targetId: 'postgres',\n  version: '0.0.1',\n  capabilities: {\n    postgres: {\n      'pgvector/cosine': true,\n    },\n  },\n  types: {\n    codecTypes: {\n      import: {\n        package: '@prisma-next/extension-pgvector/codec-types',\n        named: 'CodecTypes',\n        alias: 'PgVectorTypes',\n      },\n      typeImports: [\n        {\n          package: '@prisma-next/extension-pgvector/codec-types',\n          named: 'Vector',\n          alias: 'Vector',\n        },\n      ],\n      parameterized: {\n        [pgvectorTypeId]: 'Vector<{{length}}>',\n      },\n    },\n    operationTypes: {\n      import: {\n        package: '@prisma-next/extension-pgvector/operation-types',\n        named: 'OperationTypes',\n        alias: 'PgVectorOperationTypes',\n      },\n    },\n    queryOperationTypes: {\n      import: {\n        package: '@prisma-next/extension-pgvector/operation-types',\n        named: 'QueryOperationTypes',\n        alias: 'PgVectorQueryOperationTypes',\n      },\n    },\n    storage: [\n      { typeId: pgvectorTypeId, familyId: 'sql', targetId: 'postgres', nativeType: 'vector' },\n    ],\n  },\n} as const;\n"],"mappings":";AAGA,MAAM,iBAAiB;AAEvB,MAAM,yBAAyB;CAC7B,cAAc;CACd,UAAU;CACV,UAAU;CACX;AAED,MAAM,2BAA2B;CAC/B,cAAc;CACd,UAAU;CACV,UAAU;CACX;AAED,MAAM,0BAA0B,OAAO,OAAO;CAC5C,QAAQ;CACR,MAAM,CAAC,EAAE,MAAM,SAAS,CAAC;CACzB,SAAS;EAAE,MAAM;EAAW,MAAM;EAAU;CAC5C,UAAU;CACX,CAAU;AAEX,MAAM,4BAA4B,OAAO,OAAO;CAC9C,QAAQ;CACR,MAAM,CAAC,EAAE,MAAM,SAAS,CAAC;CACzB,SAAS;EAAE,MAAM;EAAW,MAAM;EAAU;CAC5C,UAAU;CACX,CAAU;AAEX,MAAaA,8BAAgE,CAC3E;CAAE,WAAW;CAAgB,GAAG;CAAyB,EACzD;CAAE,WAAW;CAAgB,GAAG;CAA2B,CAC5D;AAED,MAAaC,0BAA+D,CAC1E;CACE,QAAQ;CACR,MAAM,CACJ;EAAE,SAAS;EAAgB,UAAU;EAAO,EAC5C;EAAE,SAAS;EAAgB,UAAU;EAAO,CAC7C;CACD,SAAS;EAAE,SAAS;EAAe,UAAU;EAAO;CACpD,UAAU;EACR,cAAc;EACd,UAAU;EACV,UAAU;EACX;CACF,EACD;CACE,QAAQ;CACR,MAAM,CACJ;EAAE,SAAS;EAAgB,UAAU;EAAO,EAC5C;EAAE,SAAS;EAAgB,UAAU;EAAO,CAC7C;CACD,SAAS;EAAE,SAAS;EAAe,UAAU;EAAO;CACpD,UAAU;EACR,cAAc;EACd,UAAU;EACV,UAAU;EACX;CACF,CACF;AAED,MAAa,mBAAmB;CAC9B,MAAM;CACN,IAAI;CACJ,UAAU;CACV,UAAU;CACV,SAAS;CACT,cAAc,EACZ,UAAU,EACR,mBAAmB,MACpB,EACF;CACD,OAAO;EACL,YAAY;GACV,QAAQ;IACN,SAAS;IACT,OAAO;IACP,OAAO;IACR;GACD,aAAa,CACX;IACE,SAAS;IACT,OAAO;IACP,OAAO;IACR,CACF;GACD,eAAe,GACZ,iBAAiB,sBACnB;GACF;EACD,gBAAgB,EACd,QAAQ;GACN,SAAS;GACT,OAAO;GACP,OAAO;GACR,EACF;EACD,qBAAqB,EACnB,QAAQ;GACN,SAAS;GACT,OAAO;GACP,OAAO;GACR,EACF;EACD,SAAS,CACP;GAAE,QAAQ;GAAgB,UAAU;GAAO,UAAU;GAAY,YAAY;GAAU,CACxF;EACF;CACF"}

package/dist/operation-types.d.mts

@@ -0,0 +1,78 @@
+import { SqlQueryOperationTypes } from "@prisma-next/sql-contract/types";
+
+//#region src/types/operation-types.d.ts
+
+/**
+ * Operation type definitions for pgvector extension.
+ *
+ * This file exports type-only definitions for operation method signatures.
+ * These types are imported by contract.d.ts files for compile-time type inference.
+ */
+/**
+ * Operation types for pgvector extension.
+ * Maps typeId to operation methods.
+ */
+type OperationTypes = {
+  readonly 'pg/vector@1': {
+    readonly cosineDistance: {
+      readonly args: readonly [{
+        readonly kind: 'param';
+      }];
+      readonly returns: {
+        readonly kind: 'builtin';
+        readonly type: 'number';
+      };
+      readonly lowering: {
+        readonly targetFamily: 'sql';
+        readonly strategy: 'function';
+        readonly template: string;
+      };
+    };
+    readonly cosineSimilarity: {
+      readonly args: readonly [{
+        readonly kind: 'param';
+      }];
+      readonly returns: {
+        readonly kind: 'builtin';
+        readonly type: 'number';
+      };
+      readonly lowering: {
+        readonly targetFamily: 'sql';
+        readonly strategy: 'function';
+        readonly template: string;
+      };
+    };
+  };
+};
+/** Flat operation signatures for the query builder. */
+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;
+    };
+  };
+  readonly cosineSimilarity: {
+    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;
+    };
+  };
+}>;
+//#endregion
+export { type OperationTypes, type QueryOperationTypes };
+//# sourceMappingURL=operation-types.d.mts.map

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;AAsCA;;;;;;;KAtCY,cAAA;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;KAsCA,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-IXkObAr0.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 pgvectorOperationSignatures } from "./descriptor-meta-IXkObAr0.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: () => pgvectorOperationSignatures,
+	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  pgvectorOperationSignatures,\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: () => pgvectorOperationSignatures,\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;CAC3B,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.12",
+  "version": "0.3.0-dev.123",
   "type": "module",
   "sideEffects": false,
   "dependencies": {
     "arktype": "^2.0.0",
-    "@prisma-next/cli": "0.3.0-dev.12",
-    "@prisma-next/contract": "0.3.0-dev.12",
-    "@prisma-next/contract-authoring": "0.3.0-dev.12",
-    "@prisma-next/core-control-plane": "0.3.0-dev.12",
-    "@prisma-next/family-sql": "0.3.0-dev.12",
-    "@prisma-next/sql-operations": "0.3.0-dev.12",
-    "@prisma-next/sql-relational-core": "0.3.0-dev.12",
-    "@prisma-next/sql-runtime": "0.3.0-dev.12",
-    "@prisma-next/sql-schema-ir": "0.3.0-dev.12"
+    "@prisma-next/cli": "0.3.0-dev.123",
+    "@prisma-next/contract-authoring": "0.3.0-dev.123",
+    "@prisma-next/contract": "0.3.0-dev.123",
+    "@prisma-next/family-sql": "0.3.0-dev.123",
+    "@prisma-next/sql-operations": "0.3.0-dev.123",
+    "@prisma-next/core-control-plane": "0.3.0-dev.123",
+    "@prisma-next/sql-relational-core": "0.3.0-dev.123",
+    "@prisma-next/sql-runtime": "0.3.0-dev.123",
+    "@prisma-next/sql-schema-ir": "0.3.0-dev.123"
   },
   "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.12",
-    "@prisma-next/operations": "0.3.0-dev.12",
-    "@prisma-next/sql-contract": "0.3.0-dev.12",
-    "@prisma-next/sql-contract-ts": "0.3.0-dev.12",
-    "@prisma-next/sql-lane": "0.3.0-dev.12",
-    "@prisma-next/test-utils": "0.0.1"
+    "vitest": "4.0.17",
+    "@prisma-next/adapter-postgres": "0.3.0-dev.123",
+    "@prisma-next/operations": "0.3.0-dev.123",
+    "@prisma-next/sql-contract": "0.3.0-dev.123",
+    "@prisma-next/sql-contract-ts": "0.3.0-dev.123",
+    "@prisma-next/sql-lane": "0.3.0-dev.123",
+    "@prisma-next/tsconfig": "0.0.0",
+    "@prisma-next/test-utils": "0.0.1",
+    "@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 . --error-on-warnings",
     "lint:fix": "biome check --write .",
     "lint:fix:unsafe": "biome check --write --unsafe .",
-    "clean": "node ../../../scripts/clean.mjs"
+    "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,25 +1,68 @@
-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;
 
-const cosineLowering = {
+const cosineDistanceLowering = {
+  targetFamily: 'sql',
+  strategy: 'function',
+  template: '{{self}} <=> {{arg0}}',
+} as const;
+
+const cosineSimilarityLowering = {
   targetFamily: 'sql',
   strategy: 'function',
   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' }],
   returns: { kind: 'builtin', type: 'number' },
-  lowering: cosineLowering,
+  lowering: cosineDistanceLowering,
+} as const);
+
+const cosineSimilarityOperation = Object.freeze({
+  method: 'cosineSimilarity',
+  args: [{ kind: 'param' }],
+  returns: { kind: 'builtin', type: 'number' },
+  lowering: cosineSimilarityLowering,
 } as const);
 
+export const pgvectorOperationSignatures: readonly SqlOperationSignature[] = [
+  { forTypeId: pgvectorTypeId, ...cosineDistanceOperation },
+  { forTypeId: pgvectorTypeId, ...cosineSimilarityOperation },
+];
+
+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: '{{self}} <=> {{arg0}}',
+    },
+  },
+  {
+    method: 'cosineSimilarity',
+    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 +81,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 +99,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;
+}