@nicia-ai/typegraph 0.24.0 → 0.25.0

package/dist/chunk-WVQXAFP3.js

@@ -0,0 +1,79 @@
+import { z } from 'zod';
+
+// src/core/embedding.ts
+var EMBEDDING_DIMENSIONS_KEY = "_embeddingDimensions";
+var EMBEDDING_INDEX_KEY = "_embeddingIndex";
+var DEFAULT_HNSW_M = 16;
+var DEFAULT_HNSW_EF_CONSTRUCTION = 64;
+function embedding(dimensions, options = {}) {
+  if (!Number.isInteger(dimensions) || dimensions <= 0) {
+    throw new Error(
+      `Embedding dimensions must be a positive integer, got: ${dimensions}`
+    );
+  }
+  const indexConfig = resolveEmbeddingIndex(options);
+  const schema = z.custom(
+    (value) => {
+      if (!Array.isArray(value)) {
+        return false;
+      }
+      if (value.length !== dimensions) {
+        return false;
+      }
+      return value.every((n) => typeof n === "number" && Number.isFinite(n));
+    },
+    {
+      message: `Expected an array of ${dimensions} finite numbers`
+    }
+  );
+  return Object.assign(schema, {
+    [EMBEDDING_DIMENSIONS_KEY]: dimensions,
+    [EMBEDDING_INDEX_KEY]: indexConfig
+  });
+}
+function resolveEmbeddingIndex(options) {
+  const indexType = options.indexType ?? "hnsw";
+  const m = options.m ?? DEFAULT_HNSW_M;
+  const efConstruction = options.efConstruction ?? DEFAULT_HNSW_EF_CONSTRUCTION;
+  if (m <= 0 || !Number.isInteger(m)) {
+    throw new Error(
+      `embedding() index option 'm' must be a positive integer, got: ${m}`
+    );
+  }
+  if (efConstruction <= 0 || !Number.isInteger(efConstruction)) {
+    throw new Error(
+      `embedding() index option 'efConstruction' must be a positive integer, got: ${efConstruction}`
+    );
+  }
+  if (options.lists !== void 0 && (options.lists <= 0 || !Number.isInteger(options.lists))) {
+    throw new Error(
+      `embedding() index option 'lists' must be a positive integer, got: ${options.lists}`
+    );
+  }
+  return Object.freeze({
+    metric: options.metric ?? "cosine",
+    indexType,
+    m,
+    efConstruction,
+    lists: options.lists
+  });
+}
+function isEmbeddingSchema(value) {
+  return typeof value === "object" && value !== null && EMBEDDING_DIMENSIONS_KEY in value && typeof value[EMBEDDING_DIMENSIONS_KEY] === "number";
+}
+function getEmbeddingDimensions(schema) {
+  if (isEmbeddingSchema(schema)) {
+    return schema[EMBEDDING_DIMENSIONS_KEY];
+  }
+  return void 0;
+}
+function getEmbeddingIndex(schema) {
+  if (!isEmbeddingSchema(schema)) return void 0;
+  const candidate = schema[EMBEDDING_INDEX_KEY];
+  if (candidate === void 0) return void 0;
+  return candidate;
+}
+
+export { embedding, getEmbeddingDimensions, getEmbeddingIndex, isEmbeddingSchema };
+//# sourceMappingURL=chunk-WVQXAFP3.js.map
+//# sourceMappingURL=chunk-WVQXAFP3.js.map

