@nicia-ai/typegraph 0.44.0 → 0.45.0

package/dist/{chunk-SQRD2CYF.js → chunk-DXEEHSGN.js}

@@ -1,11 +1,12 @@
-import { formatPostgresTimestamp, coerceNumericScore, nowIso } from './chunk-PHXITZE4.js';
-import { runtimeStrategyContributions } from './chunk-7G32LIQM.js';
-import { toDrizzleSql } from './chunk-GWEDD3DH.js';
+import { formatPostgresTimestamp, coerceNumericScore, nowIso } from './chunk-YMTXMPV6.js';
+import { runtimeStrategyContributions } from './chunk-BGYJW37L.js';
+import { toDrizzleSql } from './chunk-EFEOOCJ3.js';
 import { chunk } from './chunk-35S6P5ZU.js';
-import { createBackendOverlay } from './chunk-FMMIX6N4.js';
+import { createBackendOverlay } from './chunk-55ITBQJI.js';
 import { sortedReplacer } from './chunk-UR7W5YQK.js';
-import { parseDimensionMismatch, isMissingTableError } from './chunk-B75XRV3O.js';
+import { parseDimensionMismatch, isMissingTableError } from './chunk-V34A6R5V.js';
 import { sha256Hex } from './chunk-K34OPSJK.js';
+import { getDialect } from './chunk-PVUI6KGR.js';
 import { sql as sql$1, asCompiledStatementSql, isSqlFragment } from './chunk-6DCB664D.js';
 import { isPresent } from './chunk-QMQLUA66.js';
 import { EmbeddingDimensionChangedError, ConfigurationError, StaleVersionError, MigrationError, SchemaContentConflictError, DatabaseOperationError, UniquenessError, TrustedImportError, StoreNotInitializedError } from './chunk-4MB3AQAQ.js';
@@ -13,23 +14,68 @@ import { sql, is, getTableName } from 'drizzle-orm';
 
 // src/backend/edge-endpoint-sets.ts
 var FIND_EDGES_ENDPOINT_FIXED_PARAM_COUNT = 6;
