@nicia-ai/typegraph 0.36.0 → 0.37.1

package/dist/{chunk-VQEXF5D4.js → chunk-IXGYRP2P.js}

@@ -1,12 +1,11 @@
-import { tables, createVectorSlotLatch, createPostgresOperationStrategy, createCachedTableExistence, createContributionMaterializer, gateFulltextMethods, EMBEDDING_UPSERT_PARAM_COUNT, createCommonOperationBackend, vectorSlotFromParams, hybridCandidatesRef, mapVectorWriteError, mapHybridSearchRow, generatePgCreateTableSQL, mapContributionMaterializationRow, POSTGRES_CONTRIBUTION_MAT_TIMESTAMPS, buildContributionInsertValues, buildContributionOnConflictSet, assertAdoptedDialect, buildKindRemovalInsertValues, POSTGRES_KIND_REMOVAL_TIMESTAMPS, buildKindRemovalOnConflictSet, mapKindRemovalRow, buildMaterializationInsertValues, POSTGRES_INDEX_MAT_TIMESTAMPS, buildMaterializationOnConflictSet, mapMaterializationRow, generatePostgresDDL, vectorSlotFromDropIndexParams, vectorSlotFromCreateIndexParams, gateFulltext, compileQueryWithDialect, getOrCreateLru } from './chunk-4GX2JJXS.js';
-import { pgvectorStrategy, assertPgvectorEfSearch } from './chunk-Q5TEXCTB.js';
-import { createNodeRowMapper, POSTGRES_ROW_MAPPER_CONFIG, createEdgeRowMapper, createUniqueRowMapper, createSchemaVersionRowMapper, quoteIdentifier, nowIso, coerceNumericScore } from './chunk-RAXU3RMB.js';
-import { tsvectorStrategy, assertVectorSearchLimit, buildFulltextCapabilities, buildVectorCapabilities } from './chunk-FPIZFYHE.js';
-import { shouldForceCustomPlan, annIndexScanTypes } from './chunk-UT3S4SEN.js';
-import { POSTGRES_MAX_BIND_PARAMETERS, POSTGRES_CAPABILITIES, chunk } from './chunk-IU46OHVW.js';
-import { isMissingTableError, isInsufficientResourcesError } from './chunk-JFUPDJXD.js';
-import { ConfigurationError, CompilerInvariantError, TransactionClosedError } from './chunk-WI7ISNQF.js';
-import { getTableName, sql, and, eq, isNull, inArray } from 'drizzle-orm';
+import { tables, createPostgresOperationStrategy, createCachedTableExistence, createContributionMaterializer, gateFulltextMethods, lockPostgresTrustedImportTables, assertTrustedImportDatabaseEmpty, suspendPostgresSecondaryIndexes, createPostgresTrustedImportSession, restoreSecondaryIndexes, analyzeImportedTables, EMBEDDING_UPSERT_PARAM_COUNT, createCommonOperationBackend, vectorSlotFromParams, hybridCandidatesRef, mapVectorWriteError, mapHybridSearchRow, generatePgCreateTableSQL, mapContributionMaterializationRow, POSTGRES_CONTRIBUTION_MAT_TIMESTAMPS, buildContributionInsertValues, buildContributionOnConflictSet, assertAdoptedDialect, buildKindRemovalInsertValues, POSTGRES_KIND_REMOVAL_TIMESTAMPS, buildKindRemovalOnConflictSet, mapKindRemovalRow, buildMaterializationInsertValues, POSTGRES_INDEX_MAT_TIMESTAMPS, buildMaterializationOnConflictSet, mapMaterializationRow, generatePostgresDDL, vectorSlotFromDropIndexParams, vectorSlotFromCreateIndexParams, gateFulltext, compileQueryWithDialect, getOrCreateLru } from './chunk-K3Q3LJY3.js';
+import { pgvectorStrategy, assertPgvectorEfSearch } from './chunk-5KPNWT6X.js';
+import { createNodeRowMapper, POSTGRES_ROW_MAPPER_CONFIG, createEdgeRowMapper, createUniqueRowMapper, createSchemaVersionRowMapper, quoteIdentifier, nowIso, coerceNumericScore } from './chunk-M6VES7V6.js';
+import { tsvectorStrategy, assertVectorSearchLimit, buildFulltextCapabilities, buildVectorCapabilities } from './chunk-EXU3RASM.js';
+import { normalizeGraphAnalyticsCapabilities, createBackendOverlay, POSTGRES_MAX_BIND_PARAMETERS, INTERNAL_TEMPORARY_WRITES, POSTGRES_CAPABILITIES, shouldForceCustomPlan, chunk, annIndexScanTypes } from './chunk-44TA22QT.js';
+import { isMissingTableError, isInsufficientResourcesError } from './chunk-XGPUQ56G.js';
+import { ConfigurationError, CompilerInvariantError, TransactionClosedError } from './chunk-5PNMDJ7J.js';
+import { getTableName, sql, eq, and, isNull, inArray } from 'drizzle-orm';
 import { PgDatabase, PgTransaction } from 'drizzle-orm/pg-core';
 
 // src/backend/drizzle/execution/postgres-execution.ts
