@nicia-ai/typegraph 0.24.1 → 0.25.0

package/dist/backend/postgres/index.cjs

@@ -1,12 +1,14 @@
 'use strict';
 
 var chunkHTGDWIDE_cjs = require('../../chunk-HTGDWIDE.cjs');
-var chunk5HVDOPZU_cjs = require('../../chunk-5HVDOPZU.cjs');
-var chunkSUHW2UYL_cjs = require('../../chunk-SUHW2UYL.cjs');
-require('../../chunk-O5PST6TB.cjs');
-require('../../chunk-PEE4EWXZ.cjs');
-var chunk3JK66E3I_cjs = require('../../chunk-3JK66E3I.cjs');
-require('../../chunk-UVZHTBZ5.cjs');
+var chunkO2P3KAUT_cjs = require('../../chunk-O2P3KAUT.cjs');
+var chunk7CDXX3QZ_cjs = require('../../chunk-7CDXX3QZ.cjs');
+var chunkJTI5WT4P_cjs = require('../../chunk-JTI5WT4P.cjs');
+require('../../chunk-CTHNVU6K.cjs');
+require('../../chunk-EK6RJ373.cjs');
+require('../../chunk-X3244EKP.cjs');
+var chunkW5MTHMTE_cjs = require('../../chunk-W5MTHMTE.cjs');
+require('../../chunk-7V335GHG.cjs');
 var drizzleOrm = require('drizzle-orm');
 
 // src/backend/drizzle/execution/postgres-execution.ts