package/dist/chunk-WVQXAFP3.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/core/embedding.ts"],"names":[],"mappings":";;;AAqCO,IAAM,wBAAA,GAA2B,sBAAA;AAOjC,IAAM,mBAAA,GAAsB,iBAAA;AA4DnC,IAAM,cAAA,GAAiB,EAAA;AACvB,IAAM,4BAAA,GAA+B,EAAA;AAyD9B,SAAS,SAAA,CACd,UAAA,EACA,OAAA,GAAiC,EAAC,EACd;AACpB,EAAA,IAAI,CAAC,MAAA,CAAO,SAAA,CAAU,UAAU,CAAA,IAAK,cAAc,CAAA,EAAG;AACpD,IAAA,MAAM,IAAI,KAAA;AAAA,MACR,yDAAyD,UAAU,CAAA;AAAA,KACrE;AAAA,EACF;AAEA,EAAA,MAAM,WAAA,GAAc,sBAAsB,OAAO,CAAA;AAIjD,EAAA,MAAM,SAAS,CAAA,CAAE,MAAA;AAAA,IACf,CAAC,KAAA,KAAmC;AAClC,MAAA,IAAI,CAAC,KAAA,CAAM,OAAA,CAAQ,KAAK,CAAA,EAAG;AACzB,QAAA,OAAO,KAAA;AAAA,MACT;AACA,MAAA,IAAI,KAAA,CAAM,WAAW,UAAA,EAAY;AAC/B,QAAA,OAAO,KAAA;AAAA,MACT;AACA,MAAA,OAAO,KAAA,CAAM,KAAA,CAAM,CAAC,CAAA,KAAM,OAAO,MAAM,QAAA,IAAY,MAAA,CAAO,QAAA,CAAS,CAAC,CAAC,CAAA;AAAA,IACvE,CAAA;AAAA,IACA;AAAA,MACE,OAAA,EAAS,wBAAwB,UAAU,CAAA,eAAA;AAAA;AAC7C,GACF;AAEA,EAAA,OAAO,MAAA,CAAO,OAAO,MAAA,EAAQ;AAAA,IAC3B,CAAC,wBAAwB,GAAG,UAAA;AAAA,IAC5B,CAAC,mBAAmB,GAAG;AAAA,GACxB,CAAA;AACH;AAEA,SAAS,sBACP,OAAA,EACwB;AACxB,EAAA,MAAM,SAAA,GAAY,QAAQ,SAAA,IAAa,MAAA;AACvC,EAAA,MAAM,CAAA,GAAI,QAAQ,CAAA,IAAK,cAAA;AACvB,EAAA,MAAM,cAAA,GAAiB,QAAQ,cAAA,IAAkB,4BAAA;AAEjD,EAAA,IAAI,KAAK,CAAA,IAAK,CAAC,MAAA,CAAO,SAAA,CAAU,CAAC,CAAA,EAAG;AAClC,IAAA,MAAM,IAAI,KAAA;AAAA,MACR,iEAAiE,CAAC,CAAA;AAAA,KACpE;AAAA,EACF;AACA,EAAA,IAAI,kBAAkB,CAAA,IAAK,CAAC,MAAA,CAAO,SAAA,CAAU,cAAc,CAAA,EAAG;AAC5D,IAAA,MAAM,IAAI,KAAA;AAAA,MACR,8EAA8E,cAAc,CAAA;AAAA,KAC9F;AAAA,EACF;AACA,EAAA,IACE,OAAA,CAAQ,KAAA,KAAU,MAAA,KACjB,OAAA,CAAQ,KAAA,IAAS,CAAA,IAAK,CAAC,MAAA,CAAO,SAAA,CAAU,OAAA,CAAQ,KAAK,CAAA,CAAA,EACtD;AACA,IAAA,MAAM,IAAI,KAAA;AAAA,MACR,CAAA,kEAAA,EAAqE,QAAQ,KAAK,CAAA;AAAA,KACpF;AAAA,EACF;AAEA,EAAA,OAAO,OAAO,MAAA,CAAO;AAAA,IACnB,MAAA,EAAQ,QAAQ,MAAA,IAAU,QAAA;AAAA,IAC1B,SAAA;AAAA,IACA,CAAA;AAAA,IACA,cAAA;AAAA,IACA,OAAO,OAAA,CAAQ;AAAA,GAChB,CAAA;AACH;AASO,SAAS,kBAAkB,KAAA,EAA0C;AAC1E,EAAA,OACE,OAAO,KAAA,KAAU,QAAA,IACjB,KAAA,KAAU,IAAA,IACV,4BAA4B,KAAA,IAC5B,OAAQ,KAAA,CAAkC,wBAAwB,CAAA,KAChE,QAAA;AAEN;AAMO,SAAS,uBAAuB,MAAA,EAAuC;AAC5E,EAAA,IAAI,iBAAA,CAAkB,MAAM,CAAA,EAAG;AAC7B,IAAA,OAAO,OAAO,wBAAwB,CAAA;AAAA,EACxC;AACA,EAAA,OAAO,MAAA;AACT;AAQO,SAAS,kBACd,MAAA,EACoC;AACpC,EAAA,IAAI,CAAC,iBAAA,CAAkB,MAAM,CAAA,EAAG,OAAO,MAAA;AACvC,EAAA,MAAM,SAAA,GAAa,OACjB,mBACF,CAAA;AACA,EAAA,IAAI,SAAA,KAAc,QAAW,OAAO,MAAA;AACpC,EAAA,OAAO,SAAA;AACT","file":"chunk-WVQXAFP3.js","sourcesContent":["/**\n * Embedding type for vector search.\n *\n * Creates a Zod-compatible schema for vector embeddings with\n * dimension validation and metadata for query compilation.\n */\nimport { z } from \"zod\";\n\n// ============================================================\n// Embedding Brand Symbol\n// ============================================================\n\n/**\n * Symbol used to brand embedding values at the type level.\n * This allows TypeScript to distinguish embedding arrays from regular\n * number arrays in the query builder type system.\n */\ndeclare const EMBEDDING_BRAND: unique symbol;\n\n/**\n * Branded embedding type for type-level distinction.\n * At runtime this is just `readonly number[]`, but TypeScript can\n * distinguish it from regular arrays for query builder typing.\n */\nexport type EmbeddingValue = readonly number[] & {\n  readonly [EMBEDDING_BRAND]: true;\n};\n\n// ============================================================\n// Embedding Metadata Symbol\n// ============================================================\n\n/**\n * Symbol key for storing embedding dimensions on the schema.\n * This allows the schema introspector to detect embedding types\n * and extract dimension information.\n */\nexport const EMBEDDING_DIMENSIONS_KEY = \"_embeddingDimensions\" as const;\n\n/**\n * Symbol key for storing the embedding's preferred index configuration\n * on the schema. The schema introspector reads this when auto-deriving\n * `VectorIndexDeclaration` entries at `defineGraph()` time.\n */\nexport const EMBEDDING_INDEX_KEY = \"_embeddingIndex\" as const;\n\n/**\n * Distance metric used for vector similarity. Pinned at the embedding\n * brand because changing the metric usually means generating different\n * model output (e.g. cosine-normalized vs. raw inner-product), and the\n * stored index needs the same metric to score correctly.\n *\n * - `cosine`: cosine similarity. The default — works for most\n *   sentence-transformer / OpenAI-style embeddings that are already\n *   length-normalized.\n * - `l2`: Euclidean distance. Use when your model outputs vectors\n *   trained for L2 (e.g. some image embedding models).\n * - `inner_product`: dot-product distance. Use when your model outputs\n *   pre-normalized vectors AND you want to skip the cosine\n *   normalization step at query time.\n */\nexport type EmbeddingMetric = \"cosine\" | \"l2\" | \"inner_product\";\n\n/**\n * Vector index implementation. `hnsw` is the default and what pgvector\n * recommends for most workloads. `ivfflat` is the alternative for\n * larger datasets where memory is the bottleneck. `none` disables\n * automatic index creation for this embedding (the operator can still\n * call `backend.createVectorIndex` manually).\n */\nexport type EmbeddingIndexType = \"hnsw\" | \"ivfflat\" | \"none\";\n\n/**\n * Per-embedding configuration for the auto-derived\n * `VectorIndexDeclaration`. All fields are optional with sensible\n * defaults (`cosine` / `hnsw` / pgvector defaults: `m=16`,\n * `ef_construction=64`). Override only when your model or dataset has\n * a known reason to.\n */\nexport type EmbeddingIndexOptions = Readonly<{\n  /** Distance metric. Default `\"cosine\"`. */\n  metric?: EmbeddingMetric;\n  /** Vector index implementation. Default `\"hnsw\"`. */\n  indexType?: EmbeddingIndexType;\n  /** HNSW `m` parameter — max connections per layer. Default `16`. */\n  m?: number;\n  /** HNSW `ef_construction` parameter — build-time search depth. Default `64`. */\n  efConstruction?: number;\n  /** IVFFlat `lists` parameter — number of inverted-list partitions. */\n  lists?: number;\n}>;\n\n/**\n * Resolved embedding index configuration with all defaults applied.\n * What gets attached to the brand and read by the auto-derivation pass.\n */\nexport type ResolvedEmbeddingIndex = Readonly<{\n  metric: EmbeddingMetric;\n  indexType: EmbeddingIndexType;\n  m: number;\n  efConstruction: number;\n  lists: number | undefined;\n}>;\n\nconst DEFAULT_HNSW_M = 16;\nconst DEFAULT_HNSW_EF_CONSTRUCTION = 64;\n\n// ============================================================\n// Embedding Schema Type\n// ============================================================\n\n/**\n * A Zod schema for vector embeddings with attached dimension metadata.\n * Uses the branded EmbeddingValue type for type-level distinction.\n */\nexport type EmbeddingSchema<D extends number = number> =\n  z.ZodType<EmbeddingValue> &\n    Readonly<{\n      [EMBEDDING_DIMENSIONS_KEY]: D;\n      [EMBEDDING_INDEX_KEY]: ResolvedEmbeddingIndex;\n    }>;\n\n// ============================================================\n// Embedding Factory\n// ============================================================\n\n/**\n * Creates a Zod schema for vector embeddings.\n *\n * The dimension is validated at runtime and attached as metadata\n * for the schema introspector and query compiler.\n *\n * @param dimensions - The number of dimensions (e.g., 384, 512, 768, 1536, 3072)\n * @returns A Zod schema that validates embedding arrays\n *\n * @example\n * ```typescript\n * // OpenAI ada-002 embeddings\n * const Document = defineNode(\"Document\", {\n *   schema: z.object({\n *     title: z.string(),\n *     embedding: embedding(1536),\n *   }),\n * });\n *\n * // Sentence transformers\n * const Sentence = defineNode(\"Sentence\", {\n *   schema: z.object({\n *     text: z.string(),\n *     embedding: embedding(384), // all-MiniLM-L6-v2\n *   }),\n * });\n *\n * // Optional embeddings are supported\n * const Article = defineNode(\"Article\", {\n *   schema: z.object({\n *     content: z.string(),\n *     embedding: embedding(1536).optional(),\n *   }),\n * });\n * ```\n */\nexport function embedding<D extends number>(\n  dimensions: D,\n  options: EmbeddingIndexOptions = {},\n): EmbeddingSchema<D> {\n  if (!Number.isInteger(dimensions) || dimensions <= 0) {\n    throw new Error(\n      `Embedding dimensions must be a positive integer, got: ${dimensions}`,\n    );\n  }\n\n  const indexConfig = resolveEmbeddingIndex(options);\n\n  // Use EmbeddingValue as the output type for proper type branding.\n  // At runtime, validation accepts any number array with correct dimensions.\n  const schema = z.custom<EmbeddingValue>(\n    (value): value is EmbeddingValue => {\n      if (!Array.isArray(value)) {\n        return false;\n      }\n      if (value.length !== dimensions) {\n        return false;\n      }\n      return value.every((n) => typeof n === \"number\" && Number.isFinite(n));\n    },\n    {\n      message: `Expected an array of ${dimensions} finite numbers`,\n    },\n  );\n\n  return Object.assign(schema, {\n    [EMBEDDING_DIMENSIONS_KEY]: dimensions,\n    [EMBEDDING_INDEX_KEY]: indexConfig,\n  });\n}\n\nfunction resolveEmbeddingIndex(\n  options: EmbeddingIndexOptions,\n): ResolvedEmbeddingIndex {\n  const indexType = options.indexType ?? \"hnsw\";\n  const m = options.m ?? DEFAULT_HNSW_M;\n  const efConstruction = options.efConstruction ?? DEFAULT_HNSW_EF_CONSTRUCTION;\n\n  if (m <= 0 || !Number.isInteger(m)) {\n    throw new Error(\n      `embedding() index option 'm' must be a positive integer, got: ${m}`,\n    );\n  }\n  if (efConstruction <= 0 || !Number.isInteger(efConstruction)) {\n    throw new Error(\n      `embedding() index option 'efConstruction' must be a positive integer, got: ${efConstruction}`,\n    );\n  }\n  if (\n    options.lists !== undefined &&\n    (options.lists <= 0 || !Number.isInteger(options.lists))\n  ) {\n    throw new Error(\n      `embedding() index option 'lists' must be a positive integer, got: ${options.lists}`,\n    );\n  }\n\n  return Object.freeze({\n    metric: options.metric ?? \"cosine\",\n    indexType,\n    m,\n    efConstruction,\n    lists: options.lists,\n  });\n}\n\n// ============================================================\n// Type Guards\n// ============================================================\n\n/**\n * Checks if a value is an embedding schema.\n */\nexport function isEmbeddingSchema(value: unknown): value is EmbeddingSchema {\n  return (\n    typeof value === \"object\" &&\n    value !== null &&\n    EMBEDDING_DIMENSIONS_KEY in value &&\n    typeof (value as Record<string, unknown>)[EMBEDDING_DIMENSIONS_KEY] ===\n      \"number\"\n  );\n}\n\n/**\n * Gets the dimensions from an embedding schema.\n * Returns undefined if the schema is not an embedding schema.\n */\nexport function getEmbeddingDimensions(schema: z.ZodType): number | undefined {\n  if (isEmbeddingSchema(schema)) {\n    return schema[EMBEDDING_DIMENSIONS_KEY];\n  }\n  return undefined;\n}\n\n/**\n * Gets the resolved index configuration from an embedding schema.\n * Returns undefined if the schema is not an embedding schema. Used by\n * the auto-derivation pass at `defineGraph()` time to build vector\n * `IndexDeclaration` entries.\n */\nexport function getEmbeddingIndex(\n  schema: z.ZodType,\n): ResolvedEmbeddingIndex | undefined {\n  if (!isEmbeddingSchema(schema)) return undefined;\n  const candidate = (schema as unknown as Record<string, unknown>)[\n    EMBEDDING_INDEX_KEY\n  ];\n  if (candidate === undefined) return undefined;\n  return candidate as ResolvedEmbeddingIndex;\n}\n"]}

