@prisma-next/extension-pgvector 0.5.0-dev.14 → 0.5.0-dev.15

package/dist/control.mjs

@@ -1,4 +1,4 @@
-import { n as pgvectorQueryOperations, t as pgvectorPackMeta } from "./descriptor-meta-BQbvJJxu.mjs";
+import { n as pgvectorQueryOperations, t as pgvectorPackMeta } from "./descriptor-meta-DTopW_37.mjs";
 
 //#region src/exports/control.ts
 const PGVECTOR_CODEC_ID = "pg/vector@1";
@@ -47,7 +47,7 @@ const pgvectorExtensionDescriptor = {
 			controlPlaneHooks: { [PGVECTOR_CODEC_ID]: vectorControlPlaneHooks }
 		}
 	},
-	queryOperations: () => pgvectorQueryOperations,
+	queryOperations: () => pgvectorQueryOperations(),
 	databaseDependencies: pgvectorDatabaseDependencies,
 	create: () => ({
 		familyId: "sql",

package/dist/control.mjs.map

@@ -1 +1 @@
-{"version":3,"file":"control.mjs","names":["vectorControlPlaneHooks: CodecControlHooks","pgvectorDatabaseDependencies: ComponentDatabaseDependencies<unknown>","pgvectorExtensionDescriptor: SqlControlExtensionDescriptor<'postgres'>"],"sources":["../src/exports/control.ts"],"sourcesContent":["import type {\n  CodecControlHooks,\n  ComponentDatabaseDependencies,\n  SqlControlExtensionDescriptor,\n} from '@prisma-next/family-sql/control';\nimport { pgvectorPackMeta, pgvectorQueryOperations } from '../core/descriptor-meta';\n\nconst PGVECTOR_CODEC_ID = 'pg/vector@1' as const;\n\nfunction buildVectorIdentityValue(typeParams: Record<string, unknown> | undefined): string | null {\n  const length = typeParams?.['length'];\n  if (typeof length !== 'number' || !Number.isInteger(length) || length <= 0) {\n    return null;\n  }\n\n  const zeroVector = `[${new Array(length).fill('0').join(',')}]`;\n  return `'${zeroVector}'::vector`;\n}\n\nconst vectorControlPlaneHooks: CodecControlHooks = {\n  expandNativeType: ({ nativeType, typeParams }) => {\n    const length = typeParams?.['length'];\n    if (typeof length === 'number' && Number.isInteger(length) && length > 0) {\n      return `${nativeType}(${length})`;\n    }\n    return nativeType;\n  },\n  resolveIdentityValue: ({ typeParams }) => buildVectorIdentityValue(typeParams),\n};\n\nconst pgvectorDatabaseDependencies: ComponentDatabaseDependencies<unknown> = {\n  init: [\n    {\n      id: 'postgres.extension.vector',\n      label: 'Enable vector extension',\n      install: [\n        {\n          id: 'extension.vector',\n          label: 'Enable extension \"vector\"',\n          summary: 'Ensures the vector extension is available for pgvector operations',\n          operationClass: 'additive',\n          target: { id: 'postgres' },\n          precheck: [\n            {\n              description: 'verify extension \"vector\" is not already enabled',\n              sql: \"SELECT NOT EXISTS (SELECT 1 FROM pg_extension WHERE extname = 'vector')\",\n            },\n          ],\n          execute: [\n            {\n              description: 'create extension \"vector\"',\n              sql: 'CREATE EXTENSION IF NOT EXISTS vector',\n            },\n          ],\n          postcheck: [\n            {\n              description: 'confirm extension \"vector\" is enabled',\n              sql: \"SELECT EXISTS (SELECT 1 FROM pg_extension WHERE extname = 'vector')\",\n            },\n          ],\n        },\n      ],\n    },\n  ],\n};\n\nconst pgvectorExtensionDescriptor: SqlControlExtensionDescriptor<'postgres'> = {\n  ...pgvectorPackMeta,\n  types: {\n    ...pgvectorPackMeta.types,\n    codecTypes: {\n      ...pgvectorPackMeta.types.codecTypes,\n      controlPlaneHooks: {\n        [PGVECTOR_CODEC_ID]: vectorControlPlaneHooks,\n      },\n    },\n  },\n  queryOperations: () => pgvectorQueryOperations,\n  databaseDependencies: pgvectorDatabaseDependencies,\n  create: () => ({\n    familyId: 'sql' as const,\n    targetId: 'postgres' as const,\n  }),\n};\n\nexport { pgvectorExtensionDescriptor };\nexport default pgvectorExtensionDescriptor;\n"],"mappings":";;;AAOA,MAAM,oBAAoB;AAE1B,SAAS,yBAAyB,YAAgE;CAChG,MAAM,SAAS,aAAa;AAC5B,KAAI,OAAO,WAAW,YAAY,CAAC,OAAO,UAAU,OAAO,IAAI,UAAU,EACvE,QAAO;AAIT,QAAO,IADY,IAAI,IAAI,MAAM,OAAO,CAAC,KAAK,IAAI,CAAC,KAAK,IAAI,CAAC,GACvC;;AAGxB,MAAMA,0BAA6C;CACjD,mBAAmB,EAAE,YAAY,iBAAiB;EAChD,MAAM,SAAS,aAAa;AAC5B,MAAI,OAAO,WAAW,YAAY,OAAO,UAAU,OAAO,IAAI,SAAS,EACrE,QAAO,GAAG,WAAW,GAAG,OAAO;AAEjC,SAAO;;CAET,uBAAuB,EAAE,iBAAiB,yBAAyB,WAAW;CAC/E;AAED,MAAMC,+BAAuE,EAC3E,MAAM,CACJ;CACE,IAAI;CACJ,OAAO;CACP,SAAS,CACP;EACE,IAAI;EACJ,OAAO;EACP,SAAS;EACT,gBAAgB;EAChB,QAAQ,EAAE,IAAI,YAAY;EAC1B,UAAU,CACR;GACE,aAAa;GACb,KAAK;GACN,CACF;EACD,SAAS,CACP;GACE,aAAa;GACb,KAAK;GACN,CACF;EACD,WAAW,CACT;GACE,aAAa;GACb,KAAK;GACN,CACF;EACF,CACF;CACF,CACF,EACF;AAED,MAAMC,8BAAyE;CAC7E,GAAG;CACH,OAAO;EACL,GAAG,iBAAiB;EACpB,YAAY;GACV,GAAG,iBAAiB,MAAM;GAC1B,mBAAmB,GAChB,oBAAoB,yBACtB;GACF;EACF;CACD,uBAAuB;CACvB,sBAAsB;CACtB,eAAe;EACb,UAAU;EACV,UAAU;EACX;CACF;AAGD,sBAAe"}
+{"version":3,"file":"control.mjs","names":["vectorControlPlaneHooks: CodecControlHooks","pgvectorDatabaseDependencies: ComponentDatabaseDependencies<unknown>","pgvectorExtensionDescriptor: SqlControlExtensionDescriptor<'postgres'>"],"sources":["../src/exports/control.ts"],"sourcesContent":["import type {\n  CodecControlHooks,\n  ComponentDatabaseDependencies,\n  SqlControlExtensionDescriptor,\n} from '@prisma-next/family-sql/control';\nimport { pgvectorPackMeta, pgvectorQueryOperations } from '../core/descriptor-meta';\n\nconst PGVECTOR_CODEC_ID = 'pg/vector@1' as const;\n\nfunction buildVectorIdentityValue(typeParams: Record<string, unknown> | undefined): string | null {\n  const length = typeParams?.['length'];\n  if (typeof length !== 'number' || !Number.isInteger(length) || length <= 0) {\n    return null;\n  }\n\n  const zeroVector = `[${new Array(length).fill('0').join(',')}]`;\n  return `'${zeroVector}'::vector`;\n}\n\nconst vectorControlPlaneHooks: CodecControlHooks = {\n  expandNativeType: ({ nativeType, typeParams }) => {\n    const length = typeParams?.['length'];\n    if (typeof length === 'number' && Number.isInteger(length) && length > 0) {\n      return `${nativeType}(${length})`;\n    }\n    return nativeType;\n  },\n  resolveIdentityValue: ({ typeParams }) => buildVectorIdentityValue(typeParams),\n};\n\nconst pgvectorDatabaseDependencies: ComponentDatabaseDependencies<unknown> = {\n  init: [\n    {\n      id: 'postgres.extension.vector',\n      label: 'Enable vector extension',\n      install: [\n        {\n          id: 'extension.vector',\n          label: 'Enable extension \"vector\"',\n          summary: 'Ensures the vector extension is available for pgvector operations',\n          operationClass: 'additive',\n          target: { id: 'postgres' },\n          precheck: [\n            {\n              description: 'verify extension \"vector\" is not already enabled',\n              sql: \"SELECT NOT EXISTS (SELECT 1 FROM pg_extension WHERE extname = 'vector')\",\n            },\n          ],\n          execute: [\n            {\n              description: 'create extension \"vector\"',\n              sql: 'CREATE EXTENSION IF NOT EXISTS vector',\n            },\n          ],\n          postcheck: [\n            {\n              description: 'confirm extension \"vector\" is enabled',\n              sql: \"SELECT EXISTS (SELECT 1 FROM pg_extension WHERE extname = 'vector')\",\n            },\n          ],\n        },\n      ],\n    },\n  ],\n};\n\nconst pgvectorExtensionDescriptor: SqlControlExtensionDescriptor<'postgres'> = {\n  ...pgvectorPackMeta,\n  types: {\n    ...pgvectorPackMeta.types,\n    codecTypes: {\n      ...pgvectorPackMeta.types.codecTypes,\n      controlPlaneHooks: {\n        [PGVECTOR_CODEC_ID]: vectorControlPlaneHooks,\n      },\n    },\n  },\n  queryOperations: () => pgvectorQueryOperations(),\n  databaseDependencies: pgvectorDatabaseDependencies,\n  create: () => ({\n    familyId: 'sql' as const,\n    targetId: 'postgres' as const,\n  }),\n};\n\nexport { pgvectorExtensionDescriptor };\nexport default pgvectorExtensionDescriptor;\n"],"mappings":";;;AAOA,MAAM,oBAAoB;AAE1B,SAAS,yBAAyB,YAAgE;CAChG,MAAM,SAAS,aAAa;AAC5B,KAAI,OAAO,WAAW,YAAY,CAAC,OAAO,UAAU,OAAO,IAAI,UAAU,EACvE,QAAO;AAIT,QAAO,IADY,IAAI,IAAI,MAAM,OAAO,CAAC,KAAK,IAAI,CAAC,KAAK,IAAI,CAAC,GACvC;;AAGxB,MAAMA,0BAA6C;CACjD,mBAAmB,EAAE,YAAY,iBAAiB;EAChD,MAAM,SAAS,aAAa;AAC5B,MAAI,OAAO,WAAW,YAAY,OAAO,UAAU,OAAO,IAAI,SAAS,EACrE,QAAO,GAAG,WAAW,GAAG,OAAO;AAEjC,SAAO;;CAET,uBAAuB,EAAE,iBAAiB,yBAAyB,WAAW;CAC/E;AAED,MAAMC,+BAAuE,EAC3E,MAAM,CACJ;CACE,IAAI;CACJ,OAAO;CACP,SAAS,CACP;EACE,IAAI;EACJ,OAAO;EACP,SAAS;EACT,gBAAgB;EAChB,QAAQ,EAAE,IAAI,YAAY;EAC1B,UAAU,CACR;GACE,aAAa;GACb,KAAK;GACN,CACF;EACD,SAAS,CACP;GACE,aAAa;GACb,KAAK;GACN,CACF;EACD,WAAW,CACT;GACE,aAAa;GACb,KAAK;GACN,CACF;EACF,CACF;CACF,CACF,EACF;AAED,MAAMC,8BAAyE;CAC7E,GAAG;CACH,OAAO;EACL,GAAG,iBAAiB;EACpB,YAAY;GACV,GAAG,iBAAiB,MAAM;GAC1B,mBAAmB,GAChB,oBAAoB,yBACtB;GACF;EACF;CACD,uBAAuB,yBAAyB;CAChD,sBAAsB;CACtB,eAAe;EACb,UAAU;EACV,UAAU;EACX;CACF;AAGD,sBAAe"}

package/dist/{descriptor-meta-BQbvJJxu.mjs → descriptor-meta-DTopW_37.mjs}

@@ -1,4 +1,5 @@
 import { n as VECTOR_MAX_DIM } from "./constants-Co5golCK.mjs";
+import { buildOperation, toExpr } from "@prisma-next/sql-relational-core/expression";
 import { codec, defineCodecs } from "@prisma-next/sql-relational-core/ast";
 
 //#region src/core/authoring.ts
@@ -64,43 +65,41 @@ const dataTypes = codecs.dataTypes;
 //#endregion
 //#region src/core/descriptor-meta.ts
 const pgvectorTypeId = "pg/vector@1";
-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
+function pgvectorQueryOperations() {
+	return [{
+		method: "cosineDistance",
+		self: { codecId: pgvectorTypeId },
+		impl: (self, other) => buildOperation({
+			method: "cosineDistance",
+			args: [toExpr(self, pgvectorTypeId), toExpr(other, pgvectorTypeId)],
+			returns: {
+				codecId: "pg/float8@1",
+				nullable: false
+			},
+			lowering: {
+				targetFamily: "sql",
+				strategy: "function",
+				template: "{{self}} <=> {{arg0}}"
+			}
+		})
 	}, {
-		codecId: pgvectorTypeId,
-		nullable: false
-	}],
-	returns: {
-		codecId: "pg/float8@1",
-		nullable: false
-	},
-	lowering: {
-		targetFamily: "sql",
-		strategy: "function",
-		template: "1 - ({{self}} <=> {{arg0}})"
-	}
-}];
+		method: "cosineSimilarity",
+		self: { codecId: pgvectorTypeId },
+		impl: (self, other) => buildOperation({
+			method: "cosineSimilarity",
+			args: [toExpr(self, pgvectorTypeId), toExpr(other, pgvectorTypeId)],
+			returns: {
+				codecId: "pg/float8@1",
+				nullable: false
+			},
+			lowering: {
+				targetFamily: "sql",
+				strategy: "function",
+				template: "1 - ({{self}} <=> {{arg0}})"
+			}
+		})
+	}];
+}
 const pgvectorPackMetaBase = {
 	kind: "extension",
 	id: "pgvector",
@@ -145,4 +144,4 @@ const pgvectorPackMeta = pgvectorPackMetaBase;
 
 //#endregion
 export { pgvectorQueryOperations as n, codecDefinitions as r, pgvectorPackMeta as t };
-//# sourceMappingURL=descriptor-meta-BQbvJJxu.mjs.map
+//# sourceMappingURL=descriptor-meta-DTopW_37.mjs.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"descriptor-meta-DTopW_37.mjs","names":["pgvectorPackMeta: typeof pgvectorPackMetaBase & {\n  readonly __codecTypes?: CodecTypes;\n}"],"sources":["../src/core/authoring.ts","../src/core/codecs.ts","../src/core/descriptor-meta.ts"],"sourcesContent":["import type { AuthoringTypeNamespace } from '@prisma-next/framework-components/authoring';\nimport { VECTOR_MAX_DIM } from './constants';\n\nexport const pgvectorAuthoringTypes = {\n  pgvector: {\n    Vector: {\n      kind: 'typeConstructor',\n      args: [\n        { kind: 'number', name: 'length', integer: true, minimum: 1, maximum: VECTOR_MAX_DIM },\n      ],\n      output: {\n        codecId: 'pg/vector@1',\n        nativeType: 'vector',\n        typeParams: {\n          length: { kind: 'arg', index: 0 },\n        },\n      },\n    },\n  },\n} as const satisfies AuthoringTypeNamespace;\n","/**\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  renderOutputType: (typeParams) => {\n    const length = typeParams['length'];\n    if (length === undefined) return undefined;\n    if (typeof length !== 'number' || !Number.isFinite(length) || !Number.isInteger(length)) {\n      throw new Error(\n        `renderOutputType: expected positive integer \"length\" in typeParams for Vector, got ${String(length)}`,\n      );\n    }\n    return `Vector<${length}>`;\n  },\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 type { SqlOperationDescriptor } from '@prisma-next/sql-operations';\nimport {\n  buildOperation,\n  type CodecExpression,\n  type Expression,\n  toExpr,\n} from '@prisma-next/sql-relational-core/expression';\nimport type { CodecTypes } from '../types/codec-types';\nimport { pgvectorAuthoringTypes } from './authoring';\nimport { codecDefinitions } from './codecs';\n\nconst pgvectorTypeId = 'pg/vector@1' as const;\n\ntype CodecTypesBase = Record<string, { readonly input: unknown; readonly output: unknown }>;\n\nexport function pgvectorQueryOperations<\n  CT extends CodecTypesBase,\n>(): readonly SqlOperationDescriptor[] {\n  return [\n    {\n      method: 'cosineDistance',\n      self: { codecId: pgvectorTypeId },\n      impl: (\n        self: CodecExpression<'pg/vector@1', boolean, CT>,\n        other: CodecExpression<'pg/vector@1', boolean, CT>,\n      ): Expression<{ codecId: 'pg/float8@1'; nullable: false }> =>\n        buildOperation({\n          method: 'cosineDistance',\n          args: [toExpr(self, pgvectorTypeId), toExpr(other, pgvectorTypeId)],\n          returns: { codecId: 'pg/float8@1', nullable: false },\n          lowering: {\n            targetFamily: 'sql',\n            strategy: 'function',\n            template: '{{self}} <=> {{arg0}}',\n          },\n        }),\n    },\n    {\n      method: 'cosineSimilarity',\n      self: { codecId: pgvectorTypeId },\n      impl: (\n        self: CodecExpression<'pg/vector@1', boolean, CT>,\n        other: CodecExpression<'pg/vector@1', boolean, CT>,\n      ): Expression<{ codecId: 'pg/float8@1'; nullable: false }> =>\n        buildOperation({\n          method: 'cosineSimilarity',\n          args: [toExpr(self, pgvectorTypeId), toExpr(other, pgvectorTypeId)],\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  ];\n}\n\nconst pgvectorPackMetaBase = {\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  authoring: {\n    type: pgvectorAuthoringTypes,\n  },\n  types: {\n    codecTypes: {\n      codecInstances: Object.values(codecDefinitions).map((def) => def.codec),\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    },\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\nexport const pgvectorPackMeta: typeof pgvectorPackMetaBase & {\n  readonly __codecTypes?: CodecTypes;\n} = pgvectorPackMetaBase;\n"],"mappings":";;;;;AAGA,MAAa,yBAAyB,EACpC,UAAU,EACR,QAAQ;CACN,MAAM;CACN,MAAM,CACJ;EAAE,MAAM;EAAU,MAAM;EAAU,SAAS;EAAM,SAAS;EAAG,SAAS;EAAgB,CACvF;CACD,QAAQ;EACN,SAAS;EACT,YAAY;EACZ,YAAY,EACV,QAAQ;GAAE,MAAM;GAAO,OAAO;GAAG,EAClC;EACF;CACF,EACF,EACF;;;;;;;;;;ACVD,MAAM,gBAAgB,MAAM;CAC1B,QAAQ;CACR,aAAa,CAAC,SAAS;CACvB,QAAQ,CAAC,WAAW;CACpB,mBAAmB,eAAe;EAChC,MAAM,SAAS,WAAW;AAC1B,MAAI,WAAW,OAAW,QAAO;AACjC,MAAI,OAAO,WAAW,YAAY,CAAC,OAAO,SAAS,OAAO,IAAI,CAAC,OAAO,UAAU,OAAO,CACrF,OAAM,IAAI,MACR,sFAAsF,OAAO,OAAO,GACrG;AAEH,SAAO,UAAU,OAAO;;CAE1B,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;;;;AC9DhC,MAAM,iBAAiB;AAIvB,SAAgB,0BAEuB;AACrC,QAAO,CACL;EACE,QAAQ;EACR,MAAM,EAAE,SAAS,gBAAgB;EACjC,OACE,MACA,UAEA,eAAe;GACb,QAAQ;GACR,MAAM,CAAC,OAAO,MAAM,eAAe,EAAE,OAAO,OAAO,eAAe,CAAC;GACnE,SAAS;IAAE,SAAS;IAAe,UAAU;IAAO;GACpD,UAAU;IACR,cAAc;IACd,UAAU;IACV,UAAU;IACX;GACF,CAAC;EACL,EACD;EACE,QAAQ;EACR,MAAM,EAAE,SAAS,gBAAgB;EACjC,OACE,MACA,UAEA,eAAe;GACb,QAAQ;GACR,MAAM,CAAC,OAAO,MAAM,eAAe,EAAE,OAAO,OAAO,eAAe,CAAC;GACnE,SAAS;IAAE,SAAS;IAAe,UAAU;IAAO;GACpD,UAAU;IACR,cAAc;IACd,UAAU;IACV,UAAU;IACX;GACF,CAAC;EACL,CACF;;AAGH,MAAM,uBAAuB;CAC3B,MAAM;CACN,IAAI;CACJ,UAAU;CACV,UAAU;CACV,SAAS;CACT,cAAc,EACZ,UAAU,EACR,mBAAmB,MACpB,EACF;CACD,WAAW,EACT,MAAM,wBACP;CACD,OAAO;EACL,YAAY;GACV,gBAAgB,OAAO,OAAO,iBAAiB,CAAC,KAAK,QAAQ,IAAI,MAAM;GACvE,QAAQ;IACN,SAAS;IACT,OAAO;IACP,OAAO;IACR;GACD,aAAa,CACX;IACE,SAAS;IACT,OAAO;IACP,OAAO;IACR,CACF;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;AAED,MAAaA,mBAET"}

package/dist/operation-types.d.mts

@@ -1,7 +1,11 @@
+import { CodecExpression, Expression } from "@prisma-next/sql-relational-core/expression";
 import { SqlQueryOperationTypes } from "@prisma-next/sql-contract/types";
 
 //#region src/types/operation-types.d.ts
-
+type CodecTypesBase = Record<string, {
+  readonly input: unknown;
+  readonly output: unknown;
+}>;
 /**
  * Operation type definitions for pgvector extension.
  *
@@ -11,64 +15,36 @@ import { SqlQueryOperationTypes } from "@prisma-next/sql-contract/types";
 type OperationTypes = {
   readonly 'pg/vector@1': {
     readonly cosineDistance: {
-      readonly args: readonly [{
+      readonly self: {
         readonly codecId: 'pg/vector@1';
-        readonly nullable: false;
-      }];
-      readonly returns: {
-        readonly codecId: 'pg/float8@1';
-        readonly nullable: false;
-      };
-      readonly lowering: {
-        readonly targetFamily: 'sql';
-        readonly strategy: 'function';
-        readonly template: string;
       };
     };
     readonly cosineSimilarity: {
-      readonly args: readonly [{
+      readonly self: {
         readonly codecId: 'pg/vector@1';
-        readonly nullable: false;
-      }];
-      readonly returns: {
-        readonly codecId: 'pg/float8@1';
-        readonly nullable: false;
-      };
-      readonly lowering: {
-        readonly targetFamily: 'sql';
-        readonly strategy: 'function';
-        readonly template: string;
       };
     };
   };
 };
 /** Flat operation signatures for the query builder. */
-type QueryOperationTypes = SqlQueryOperationTypes<{
+type QueryOperationTypes<CT extends CodecTypesBase> = SqlQueryOperationTypes<CT, {
   readonly cosineDistance: {
-    readonly args: readonly [{
-      readonly codecId: 'pg/vector@1';
-      readonly nullable: boolean;
-    }, {
+    readonly self: {
       readonly codecId: 'pg/vector@1';
-      readonly nullable: boolean;
-    }];
-    readonly returns: {
-      readonly codecId: 'pg/float8@1';
-      readonly nullable: false;
     };
+    readonly impl: (self: CodecExpression<'pg/vector@1', boolean, CT>, other: CodecExpression<'pg/vector@1', boolean, CT>) => Expression<{
+      codecId: 'pg/float8@1';
+      nullable: false;
+    }>;
   };
   readonly cosineSimilarity: {
-    readonly args: readonly [{
-      readonly codecId: 'pg/vector@1';
-      readonly nullable: boolean;
-    }, {
+    readonly self: {
       readonly codecId: 'pg/vector@1';
-      readonly nullable: boolean;
-    }];
-    readonly returns: {
-      readonly codecId: 'pg/float8@1';
-      readonly nullable: false;
     };
+    readonly impl: (self: CodecExpression<'pg/vector@1', boolean, CT>, other: CodecExpression<'pg/vector@1', boolean, CT>) => Expression<{
+      codecId: 'pg/float8@1';
+      nullable: false;
+    }>;
   };
 }>;
 //#endregion

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

@@ -1 +1 @@
-{"version":3,"file":"operation-types.d.mts","names":[],"sources":["../src/types/operation-types.ts"],"sourcesContent":[],"mappings":";;;;;;AASA;AAwBA;;;KAxBY,cAAA;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;KAwBA,mBAAA,GAAsB"}
+{"version":3,"file":"operation-types.d.mts","names":[],"sources":["../src/types/operation-types.ts"],"sourcesContent":[],"mappings":";;;;KAGK,cAAA,GAAiB;;EAAjB,SAAA,MAAA,EAAc,OAAA;AASnB,CAAA,CAAA;AAYA;;;;;;AAOe,KAnBH,cAAA,GAmBG;EACJ,SAAA,aAAA,EAAA;IAK2C,SAAA,cAAA,EAAA;MAAxC,SAAA,IAAA,EAAA;QACyC,SAAA,OAAA,EAAA,aAAA;MAAxC,CAAA;IACJ,CAAA;IAfkD,SAAA,gBAAA,EAAA;MAAsB,SAAA,IAAA,EAAA;;;;;;;KAAvE,+BAA+B,kBAAkB,uBAC3D;;;;;0BAKY,wCAAwC,YACvC,wCAAwC,QAC5C;;;;;;;;;0BAKG,wCAAwC,YACvC,wCAAwC,QAC5C"}

package/dist/pack.d.mts

@@ -1,7 +1,8 @@
 import { t as CodecTypes } from "./codec-types-BifaP625.mjs";
-import * as _prisma_next_sql_relational_core_ast0 from "@prisma-next/sql-relational-core/ast";
+import * as _prisma_next_sql_relational_core_ast2 from "@prisma-next/sql-relational-core/ast";
 
 //#region src/core/descriptor-meta.d.ts
+
 declare const pgvectorPackMetaBase: {
   readonly kind: "extension";
   readonly id: "pgvector";
@@ -41,7 +42,7 @@ declare const pgvectorPackMetaBase: {
   };
   readonly types: {
     readonly codecTypes: {
-      readonly codecInstances: _prisma_next_sql_relational_core_ast0.Codec<"pg/vector@1", readonly ["equality"], string, number[], Record<string, unknown>, unknown>[];
+      readonly codecInstances: _prisma_next_sql_relational_core_ast2.Codec<"pg/vector@1", readonly ["equality"], string, number[], Record<string, unknown>, unknown>[];
       readonly import: {
         readonly package: "@prisma-next/extension-pgvector/codec-types";
         readonly named: "CodecTypes";

package/dist/pack.d.mts.map

@@ -1 +1 @@
-{"version":3,"file":"pack.d.mts","names":[],"sources":["../src/core/descriptor-meta.ts"],"sourcesContent":[],"mappings":";;;;cAoCM;EAAA,SAAA,IAAA,EAAA,WAgDI;EAEG,SAAA,EAAA,EAAA,UAEW;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;+BAJd,qCAAA,CAAA,8DAAA;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;cAEG,yBAAyB;0BACZ"}
+{"version":3,"file":"pack.d.mts","names":[],"sources":["../src/core/descriptor-meta.ts"],"sourcesContent":[],"mappings":";;;;;cA0DM;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;+BAgDI,qCAAA,CAAA,8DAAA;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;cAEG,yBAAyB;0BACZ"}

package/dist/pack.mjs

@@ -1,3 +1,3 @@
-import { t as pgvectorPackMeta } from "./descriptor-meta-BQbvJJxu.mjs";
+import { t as pgvectorPackMeta } from "./descriptor-meta-DTopW_37.mjs";
 
 export { pgvectorPackMeta as default };

package/dist/runtime.mjs

@@ -1,5 +1,5 @@
 import { n as VECTOR_MAX_DIM, t as VECTOR_CODEC_ID } from "./constants-Co5golCK.mjs";
-import { n as pgvectorQueryOperations, r as codecDefinitions, t as pgvectorPackMeta } from "./descriptor-meta-BQbvJJxu.mjs";
+import { n as pgvectorQueryOperations, r as codecDefinitions, t as pgvectorPackMeta } from "./descriptor-meta-DTopW_37.mjs";
 import { createCodecRegistry } from "@prisma-next/sql-relational-core/ast";
 import { type } from "arktype";
 
@@ -25,7 +25,7 @@ const pgvectorRuntimeDescriptor = {
 	familyId: "sql",
 	targetId: "postgres",
 	codecs: createPgvectorCodecRegistry,
-	queryOperations: () => pgvectorQueryOperations,
+	queryOperations: () => pgvectorQueryOperations(),
 	parameterizedCodecs: () => parameterizedCodecDescriptors,
 	create() {
 		return {

package/dist/runtime.mjs.map

@@ -1 +1 @@
-{"version":3,"file":"runtime.mjs","names":["arktype","pgvectorRuntimeDescriptor: SqlRuntimeExtensionDescriptor<'postgres'>"],"sources":["../src/exports/runtime.ts"],"sourcesContent":["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 { pgvectorPackMeta, pgvectorQueryOperations } 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  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":";;;;;;AAuBA,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,uBAAuB;CACvB,2BAA2B;CAC3B,SAAS;AACP,SAAO;GACL,UAAU;GACV,UAAU;GACX;;CAEJ;AAED,sBAAe"}
+{"version":3,"file":"runtime.mjs","names":["arktype","pgvectorRuntimeDescriptor: SqlRuntimeExtensionDescriptor<'postgres'>"],"sources":["../src/exports/runtime.ts"],"sourcesContent":["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 { pgvectorPackMeta, pgvectorQueryOperations } 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  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":";;;;;;AAuBA,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,uBAAuB,yBAAyB;CAChD,2BAA2B;CAC3B,SAAS;AACP,SAAO;GACL,UAAU;GACV,UAAU;GACX;;CAEJ;AAED,sBAAe"}

package/package.json

@@ -1,29 +1,29 @@
 {
   "name": "@prisma-next/extension-pgvector",
-  "version": "0.5.0-dev.14",
+  "version": "0.5.0-dev.15",
   "type": "module",
   "sideEffects": false,
   "dependencies": {
     "arktype": "^2.0.0",
-    "@prisma-next/contract-authoring": "0.5.0-dev.14",
-    "@prisma-next/family-sql": "0.5.0-dev.14",
-    "@prisma-next/sql-operations": "0.5.0-dev.14",
-    "@prisma-next/sql-relational-core": "0.5.0-dev.14",
-    "@prisma-next/contract": "0.5.0-dev.14",
-    "@prisma-next/framework-components": "0.5.0-dev.14",
-    "@prisma-next/sql-runtime": "0.5.0-dev.14",
-    "@prisma-next/sql-schema-ir": "0.5.0-dev.14"
+    "@prisma-next/contract-authoring": "0.5.0-dev.15",
+    "@prisma-next/contract": "0.5.0-dev.15",
+    "@prisma-next/framework-components": "0.5.0-dev.15",
+    "@prisma-next/family-sql": "0.5.0-dev.15",
+    "@prisma-next/sql-relational-core": "0.5.0-dev.15",
+    "@prisma-next/sql-runtime": "0.5.0-dev.15",
+    "@prisma-next/sql-schema-ir": "0.5.0-dev.15",
+    "@prisma-next/sql-operations": "0.5.0-dev.15"
   },
   "devDependencies": {
     "tsdown": "0.18.4",
     "typescript": "5.9.3",
     "vitest": "4.0.17",
-    "@prisma-next/operations": "0.5.0-dev.14",
-    "@prisma-next/sql-contract": "0.5.0-dev.14",
-    "@prisma-next/sql-contract-ts": "0.5.0-dev.14",
-    "@prisma-next/tsdown": "0.0.0",
+    "@prisma-next/operations": "0.5.0-dev.15",
+    "@prisma-next/sql-contract": "0.5.0-dev.15",
     "@prisma-next/test-utils": "0.0.1",
-    "@prisma-next/tsconfig": "0.0.0"
+    "@prisma-next/sql-contract-ts": "0.5.0-dev.15",
+    "@prisma-next/tsconfig": "0.0.0",
+    "@prisma-next/tsdown": "0.0.0"
   },
   "files": [
     "dist",

package/src/core/descriptor-meta.ts

@@ -1,38 +1,60 @@
 import type { SqlOperationDescriptor } from '@prisma-next/sql-operations';
+import {
+  buildOperation,
+  type CodecExpression,
+  type Expression,
+  toExpr,
+} from '@prisma-next/sql-relational-core/expression';
 import type { CodecTypes } from '../types/codec-types';
 import { pgvectorAuthoringTypes } from './authoring';
 import { codecDefinitions } from './codecs';
 
 const pgvectorTypeId = 'pg/vector@1' as const;
 
-export const pgvectorQueryOperations: readonly SqlOperationDescriptor[] = [
-  {
-    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}}',
+type CodecTypesBase = Record<string, { readonly input: unknown; readonly output: unknown }>;
+
+export function pgvectorQueryOperations<
+  CT extends CodecTypesBase,
+>(): readonly SqlOperationDescriptor[] {
+  return [
+    {
+      method: 'cosineDistance',
+      self: { codecId: pgvectorTypeId },
+      impl: (
+        self: CodecExpression<'pg/vector@1', boolean, CT>,
+        other: CodecExpression<'pg/vector@1', boolean, CT>,
+      ): Expression<{ codecId: 'pg/float8@1'; nullable: false }> =>
+        buildOperation({
+          method: 'cosineDistance',
+          args: [toExpr(self, pgvectorTypeId), toExpr(other, pgvectorTypeId)],
+          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}})',
+    {
+      method: 'cosineSimilarity',
+      self: { codecId: pgvectorTypeId },
+      impl: (
+        self: CodecExpression<'pg/vector@1', boolean, CT>,
+        other: CodecExpression<'pg/vector@1', boolean, CT>,
+      ): Expression<{ codecId: 'pg/float8@1'; nullable: false }> =>
+        buildOperation({
+          method: 'cosineSimilarity',
+          args: [toExpr(self, pgvectorTypeId), toExpr(other, pgvectorTypeId)],
+          returns: { codecId: 'pg/float8@1', nullable: false },
+          lowering: {
+            targetFamily: 'sql',
+            strategy: 'function',
+            template: '1 - ({{self}} <=> {{arg0}})',
+          },
+        }),
     },
-  },
-];
+  ];
+}
 
 const pgvectorPackMetaBase = {
   kind: 'extension',

package/src/exports/control.ts

@@ -75,7 +75,7 @@ const pgvectorExtensionDescriptor: SqlControlExtensionDescriptor<'postgres'> = {
       },
     },
   },
-  queryOperations: () => pgvectorQueryOperations,
+  queryOperations: () => pgvectorQueryOperations(),
   databaseDependencies: pgvectorDatabaseDependencies,
   create: () => ({
     familyId: 'sql' as const,

package/src/exports/runtime.ts

@@ -45,7 +45,7 @@ const pgvectorRuntimeDescriptor: SqlRuntimeExtensionDescriptor<'postgres'> = {
   familyId: 'sql' as const,
   targetId: 'postgres' as const,
   codecs: createPgvectorCodecRegistry,
-  queryOperations: () => pgvectorQueryOperations,
+  queryOperations: () => pgvectorQueryOperations(),
   parameterizedCodecs: () => parameterizedCodecDescriptors,
   create() {
     return {

package/src/types/operation-types.ts

@@ -1,4 +1,7 @@
 import type { SqlQueryOperationTypes } from '@prisma-next/sql-contract/types';
+import type { CodecExpression, Expression } from '@prisma-next/sql-relational-core/expression';
+
+type CodecTypesBase = Record<string, { readonly input: unknown; readonly output: unknown }>;
 
 /**
  * Operation type definitions for pgvector extension.
@@ -10,40 +13,31 @@ import type { SqlQueryOperationTypes } from '@prisma-next/sql-contract/types';
 export type OperationTypes = {
   readonly 'pg/vector@1': {
     readonly cosineDistance: {
-      readonly args: readonly [{ readonly codecId: 'pg/vector@1'; readonly nullable: false }];
-      readonly returns: { readonly codecId: 'pg/float8@1'; readonly nullable: false };
-      readonly lowering: {
-        readonly targetFamily: 'sql';
-        readonly strategy: 'function';
-        readonly template: string;
-      };
+      readonly self: { readonly codecId: 'pg/vector@1' };
     };
     readonly cosineSimilarity: {
-      readonly args: readonly [{ readonly codecId: 'pg/vector@1'; readonly nullable: false }];
-      readonly returns: { readonly codecId: 'pg/float8@1'; readonly nullable: false };
-      readonly lowering: {
-        readonly targetFamily: 'sql';
-        readonly strategy: 'function';
-        readonly template: string;
-      };
+      readonly self: { readonly codecId: 'pg/vector@1' };
     };
   };
 };
 
 /** 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 };
-  };
-  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 };
-  };
-}>;
+export type QueryOperationTypes<CT extends CodecTypesBase> = SqlQueryOperationTypes<
+  CT,
+  {
+    readonly cosineDistance: {
+      readonly self: { readonly codecId: 'pg/vector@1' };
+      readonly impl: (
+        self: CodecExpression<'pg/vector@1', boolean, CT>,
+        other: CodecExpression<'pg/vector@1', boolean, CT>,
+      ) => Expression<{ codecId: 'pg/float8@1'; nullable: false }>;
+    };
+    readonly cosineSimilarity: {
+      readonly self: { readonly codecId: 'pg/vector@1' };
+      readonly impl: (
+        self: CodecExpression<'pg/vector@1', boolean, CT>,
+        other: CodecExpression<'pg/vector@1', boolean, CT>,
+      ) => Expression<{ codecId: 'pg/float8@1'; nullable: false }>;
+    };
+  }
+>;

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

@@ -1 +0,0 @@
-{"version":3,"file":"descriptor-meta-BQbvJJxu.mjs","names":["pgvectorQueryOperations: readonly SqlOperationDescriptor[]","pgvectorPackMeta: typeof pgvectorPackMetaBase & {\n  readonly __codecTypes?: CodecTypes;\n}"],"sources":["../src/core/authoring.ts","../src/core/codecs.ts","../src/core/descriptor-meta.ts"],"sourcesContent":["import type { AuthoringTypeNamespace } from '@prisma-next/framework-components/authoring';\nimport { VECTOR_MAX_DIM } from './constants';\n\nexport const pgvectorAuthoringTypes = {\n  pgvector: {\n    Vector: {\n      kind: 'typeConstructor',\n      args: [\n        { kind: 'number', name: 'length', integer: true, minimum: 1, maximum: VECTOR_MAX_DIM },\n      ],\n      output: {\n        codecId: 'pg/vector@1',\n        nativeType: 'vector',\n        typeParams: {\n          length: { kind: 'arg', index: 0 },\n        },\n      },\n    },\n  },\n} as const satisfies AuthoringTypeNamespace;\n","/**\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  renderOutputType: (typeParams) => {\n    const length = typeParams['length'];\n    if (length === undefined) return undefined;\n    if (typeof length !== 'number' || !Number.isFinite(length) || !Number.isInteger(length)) {\n      throw new Error(\n        `renderOutputType: expected positive integer \"length\" in typeParams for Vector, got ${String(length)}`,\n      );\n    }\n    return `Vector<${length}>`;\n  },\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 type { SqlOperationDescriptor } from '@prisma-next/sql-operations';\nimport type { CodecTypes } from '../types/codec-types';\nimport { pgvectorAuthoringTypes } from './authoring';\nimport { codecDefinitions } from './codecs';\n\nconst pgvectorTypeId = 'pg/vector@1' as const;\n\nexport const pgvectorQueryOperations: readonly SqlOperationDescriptor[] = [\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\nconst pgvectorPackMetaBase = {\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  authoring: {\n    type: pgvectorAuthoringTypes,\n  },\n  types: {\n    codecTypes: {\n      codecInstances: Object.values(codecDefinitions).map((def) => def.codec),\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    },\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\nexport const pgvectorPackMeta: typeof pgvectorPackMetaBase & {\n  readonly __codecTypes?: CodecTypes;\n} = pgvectorPackMetaBase;\n"],"mappings":";;;;AAGA,MAAa,yBAAyB,EACpC,UAAU,EACR,QAAQ;CACN,MAAM;CACN,MAAM,CACJ;EAAE,MAAM;EAAU,MAAM;EAAU,SAAS;EAAM,SAAS;EAAG,SAAS;EAAgB,CACvF;CACD,QAAQ;EACN,SAAS;EACT,YAAY;EACZ,YAAY,EACV,QAAQ;GAAE,MAAM;GAAO,OAAO;GAAG,EAClC;EACF;CACF,EACF,EACF;;;;;;;;;;ACVD,MAAM,gBAAgB,MAAM;CAC1B,QAAQ;CACR,aAAa,CAAC,SAAS;CACvB,QAAQ,CAAC,WAAW;CACpB,mBAAmB,eAAe;EAChC,MAAM,SAAS,WAAW;AAC1B,MAAI,WAAW,OAAW,QAAO;AACjC,MAAI,OAAO,WAAW,YAAY,CAAC,OAAO,SAAS,OAAO,IAAI,CAAC,OAAO,UAAU,OAAO,CACrF,OAAM,IAAI,MACR,sFAAsF,OAAO,OAAO,GACrG;AAEH,SAAO,UAAU,OAAO;;CAE1B,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;;;;ACpEhC,MAAM,iBAAiB;AAEvB,MAAaA,0BAA6D,CACxE;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,MAAM,uBAAuB;CAC3B,MAAM;CACN,IAAI;CACJ,UAAU;CACV,UAAU;CACV,SAAS;CACT,cAAc,EACZ,UAAU,EACR,mBAAmB,MACpB,EACF;CACD,WAAW,EACT,MAAM,wBACP;CACD,OAAO;EACL,YAAY;GACV,gBAAgB,OAAO,OAAO,iBAAiB,CAAC,KAAK,QAAQ,IAAI,MAAM;GACvE,QAAQ;IACN,SAAS;IACT,OAAO;IACP,OAAO;IACR;GACD,aAAa,CACX;IACE,SAAS;IACT,OAAO;IACP,OAAO;IACR,CACF;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;AAED,MAAaC,mBAET"}