@@ -31,7 +33,7 @@ var DEFAULT_POSTGRES_STATEMENT_CACHE_MAX = 256;
 var STATEMENT_NAME_CACHE = /* @__PURE__ */ new Map();
 var nextStatementId = 0;
 function statementNameFor(sqlText, cacheMax) {
-  return chunk5HVDOPZU_cjs.getOrCreateLru(STATEMENT_NAME_CACHE, sqlText, cacheMax, () => {
+  return chunkO2P3KAUT_cjs.getOrCreateLru(STATEMENT_NAME_CACHE, sqlText, cacheMax, () => {
     const name = `tg_${nextStatementId}`;
     nextStatementId += 1;
     return name;
@@ -122,7 +124,7 @@ function createPostgresExecutionAdapter(db, options = {}) {
   const cacheMax = options.preparedStatementCacheMax ?? DEFAULT_POSTGRES_STATEMENT_CACHE_MAX;
   const pgClient = resolvePgClient(db, prepareStatements, cacheMax);
   function compile(query) {
-    return chunk5HVDOPZU_cjs.compileQueryWithDialect(db, query, "PostgreSQL");
+    return chunkO2P3KAUT_cjs.compileQueryWithDialect(db, query, "PostgreSQL");
   }
   if (pgClient === void 0) {
     return {
@@ -178,10 +180,10 @@ var POSTGRES_CHECK_UNIQUE_BATCH_CHUNK_SIZE = Math.max(
   1,
   POSTGRES_MAX_BIND_PARAMETERS - CHECK_UNIQUE_BATCH_FIXED_PARAM_COUNT
 );
-var toNodeRow = chunk5HVDOPZU_cjs.createNodeRowMapper(chunk5HVDOPZU_cjs.POSTGRES_ROW_MAPPER_CONFIG);
-var toEdgeRow = chunk5HVDOPZU_cjs.createEdgeRowMapper(chunk5HVDOPZU_cjs.POSTGRES_ROW_MAPPER_CONFIG);
-var toUniqueRow = chunk5HVDOPZU_cjs.createUniqueRowMapper(chunk5HVDOPZU_cjs.POSTGRES_ROW_MAPPER_CONFIG);
-var toSchemaVersionRow = chunk5HVDOPZU_cjs.createSchemaVersionRowMapper(chunk5HVDOPZU_cjs.POSTGRES_ROW_MAPPER_CONFIG);
+var toNodeRow = chunkO2P3KAUT_cjs.createNodeRowMapper(chunkO2P3KAUT_cjs.POSTGRES_ROW_MAPPER_CONFIG);
+var toEdgeRow = chunkO2P3KAUT_cjs.createEdgeRowMapper(chunkO2P3KAUT_cjs.POSTGRES_ROW_MAPPER_CONFIG);
+var toUniqueRow = chunkO2P3KAUT_cjs.createUniqueRowMapper(chunkO2P3KAUT_cjs.POSTGRES_ROW_MAPPER_CONFIG);
+var toSchemaVersionRow = chunkO2P3KAUT_cjs.createSchemaVersionRowMapper(chunkO2P3KAUT_cjs.POSTGRES_ROW_MAPPER_CONFIG);
 function toEmbeddingRow(row) {
   let embedding;
   if (typeof row.embedding === "string") {
@@ -199,8 +201,8 @@ function toEmbeddingRow(row) {
     field_path: row.field_path,
     embedding,
     dimensions: row.dimensions,
-    created_at: chunk5HVDOPZU_cjs.formatPostgresTimestamp(row.created_at) ?? "",
-    updated_at: chunk5HVDOPZU_cjs.formatPostgresTimestamp(row.updated_at) ?? ""
+    created_at: chunkO2P3KAUT_cjs.formatPostgresTimestamp(row.created_at) ?? "",
+    updated_at: chunkO2P3KAUT_cjs.formatPostgresTimestamp(row.updated_at) ?? ""
   };
 }
 function coerceNumericScore(value) {
@@ -215,7 +217,7 @@ function coerceNumericScore(value) {
 }
 function buildPostgresCapabilities(strategy) {
   return {
-    ...chunkSUHW2UYL_cjs.POSTGRES_CAPABILITIES,
+    ...chunk7CDXX3QZ_cjs.POSTGRES_CAPABILITIES,
     vector: {
       supported: true,
       metrics: ["cosine", "l2", "inner_product"],
@@ -223,12 +225,12 @@ function buildPostgresCapabilities(strategy) {
       maxDimensions: 16e3
       // pgvector limit
     },
-    fulltext: chunk3JK66E3I_cjs.buildFulltextCapabilities(strategy)
+    fulltext: chunkJTI5WT4P_cjs.buildFulltextCapabilities(strategy)
   };
 }
 function createPostgresBackend(db, options = {}) {
-  const tables2 = options.tables ?? chunk5HVDOPZU_cjs.tables;
-  const fulltextStrategy = options.fulltext ?? chunk3JK66E3I_cjs.tsvectorStrategy;
+  const tables2 = options.tables ?? chunkO2P3KAUT_cjs.tables;
+  const fulltextStrategy = options.fulltext ?? chunkJTI5WT4P_cjs.tsvectorStrategy;
   const baseCapabilities = buildPostgresCapabilities(fulltextStrategy);
   const httpOnlyOverrides = isNeonHttpClient(db) ? { transactions: false } : {};
   const capabilities = {
@@ -245,7 +247,8 @@ function createPostgresBackend(db, options = {}) {
     nodes: drizzleOrm.getTableName(tables2.nodes),
     edges: drizzleOrm.getTableName(tables2.edges),
     embeddings: drizzleOrm.getTableName(tables2.embeddings),
-    fulltext: tables2.fulltextTableName
+    fulltext: tables2.fulltextTableName,
+    uniques: drizzleOrm.getTableName(tables2.uniques)
   };
   const analyzeStatement = drizzleOrm.sql`ANALYZE ${drizzleOrm.sql.join(
     [
@@ -257,7 +260,7 @@ function createPostgresBackend(db, options = {}) {
     ].map((name) => chunkHTGDWIDE_cjs.quoteIdentifier(name)),
     drizzleOrm.sql`, `
   )}`;
-  const operationStrategy = chunk5HVDOPZU_cjs.createPostgresOperationStrategy(
+  const operationStrategy = chunkO2P3KAUT_cjs.createPostgresOperationStrategy(
     tables2,
     fulltextStrategy
   );
@@ -269,22 +272,139 @@ function createPostgresBackend(db, options = {}) {
     capabilities,
     fulltextStrategy
   });
+  function runSchemaWriteTransaction(graphId, fn) {
+    if (!capabilities.transactions) {
+      throw new chunkW5MTHMTE_cjs.ConfigurationError(
+        "commitSchemaVersion and setActiveVersion require atomic transactions, but this Postgres backend does not provide them. The drizzle-orm/neon-http driver communicates over HTTP and cannot hold a session across statements; use drizzle-orm/neon-serverless (websocket) for transactional writes.",
+        {
+          backend: "postgres",
+          capability: "transactions",
+          supportsTransactions: false
+        }
+      );
+    }
+    return db.transaction(async (tx) => {
+      await tx.execute(
+        drizzleOrm.sql`SELECT pg_advisory_xact_lock(hashtext(${graphId}))`
+      );
+      const txBackend = createTransactionBackend({
+        db: tx,
+        adapterOptions,
+        operationStrategy,
+        tableNames,
+        capabilities,
+        fulltextStrategy
+      });
+      return fn(txBackend);
+    });
+  }
   const backend = {
     ...operations,
     async bootstrapTables() {
-      const statements = chunk5HVDOPZU_cjs.generatePostgresDDL(tables2, fulltextStrategy);
+      const statements = chunkO2P3KAUT_cjs.generatePostgresDDL(tables2, fulltextStrategy);
       for (const statement of statements) {
         await db.execute(drizzleOrm.sql.raw(statement));
       }
     },
+    async executeDdl(ddl) {
+      await db.execute(drizzleOrm.sql.raw(ddl));
+    },
+    async ensureIndexMaterializationsTable() {
+      await db.execute(
+        drizzleOrm.sql.raw(chunkO2P3KAUT_cjs.generatePgCreateTableSQL(tables2.indexMaterializations))
+      );
+    },
+    async getIndexMaterialization(indexName) {
+      const t = tables2.indexMaterializations;
+      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 chunkO2P3KAUT_cjs.mapMaterializationRow(row, chunkO2P3KAUT_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) => chunkO2P3KAUT_cjs.mapMaterializationRow(row, chunkO2P3KAUT_cjs.POSTGRES_INDEX_MAT_TIMESTAMPS.decode)
+      );
+    },
+    async recordIndexMaterialization(params) {
+      const t = tables2.indexMaterializations;
+      await db.insert(t).values(
+        chunkO2P3KAUT_cjs.buildMaterializationInsertValues(
+          params,
+          chunkO2P3KAUT_cjs.POSTGRES_INDEX_MAT_TIMESTAMPS.encode
+        )
+      ).onConflictDoUpdate({
+        target: t.indexName,
+        set: chunkO2P3KAUT_cjs.buildMaterializationOnConflictSet(
+          t.materializedAt,
+          params.materializedAt
+        )
+      });
+    },
+    async ensureKindRemovalsTable() {
+      await db.execute(
+        drizzleOrm.sql.raw(chunkO2P3KAUT_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) => chunkO2P3KAUT_cjs.mapKindRemovalRow(row, chunkO2P3KAUT_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) => chunkO2P3KAUT_cjs.mapKindRemovalRow(row, chunkO2P3KAUT_cjs.POSTGRES_KIND_REMOVAL_TIMESTAMPS.decode)
+      );
+    },
+    async recordKindRemoval(params) {
+      const t = tables2.kindRemovals;
+      await db.insert(t).values(
+        chunkO2P3KAUT_cjs.buildKindRemovalInsertValues(
+          params,
+          chunkO2P3KAUT_cjs.POSTGRES_KIND_REMOVAL_TIMESTAMPS.encode
+        )
+      ).onConflictDoUpdate({
+        target: [t.graphId, t.kindName, t.entity, t.schemaVersion],
+        set: chunkO2P3KAUT_cjs.buildKindRemovalOnConflictSet(t.removedAt, params.removedAt)
+      });
+    },
+    async ensureReconciliationMarkersTable() {
+      await db.execute(
+        drizzleOrm.sql.raw(chunkO2P3KAUT_cjs.generatePgCreateTableSQL(tables2.reconciliationMarkers))
+      );
+    },
+    async getReconciliationMarker(graphId) {
+      const t = tables2.reconciliationMarkers;
+      const rows = await db.select().from(t).where(drizzleOrm.eq(t.graphId, graphId));
+      return rows[0]?.reconciledToVersion;
+    },
+    async setReconciliationMarker(graphId, version) {
+      const t = tables2.reconciliationMarkers;
+      await db.insert(t).values({ graphId, reconciledToVersion: version }).onConflictDoUpdate({
+        target: t.graphId,
+        set: { reconciledToVersion: version }
+      });
+    },
     async refreshStatistics() {
       await db.execute(analyzeStatement);
     },
-    async setActiveSchema(graphId, version) {
-      await chunkSUHW2UYL_cjs.runOptionallyInTransaction(
-        backend,
-        (target) => target.setActiveSchema(graphId, version),
-        operations
+    async commitSchemaVersion(params) {
+      return runSchemaWriteTransaction(
+        params.graphId,
+        (target) => target.commitSchemaVersion(params)
+      );
+    },
+    async setActiveVersion(params) {
+      await runSchemaWriteTransaction(
+        params.graphId,
+        (target) => target.setActiveVersion(params)
       );
     },
     async transaction(fn, options2) {
@@ -327,7 +447,7 @@ function createPostgresOperationBackend(options) {
   async function execRun(query) {
     await executionAdapter.execute(query);
   }
-  const commonBackend = chunk5HVDOPZU_cjs.createCommonOperationBackend({
+  const commonBackend = chunkO2P3KAUT_cjs.createCommonOperationBackend({
     batchConfig: {
       checkUniqueBatchChunkSize: POSTGRES_CHECK_UNIQUE_BATCH_CHUNK_SIZE,
       edgeInsertBatchSize: POSTGRES_EDGE_INSERT_BATCH_SIZE,
@@ -340,7 +460,7 @@ function createPostgresOperationBackend(options) {
       execGet,
       execRun
     },
-    nowIso: chunk5HVDOPZU_cjs.nowIso,
+    nowIso: chunkO2P3KAUT_cjs.nowIso,
     operationStrategy,
     rowMappers: {
       toEdgeRow,
@@ -364,7 +484,7 @@ function createPostgresOperationBackend(options) {
     tableNames,
     // === Embedding Operations ===
     async upsertEmbedding(params) {
-      const timestamp = chunk5HVDOPZU_cjs.nowIso();
+      const timestamp = chunkO2P3KAUT_cjs.nowIso();
       const query = operationStrategy.buildUpsertEmbedding(params, timestamp);
       await execRun(query);
     },
@@ -401,9 +521,10 @@ function createPostgresOperationBackend(options) {
         metric: params.metric,
         ...params.indexParams?.m === void 0 ? {} : { hnswM: params.indexParams.m },
         ...params.indexParams?.efConstruction === void 0 ? {} : { hnswEfConstruction: params.indexParams.efConstruction },
-        ...params.indexParams?.lists === void 0 ? {} : { ivfflatLists: params.indexParams.lists }
+        ...params.indexParams?.lists === void 0 ? {} : { ivfflatLists: params.indexParams.lists },
+        ...params.concurrent === true ? { concurrent: true } : {}
       };
-      const result = await chunk5HVDOPZU_cjs.createPostgresVectorIndex(db, indexOptions);
+      const result = await chunkO2P3KAUT_cjs.createPostgresVectorIndex(db, indexOptions);
       if (!result.success) {
         throw new Error(
           result.message ?? "Failed to create PostgreSQL vector index"
@@ -412,7 +533,7 @@ function createPostgresOperationBackend(options) {
     },
     // === Fulltext Operations ===
     async upsertFulltext(params) {
-      const timestamp = chunk5HVDOPZU_cjs.nowIso();
+      const timestamp = chunkO2P3KAUT_cjs.nowIso();
       const statements = operationStrategy.buildUpsertFulltext(
         params,
         timestamp
@@ -429,7 +550,7 @@ function createPostgresOperationBackend(options) {
     },
     async upsertFulltextBatch(params) {
       if (params.rows.length === 0) return;
-      const timestamp = chunk5HVDOPZU_cjs.nowIso();
+      const timestamp = chunkO2P3KAUT_cjs.nowIso();
       const statements = operationStrategy.buildUpsertFulltextBatch(
         params,
         timestamp
@@ -457,20 +578,22 @@ function createPostgresOperationBackend(options) {
     },
     async dropVectorIndex(params) {
       const metrics = capabilities.vector?.metrics ?? ["cosine"];
-      for (const metric of metrics) {
-        const indexName = chunk5HVDOPZU_cjs.generateVectorIndexName(
-          params.graphId,
-          params.nodeKind,
-          params.fieldPath,
-          metric
-        );
-        const result = await chunk5HVDOPZU_cjs.dropPostgresVectorIndex(db, indexName);
-        if (!result.success) {
-          throw new Error(
-            result.message ?? "Failed to drop PostgreSQL vector index"
+      await Promise.all(
+        metrics.map(async (metric) => {
+          const indexName = chunkO2P3KAUT_cjs.generateVectorIndexName(
+            params.graphId,
+            params.nodeKind,
+            params.fieldPath,
+            metric
           );
-        }
-      }
+          const result = await chunkO2P3KAUT_cjs.dropPostgresVectorIndex(db, indexName);
+          if (!result.success) {
+            throw new Error(
+              result.message ?? "Failed to drop PostgreSQL vector index"
+            );
+          }
+        })
+      );
     },
     // === Query Execution ===
     async execute(query) {
@@ -496,39 +619,39 @@ function createTransactionBackend(options) {
 
 Object.defineProperty(exports, "createPostgresTables", {
   enumerable: true,
-  get: function () { return chunk5HVDOPZU_cjs.createPostgresTables; }
+  get: function () { return chunkO2P3KAUT_cjs.createPostgresTables; }
 });
 Object.defineProperty(exports, "edges", {
   enumerable: true,
-  get: function () { return chunk5HVDOPZU_cjs.edges; }
+  get: function () { return chunkO2P3KAUT_cjs.edges; }
 });
 Object.defineProperty(exports, "embeddings", {
   enumerable: true,
-  get: function () { return chunk5HVDOPZU_cjs.embeddings; }
+  get: function () { return chunkO2P3KAUT_cjs.embeddings; }
 });
 Object.defineProperty(exports, "generatePostgresDDL", {
   enumerable: true,
-  get: function () { return chunk5HVDOPZU_cjs.generatePostgresDDL; }
+  get: function () { return chunkO2P3KAUT_cjs.generatePostgresDDL; }
 });
 Object.defineProperty(exports, "generatePostgresMigrationSQL", {
   enumerable: true,
-  get: function () { return chunk5HVDOPZU_cjs.generatePostgresMigrationSQL; }
+  get: function () { return chunkO2P3KAUT_cjs.generatePostgresMigrationSQL; }
 });
 Object.defineProperty(exports, "nodes", {
   enumerable: true,
-  get: function () { return chunk5HVDOPZU_cjs.nodes; }
+  get: function () { return chunkO2P3KAUT_cjs.nodes; }
 });
 Object.defineProperty(exports, "schemaVersions", {
   enumerable: true,
-  get: function () { return chunk5HVDOPZU_cjs.schemaVersions; }
+  get: function () { return chunkO2P3KAUT_cjs.schemaVersions; }
 });
 Object.defineProperty(exports, "tables", {
   enumerable: true,
-  get: function () { return chunk5HVDOPZU_cjs.tables; }
+  get: function () { return chunkO2P3KAUT_cjs.tables; }
 });
 Object.defineProperty(exports, "uniques", {
   enumerable: true,
-  get: function () { return chunk5HVDOPZU_cjs.uniques; }
+  get: function () { return chunkO2P3KAUT_cjs.uniques; }
 });
 exports.createPostgresBackend = createPostgresBackend;
 //# sourceMappingURL=index.cjs.map