package/dist/chunk-X3244EKP.cjs

@@ -0,0 +1,84 @@
+'use strict';
+
+var zod = require('zod');
+
+// src/core/embedding.ts
+var EMBEDDING_DIMENSIONS_KEY = "_embeddingDimensions";
+var EMBEDDING_INDEX_KEY = "_embeddingIndex";
+var DEFAULT_HNSW_M = 16;
+var DEFAULT_HNSW_EF_CONSTRUCTION = 64;
+function embedding(dimensions, options = {}) {
+  if (!Number.isInteger(dimensions) || dimensions <= 0) {
+    throw new Error(
+      `Embedding dimensions must be a positive integer, got: ${dimensions}`
+    );
+  }
+  const indexConfig = resolveEmbeddingIndex(options);
+  const schema = zod.z.custom(
+    (value) => {
+      if (!Array.isArray(value)) {
+        return false;
+      }
+      if (value.length !== dimensions) {
+        return false;
+      }
+      return value.every((n) => typeof n === "number" && Number.isFinite(n));
+    },
+    {
+      message: `Expected an array of ${dimensions} finite numbers`
+    }
+  );
+  return Object.assign(schema, {
+    [EMBEDDING_DIMENSIONS_KEY]: dimensions,
+    [EMBEDDING_INDEX_KEY]: indexConfig
+  });
+}
+function resolveEmbeddingIndex(options) {
+  const indexType = options.indexType ?? "hnsw";
+  const m = options.m ?? DEFAULT_HNSW_M;
+  const efConstruction = options.efConstruction ?? DEFAULT_HNSW_EF_CONSTRUCTION;
+  if (m <= 0 || !Number.isInteger(m)) {
+    throw new Error(
+      `embedding() index option 'm' must be a positive integer, got: ${m}`
+    );
+  }
+  if (efConstruction <= 0 || !Number.isInteger(efConstruction)) {
+    throw new Error(
+      `embedding() index option 'efConstruction' must be a positive integer, got: ${efConstruction}`
+    );
+  }
+  if (options.lists !== void 0 && (options.lists <= 0 || !Number.isInteger(options.lists))) {
+    throw new Error(
+      `embedding() index option 'lists' must be a positive integer, got: ${options.lists}`
+    );
+  }
+  return Object.freeze({
+    metric: options.metric ?? "cosine",
+    indexType,
+    m,
+    efConstruction,
+    lists: options.lists
+  });
+}
+function isEmbeddingSchema(value) {
+  return typeof value === "object" && value !== null && EMBEDDING_DIMENSIONS_KEY in value && typeof value[EMBEDDING_DIMENSIONS_KEY] === "number";
+}
+function getEmbeddingDimensions(schema) {
+  if (isEmbeddingSchema(schema)) {
+    return schema[EMBEDDING_DIMENSIONS_KEY];
+  }
+  return void 0;
+}
+function getEmbeddingIndex(schema) {
+  if (!isEmbeddingSchema(schema)) return void 0;
+  const candidate = schema[EMBEDDING_INDEX_KEY];
+  if (candidate === void 0) return void 0;
+  return candidate;
+}
+
+exports.embedding = embedding;
+exports.getEmbeddingDimensions = getEmbeddingDimensions;
+exports.getEmbeddingIndex = getEmbeddingIndex;
+exports.isEmbeddingSchema = isEmbeddingSchema;
+//# sourceMappingURL=chunk-X3244EKP.cjs.map
+//# sourceMappingURL=chunk-X3244EKP.cjs.map