-function resolveEdgeEndpointIds(params) {
-  const { limitPerEndpoint } = params;
-  if (limitPerEndpoint !== void 0 && (!Number.isInteger(limitPerEndpoint) || limitPerEndpoint <= 0)) {
-    throw new ConfigurationError(
-      `findEdgesByEndpointSet \`limitPerEndpoint\` must be a positive integer, received ${String(limitPerEndpoint)}.`,
-      {
-        code: "EDGE_ENDPOINT_SET_INVALID",
-        graphId: params.graphId,
-        kind: params.kind
-      },
-      {
-        suggestion: "Pass a positive integer to cap each endpoint's rows, or omit it for an unbounded read."
-      }
-    );
+var FIND_EDGES_HETEROGENEOUS_FIXED_PARAM_COUNT = 4;
+function assertLimitPerEndpoint(operation, graphId, limitPerEndpoint, kind) {
+  if (limitPerEndpoint === void 0 || Number.isInteger(limitPerEndpoint) && limitPerEndpoint > 0) {
+    return;
   }
+  throw new ConfigurationError(
+    `${operation} \`limitPerEndpoint\` must be a positive integer, received ${String(limitPerEndpoint)}.`,
+    {
+      code: "EDGE_ENDPOINT_SET_INVALID",
+      graphId,
+      ...kind === void 0 ? {} : { kind }
+    },
+    {
+      suggestion: "Pass a positive integer to cap each endpoint's rows, or omit it for an unbounded read."
+    }
+  );
+}
+function resolveEdgeEndpointIds(params) {
+  assertLimitPerEndpoint(
+    "findEdgesByEndpointSet",
+    params.graphId,
+    params.limitPerEndpoint,
+    params.kind
+  );
   return [...new Set(params.endpointIds)];
 }
+function endpointKey(endpoint) {
+  return `${endpoint.kind}\0${endpoint.id}`;
+}
+function resolveHeterogeneousEdgeRead(params, maxBindParameters) {
+  assertLimitPerEndpoint(
+    "findEdgesByHeterogeneousEndpointSet",
+    params.graphId,
+    params.limitPerEndpoint
+  );
+  const edgeKinds = [...new Set(params.edgeKinds)];
+  const endpoints = [
+    ...new Map(
+      params.endpoints.map((endpoint) => [endpointKey(endpoint), endpoint])
+    ).values()
+  ];
+  if (edgeKinds.length === 0 || endpoints.length === 0) {
+    return { edgeKinds, endpoints, endpointChunkSize: 1 };
+  }
+  const endpointBindBudget = maxBindParameters - FIND_EDGES_HETEROGENEOUS_FIXED_PARAM_COUNT - edgeKinds.length;
+  const endpointChunkSize = Math.floor(endpointBindBudget / 2);
+  if (endpointChunkSize >= 1) {
+    return { edgeKinds, endpoints, endpointChunkSize };
+  }
+  throw new ConfigurationError(
+    "findEdgesByHeterogeneousEndpointSet cannot fit one endpoint and the selected edge kinds within the backend bind-parameter budget.",
+    {
+      code: "EDGE_HETEROGENEOUS_READ_BIND_BUDGET_EXCEEDED",
+      graphId: params.graphId,
+      edgeKindCount: edgeKinds.length,
+      maxBindParameters
+    },
+    {
+      suggestion: "Select fewer edge kinds in one call or use a backend with a larger bind-parameter budget."
+    }
+  );
+}
 
 // src/backend/live-node-candidates.ts
 function buildLiveNodeCandidates(nodesTableName, graphId, nodeKind, now) {
@@ -159,6 +205,9 @@ function diagnoseContribution(row, signature, tableExists) {
     if (row.materializedAt !== void 0) return "orphaned-marker";
     return row.lastError === void 0 ? void 0 : "failed-materialization";
   }
+  if (row?.materializedAt !== void 0 && row.signature !== signature) {
+    return "stale";
+  }
   const state = evaluateContributionState(row, signature);
   switch (state) {
     case "initialized": {
@@ -354,6 +403,7 @@ function createContributionMaterializer(deps) {
   }
   async function ensureContributions(graphId, contributions, options) {
     const force = options?.force === true;
+    const bypassCache = options?.bypassCache === true;
     const entries = await Promise.all(
       contributions.map(async (contribution) => {
         const key = contributionKey(graphId, contribution);
@@ -364,7 +414,13 @@ function createContributionMaterializer(deps) {
         };
       })
     );
-    const pending = force ? entries : entries.filter(
+    if (bypassCache) {
+      for (const entry of entries) {
+        uncacheableKeys.add(entry.key);
+        initializedSignatures.delete(entry.key);
+      }
+    }
+    const pending = force || bypassCache ? entries : entries.filter(
       (entry) => uncacheableKeys.has(entry.key) || initializedSignatures.get(entry.key) !== entry.signature
     );
     if (pending.length === 0) return;
@@ -469,11 +525,11 @@ function createContributionMaterializer(deps) {
       await assertContributions(graphId, contributions);
     }
   }
-  async function verifyContributions(graphId, vectorSlots) {
-    const targetsByGraph = /* @__PURE__ */ new Map();
+  function verificationTargetsByGraph(graphId, vectorSlots) {
+    const targets = /* @__PURE__ */ new Map();
     function addTarget(id, target) {
-      const existing = targetsByGraph.get(id);
-      if (existing === void 0) targetsByGraph.set(id, [target]);
+      const existing = targets.get(id);
+      if (existing === void 0) targets.set(id, [target]);
       else existing.push(target);
     }
     for (const contribution of runtimeContributions()) {
@@ -491,6 +547,10 @@ function createContributionMaterializer(deps) {
         }
       }
     }
+    return targets;
+  }
+  async function diagnoseContributions(graphId, vectorSlots) {
+    const targetsByGraph = verificationTargetsByGraph(graphId, vectorSlots);
     const probes = /* @__PURE__ */ new Map();
     function tableExists(tableName) {
       const pending = probes.get(tableName);
@@ -499,7 +559,7 @@ function createContributionMaterializer(deps) {
       probes.set(tableName, probe);
       return probe;
     }
-    const diagnostics = [];
+    const diagnosed = [];
     for (const [id, targets] of targetsByGraph) {
       const read = await readMarkerRows(id);
       const rows = read.kind === "rows" ? read.rows : /* @__PURE__ */ new Map();
@@ -512,22 +572,56 @@ function createContributionMaterializer(deps) {
           await tableExists(contribution.tableName)
         );
         if (state === void 0) continue;
-        diagnostics.push({
-          owner: contribution.owner,
-          logicalName: contribution.logicalName,
-          physicalName: contribution.tableName,
-          ...kind === void 0 ? {} : { kind },
-          ...fieldPath === void 0 ? {} : { fieldPath },
-          state,
-          // The recorded reason, carried through verbatim. `state` folds
-          // several marker verdicts together because they share a repair;
-          // this is what keeps the fold from also discarding the one thing
-          // the catalog can never tell the operator.
-          ...row?.lastError === void 0 ? {} : { lastError: row.lastError }
+        diagnosed.push({
+          graphId: id,
+          contribution,
+          diagnostic: {
+            owner: contribution.owner,
+            logicalName: contribution.logicalName,
+            physicalName: contribution.tableName,
+            ...kind === void 0 ? {} : { kind },
+            ...fieldPath === void 0 ? {} : { fieldPath },
+            state,
+            // The recorded reason, carried through verbatim. `state` folds
+            // several marker verdicts together because they share a repair;
+            // this is what keeps the fold from also discarding the one thing
+            // the catalog can never tell the operator.
+            ...row?.lastError === void 0 ? {} : { lastError: row.lastError }
+          }
+        });
+      }
+    }
+    return diagnosed;
+  }
+  async function verifyContributions(graphId, vectorSlots) {
+    const diagnosed = await diagnoseContributions(graphId, vectorSlots);
+    return diagnosed.map((entry) => entry.diagnostic);
+  }
+  async function repairContributions(graphId, vectorSlots) {
+    const diagnosed = await diagnoseContributions(graphId, vectorSlots);
+    const results = [];
+    for (const { graphId: targetGraphId, contribution, diagnostic } of diagnosed) {
+      if (diagnostic.state === "stale" || diagnostic.state === "orphaned-marker") {
+        results.push({ diagnostic, status: "requires-rebuild" });
+        continue;
+      }
+      try {
+        await ensureContributions(targetGraphId, [contribution], {
+          bypassCache: true
+        });
+        results.push({ diagnostic, status: "repaired" });
+      } catch (error) {
+        results.push({
+          diagnostic,
+          status: "failed",
+          error: error instanceof Error ? error.message : String(error)
         });
       }
     }
-    return diagnostics;
+    return {
+      results,
+      remaining: await verifyContributions(graphId, vectorSlots)
+    };
   }
   function evictVectorSlot(slot) {
     if (deps.vectorStrategy === void 0) return;
@@ -557,7 +651,8 @@ function createContributionMaterializer(deps) {
     assertVectorSlots,
     dropVectorSlot,
     evictVectorSlot,
-    verifyContributions
+    verifyContributions,
+    repairContributions
   };
 }
 var SQLITE_INDEX_MAT_TIMESTAMPS = {
@@ -1007,6 +1102,58 @@ function buildFindEdgesByEndpointSet(tables, params, endpointIds) {
     ORDER BY ${derivedOrderByClause}
   `;
 }
+function buildFindEdgesByHeterogeneousEndpointSet(tables, params, endpoints, edgeKinds) {
+  const { edges } = tables;
+  const fromSide = params.side === "from";
+  const kindColumn = fromSide ? edges.fromKind : edges.toKind;
+  const idColumn = fromSide ? edges.fromId : edges.toId;
+  const requestedKind = sql.raw(`requested_endpoints."endpoint_kind"`);
+  const requestedId = sql.raw(`requested_endpoints."endpoint_id"`);
+  const requestedRows = sql.join(
+    endpoints.map((endpoint) => sql`(${endpoint.kind}, ${endpoint.id})`),
+    sql`, `
+  );
+  const whereClause = sql.join(
+    [
+      sql`${edges.graphId} = ${params.graphId}`,
+      buildIdSetCondition(edges.kind, edgeKinds),
+      ...buildTemporalConditions(edges, params)
+    ],
+    sql` AND `
+  );
+  const joinClause = sql`
+    FROM ${edges}
+    INNER JOIN requested_endpoints
+      ON ${kindColumn} = ${requestedKind}
+     AND ${idColumn} = ${requestedId}
+    WHERE ${whereClause}
+  `;
+  const { qualified: orderByClause, bare: derivedOrderByClause } = buildEdgeOrdering(tables, void 0);
+  if (params.limitPerEndpoint === void 0) {
+    return sql`
+      WITH requested_endpoints(endpoint_kind, endpoint_id) AS (
+        VALUES ${requestedRows}
+      )
+      SELECT ${edges}.*
+      ${joinClause}
+      ORDER BY ${orderByClause}
+    `;
+  }
+  const ranked = sql`
+    SELECT ${edges}.*, ROW_NUMBER() OVER (
+      PARTITION BY ${kindColumn}, ${idColumn} ORDER BY ${orderByClause}
+    ) AS endpoint_rank
+    ${joinClause}
+  `;
+  return sql`
+    WITH requested_endpoints(endpoint_kind, endpoint_id) AS (
+      VALUES ${requestedRows}
+    )
+    SELECT * FROM (${ranked}) AS ranked_edges
+    WHERE endpoint_rank <= ${params.limitPerEndpoint}
+    ORDER BY ${derivedOrderByClause}
+  `;
+}
 function buildCountEdgesByKind(tables, params) {
   const { edges } = tables;
   const conditions = [
@@ -1416,6 +1563,37 @@ function buildUpdateNode(tables, params, timestamp) {
     RETURNING *
   `;
 }
+function buildUpdateNodeSet(tables, dialect, params, timestamp) {
+  const { nodes } = tables;
+  const adapter = getDialect(dialect);
+  const patchedProps = toDrizzleSql(
+    adapter.jsonSetProperties(
+      sql$1.identifier(nodes.props.name),
+      params.patch,
+      params.unsetProperties ?? []
+    ),
+    dialect
+  );
+  const candidateIds = toDrizzleSql(params.candidateIds, dialect);
+  const candidateIdColumn = toDrizzleSql(
+    sql$1.identifier(params.candidateIdColumn),
+    dialect
+  );
+  return sql`
+    UPDATE ${nodes}
+    SET ${quotedColumn(nodes.props)} = ${patchedProps},
+        ${quotedColumn(nodes.updatedAt)} = ${timestamp},
+        ${quotedColumn(nodes.version)} = ${quotedColumn(nodes.version)} + 1
+    WHERE ${nodes.graphId} = ${params.graphId}
+      AND ${nodes.kind} = ${params.kind}
+      AND ${nodes.deletedAt} IS NULL
+      AND ${nodes.id} IN (
+        SELECT ${candidateIdColumn}
+        FROM (${candidateIds}) AS tg_set_candidates
+      )
+    RETURNING *
+  `;
+}
 function buildDeleteNode(tables, params, timestamp) {
   const { nodes } = tables;
   return sql`
@@ -1651,12 +1829,23 @@ function buildDeleteUnique(tables, params, timestamp) {
       AND ${uniques.deletedAt} IS NULL
   `;
 }
-function buildHardDeleteUniquesByNode(tables, graphId, nodeId) {
+function buildHardDeleteUniquesByNode(tables, graphId, concreteKind, nodeId) {
+  return buildHardDeleteUniquesByNodeIds(tables, {
+    graphId,
+    concreteKind,
+    nodeIds: [nodeId]
+  });
+}
+function buildHardDeleteUniquesByNodeIds(tables, params) {
   const { uniques } = tables;
   return sql`
     DELETE FROM ${uniques}
-    WHERE ${uniques.graphId} = ${graphId}
-      AND ${uniques.nodeId} = ${nodeId}
+    WHERE ${uniques.graphId} = ${params.graphId}
+      AND ${uniques.concreteKind} = ${params.concreteKind}
+      AND ${uniques.nodeId} IN (${sql.join(
+    params.nodeIds.map((nodeId) => sql`${nodeId}`),
+    sql`, `
+  )})
   `;
 }
 function buildCheckUnique(tables, params) {
@@ -1745,9 +1934,11 @@ var COMMON_TABLE_OPERATION_BUILDERS = {
   buildCountNodesByKind,
   buildFindEdgesByKind,
   buildFindEdgesByEndpointSet,
+  buildFindEdgesByHeterogeneousEndpointSet,
   buildCountEdgesByKind,
   buildDeleteUnique,
   buildHardDeleteUniquesByNode,
+  buildHardDeleteUniquesByNodeIds,
   buildCheckUnique,
   buildCheckUniqueBatch,
   buildGetSchemaVersion
@@ -1837,6 +2028,9 @@ function createCommonOperationStrategy(tables, dialect, fulltextStrategy) {
   return {
     ...tableOperations,
     ...fulltextBuilders,
+    buildUpdateNodeSet(params, timestamp) {
+      return buildUpdateNodeSet(tables, dialect, params, timestamp);
+    },
     buildDeleteContributionMaterialization,
     buildInsertUnique(params) {
       return buildInsertUnique(tables, dialect, params);
@@ -1967,7 +2161,13 @@ async function commitSchemaVersionIfKindsEmpty(backend, params, probes) {
   };
 }
 function createCommonOperationBackend(options) {
-  const { batchConfig, execution, operationStrategy, rowMappers } = options;
+  const {
+    batchConfig,
+    execution,
+    maxBindParameters,
+    operationStrategy,
+    rowMappers
+  } = options;
   const nowIso2 = options.nowIso ?? nowIso;
   const tableExists = createCachedTableExistence(
     (tableName) => execution.execGet(
@@ -2069,6 +2269,25 @@ function createCommonOperationBackend(options) {
         );
       return rowMappers.toNodeRow(row);
     },
+    async updateNodeSet(params) {
+      if (Object.keys(params.patch).length === 0 && (params.unsetProperties?.length ?? 0) === 0) {
+        throw new ConfigurationError(
+          "Set-based node update requires at least one property",
+          { operation: "updateNodeSet", kind: params.kind }
+        );
+      }
+      if (params.candidateIdColumn.length === 0) {
+        throw new ConfigurationError(
+          "Set-based node update requires a candidate id column",
+          { operation: "updateNodeSet", kind: params.kind }
+        );
+      }
+      const timestamp = nowIso2();
+      const query = operationStrategy.buildUpdateNodeSet(params, timestamp);
+      const rows = await execution.execAll(query);
+      const updatedRows = rows.map((row) => rowMappers.toNodeRow(row));
+      return { affectedCount: updatedRows.length, rows: updatedRows };
+    },
     async deleteNode(params) {
       const timestamp = nowIso2();
       const query = operationStrategy.buildDeleteNode(params, timestamp);
@@ -2086,6 +2305,7 @@ function createCommonOperationBackend(options) {
     async hardDeleteNode(params) {
       const deleteUniquesQuery = operationStrategy.buildHardDeleteUniquesByNode(
         params.graphId,
+        params.kind,
         params.id
       );
       await execution.execRun(deleteUniquesQuery);
@@ -2260,6 +2480,21 @@ function createCommonOperationBackend(options) {
       }
       return edgeRows;
     },
+    async findEdgesByHeterogeneousEndpointSet(params) {
+      const { edgeKinds, endpoints, endpointChunkSize } = resolveHeterogeneousEdgeRead(params, maxBindParameters);
+      if (edgeKinds.length === 0 || endpoints.length === 0) return [];
+      const edgeRows = [];
+      for (const endpointChunk of chunk(endpoints, endpointChunkSize)) {
+        const query = operationStrategy.buildFindEdgesByHeterogeneousEndpointSet(
+          params,
+          endpointChunk,
+          edgeKinds
+        );
+        const rows = await execution.execAll(query);
+        for (const row of rows) edgeRows.push(rowMappers.toEdgeRow(row));
+      }
+      return edgeRows;
+    },
     async countEdgesByKind(params) {
       const query = operationStrategy.buildCountEdgesByKind(params);
       const row = await execution.execGet(query);
@@ -2330,6 +2565,19 @@ function createCommonOperationBackend(options) {
       const query = operationStrategy.buildDeleteUnique(params, timestamp);
       await execution.execRun(query);
     },
+    async hardDeleteUniquesByNodeIds(params) {
+      const nodeIds = [...new Set(params.nodeIds)];
+      for (const chunk2 of chunk(
+        nodeIds,
+        batchConfig.uniqueDeleteChunkSize
+      )) {
+        const query = operationStrategy.buildHardDeleteUniquesByNodeIds({
+          ...params,
+          nodeIds: chunk2
+        });
+        await execution.execRun(query);
+      }
+    },
     async checkUnique(params) {
       const query = operationStrategy.buildCheckUnique(params);
       const row = await execution.execGet(query);
@@ -2731,5 +2979,5 @@ function getOrCreateLru(cache, key, cacheMax, create) {
 }
 
 export { EMBEDDING_UPSERT_PARAM_COUNT, FIND_EDGES_ENDPOINT_FIXED_PARAM_COUNT, POSTGRES_CONTRIBUTION_MAT_TIMESTAMPS, POSTGRES_INDEX_MAT_TIMESTAMPS, POSTGRES_KIND_REMOVAL_TIMESTAMPS, SQLITE_CONTRIBUTION_MAT_TIMESTAMPS, SQLITE_INDEX_MAT_TIMESTAMPS, SQLITE_KIND_REMOVAL_TIMESTAMPS, analyzeImportedTables, assertActiveSchemaVersion, assertAdoptedDialect, assertTrustedImportDatabaseEmpty, buildContributionInsertValues, buildContributionOnConflictSet, buildKindRemovalInsertValues, buildKindRemovalOnConflictSet, buildLiveNodeCandidates, buildMaterializationInsertValues, buildMaterializationOnConflictSet, commitSchemaVersionIfKindsEmpty, createCachedTableExistence, createCommonOperationBackend, createContributionMaterializer, createPostgresOperationStrategy, createPostgresTrustedImportSession, createSqliteOperationStrategy, createSqliteTrustedImportSession, gateFulltext, gateFulltextMethods, getOrCreateLru, lockPostgresTrustedImportTables, mapContributionMaterializationRow, mapHybridSearchRow, mapKindRemovalRow, mapMaterializationRow, mapVectorWriteError, restoreSecondaryIndexes, suspendPostgresSecondaryIndexes, suspendSqliteSecondaryIndexes, tableExistsFromRow, vectorSlotFromCreateIndexParams, vectorSlotFromDropIndexParams, vectorSlotFromParams };
-//# sourceMappingURL=chunk-SQRD2CYF.js.map
-//# sourceMappingURL=chunk-SQRD2CYF.js.map
+//# sourceMappingURL=chunk-DXEEHSGN.js.map
+//# sourceMappingURL=chunk-DXEEHSGN.js.map