@nicia-ai/typegraph 0.26.0 → 0.28.0

package/dist/backend/postgres/index.cjs

@@ -1,8 +1,8 @@
 'use strict';
 
 var chunkHTGDWIDE_cjs = require('../../chunk-HTGDWIDE.cjs');
-var chunkTVUARC7S_cjs = require('../../chunk-TVUARC7S.cjs');
-var chunkDN4RUDF3_cjs = require('../../chunk-DN4RUDF3.cjs');
+var chunk2E4VTD2E_cjs = require('../../chunk-2E4VTD2E.cjs');
+var chunkLYRUNG6E_cjs = require('../../chunk-LYRUNG6E.cjs');
 require('../../chunk-IUIWCFG3.cjs');
 var chunkPN37HVAV_cjs = require('../../chunk-PN37HVAV.cjs');
 require('../../chunk-CTHNVU6K.cjs');
@@ -35,7 +35,7 @@ var DEFAULT_POSTGRES_STATEMENT_CACHE_MAX = 256;
 var STATEMENT_NAME_CACHE = /* @__PURE__ */ new Map();
 var nextStatementId = 0;
 function statementNameFor(sqlText, cacheMax) {
-  return chunkTVUARC7S_cjs.getOrCreateLru(STATEMENT_NAME_CACHE, sqlText, cacheMax, () => {
+  return chunk2E4VTD2E_cjs.getOrCreateLru(STATEMENT_NAME_CACHE, sqlText, cacheMax, () => {
     const name = `tg_${nextStatementId}`;
     nextStatementId += 1;
     return name;
@@ -126,7 +126,7 @@ function createPostgresExecutionAdapter(db, options = {}) {
   const cacheMax = options.preparedStatementCacheMax ?? DEFAULT_POSTGRES_STATEMENT_CACHE_MAX;
   const pgClient = resolvePgClient(db, prepareStatements, cacheMax);
   function compile(query) {
-    return chunkTVUARC7S_cjs.compileQueryWithDialect(db, query, "PostgreSQL");
+    return chunk2E4VTD2E_cjs.compileQueryWithDialect(db, query, "PostgreSQL");
   }
   if (pgClient === void 0) {
     return {
@@ -182,10 +182,10 @@ var POSTGRES_CHECK_UNIQUE_BATCH_CHUNK_SIZE = Math.max(
   1,
   POSTGRES_MAX_BIND_PARAMETERS - CHECK_UNIQUE_BATCH_FIXED_PARAM_COUNT
 );
-var toNodeRow = chunkTVUARC7S_cjs.createNodeRowMapper(chunkTVUARC7S_cjs.POSTGRES_ROW_MAPPER_CONFIG);
-var toEdgeRow = chunkTVUARC7S_cjs.createEdgeRowMapper(chunkTVUARC7S_cjs.POSTGRES_ROW_MAPPER_CONFIG);
-var toUniqueRow = chunkTVUARC7S_cjs.createUniqueRowMapper(chunkTVUARC7S_cjs.POSTGRES_ROW_MAPPER_CONFIG);
-var toSchemaVersionRow = chunkTVUARC7S_cjs.createSchemaVersionRowMapper(chunkTVUARC7S_cjs.POSTGRES_ROW_MAPPER_CONFIG);
+var toNodeRow = chunk2E4VTD2E_cjs.createNodeRowMapper(chunk2E4VTD2E_cjs.POSTGRES_ROW_MAPPER_CONFIG);
+var toEdgeRow = chunk2E4VTD2E_cjs.createEdgeRowMapper(chunk2E4VTD2E_cjs.POSTGRES_ROW_MAPPER_CONFIG);
+var toUniqueRow = chunk2E4VTD2E_cjs.createUniqueRowMapper(chunk2E4VTD2E_cjs.POSTGRES_ROW_MAPPER_CONFIG);
+var toSchemaVersionRow = chunk2E4VTD2E_cjs.createSchemaVersionRowMapper(chunk2E4VTD2E_cjs.POSTGRES_ROW_MAPPER_CONFIG);
 function toEmbeddingRow(row) {
   let embedding;
   if (typeof row.embedding === "string") {
@@ -203,13 +203,13 @@ function toEmbeddingRow(row) {
     field_path: row.field_path,
     embedding,
     dimensions: row.dimensions,
-    created_at: chunkTVUARC7S_cjs.formatPostgresTimestamp(row.created_at) ?? "",
-    updated_at: chunkTVUARC7S_cjs.formatPostgresTimestamp(row.updated_at) ?? ""
+    created_at: chunk2E4VTD2E_cjs.formatPostgresTimestamp(row.created_at) ?? "",
+    updated_at: chunk2E4VTD2E_cjs.formatPostgresTimestamp(row.updated_at) ?? ""
   };
 }
 function buildPostgresCapabilities(strategy) {
   return {
-    ...chunkDN4RUDF3_cjs.POSTGRES_CAPABILITIES,
+    ...chunkLYRUNG6E_cjs.POSTGRES_CAPABILITIES,
     vector: {
       supported: true,
       metrics: ["cosine", "l2", "inner_product"],
@@ -221,7 +221,7 @@ function buildPostgresCapabilities(strategy) {
   };
 }
 function createPostgresBackend(db, options = {}) {
-  const tables2 = options.tables ?? chunkTVUARC7S_cjs.tables;
+  const tables2 = options.tables ?? chunk2E4VTD2E_cjs.tables;
   const fulltextStrategy = options.fulltext ?? chunkPN37HVAV_cjs.tsvectorStrategy;
   const baseCapabilities = buildPostgresCapabilities(fulltextStrategy);
   const httpOnlyOverrides = isNeonHttpClient(db) ? { transactions: false } : {};
@@ -252,13 +252,14 @@ function createPostgresBackend(db, options = {}) {
     ].map((name) => chunkHTGDWIDE_cjs.quoteIdentifier(name)),
     drizzleOrm.sql`, `
   )}`;
-  const operationStrategy = chunkTVUARC7S_cjs.createPostgresOperationStrategy(
+  const operationStrategy = chunk2E4VTD2E_cjs.createPostgresOperationStrategy(
     tables2,
     fulltextStrategy
   );
   const operations = createPostgresOperationBackend({
     db,
     executionAdapter,
+    adapterOptions,
     operationStrategy,
     tableNames,
     capabilities,
@@ -292,7 +293,7 @@ function createPostgresBackend(db, options = {}) {
   }
   const matTable = tables2.contributionMaterializations;
   async function ensureContributionMaterializationsTableImpl() {
-    await db.execute(drizzleOrm.sql.raw(chunkTVUARC7S_cjs.generatePgCreateTableSQL(matTable)));
+    await db.execute(drizzleOrm.sql.raw(chunk2E4VTD2E_cjs.generatePgCreateTableSQL(matTable)));
   }
   async function getContributionMaterializationRow(identity) {
     const rows = await db.select().from(matTable).where(
@@ -305,16 +306,16 @@ function createPostgresBackend(db, options = {}) {
     );
     const row = rows[0];
     if (row === void 0) return void 0;
-    return chunkTVUARC7S_cjs.mapContributionMaterializationRow(
+    return chunk2E4VTD2E_cjs.mapContributionMaterializationRow(
       row,
-      chunkTVUARC7S_cjs.POSTGRES_CONTRIBUTION_MAT_TIMESTAMPS.decode
+      chunk2E4VTD2E_cjs.POSTGRES_CONTRIBUTION_MAT_TIMESTAMPS.decode
     );
   }
   async function recordContributionMaterializationRow(params) {
     await db.insert(matTable).values(
-      chunkTVUARC7S_cjs.buildContributionInsertValues(
+      chunk2E4VTD2E_cjs.buildContributionInsertValues(
         params,
-        chunkTVUARC7S_cjs.POSTGRES_CONTRIBUTION_MAT_TIMESTAMPS.encode
+        chunk2E4VTD2E_cjs.POSTGRES_CONTRIBUTION_MAT_TIMESTAMPS.encode
       )
     ).onConflictDoUpdate({
       target: [
@@ -323,13 +324,13 @@ function createPostgresBackend(db, options = {}) {
         matTable.owner,
         matTable.tableName
       ],
-      set: chunkTVUARC7S_cjs.buildContributionOnConflictSet(
+      set: chunk2E4VTD2E_cjs.buildContributionOnConflictSet(
         matTable.materializedAt,
         params.materializedAt
       )
     });
   }
-  const contributionMaterializer = chunkTVUARC7S_cjs.createContributionMaterializer({
+  const contributionMaterializer = chunk2E4VTD2E_cjs.createContributionMaterializer({
     dialect: "postgres",
     fulltextStrategy,
     fulltextTableName: tables2.fulltextTableName,
@@ -349,12 +350,12 @@ function createPostgresBackend(db, options = {}) {
       capabilities,
       fulltextStrategy
     });
-    return chunkTVUARC7S_cjs.gateFulltext(txBackend, contributionMaterializer.assertInitialized);
+    return chunk2E4VTD2E_cjs.gateFulltext(txBackend, contributionMaterializer.assertInitialized);
   }
   const backend = {
     ...operations,
     async bootstrapTables() {
-      const statements = chunkTVUARC7S_cjs.generatePostgresDDL(tables2, fulltextStrategy);
+      const statements = chunk2E4VTD2E_cjs.generatePostgresDDL(tables2, fulltextStrategy);
       for (const statement of statements) {
         await db.execute(drizzleOrm.sql.raw(statement));
       }
@@ -364,7 +365,7 @@ function createPostgresBackend(db, options = {}) {
     // uninitialized database throws `StoreNotInitializedError` loudly
     // rather than self-healing on a read/write path (#135). Shared
     // verbatim with the tx-scoped gate via `gateFulltextMethods`.
-    ...chunkTVUARC7S_cjs.gateFulltextMethods(
+    ...chunk2E4VTD2E_cjs.gateFulltextMethods(
       operations,
       contributionMaterializer.assertInitialized
     ),
@@ -373,7 +374,7 @@ function createPostgresBackend(db, options = {}) {
     },
     async ensureIndexMaterializationsTable() {
       await db.execute(
-        drizzleOrm.sql.raw(chunkTVUARC7S_cjs.generatePgCreateTableSQL(tables2.indexMaterializations))
+        drizzleOrm.sql.raw(chunk2E4VTD2E_cjs.generatePgCreateTableSQL(tables2.indexMaterializations))
       );
     },
     async getIndexMaterialization(indexName) {
@@ -381,26 +382,26 @@ function createPostgresBackend(db, options = {}) {
       const rows = await db.select().from(t).where(drizzleOrm.eq(t.indexName, indexName));
       const row = rows[0];
       if (row === void 0) return void 0;
-      return chunkTVUARC7S_cjs.mapMaterializationRow(row, chunkTVUARC7S_cjs.POSTGRES_INDEX_MAT_TIMESTAMPS.decode);
+      return chunk2E4VTD2E_cjs.mapMaterializationRow(row, chunk2E4VTD2E_cjs.POSTGRES_INDEX_MAT_TIMESTAMPS.decode);
     },
     async getIndexMaterializations(statusKeys) {
       if (statusKeys.length === 0) return [];
       const t = tables2.indexMaterializations;
       const rows = await db.select().from(t).where(drizzleOrm.inArray(t.indexName, [...statusKeys]));
       return rows.map(
-        (row) => chunkTVUARC7S_cjs.mapMaterializationRow(row, chunkTVUARC7S_cjs.POSTGRES_INDEX_MAT_TIMESTAMPS.decode)
+        (row) => chunk2E4VTD2E_cjs.mapMaterializationRow(row, chunk2E4VTD2E_cjs.POSTGRES_INDEX_MAT_TIMESTAMPS.decode)
       );
     },
     async recordIndexMaterialization(params) {
       const t = tables2.indexMaterializations;
       await db.insert(t).values(
-        chunkTVUARC7S_cjs.buildMaterializationInsertValues(
+        chunk2E4VTD2E_cjs.buildMaterializationInsertValues(
           params,
-          chunkTVUARC7S_cjs.POSTGRES_INDEX_MAT_TIMESTAMPS.encode
+          chunk2E4VTD2E_cjs.POSTGRES_INDEX_MAT_TIMESTAMPS.encode
         )
       ).onConflictDoUpdate({
         target: t.indexName,
-        set: chunkTVUARC7S_cjs.buildMaterializationOnConflictSet(
+        set: chunk2E4VTD2E_cjs.buildMaterializationOnConflictSet(
           t.materializedAt,
           params.materializedAt
         )
@@ -420,38 +421,38 @@ function createPostgresBackend(db, options = {}) {
     },
     async ensureKindRemovalsTable() {
       await db.execute(
-        drizzleOrm.sql.raw(chunkTVUARC7S_cjs.generatePgCreateTableSQL(tables2.kindRemovals))
+        drizzleOrm.sql.raw(chunk2E4VTD2E_cjs.generatePgCreateTableSQL(tables2.kindRemovals))
       );
     },
     async getPendingKindRemovals(graphId) {
       const t = tables2.kindRemovals;
       const rows = await db.select().from(t).where(drizzleOrm.and(drizzleOrm.eq(t.graphId, graphId), drizzleOrm.isNull(t.removedAt)));
       return rows.map(
-        (row) => chunkTVUARC7S_cjs.mapKindRemovalRow(row, chunkTVUARC7S_cjs.POSTGRES_KIND_REMOVAL_TIMESTAMPS.decode)
+        (row) => chunk2E4VTD2E_cjs.mapKindRemovalRow(row, chunk2E4VTD2E_cjs.POSTGRES_KIND_REMOVAL_TIMESTAMPS.decode)
       );
     },
     async getAllKindRemovals(graphId) {
       const t = tables2.kindRemovals;
       const rows = await db.select().from(t).where(drizzleOrm.eq(t.graphId, graphId));
       return rows.map(
-        (row) => chunkTVUARC7S_cjs.mapKindRemovalRow(row, chunkTVUARC7S_cjs.POSTGRES_KIND_REMOVAL_TIMESTAMPS.decode)
+        (row) => chunk2E4VTD2E_cjs.mapKindRemovalRow(row, chunk2E4VTD2E_cjs.POSTGRES_KIND_REMOVAL_TIMESTAMPS.decode)
       );
     },
     async recordKindRemoval(params) {
       const t = tables2.kindRemovals;
       await db.insert(t).values(
-        chunkTVUARC7S_cjs.buildKindRemovalInsertValues(
+        chunk2E4VTD2E_cjs.buildKindRemovalInsertValues(
           params,
-          chunkTVUARC7S_cjs.POSTGRES_KIND_REMOVAL_TIMESTAMPS.encode
+          chunk2E4VTD2E_cjs.POSTGRES_KIND_REMOVAL_TIMESTAMPS.encode
         )
       ).onConflictDoUpdate({
         target: [t.graphId, t.kindName, t.entity, t.schemaVersion],
-        set: chunkTVUARC7S_cjs.buildKindRemovalOnConflictSet(t.removedAt, params.removedAt)
+        set: chunk2E4VTD2E_cjs.buildKindRemovalOnConflictSet(t.removedAt, params.removedAt)
       });
     },
     async ensureReconciliationMarkersTable() {
       await db.execute(
-        drizzleOrm.sql.raw(chunkTVUARC7S_cjs.generatePgCreateTableSQL(tables2.reconciliationMarkers))
+        drizzleOrm.sql.raw(chunk2E4VTD2E_cjs.generatePgCreateTableSQL(tables2.reconciliationMarkers))
       );
     },
     async ensureRuntimeContributions(graphId) {
@@ -512,7 +513,7 @@ function createPostgresBackend(db, options = {}) {
           }
         );
       }
-      chunkTVUARC7S_cjs.assertAdoptedDialect(externalTx, pgCore.PgDatabase, "postgres");
+      chunk2E4VTD2E_cjs.assertAdoptedDialect(externalTx, pgCore.PgDatabase, "postgres");
       return bindTransactionBackend(externalTx);
     },
     async close() {
@@ -524,6 +525,7 @@ function createPostgresOperationBackend(options) {
   const {
     db,
     executionAdapter,
+    adapterOptions,
     operationStrategy,
     tableNames,
     capabilities,
@@ -539,7 +541,46 @@ function createPostgresOperationBackend(options) {
   async function execRun(query) {
     await executionAdapter.execute(query);
   }
-  const commonBackend = chunkTVUARC7S_cjs.createCommonOperationBackend({
+  let efSearchUnsupportedWarned = false;
+  function warnEfSearchUnsupported() {
+    if (efSearchUnsupportedWarned) return;
+    efSearchUnsupportedWarned = true;
+    if (typeof console === "undefined" || typeof console.warn !== "function") {
+      return;
+    }
+    console.warn(
+      "[typegraph] efSearch (hnsw.ef_search override) was ignored: this Postgres backend has transactions disabled (e.g. drizzle-orm/neon-http), and SET LOCAL needs a transaction to scope the override. Use a transactional driver (node-postgres / neon-serverless / postgres-js) to apply efSearch."
+    );
+  }
+  async function runVectorSearch(efSearch, query) {
+    if (efSearch === void 0) {
+      return execAll(query);
+    }
+    if (!capabilities.transactions) {
+      warnEfSearchUnsupported();
+      return execAll(query);
+    }
+    const setEfSearch = drizzleOrm.sql.raw(`SET LOCAL hnsw.ef_search = ${efSearch}`);
+    if (db instanceof pgCore.PgTransaction) {
+      const [setting] = await execAll(
+        drizzleOrm.sql`SELECT current_setting('hnsw.ef_search') AS ef_search`
+      );
+      await execRun(setEfSearch);
+      const rows = await execAll(query);
+      if (setting !== void 0) {
+        await execRun(
+          drizzleOrm.sql`SELECT set_config('hnsw.ef_search', ${setting.ef_search}, true)`
+        );
+      }
+      return rows;
+    }
+    return db.transaction(async (tx) => {
+      const txAdapter = createPostgresExecutionAdapter(tx, adapterOptions);
+      await txAdapter.execute(setEfSearch);
+      return txAdapter.execute(query);
+    });
+  }
+  const commonBackend = chunk2E4VTD2E_cjs.createCommonOperationBackend({
     batchConfig: {
       checkUniqueBatchChunkSize: POSTGRES_CHECK_UNIQUE_BATCH_CHUNK_SIZE,
       edgeInsertBatchSize: POSTGRES_EDGE_INSERT_BATCH_SIZE,
@@ -552,7 +593,7 @@ function createPostgresOperationBackend(options) {
       execGet,
       execRun
     },
-    nowIso: chunkTVUARC7S_cjs.nowIso,
+    nowIso: chunk2E4VTD2E_cjs.nowIso,
     operationStrategy,
     rowMappers: {
       toEdgeRow,
@@ -576,7 +617,7 @@ function createPostgresOperationBackend(options) {
     tableNames,
     // === Embedding Operations ===
     async upsertEmbedding(params) {
-      const timestamp = chunkTVUARC7S_cjs.nowIso();
+      const timestamp = chunk2E4VTD2E_cjs.nowIso();
       const query = operationStrategy.buildUpsertEmbedding(params, timestamp);
       await execRun(query);
     },
@@ -596,7 +637,7 @@ function createPostgresOperationBackend(options) {
     },
     async vectorSearch(params) {
       const query = operationStrategy.buildVectorSearch(params);
-      const rows = await execAll(query);
+      const rows = await runVectorSearch(params.efSearch, query);
       return rows.map((row) => ({
         nodeId: row.node_id,
         score: row.score
@@ -616,7 +657,7 @@ function createPostgresOperationBackend(options) {
         ...params.indexParams?.lists === void 0 ? {} : { ivfflatLists: params.indexParams.lists },
         ...params.concurrent === true ? { concurrent: true } : {}
       };
-      const result = await chunkTVUARC7S_cjs.createPostgresVectorIndex(db, indexOptions);
+      const result = await chunk2E4VTD2E_cjs.createPostgresVectorIndex(db, indexOptions);
       if (!result.success) {
         throw new Error(
           result.message ?? "Failed to create PostgreSQL vector index"
@@ -625,7 +666,7 @@ function createPostgresOperationBackend(options) {
     },
     // === Fulltext Operations ===
     async upsertFulltext(params) {
-      const timestamp = chunkTVUARC7S_cjs.nowIso();
+      const timestamp = chunk2E4VTD2E_cjs.nowIso();
       const statements = operationStrategy.buildUpsertFulltext(
         params,
         timestamp
@@ -642,7 +683,7 @@ function createPostgresOperationBackend(options) {
     },
     async upsertFulltextBatch(params) {
       if (params.rows.length === 0) return;
-      const timestamp = chunkTVUARC7S_cjs.nowIso();
+      const timestamp = chunk2E4VTD2E_cjs.nowIso();
       const statements = operationStrategy.buildUpsertFulltextBatch(
         params,
         timestamp
@@ -663,7 +704,7 @@ function createPostgresOperationBackend(options) {
       const rows = await execAll(query);
       return rows.map((row, index) => ({
         nodeId: row.node_id,
-        score: chunkTVUARC7S_cjs.coerceNumericScore(row.score),
+        score: chunk2E4VTD2E_cjs.coerceNumericScore(row.score),
         rank: index + 1,
         ...row.snippet === null ? {} : { snippet: row.snippet }
       }));
@@ -672,13 +713,13 @@ function createPostgresOperationBackend(options) {
       const metrics = capabilities.vector?.metrics ?? ["cosine"];
       await Promise.all(
         metrics.map(async (metric) => {
-          const indexName = chunkTVUARC7S_cjs.generateVectorIndexName(
+          const indexName = chunk2E4VTD2E_cjs.generateVectorIndexName(
             params.graphId,
             params.nodeKind,
             params.fieldPath,
             metric
           );
-          const result = await chunkTVUARC7S_cjs.dropPostgresVectorIndex(db, indexName);
+          const result = await chunk2E4VTD2E_cjs.dropPostgresVectorIndex(db, indexName);
           if (!result.success) {
             throw new Error(
               result.message ?? "Failed to drop PostgreSQL vector index"
@@ -702,6 +743,7 @@ function createTransactionBackend(options) {
   return createPostgresOperationBackend({
     db: options.db,
     executionAdapter: txExecutionAdapter,
+    adapterOptions: options.adapterOptions,
     operationStrategy: options.operationStrategy,
     tableNames: options.tableNames,
     capabilities: options.capabilities,
@@ -711,43 +753,43 @@ function createTransactionBackend(options) {
 
 Object.defineProperty(exports, "createPostgresTables", {
   enumerable: true,
-  get: function () { return chunkTVUARC7S_cjs.createPostgresTables; }
+  get: function () { return chunk2E4VTD2E_cjs.createPostgresTables; }
 });
 Object.defineProperty(exports, "edges", {
   enumerable: true,
-  get: function () { return chunkTVUARC7S_cjs.edges; }
+  get: function () { return chunk2E4VTD2E_cjs.edges; }
 });
 Object.defineProperty(exports, "embeddings", {
   enumerable: true,
-  get: function () { return chunkTVUARC7S_cjs.embeddings; }
+  get: function () { return chunk2E4VTD2E_cjs.embeddings; }
 });
 Object.defineProperty(exports, "fulltext", {
   enumerable: true,
-  get: function () { return chunkTVUARC7S_cjs.fulltext; }
+  get: function () { return chunk2E4VTD2E_cjs.fulltext; }
 });
 Object.defineProperty(exports, "generatePostgresDDL", {
   enumerable: true,
-  get: function () { return chunkTVUARC7S_cjs.generatePostgresDDL; }
+  get: function () { return chunk2E4VTD2E_cjs.generatePostgresDDL; }
 });
 Object.defineProperty(exports, "generatePostgresMigrationSQL", {
   enumerable: true,
-  get: function () { return chunkTVUARC7S_cjs.generatePostgresMigrationSQL; }
+  get: function () { return chunk2E4VTD2E_cjs.generatePostgresMigrationSQL; }
 });
 Object.defineProperty(exports, "nodes", {
   enumerable: true,
-  get: function () { return chunkTVUARC7S_cjs.nodes; }
+  get: function () { return chunk2E4VTD2E_cjs.nodes; }
 });
 Object.defineProperty(exports, "schemaVersions", {
   enumerable: true,
-  get: function () { return chunkTVUARC7S_cjs.schemaVersions; }
+  get: function () { return chunk2E4VTD2E_cjs.schemaVersions; }
 });
 Object.defineProperty(exports, "tables", {
   enumerable: true,
-  get: function () { return chunkTVUARC7S_cjs.tables; }
+  get: function () { return chunk2E4VTD2E_cjs.tables; }
 });
 Object.defineProperty(exports, "uniques", {
   enumerable: true,
-  get: function () { return chunkTVUARC7S_cjs.uniques; }
+  get: function () { return chunk2E4VTD2E_cjs.uniques; }
 });
 exports.createPostgresBackend = createPostgresBackend;
 //# sourceMappingURL=index.cjs.map