@@ -23,8 +22,14 @@ function isPgNativeClient(candidate) {
 function isPgliteClient(candidate) {
   return typeof candidate === "object" && hasFunctionProperty(candidate, "query") && hasFunctionProperty(candidate, "dumpDataDir");
 }
+function transactionSession(carrier) {
+  return carrier.session ?? carrier._?.session;
+}
+function hasPgliteSession(carrier) {
+  return transactionSession(carrier)?.constructor?.name === "PgliteSession";
+}
 function isPostgresJsClient(candidate) {
-  return typeof candidate === "function" && hasFunctionProperty(candidate, "unsafe") && hasFunctionProperty(candidate, "begin");
+  return typeof candidate === "function" && hasFunctionProperty(candidate, "unsafe") && (hasFunctionProperty(candidate, "begin") || hasFunctionProperty(candidate, "savepoint"));
 }
 function isNeonHttpClient(db) {
   const candidate = db.$client;
@@ -103,12 +108,14 @@ function adaptPostgresJsClient(sql2) {
     }
   };
 }
-function resolvePgClient(db, prepareStatements, cacheMax) {
+function resolvePgClient(db, prepareStatements, cacheMax, useTransactionClient) {
   if (isNeonHttpClient(db)) {
     return void 0;
   }
-  const client = db.$client;
-  if (isPgliteClient(client)) {
+  const carrier = db;
+  const client = carrier.$client ?? (useTransactionClient ? transactionSession(carrier)?.client : void 0);
+  if (isPgliteClient(client) || useTransactionClient && hasPgliteSession(carrier)) {
+    if (!isPgNativeClient(client)) return void 0;
     return wrapNodePgClientUnnamed(client);
   }
   if (isPgNativeClient(client)) {
@@ -137,7 +144,12 @@ function createPgPreparedStatement(pgClient, sqlText) {
 function createPostgresExecutionAdapter(db, options = {}) {
   const prepareStatements = options.prepareStatements ?? true;
   const cacheMax = options.preparedStatementCacheMax ?? DEFAULT_POSTGRES_STATEMENT_CACHE_MAX;
-  const pgClient = resolvePgClient(db, prepareStatements, cacheMax);
+  const pgClient = resolvePgClient(
+    db,
+    prepareStatements,
+    cacheMax,
+    options.useTransactionClient ?? false
+  );
   function compile(query) {
     return compileQueryWithDialect(db, query, "PostgreSQL");
   }
@@ -284,18 +296,23 @@ function createPostgresBackend(db, options = {}) {
   const tables2 = options.tables ?? tables;
   const fulltextStrategy = options.fulltext ?? tsvectorStrategy;
   const vectorStrategy = options.vector === false ? void 0 : options.vector ?? pgvectorStrategy;
-  const vectorSlotLatch = vectorStrategy === void 0 ? void 0 : createVectorSlotLatch();
   const iterativeScanProbe = createIterativeScanProbe();
   const baseCapabilities = buildPostgresCapabilities(
     fulltextStrategy,
     vectorStrategy
   );
-  const httpOnlyOverrides = isNeonHttpClient(db) ? { transactions: false } : {};
-  const capabilities = {
+  const httpOnlyOverrides = isNeonHttpClient(db) ? {
+    transactions: false,
+    graphAnalytics: {
+      ...baseCapabilities.graphAnalytics ?? { mathFunctions: true },
+      supported: false
+    }
+  } : {};
+  const capabilities = normalizeGraphAnalyticsCapabilities({
     ...baseCapabilities,
     ...httpOnlyOverrides,
     ...options.capabilities
-  };
+  });
   const adapterOptions = {
     ...options.prepareStatements === void 0 ? {} : { prepareStatements: options.prepareStatements },
     ...options.preparedStatementCacheMax === void 0 ? {} : { preparedStatementCacheMax: options.preparedStatementCacheMax }
@@ -326,18 +343,6 @@ function createPostgresBackend(db, options = {}) {
     tables2,
     fulltextStrategy
   );
-  const operations = createPostgresOperationBackend({
-    db,
-    executionAdapter,
-    adapterOptions,
-    operationStrategy,
-    tableNames,
-    capabilities,
-    fulltextStrategy,
-    vectorStrategy,
-    vectorSlotLatch,
-    iterativeScanProbe
-  });
   const recordedTableExists = createCachedTableExistence(
     async (tableName) => {
       const rows = await executionAdapter.execute(
@@ -347,36 +352,6 @@ function createPostgresBackend(db, options = {}) {
     },
     { cacheExisting: false }
   );
-  function runSchemaWriteTransaction(graphId, fn) {
-    if (!capabilities.transactions) {
-      throw new 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(sql`SELECT pg_advisory_xact_lock(hashtext(${graphId}))`);
-      const { backend: txBackend, drainAndClose } = createTransactionBackend({
-        db: tx,
-        adapterOptions,
-        operationStrategy,
-        tableNames,
-        capabilities,
-        fulltextStrategy,
-        vectorStrategy,
-        iterativeScanProbe
-      });
-      try {
-        return await fn(txBackend);
-      } finally {
-        await drainAndClose();
-      }
-    });
-  }
   const matTable = tables2.contributionMaterializations;
   async function ensureContributionMaterializationsTableImpl() {
     await db.execute(sql.raw(generatePgCreateTableSQL(matTable)));
@@ -397,6 +372,15 @@ function createPostgresBackend(db, options = {}) {
       POSTGRES_CONTRIBUTION_MAT_TIMESTAMPS.decode
     );
   }
+  async function getContributionMaterializationRows(graphId) {
+    const rows = await db.select().from(matTable).where(eq(matTable.graphId, graphId));
+    return rows.map(
+      (row) => mapContributionMaterializationRow(
+        row,
+        POSTGRES_CONTRIBUTION_MAT_TIMESTAMPS.decode
+      )
+    );
+  }
   async function recordContributionMaterializationRow(params) {
     await db.insert(matTable).values(
       buildContributionInsertValues(
@@ -416,17 +400,72 @@ function createPostgresBackend(db, options = {}) {
       )
     });
   }
+  async function deleteContributionMaterializationRow(identity) {
+    await db.delete(matTable).where(
+      and(
+        eq(matTable.graphId, identity.graphId),
+        eq(matTable.logicalName, identity.logicalName),
+        eq(matTable.owner, identity.owner),
+        eq(matTable.tableName, identity.tableName)
+      )
+    );
+  }
   const contributionMaterializer = createContributionMaterializer({
     dialect: "postgres",
     fulltextStrategy,
     fulltextTableName: tables2.fulltextTableName,
+    vectorStrategy,
     execDdl: async (statement) => {
       await db.execute(sql.raw(statement));
     },
     ensureMarkerTable: ensureContributionMaterializationsTableImpl,
-    getMarker: getContributionMaterializationRow,
-    recordMarker: recordContributionMaterializationRow
+    getMarkers: getContributionMaterializationRows,
+    recordMarker: recordContributionMaterializationRow,
+    deleteMarker: deleteContributionMaterializationRow
   });
+  const operations = createPostgresOperationBackend({
+    db,
+    executionAdapter,
+    adapterOptions,
+    operationStrategy,
+    tableNames,
+    capabilities,
+    fulltextStrategy,
+    vectorStrategy,
+    contributionMaterializer,
+    iterativeScanProbe
+  });
+  function runSchemaWriteTransaction(graphId, fn) {
+    if (!capabilities.transactions) {
+      throw new 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(sql`SELECT pg_advisory_xact_lock(hashtext(${graphId}))`);
+      const { backend: txBackend, drainAndClose } = createTransactionBackend({
+        db: tx,
+        adapterOptions,
+        operationStrategy,
+        tableNames,
+        capabilities,
+        fulltextStrategy,
+        vectorStrategy,
+        contributionMaterializer,
+        iterativeScanProbe
+      });
+      try {
+        return await fn(txBackend);
+      } finally {
+        await drainAndClose();
+      }
+    });
+  }
   function bindTransactionBackend(tx) {
     const { backend: backend2, drainAndClose } = createTransactionBackend({
       db: tx,
@@ -436,6 +475,7 @@ function createPostgresBackend(db, options = {}) {
       capabilities,
       fulltextStrategy,
       vectorStrategy,
+      contributionMaterializer,
       iterativeScanProbe
     });
     return {
@@ -448,6 +488,40 @@ function createPostgresBackend(db, options = {}) {
   }
   const backend = {
     ...operations,
+    ...capabilities.transactions && executionAdapter.executeCompiled !== void 0 ? {
+      async trustedImport(fn) {
+        return backend.transaction(async (tx, rawSql) => {
+          const trustedExecutionAdapter = createPostgresExecutionAdapter(
+            rawSql,
+            { ...adapterOptions, useTransactionClient: true }
+          );
+          const executeCompiled = trustedExecutionAdapter.executeCompiled;
+          if (executeCompiled === void 0) {
+            throw new ConfigurationError(
+              "Trusted import could not bind raw execution to the PostgreSQL transaction.",
+              { capability: "trustedImport", dialect: "postgres" }
+            );
+          }
+          const trustedTx = createBackendOverlay(tx, {
+            executeRaw(sqlText, params) {
+              return executeCompiled({ params, sql: sqlText });
+            }
+          });
+          await lockPostgresTrustedImportTables(trustedTx, tableNames);
+          await assertTrustedImportDatabaseEmpty(trustedTx, tableNames);
+          const indexDefinitions = await suspendPostgresSecondaryIndexes(
+            trustedTx,
+            tableNames
+          );
+          const result = await fn(
+            createPostgresTrustedImportSession(trustedTx, tableNames)
+          );
+          await restoreSecondaryIndexes(trustedTx, indexDefinitions);
+          await analyzeImportedTables(trustedTx, tableNames);
+          return result;
+        });
+      }
+    } : {},
     async bootstrapTables() {
       const statements = generatePostgresDDL(tables2, fulltextStrategy);
       for (const statement of statements) {
@@ -544,10 +618,7 @@ function createPostgresBackend(db, options = {}) {
         )
       ).onConflictDoUpdate({
         target: t.indexName,
-        set: buildMaterializationOnConflictSet(
-          t.materializedAt,
-          params.materializedAt
-        )
+        set: buildMaterializationOnConflictSet(params.materializedAt)
       });
     },
     async ensureContributionMaterializationsTable() {
@@ -607,6 +678,27 @@ function createPostgresBackend(db, options = {}) {
     async ensureFulltextTable(graphId) {
       await contributionMaterializer.ensureRuntimeContributions(graphId);
     },
+    // Vector counterparts of the runtime-contribution methods. Present
+    // only when a vector strategy is wired (omitted under `vector: false`,
+    // mirroring the embedding/search methods), so a no-vector backend
+    // doesn't advertise vector materialization it can't perform.
+    ...vectorStrategy === void 0 ? {} : {
+      async ensureVectorSlotContribution(slot, options_) {
+        await contributionMaterializer.ensureVectorSlot(slot, options_);
+      },
+      async ensureVectorSlotContributions(slots, options_) {
+        await contributionMaterializer.ensureVectorSlots(slots, options_);
+      },
+      async assertVectorSlotInitialized(slot) {
+        await contributionMaterializer.assertVectorSlot(slot);
+      },
+      async assertVectorSlotsInitialized(slots) {
+        await contributionMaterializer.assertVectorSlots(slots);
+      },
+      async deleteVectorSlotContribution(slot) {
+        await contributionMaterializer.dropVectorSlot(slot);
+      }
+    },
     async getReconciliationMarker(graphId) {
       const t = tables2.reconciliationMarkers;
       const rows = await db.select().from(t).where(eq(t.graphId, graphId));
@@ -649,8 +741,20 @@ function createPostgresBackend(db, options = {}) {
       );
     },
     async transaction(fn, options2) {
-      const txConfig = options2?.isolationLevel ? {
-        isolationLevel: options2.isolationLevel.replace("_", " ")
+      const temporaryWrites = options2?.temporaryWrites === INTERNAL_TEMPORARY_WRITES;
+      if (temporaryWrites && options2.accessMode !== "read_only") {
+        throw new ConfigurationError(
+          "Temporary-write transactions must be semantically read-only.",
+          { dialect: "postgres" }
+        );
+      }
+      const txConfig = options2?.isolationLevel !== void 0 || options2?.accessMode !== void 0 || temporaryWrites ? {
+        ...options2.isolationLevel === void 0 ? {} : {
+          isolationLevel: options2.isolationLevel.replace("_", " ")
+        },
+        ...options2.accessMode === void 0 && !temporaryWrites ? {} : {
+          accessMode: temporaryWrites ? "read write" : options2.accessMode.replace("_", " ")
+        }
       } : void 0;
       return db.transaction(async (tx) => {
         const { backend: txBackend, drainAndClose } = bindTransactionBackend(tx);
@@ -723,7 +827,7 @@ function createPostgresOperationBackend(options) {
     capabilities,
     fulltextStrategy,
     vectorStrategy,
-    vectorSlotLatch,
+    contributionMaterializer,
     iterativeScanProbe
   } = options;
   async function execAll(query) {
@@ -819,12 +923,6 @@ function createPostgresOperationBackend(options) {
       return txAdapter.execute(query);
     });
   }
-  async function ensureVectorSlotStorage(slot) {
-    if (vectorStrategy === void 0 || vectorSlotLatch === void 0) return;
-    await vectorSlotLatch.ensure(vectorStrategy, slot, async (statement) => {
-      await execRun(sql.raw(statement));
-    });
-  }
   const batchConfig = {
     checkUniqueBatchChunkSize: POSTGRES_CHECK_UNIQUE_BATCH_CHUNK_SIZE,
     edgeInsertBatchSize: POSTGRES_EDGE_INSERT_BATCH_SIZE,
@@ -868,7 +966,7 @@ function createPostgresOperationBackend(options) {
   const vectorEmbeddingMethods = vectorStrategy === void 0 ? {} : {
     async upsertEmbedding(params) {
       const slot = vectorSlotFromParams(params);
-      await ensureVectorSlotStorage(slot);
+      await contributionMaterializer.assertVectorSlot(slot);
       const statements = vectorStrategy.buildUpsert(slot, params, nowIso());
       try {
         for (const statement of statements) {
@@ -881,7 +979,7 @@ function createPostgresOperationBackend(options) {
     async upsertEmbeddingBatch(params) {
       if (params.rows.length === 0) return;
       const slot = vectorSlotFromParams(params);
-      await ensureVectorSlotStorage(slot);
+      await contributionMaterializer.assertVectorSlot(slot);
       const rowsById = new Map(
         params.rows.map((row) => [row.nodeId, row])
       );
@@ -922,7 +1020,7 @@ function createPostgresOperationBackend(options) {
     },
     async deleteEmbedding(params) {
       const slot = vectorSlotFromParams(params);
-      await ensureVectorSlotStorage(slot);
+      await contributionMaterializer.assertVectorSlot(slot);
       const statements = vectorStrategy.buildDelete(slot, params);
       for (const statement of statements) {
         await execRun(statement);
@@ -932,7 +1030,6 @@ function createPostgresOperationBackend(options) {
       assertVectorSearchLimit(params.limit);
       assertPgvectorEfSearch(params.efSearch);
       const slot = vectorSlotFromParams(params);
-      await ensureVectorSlotStorage(slot);
       const query = vectorStrategy.buildSearch(
         slot,
         params,
@@ -971,7 +1068,6 @@ function createPostgresOperationBackend(options) {
           metric: params.vector.metric,
           indexType: params.vector.indexType
         });
-        await ensureVectorSlotStorage(slot);
         const candidates = params.candidates ?? operationStrategy.buildLiveNodeIds(
           params.graphId,
           params.nodeKind
@@ -1027,7 +1123,7 @@ function createPostgresOperationBackend(options) {
     async createVectorIndex(params) {
       if (vectorStrategy === void 0) return;
       const slot = vectorSlotFromCreateIndexParams(params);
-      await ensureVectorSlotStorage(slot);
+      await contributionMaterializer.ensureVectorSlot(slot);
       const indexStatement = vectorStrategy.buildCreateIndex?.(slot, {
         concurrent: params.concurrent === true
       });
@@ -1047,9 +1143,7 @@ function createPostgresOperationBackend(options) {
               slot.fieldPath
             )
           );
-          await execRun(
-            sql`ALTER TABLE ${table} SET (parallel_workers = 0)`
-          );
+          await execRun(sql`ALTER TABLE ${table} SET (parallel_workers = 0)`);
           try {
             await execRun(indexStatement);
           } finally {
@@ -1134,9 +1228,7 @@ function createPostgresOperationBackend(options) {
         const overrides = [];
         for (const indexType of annTypes) {
           if (indexType !== "hnsw" && indexType !== "ivfflat") continue;
-          overrides.push(
-            ...await vectorSearchGucOverrides({ indexType })
-          );
+          overrides.push(...await vectorSearchGucOverrides({ indexType }));
         }
         return runVectorSearch(overrides, query);
       }
@@ -1161,14 +1253,14 @@ function createTransactionBackend(options) {
     capabilities: options.capabilities,
     fulltextStrategy: options.fulltextStrategy,
     vectorStrategy: options.vectorStrategy,
-    vectorSlotLatch: options.vectorStrategy === void 0 ? void 0 : createVectorSlotLatch(),
-    // The probe is process-wide truth, so — unlike the slot latch — the outer
-    // instance's is reused rather than a fresh one per transaction.
+    contributionMaterializer: options.contributionMaterializer,
+    // The probe is process-wide truth, so the outer instance's is reused
+    // rather than a fresh one per transaction.
     iterativeScanProbe: options.iterativeScanProbe
   });
   return { backend, drainAndClose: txExecutionAdapter.drainAndClose };
 }
 
 export { createPostgresBackend };
-//# sourceMappingURL=chunk-VQEXF5D4.js.map
-//# sourceMappingURL=chunk-VQEXF5D4.js.map
+//# sourceMappingURL=chunk-IXGYRP2P.js.map
+//# sourceMappingURL=chunk-IXGYRP2P.js.map