@prisma-next/extension-pgvector 0.3.0-dev.34 → 0.3.0-dev.37

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (68) hide show
  1. package/README.md +11 -8
  2. package/dist/codec-types.d.mts +41 -0
  3. package/dist/codec-types.d.mts.map +1 -0
  4. package/dist/codec-types.mjs +1 -0
  5. package/dist/{exports/column-types.d.ts → column-types.d.mts} +14 -15
  6. package/dist/column-types.d.mts.map +1 -0
  7. package/dist/column-types.mjs +36 -0
  8. package/dist/column-types.mjs.map +1 -0
  9. package/dist/constants-BmZpaev5.mjs +13 -0
  10. package/dist/constants-BmZpaev5.mjs.map +1 -0
  11. package/dist/control.d.mts +7 -0
  12. package/dist/control.d.mts.map +1 -0
  13. package/dist/control.mjs +49 -0
  14. package/dist/control.mjs.map +1 -0
  15. package/dist/descriptor-meta-BADTQp0u.mjs +57 -0
  16. package/dist/descriptor-meta-BADTQp0u.mjs.map +1 -0
  17. package/dist/operation-types.d.mts +32 -0
  18. package/dist/operation-types.d.mts.map +1 -0
  19. package/dist/operation-types.mjs +1 -0
  20. package/dist/pack.d.mts +46 -0
  21. package/dist/pack.d.mts.map +1 -0
  22. package/dist/pack.mjs +9 -0
  23. package/dist/pack.mjs.map +1 -0
  24. package/dist/runtime.d.mts +7 -0
  25. package/dist/runtime.d.mts.map +1 -0
  26. package/dist/runtime.mjs +74 -0
  27. package/dist/runtime.mjs.map +1 -0
  28. package/package.json +32 -44
  29. package/src/core/descriptor-meta.ts +6 -20
  30. package/src/exports/control.ts +2 -13
  31. package/src/exports/runtime.ts +14 -38
  32. package/dist/chunk-DSJTFXJR.js +0 -73
  33. package/dist/chunk-DSJTFXJR.js.map +0 -1
  34. package/dist/chunk-MUHUVAQX.js +0 -9
  35. package/dist/chunk-MUHUVAQX.js.map +0 -1
  36. package/dist/core/codecs.d.ts +0 -23
  37. package/dist/core/codecs.d.ts.map +0 -1
  38. package/dist/core/constants.d.ts +0 -9
  39. package/dist/core/constants.d.ts.map +0 -1
  40. package/dist/core/descriptor-meta.d.ts +0 -61
  41. package/dist/core/descriptor-meta.d.ts.map +0 -1
  42. package/dist/exports/codec-types.d.ts +0 -7
  43. package/dist/exports/codec-types.d.ts.map +0 -1
  44. package/dist/exports/codec-types.js +0 -1
  45. package/dist/exports/codec-types.js.map +0 -1
  46. package/dist/exports/column-types.d.ts.map +0 -1
  47. package/dist/exports/column-types.js +0 -27
  48. package/dist/exports/column-types.js.map +0 -1
  49. package/dist/exports/control.d.ts +0 -9
  50. package/dist/exports/control.d.ts.map +0 -1
  51. package/dist/exports/control.js +0 -67
  52. package/dist/exports/control.js.map +0 -1
  53. package/dist/exports/operation-types.d.ts +0 -7
  54. package/dist/exports/operation-types.d.ts.map +0 -1
  55. package/dist/exports/operation-types.js +0 -1
  56. package/dist/exports/operation-types.js.map +0 -1
  57. package/dist/exports/pack.d.ts +0 -60
  58. package/dist/exports/pack.d.ts.map +0 -1
  59. package/dist/exports/pack.js +0 -11
  60. package/dist/exports/pack.js.map +0 -1
  61. package/dist/exports/runtime.d.ts +0 -8
  62. package/dist/exports/runtime.d.ts.map +0 -1
  63. package/dist/exports/runtime.js +0 -112
  64. package/dist/exports/runtime.js.map +0 -1
  65. package/dist/types/codec-types.d.ts +0 -34
  66. package/dist/types/codec-types.d.ts.map +0 -1
  67. package/dist/types/operation-types.d.ts +0 -31
  68. package/dist/types/operation-types.d.ts.map +0 -1
