@massa-ai/tools-api 1.34.0 → 1.35.1

package/dist/index.js

@@ -121298,8 +121298,8 @@ class GraphGenerationRepositoryPg {
   async heartbeat(lease, leaseTtlMs) {
     const ttl = validateTtl2(leaseTtlMs);
     return getPrismaClient2().$transaction(async (tx) => {
-      const workspace = await lockWorkspace(tx, lease.projectId);
       const generation = await lockGeneration(tx, lease.projectId, lease.generationId);
+      const workspace = await lockWorkspace(tx, lease.projectId);
       if (!leaseMatches(workspace, generation, lease))
         return { status: "lease_lost" };
       const renewed = await tx.$queryRaw`
@@ -121322,8 +121322,8 @@ class GraphGenerationRepositoryPg {
   }
   async complete(lease) {
     return getPrismaClient2().$transaction(async (tx) => {
-      const workspace = await lockWorkspace(tx, lease.projectId);
       const generation = await lockGeneration(tx, lease.projectId, lease.generationId);
+      const workspace = await lockWorkspace(tx, lease.projectId);
       if (!leaseMatches(workspace, generation, lease))
         return { status: "lease_lost" };
       const live = await tx.$queryRaw`
@@ -121352,8 +121352,8 @@ class GraphGenerationRepositoryPg {
   }
   async activate(lease) {
     return getPrismaClient2().$transaction(async (tx) => {
-      const workspace = await lockWorkspace(tx, lease.projectId);
       const generation = await lockGeneration(tx, lease.projectId, lease.generationId);
+      const workspace = await lockWorkspace(tx, lease.projectId);
       if (!leaseMatches(workspace, generation, lease))
         return { status: "lease_lost" };
       if (workspace.active_graph_generation_id !== generation.expected_active_id) {
@@ -121403,8 +121403,8 @@ class GraphGenerationRepositoryPg {
   async abort(lease, reason) {
     const failureReason = boundedText2(reason, "reason", MAX_FAILURE_REASON);
     return getPrismaClient2().$transaction(async (tx) => {
-      const workspace = await lockWorkspace(tx, lease.projectId);
       const generation = await lockGeneration(tx, lease.projectId, lease.generationId);
+      const workspace = await lockWorkspace(tx, lease.projectId);
       if (!leaseMatches(workspace, generation, lease))
         return { status: "lease_lost" };
       const live = await tx.$queryRaw`
@@ -121480,7 +121480,7 @@ class GraphGenerationCoordinator {
   async begin(input) {
     const deadline = Date.now() + GRAPH_GENERATION_LEASE_TTL_MS;
     for (;; ) {
-      const outcome2 = await this.repository.begin({ ...input, leaseTtlMs: GRAPH_GENERATION_LEASE_TTL_MS });
+      const outcome2 = await withDeadlockRetry(() => this.repository.begin({ ...input, leaseTtlMs: GRAPH_GENERATION_LEASE_TTL_MS }), { operation: "graph_generation.begin", maxAttempts: 5 });
       if (outcome2.status === "acquired")
         return outcome2.lease;
       if (outcome2.status === "stale_active") {
@@ -121492,12 +121492,12 @@ class GraphGenerationCoordinator {
     }
   }
   async heartbeat(lease) {
-    const outcome2 = await withDeadlockRetry(() => this.repository.heartbeat(lease, GRAPH_GENERATION_LEASE_TTL_MS), { operation: "graph_generation.heartbeat", maxAttempts: 3 });
+    const outcome2 = await withDeadlockRetry(() => this.repository.heartbeat(lease, GRAPH_GENERATION_LEASE_TTL_MS), { operation: "graph_generation.heartbeat", maxAttempts: 5 });
     if (outcome2.status !== "renewed")
       throw new Error("graph_generation_lease_lost");
   }
   async activate(lease) {
-    const completeness = await this.repository.complete(lease);
+    const completeness = await withDeadlockRetry(() => this.repository.complete(lease), { operation: "graph_generation.complete", maxAttempts: 5 });
     if (completeness.status === "incomplete") {
       throw new Error(`graph_generation_incomplete:${completeness.reasons.join(",")}`);
     }
@@ -121506,7 +121506,7 @@ class GraphGenerationCoordinator {
     if (completeness.status === "stale_active") {
       throw new Error(`graph_generation_stale_active:${completeness.activeGenerationId ?? "none"}`);
     }
-    const activation = await this.repository.activate(lease);
+    const activation = await withDeadlockRetry(() => this.repository.activate(lease), { operation: "graph_generation.activate", maxAttempts: 5 });
     if (activation.status === "activated")
       return activation;
     if (activation.status === "incomplete") {
@@ -121517,7 +121517,7 @@ class GraphGenerationCoordinator {
     throw new Error(`graph_generation_stale_active:${activation.activeGenerationId ?? "none"}`);
   }
   async abort(lease, reason) {
-    const outcome2 = await this.repository.abort(lease, reason);
+    const outcome2 = await withDeadlockRetry(() => this.repository.abort(lease, reason), { operation: "graph_generation.abort", maxAttempts: 5 });
     if (outcome2.status === "lease_lost")
       throw new Error("graph_generation_lease_lost_during_abort");
   }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@massa-ai/tools-api",
-  "version": "1.34.0",
+  "version": "1.35.1",
   "author": "luizgmassa",
   "description": "massa-ai REST API server - Semantic code search, memory, and context compression",
   "type": "module",
@@ -21,8 +21,8 @@
     "test": "bun scripts/run-tests-isolated.ts"
   },
   "dependencies": {
-    "@massa-ai/core": "^1.34.0",
-    "@massa-ai/shared": "^1.34.0",
+    "@massa-ai/core": "^1.35.1",
+    "@massa-ai/shared": "^1.35.1",
     "elysia": "^1.2.25",
     "@elysiajs/swagger": "^1.2.0",
     "@elysiajs/cors": "^1.2.0",