package/dist/chunk-X3244EKP.cjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/core/embedding.ts"],"names":["z"],"mappings":";;;;;AAqCO,IAAM,wBAAA,GAA2B,sBAAA;AAOjC,IAAM,mBAAA,GAAsB,iBAAA;AA4DnC,IAAM,cAAA,GAAiB,EAAA;AACvB,IAAM,4BAAA,GAA+B,EAAA;AAyD9B,SAAS,SAAA,CACd,UAAA,EACA,OAAA,GAAiC,EAAC,EACd;AACpB,EAAA,IAAI,CAAC,MAAA,CAAO,SAAA,CAAU,UAAU,CAAA,IAAK,cAAc,CAAA,EAAG;AACpD,IAAA,MAAM,IAAI,KAAA;AAAA,MACR,yDAAyD,UAAU,CAAA;AAAA,KACrE;AAAA,EACF;AAEA,EAAA,MAAM,WAAA,GAAc,sBAAsB,OAAO,CAAA;AAIjD,EAAA,MAAM,SAASA,KAAA,CAAE,MAAA;AAAA,IACf,CAAC,KAAA,KAAmC;AAClC,MAAA,IAAI,CAAC,KAAA,CAAM,OAAA,CAAQ,KAAK,CAAA,EAAG;AACzB,QAAA,OAAO,KAAA;AAAA,MACT;AACA,MAAA,IAAI,KAAA,CAAM,WAAW,UAAA,EAAY;AAC/B,QAAA,OAAO,KAAA;AAAA,MACT;AACA,MAAA,OAAO,KAAA,CAAM,KAAA,CAAM,CAAC,CAAA,KAAM,OAAO,MAAM,QAAA,IAAY,MAAA,CAAO,QAAA,CAAS,CAAC,CAAC,CAAA;AAAA,IACvE,CAAA;AAAA,IACA;AAAA,MACE,OAAA,EAAS,wBAAwB,UAAU,CAAA,eAAA;AAAA;AAC7C,GACF;AAEA,EAAA,OAAO,MAAA,CAAO,OAAO,MAAA,EAAQ;AAAA,IAC3B,CAAC,wBAAwB,GAAG,UAAA;AAAA,IAC5B,CAAC,mBAAmB,GAAG;AAAA,GACxB,CAAA;AACH;AAEA,SAAS,sBACP,OAAA,EACwB;AACxB,EAAA,MAAM,SAAA,GAAY,QAAQ,SAAA,IAAa,MAAA;AACvC,EAAA,MAAM,CAAA,GAAI,QAAQ,CAAA,IAAK,cAAA;AACvB,EAAA,MAAM,cAAA,GAAiB,QAAQ,cAAA,IAAkB,4BAAA;AAEjD,EAAA,IAAI,KAAK,CAAA,IAAK,CAAC,MAAA,CAAO,SAAA,CAAU,CAAC,CAAA,EAAG;AAClC,IAAA,MAAM,IAAI,KAAA;AAAA,MACR,iEAAiE,CAAC,CAAA;AAAA,KACpE;AAAA,EACF;AACA,EAAA,IAAI,kBAAkB,CAAA,IAAK,CAAC,MAAA,CAAO,SAAA,CAAU,cAAc,CAAA,EAAG;AAC5D,IAAA,MAAM,IAAI,KAAA;AAAA,MACR,8EAA8E,cAAc,CAAA;AAAA,KAC9F;AAAA,EACF;AACA,EAAA,IACE,OAAA,CAAQ,KAAA,KAAU,MAAA,KACjB,OAAA,CAAQ,KAAA,IAAS,CAAA,IAAK,CAAC,MAAA,CAAO,SAAA,CAAU,OAAA,CAAQ,KAAK,CAAA,CAAA,EACtD;AACA,IAAA,MAAM,IAAI,KAAA;AAAA,MACR,CAAA,kEAAA,EAAqE,QAAQ,KAAK,CAAA;AAAA,KACpF;AAAA,EACF;AAEA,EAAA,OAAO,OAAO,MAAA,CAAO;AAAA,IACnB,MAAA,EAAQ,QAAQ,MAAA,IAAU,QAAA;AAAA,IAC1B,SAAA;AAAA,IACA,CAAA;AAAA,IACA,cAAA;AAAA,IACA,OAAO,OAAA,CAAQ;AAAA,GAChB,CAAA;AACH;AASO,SAAS,kBAAkB,KAAA,EAA0C;AAC1E,EAAA,OACE,OAAO,KAAA,KAAU,QAAA,IACjB,KAAA,KAAU,IAAA,IACV,4BAA4B,KAAA,IAC5B,OAAQ,KAAA,CAAkC,wBAAwB,CAAA,KAChE,QAAA;AAEN;AAMO,SAAS,uBAAuB,MAAA,EAAuC;AAC5E,EAAA,IAAI,iBAAA,CAAkB,MAAM,CAAA,EAAG;AAC7B,IAAA,OAAO,OAAO,wBAAwB,CAAA;AAAA,EACxC;AACA,EAAA,OAAO,MAAA;AACT;AAQO,SAAS,kBACd,MAAA,EACoC;AACpC,EAAA,IAAI,CAAC,iBAAA,CAAkB,MAAM,CAAA,EAAG,OAAO,MAAA;AACvC,EAAA,MAAM,SAAA,GAAa,OACjB,mBACF,CAAA;AACA,EAAA,IAAI,SAAA,KAAc,QAAW,OAAO,MAAA;AACpC,EAAA,OAAO,SAAA;AACT","file":"chunk-X3244EKP.cjs","sourcesContent":["/**\n * Embedding type for vector search.\n *\n * Creates a Zod-compatible schema for vector embeddings with\n * dimension validation and metadata for query compilation.\n */\nimport { z } from \"zod\";\n\n// ============================================================\n// Embedding Brand Symbol\n// ============================================================\n\n/**\n * Symbol used to brand embedding values at the type level.\n * This allows TypeScript to distinguish embedding arrays from regular\n * number arrays in the query builder type system.\n */\ndeclare const EMBEDDING_BRAND: unique symbol;\n\n/**\n * Branded embedding type for type-level distinction.\n * At runtime this is just `readonly number[]`, but TypeScript can\n * distinguish it from regular arrays for query builder typing.\n */\nexport type EmbeddingValue = readonly number[] & {\n  readonly [EMBEDDING_BRAND]: true;\n};\n\n// ============================================================\n// Embedding Metadata Symbol\n// ============================================================\n\n/**\n * Symbol key for storing embedding dimensions on the schema.\n * This allows the schema introspector to detect embedding types\n * and extract dimension information.\n */\nexport const EMBEDDING_DIMENSIONS_KEY = \"_embeddingDimensions\" as const;\n\n/**\n * Symbol key for storing the embedding's preferred index configuration\n * on the schema. The schema introspector reads this when auto-deriving\n * `VectorIndexDeclaration` entries at `defineGraph()` time.\n */\nexport const EMBEDDING_INDEX_KEY = \"_embeddingIndex\" as const;\n\n/**\n * Distance metric used for vector similarity. Pinned at the embedding\n * brand because changing the metric usually means generating different\n * model output (e.g. cosine-normalized vs. raw inner-product), and the\n * stored index needs the same metric to score correctly.\n *\n * - `cosine`: cosine similarity. The default — works for most\n *   sentence-transformer / OpenAI-style embeddings that are already\n *   length-normalized.\n * - `l2`: Euclidean distance. Use when your model outputs vectors\n *   trained for L2 (e.g. some image embedding models).\n * - `inner_product`: dot-product distance. Use when your model outputs\n *   pre-normalized vectors AND you want to skip the cosine\n *   normalization step at query time.\n */\nexport type EmbeddingMetric = \"cosine\" | \"l2\" | \"inner_product\";\n\n/**\n * Vector index implementation. `hnsw` is the default and what pgvector\n * recommends for most workloads. `ivfflat` is the alternative for\n * larger datasets where memory is the bottleneck. `none` disables\n * automatic index creation for this embedding (the operator can still\n * call `backend.createVectorIndex` manually).\n */\nexport type EmbeddingIndexType = \"hnsw\" | \"ivfflat\" | \"none\";\n\n/**\n * Per-embedding configuration for the auto-derived\n * `VectorIndexDeclaration`. All fields are optional with sensible\n * defaults (`cosine` / `hnsw` / pgvector defaults: `m=16`,\n * `ef_construction=64`). Override only when your model or dataset has\n * a known reason to.\n */\nexport type EmbeddingIndexOptions = Readonly<{\n  /** Distance metric. Default `\"cosine\"`. */\n  metric?: EmbeddingMetric;\n  /** Vector index implementation. Default `\"hnsw\"`. */\n  indexType?: EmbeddingIndexType;\n  /** HNSW `m` parameter — max connections per layer. Default `16`. */\n  m?: number;\n  /** HNSW `ef_construction` parameter — build-time search depth. Default `64`. */\n  efConstruction?: number;\n  /** IVFFlat `lists` parameter — number of inverted-list partitions. */\n  lists?: number;\n}>;\n\n/**\n * Resolved embedding index configuration with all defaults applied.\n * What gets attached to the brand and read by the auto-derivation pass.\n */\nexport type ResolvedEmbeddingIndex = Readonly<{\n  metric: EmbeddingMetric;\n  indexType: EmbeddingIndexType;\n  m: number;\n  efConstruction: number;\n  lists: number | undefined;\n}>;\n\nconst DEFAULT_HNSW_M = 16;\nconst DEFAULT_HNSW_EF_CONSTRUCTION = 64;\n\n// ============================================================\n// Embedding Schema Type\n// ============================================================\n\n/**\n * A Zod schema for vector embeddings with attached dimension metadata.\n * Uses the branded EmbeddingValue type for type-level distinction.\n */\nexport type EmbeddingSchema<D extends number = number> =\n  z.ZodType<EmbeddingValue> &\n    Readonly<{\n      [EMBEDDING_DIMENSIONS_KEY]: D;\n      [EMBEDDING_INDEX_KEY]: ResolvedEmbeddingIndex;\n    }>;\n\n// ============================================================\n// Embedding Factory\n// ============================================================\n\n/**\n * Creates a Zod schema for vector embeddings.\n *\n * The dimension is validated at runtime and attached as metadata\n * for the schema introspector and query compiler.\n *\n * @param dimensions - The number of dimensions (e.g., 384, 512, 768, 1536, 3072)\n * @returns A Zod schema that validates embedding arrays\n *\n * @example\n * ```typescript\n * // OpenAI ada-002 embeddings\n * const Document = defineNode(\"Document\", {\n *   schema: z.object({\n *     title: z.string(),\n *     embedding: embedding(1536),\n *   }),\n * });\n *\n * // Sentence transformers\n * const Sentence = defineNode(\"Sentence\", {\n *   schema: z.object({\n *     text: z.string(),\n *     embedding: embedding(384), // all-MiniLM-L6-v2\n *   }),\n * });\n *\n * // Optional embeddings are supported\n * const Article = defineNode(\"Article\", {\n *   schema: z.object({\n *     content: z.string(),\n *     embedding: embedding(1536).optional(),\n *   }),\n * });\n * ```\n */\nexport function embedding<D extends number>(\n  dimensions: D,\n  options: EmbeddingIndexOptions = {},\n): EmbeddingSchema<D> {\n  if (!Number.isInteger(dimensions) || dimensions <= 0) {\n    throw new Error(\n      `Embedding dimensions must be a positive integer, got: ${dimensions}`,\n    );\n  }\n\n  const indexConfig = resolveEmbeddingIndex(options);\n\n  // Use EmbeddingValue as the output type for proper type branding.\n  // At runtime, validation accepts any number array with correct dimensions.\n  const schema = z.custom<EmbeddingValue>(\n    (value): value is EmbeddingValue => {\n      if (!Array.isArray(value)) {\n        return false;\n      }\n      if (value.length !== dimensions) {\n        return false;\n      }\n      return value.every((n) => typeof n === \"number\" && Number.isFinite(n));\n    },\n    {\n      message: `Expected an array of ${dimensions} finite numbers`,\n    },\n  );\n\n  return Object.assign(schema, {\n    [EMBEDDING_DIMENSIONS_KEY]: dimensions,\n    [EMBEDDING_INDEX_KEY]: indexConfig,\n  });\n}\n\nfunction resolveEmbeddingIndex(\n  options: EmbeddingIndexOptions,\n): ResolvedEmbeddingIndex {\n  const indexType = options.indexType ?? \"hnsw\";\n  const m = options.m ?? DEFAULT_HNSW_M;\n  const efConstruction = options.efConstruction ?? DEFAULT_HNSW_EF_CONSTRUCTION;\n\n  if (m <= 0 || !Number.isInteger(m)) {\n    throw new Error(\n      `embedding() index option 'm' must be a positive integer, got: ${m}`,\n    );\n  }\n  if (efConstruction <= 0 || !Number.isInteger(efConstruction)) {\n    throw new Error(\n      `embedding() index option 'efConstruction' must be a positive integer, got: ${efConstruction}`,\n    );\n  }\n  if (\n    options.lists !== undefined &&\n    (options.lists <= 0 || !Number.isInteger(options.lists))\n  ) {\n    throw new Error(\n      `embedding() index option 'lists' must be a positive integer, got: ${options.lists}`,\n    );\n  }\n\n  return Object.freeze({\n    metric: options.metric ?? \"cosine\",\n    indexType,\n    m,\n    efConstruction,\n    lists: options.lists,\n  });\n}\n\n// ============================================================\n// Type Guards\n// ============================================================\n\n/**\n * Checks if a value is an embedding schema.\n */\nexport function isEmbeddingSchema(value: unknown): value is EmbeddingSchema {\n  return (\n    typeof value === \"object\" &&\n    value !== null &&\n    EMBEDDING_DIMENSIONS_KEY in value &&\n    typeof (value as Record<string, unknown>)[EMBEDDING_DIMENSIONS_KEY] ===\n      \"number\"\n  );\n}\n\n/**\n * Gets the dimensions from an embedding schema.\n * Returns undefined if the schema is not an embedding schema.\n */\nexport function getEmbeddingDimensions(schema: z.ZodType): number | undefined {\n  if (isEmbeddingSchema(schema)) {\n    return schema[EMBEDDING_DIMENSIONS_KEY];\n  }\n  return undefined;\n}\n\n/**\n * Gets the resolved index configuration from an embedding schema.\n * Returns undefined if the schema is not an embedding schema. Used by\n * the auto-derivation pass at `defineGraph()` time to build vector\n * `IndexDeclaration` entries.\n */\nexport function getEmbeddingIndex(\n  schema: z.ZodType,\n): ResolvedEmbeddingIndex | undefined {\n  if (!isEmbeddingSchema(schema)) return undefined;\n  const candidate = (schema as unknown as Record<string, unknown>)[\n    EMBEDDING_INDEX_KEY\n  ];\n  if (candidate === undefined) return undefined;\n  return candidate as ResolvedEmbeddingIndex;\n}\n"]}