@@ -1,112 +0,0 @@
1
- import {
2
- VECTOR_CODEC_ID,
3
- VECTOR_MAX_DIM
4
- } from "../chunk-MUHUVAQX.js";
5
- import {
6
- pgvectorPackMeta,
7
- pgvectorRuntimeOperation
8
- } from "../chunk-DSJTFXJR.js";
9
-
10
- // src/exports/runtime.ts
11
- import { createCodecRegistry } from "@prisma-next/sql-relational-core/ast";
12
- import { type as arktype } from "arktype";
13
-
14
- // src/core/codecs.ts
15
- import { codec, defineCodecs } from "@prisma-next/sql-relational-core/ast";
16
- var pgVectorCodec = codec({
17
- typeId: "pg/vector@1",
18
- targetTypes: ["vector"],
19
- encode: (value) => {
20
- if (!Array.isArray(value)) {
21
- throw new Error("Vector value must be an array of numbers");
22
- }
23
- if (!value.every((v) => typeof v === "number")) {
24
- throw new Error("Vector value must contain only numbers");
25
- }
26
- return `[${value.join(",")}]`;
27
- },
28
- decode: (wire) => {
29
- if (typeof wire !== "string") {
30
- throw new Error("Vector wire value must be a string");
31
- }
32
- if (!wire.startsWith("[") || !wire.endsWith("]")) {
33
- throw new Error(`Invalid vector format: expected "[...]", got "${wire}"`);
34
- }
35
- const content = wire.slice(1, -1).trim();
36
- if (content === "") {
37
- return [];
38
- }
39
- const values = content.split(",").map((v) => {
40
- const num = Number.parseFloat(v.trim());
41
- if (Number.isNaN(num)) {
42
- throw new Error(`Invalid vector value: "${v}" is not a number`);
43
- }
44
- return num;
45
- });
46
- return values;
47
- },
48
- meta: {
49
- db: {
50
- sql: {
51
- postgres: {
52
- nativeType: "vector"
53
- }
54
- }
55
- }
56
- }
57
- });
58
- var codecs = defineCodecs().add("vector", pgVectorCodec);
59
- var codecDefinitions = codecs.codecDefinitions;
60
- var dataTypes = codecs.dataTypes;
61
-
62
- // src/exports/runtime.ts
63
- var vectorParamsSchema = arktype({
64
- length: "number"
65
- }).narrow((params, ctx) => {
66
- const { length } = params;
67
- if (!Number.isInteger(length)) {
68
- return ctx.mustBe("an integer");
69
- }
70
- if (length < 1 || length > VECTOR_MAX_DIM) {
71
- return ctx.mustBe(`in the range [1, ${VECTOR_MAX_DIM}]`);
72
- }
73
- return true;
74
- });
75
- var parameterizedCodecDescriptors = [
76
- {
77
- codecId: VECTOR_CODEC_ID,
78
- paramsSchema: vectorParamsSchema
79
- }
80
- ];
81
- var PgVectorRuntimeExtensionInstance = class {
82
- familyId = "sql";
83
- targetId = "postgres";
84
- codecs() {
85
- const registry = createCodecRegistry();
86
- for (const def of Object.values(codecDefinitions)) {
87
- registry.register(def.codec);
88
- }
89
- return registry;
90
- }
91
- operations() {
92
- return [pgvectorRuntimeOperation];
93
- }
94
- parameterizedCodecs() {
95
- return parameterizedCodecDescriptors;
96
- }
97
- };
98
- var pgvectorRuntimeDescriptor = {
99
- kind: "extension",
100
- id: pgvectorPackMeta.id,
101
- version: pgvectorPackMeta.version,
102
- familyId: "sql",
103
- targetId: "postgres",
104
- create() {
105
- return new PgVectorRuntimeExtensionInstance();
106
- }
107
- };
108
- var runtime_default = pgvectorRuntimeDescriptor;
109
- export {
110
- runtime_default as default
111
- };
112
- //# sourceMappingURL=runtime.js.map
@@ -1 +0,0 @@
1
- {"version":3,"sources":["../../src/exports/runtime.ts","../../src/core/codecs.ts"],"sourcesContent":["import type { SqlOperationSignature } from '@prisma-next/sql-operations';\nimport type { CodecRegistry } from '@prisma-next/sql-relational-core/ast';\nimport { createCodecRegistry } from '@prisma-next/sql-relational-core/ast';\nimport type {\n RuntimeParameterizedCodecDescriptor,\n SqlRuntimeExtensionDescriptor,\n SqlRuntimeExtensionInstance,\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, pgvectorRuntimeOperation } 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\n/**\n * Pre-allocated parameterized codec descriptors to avoid per-call allocations.\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\n/**\n * pgvector SQL runtime extension instance.\n * Provides codecs and operations for vector data type and similarity operations.\n */\nclass PgVectorRuntimeExtensionInstance implements SqlRuntimeExtensionInstance<'postgres'> {\n readonly familyId = 'sql' as const;\n readonly targetId = 'postgres' as const;\n\n codecs(): CodecRegistry {\n const registry = createCodecRegistry();\n // Register all codecs from codecDefinitions\n for (const def of Object.values(codecDefinitions)) {\n registry.register(def.codec);\n }\n return registry;\n }\n\n operations(): ReadonlyArray<SqlOperationSignature> {\n return [pgvectorRuntimeOperation];\n }\n\n parameterizedCodecs(): ReadonlyArray<\n RuntimeParameterizedCodecDescriptor<{ readonly length: number }>\n > {\n return parameterizedCodecDescriptors;\n }\n}\n\n/**\n * pgvector SQL runtime extension descriptor.\n * Provides metadata and factory for creating runtime extension instances.\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 create(): SqlRuntimeExtensionInstance<'postgres'> {\n return new PgVectorRuntimeExtensionInstance();\n },\n};\n\nexport default pgvectorRuntimeDescriptor;\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<'pg/vector@1', string, number[]>({\n typeId: 'pg/vector@1',\n targetTypes: ['vector'],\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"],"mappings":";;;;;;;;;;AAEA,SAAS,2BAA2B;AAMpC,SAAS,QAAQ,eAAe;;;ACDhC,SAAS,OAAO,oBAAoB;AAEpC,IAAM,gBAAgB,MAAuC;AAAA,EAC3D,QAAQ;AAAA,EACR,aAAa,CAAC,QAAQ;AAAA,EACtB,QAAQ,CAAC,UAA4B;AAEnC,QAAI,CAAC,MAAM,QAAQ,KAAK,GAAG;AACzB,YAAM,IAAI,MAAM,0CAA0C;AAAA,IAC5D;AACA,QAAI,CAAC,MAAM,MAAM,CAAC,MAAM,OAAO,MAAM,QAAQ,GAAG;AAC9C,YAAM,IAAI,MAAM,wCAAwC;AAAA,IAC1D;AAGA,WAAO,IAAI,MAAM,KAAK,GAAG,CAAC;AAAA,EAC5B;AAAA,EACA,QAAQ,CAAC,SAA2B;AAElC,QAAI,OAAO,SAAS,UAAU;AAC5B,YAAM,IAAI,MAAM,oCAAoC;AAAA,IACtD;AAEA,QAAI,CAAC,KAAK,WAAW,GAAG,KAAK,CAAC,KAAK,SAAS,GAAG,GAAG;AAChD,YAAM,IAAI,MAAM,iDAAiD,IAAI,GAAG;AAAA,IAC1E;AACA,UAAM,UAAU,KAAK,MAAM,GAAG,EAAE,EAAE,KAAK;AACvC,QAAI,YAAY,IAAI;AAClB,aAAO,CAAC;AAAA,IACV;AACA,UAAM,SAAS,QAAQ,MAAM,GAAG,EAAE,IAAI,CAAC,MAAM;AAC3C,YAAM,MAAM,OAAO,WAAW,EAAE,KAAK,CAAC;AACtC,UAAI,OAAO,MAAM,GAAG,GAAG;AACrB,cAAM,IAAI,MAAM,0BAA0B,CAAC,mBAAmB;AAAA,MAChE;AACA,aAAO;AAAA,IACT,CAAC;AACD,WAAO;AAAA,EACT;AAAA,EACA,MAAM;AAAA,IACJ,IAAI;AAAA,MACF,KAAK;AAAA,QACH,UAAU;AAAA,UACR,YAAY;AAAA,QACd;AAAA,MACF;AAAA,IACF;AAAA,EACF;AACF,CAAC;AAGD,IAAM,SAAS,aAAa,EAAE,IAAI,UAAU,aAAa;AAGlD,IAAM,mBAAmB,OAAO;AAChC,IAAM,YAAY,OAAO;;;ADjDhC,IAAM,qBAAqB,QAAQ;AAAA,EACjC,QAAQ;AACV,CAAC,EAAE,OAAO,CAAC,QAAQ,QAAQ;AACzB,QAAM,EAAE,OAAO,IAAI;AACnB,MAAI,CAAC,OAAO,UAAU,MAAM,GAAG;AAC7B,WAAO,IAAI,OAAO,YAAY;AAAA,EAChC;AACA,MAAI,SAAS,KAAK,SAAS,gBAAgB;AACzC,WAAO,IAAI,OAAO,oBAAoB,cAAc,GAAG;AAAA,EACzD;AACA,SAAO;AACT,CAAC;AAKD,IAAM,gCAAgC;AAAA,EACpC;AAAA,IACE,SAAS;AAAA,IACT,cAAc;AAAA,EAChB;AACF;AAQA,IAAM,mCAAN,MAA0F;AAAA,EAC/E,WAAW;AAAA,EACX,WAAW;AAAA,EAEpB,SAAwB;AACtB,UAAM,WAAW,oBAAoB;AAErC,eAAW,OAAO,OAAO,OAAO,gBAAgB,GAAG;AACjD,eAAS,SAAS,IAAI,KAAK;AAAA,IAC7B;AACA,WAAO;AAAA,EACT;AAAA,EAEA,aAAmD;AACjD,WAAO,CAAC,wBAAwB;AAAA,EAClC;AAAA,EAEA,sBAEE;AACA,WAAO;AAAA,EACT;AACF;AAMA,IAAM,4BAAuE;AAAA,EAC3E,MAAM;AAAA,EACN,IAAI,iBAAiB;AAAA,EACrB,SAAS,iBAAiB;AAAA,EAC1B,UAAU;AAAA,EACV,UAAU;AAAA,EACV,SAAkD;AAChD,WAAO,IAAI,iCAAiC;AAAA,EAC9C;AACF;AAEA,IAAO,kBAAQ;","names":[]}
@@ -1,34 +0,0 @@
1
- /**
2
- * Codec type definitions for pgvector extension.
3
- *
4
- * This file exports type-only definitions for codec input/output types.
5
- * These types are imported by contract.d.ts files for compile-time type inference.
6
- *
7
- * Runtime codec implementations are provided by the extension's codec registry.
8
- */
9
- import type { CodecTypes as CoreCodecTypes } from '../core/codecs';
10
- /**
11
- * Type-level branded vector.
12
- *
13
- * The runtime values are plain number arrays, but parameterized column typing can
14
- * carry the dimension at the type level (e.g. Vector<1536>).
15
- */
16
- export type Vector<N extends number = number> = number[] & {
17
- readonly __vectorLength?: N;
18
- };
19
- /**
20
- * Codec types for pgvector.
21
- *
22
- * - Scalar output remains `number[]` (runtime representation).
23
- * - `parameterizedOutput` enables lane typing to compute `Vector<N>` from column `typeParams`.
24
- */
25
- export type CodecTypes = CoreCodecTypes & {
26
- readonly 'pg/vector@1': CoreCodecTypes['pg/vector@1'] & {
27
- readonly parameterizedOutput: <P extends {
28
- readonly length: number;
29
- }>(params: P) => P extends {
30
- readonly length: infer N extends number;
31
- } ? Vector<N> : Vector<number>;
32
- };
33
- };
34
- //# sourceMappingURL=codec-types.d.ts.map
@@ -1 +0,0 @@
1
- {"version":3,"file":"codec-types.d.ts","sourceRoot":"","sources":["../../src/types/codec-types.ts"],"names":[],"mappings":"AAAA;;;;;;;GAOG;AAEH,OAAO,KAAK,EAAE,UAAU,IAAI,cAAc,EAAE,MAAM,gBAAgB,CAAC;AAEnE;;;;;GAKG;AACH,MAAM,MAAM,MAAM,CAAC,CAAC,SAAS,MAAM,GAAG,MAAM,IAAI,MAAM,EAAE,GAAG;IAAE,QAAQ,CAAC,cAAc,CAAC,EAAE,CAAC,CAAA;CAAE,CAAC;AAE3F;;;;;GAKG;AACH,MAAM,MAAM,UAAU,GAAG,cAAc,GAAG;IACxC,QAAQ,CAAC,aAAa,EAAE,cAAc,CAAC,aAAa,CAAC,GAAG;QACtD,QAAQ,CAAC,mBAAmB,EAAE,CAAC,CAAC,SAAS;YAAE,QAAQ,CAAC,MAAM,EAAE,MAAM,CAAA;SAAE,EAClE,MAAM,EAAE,CAAC,KACN,CAAC,SAAS;YAAE,QAAQ,CAAC,MAAM,EAAE,MAAM,CAAC,SAAS,MAAM,CAAA;SAAE,GAAG,MAAM,CAAC,CAAC,CAAC,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC;KACzF,CAAC;CACH,CAAC"}
@@ -1,31 +0,0 @@
1
- /**
2
- * Operation type definitions for pgvector extension.
3
- *
4
- * This file exports type-only definitions for operation method signatures.
5
- * These types are imported by contract.d.ts files for compile-time type inference.
6
- */
7
- /**
8
- * Operation types for pgvector extension.
9
- * Maps typeId to operation methods.
10
- */
11
- export type OperationTypes = {
12
- readonly 'pg/vector@1': {
13
- readonly cosineDistance: {
14
- readonly args: readonly [
15
- {
16
- readonly kind: 'param';
17
- }
18
- ];
19
- readonly returns: {
20
- readonly kind: 'builtin';
21
- readonly type: 'number';
22
- };
23
- readonly lowering: {
24
- readonly targetFamily: 'sql';
25
- readonly strategy: 'function';
26
- readonly template: string;
27
- };
28
- };
29
- };
30
- };
31
- //# sourceMappingURL=operation-types.d.ts.map
@@ -1 +0,0 @@
1
- {"version":3,"file":"operation-types.d.ts","sourceRoot":"","sources":["../../src/types/operation-types.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH;;;GAGG;AACH,MAAM,MAAM,cAAc,GAAG;IAC3B,QAAQ,CAAC,aAAa,EAAE;QACtB,QAAQ,CAAC,cAAc,EAAE;YACvB,QAAQ,CAAC,IAAI,EAAE,SAAS;gBACtB;oBACE,QAAQ,CAAC,IAAI,EAAE,OAAO,CAAC;iBACxB;aACF,CAAC;YACF,QAAQ,CAAC,OAAO,EAAE;gBAChB,QAAQ,CAAC,IAAI,EAAE,SAAS,CAAC;gBACzB,QAAQ,CAAC,IAAI,EAAE,QAAQ,CAAC;aACzB,CAAC;YACF,QAAQ,CAAC,QAAQ,EAAE;gBACjB,QAAQ,CAAC,YAAY,EAAE,KAAK,CAAC;gBAC7B,QAAQ,CAAC,QAAQ,EAAE,UAAU,CAAC;gBAC9B,QAAQ,CAAC,QAAQ,EAAE,MAAM,CAAC;aAC3B,CAAC;SACH,CAAC;KACH,CAAC;CACH,CAAC"}