@mastra/convex 1.7.3-alpha.0 → 1.7.3

package/dist/server/index.cjs

@@ -1,5 +1,5 @@
 Object.defineProperty(exports, Symbol.toStringTag, { value: "Module" });
-const require_server = require("../server-cdXFJzWz.cjs");
+const require_server = require("../server-Bcgr5CcA.cjs");
 const require_schema = require("../schema.cjs");
 let _mastra_core_storage_constants = require("@mastra/core/storage/constants");
 exports.TABLE_BACKGROUND_TASKS = _mastra_core_storage_constants.TABLE_BACKGROUND_TASKS;

package/dist/server/index.js

@@ -1,3 +1,3 @@
-import { a as mastraCache, i as mastraStorage, n as mastraNativeVectorMutation, r as mastraNativeVectorQuery, t as mastraNativeVectorAction } from "../server-DozOFGfO.js";
+import { a as mastraCache, i as mastraStorage, n as mastraNativeVectorMutation, r as mastraNativeVectorQuery, t as mastraNativeVectorAction } from "../server-B8RSiOmu.js";
 import { TABLE_BACKGROUND_TASKS, TABLE_CHANNEL_CONFIG, TABLE_CHANNEL_INSTALLATIONS, TABLE_MESSAGES, TABLE_OBSERVATIONAL_MEMORY, TABLE_RESOURCES, TABLE_SCHEDULES, TABLE_SCHEDULE_TRIGGERS, TABLE_SCORERS, TABLE_THREADS, TABLE_WORKFLOW_SNAPSHOT, defineMastraNativeVectorTable, mastraBackgroundTasksTable, mastraCacheListItemsTable, mastraCacheTable, mastraChannelConfigTable, mastraChannelInstallationsTable, mastraDocumentsTable, mastraMessagesTable, mastraObservationalMemoryTable, mastraResourcesTable, mastraScheduleTriggersTable, mastraSchedulesTable, mastraScoresTable, mastraThreadsTable, mastraVectorIndexesTable, mastraVectorsTable, mastraWorkflowSnapshotsTable } from "../schema.js";
 export { TABLE_BACKGROUND_TASKS, TABLE_CHANNEL_CONFIG, TABLE_CHANNEL_INSTALLATIONS, TABLE_MESSAGES, TABLE_OBSERVATIONAL_MEMORY, TABLE_RESOURCES, TABLE_SCHEDULES, TABLE_SCHEDULE_TRIGGERS, TABLE_SCORERS, TABLE_THREADS, TABLE_WORKFLOW_SNAPSHOT, defineMastraNativeVectorTable, mastraBackgroundTasksTable, mastraCache, mastraCacheListItemsTable, mastraCacheTable, mastraChannelConfigTable, mastraChannelInstallationsTable, mastraDocumentsTable, mastraMessagesTable, mastraNativeVectorAction, mastraNativeVectorMutation, mastraNativeVectorQuery, mastraObservationalMemoryTable, mastraResourcesTable, mastraScheduleTriggersTable, mastraSchedulesTable, mastraScoresTable, mastraStorage, mastraThreadsTable, mastraVectorIndexesTable, mastraVectorsTable, mastraWorkflowSnapshotsTable };

package/dist/server/run-fencing.d.ts

@@ -0,0 +1,14 @@
+import type { GenericMutationCtx as MutationCtx } from 'convex/server';
+import type { RunClaimRequest, StorageRequest, StorageResponse } from '../storage/types.js';
+export declare const RUN_FENCE_CONFLICT_CODE = "STORAGE_RUN_FENCE_CONFLICT";
+export declare function isRunClaimRequest(request: StorageRequest | RunClaimRequest): request is RunClaimRequest;
+/**
+ * Runs a run fencing operation. Each reads and writes the run's claim in the
+ * calling mutation, so Convex serializes it against every other claim
+ * operation and fenced write on the same run. Leases are measured on the
+ * mutation's clock.
+ *
+ * `route` runs the request a fenced operation wraps.
+ */
+export declare function handleRunClaimOperation(ctx: MutationCtx<any>, request: RunClaimRequest, route: (ctx: MutationCtx<any>, request: StorageRequest) => Promise<StorageResponse>): Promise<StorageResponse>;
+//# sourceMappingURL=run-fencing.d.ts.map

package/dist/server/run-fencing.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"run-fencing.d.ts","sourceRoot":"","sources":["../../src/server/run-fencing.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,kBAAkB,IAAI,WAAW,EAAE,MAAM,eAAe,CAAC;AAGvE,OAAO,KAAK,EAEV,eAAe,EAGf,cAAc,EACd,eAAe,EAEhB,MAAM,kBAAkB,CAAC;AAI1B,eAAO,MAAM,uBAAuB,+BAA+B,CAAC;AAgBpE,wBAAgB,iBAAiB,CAAC,OAAO,EAAE,cAAc,GAAG,eAAe,GAAG,OAAO,IAAI,eAAe,CAEvG;AAiCD;;;;;;;GAOG;AACH,wBAAsB,uBAAuB,CAC3C,GAAG,EAAE,WAAW,CAAC,GAAG,CAAC,EACrB,OAAO,EAAE,eAAe,EACxB,KAAK,EAAE,CAAC,GAAG,EAAE,WAAW,CAAC,GAAG,CAAC,EAAE,OAAO,EAAE,cAAc,KAAK,OAAO,CAAC,eAAe,CAAC,GAClF,OAAO,CAAC,eAAe,CAAC,CA6F1B"}

package/dist/server/storage.d.ts

@@ -1,10 +1,10 @@
 import type { GenericMutationCtx as MutationCtx } from 'convex/server';
-import type { StorageRequest, StorageResponse } from '../storage/types.js';
+import type { RunClaimRequest, StorageRequest, StorageResponse } from '../storage/types.js';
 /**
  * Main storage mutation handler.
  * Routes operations to the appropriate typed table.
  */
-export declare const mastraStorage: import("convex/server").RegisteredMutation<"public", StorageRequest, Promise<StorageResponse>>;
+export declare const mastraStorage: import("convex/server").RegisteredMutation<"public", RunClaimRequest | StorageRequest, Promise<StorageResponse>>;
 /**
  * Handle operations on typed tables (threads, messages, etc.)
  * Records are stored with their `id` field as a regular field (not _id).

package/dist/server/storage.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"storage.d.ts","sourceRoot":"","sources":["../../src/server/storage.ts"],"names":[],"mappings":"AAYA,OAAO,KAAK,EAAE,kBAAkB,IAAI,WAAW,EAAE,MAAM,eAAe,CAAC;AAIvE,OAAO,KAAK,EAAkB,cAAc,EAAE,eAAe,EAAE,MAAM,kBAAkB,CAAC;AAkRxF;;;GAGG;AACH,eAAO,MAAM,aAAa,gGA+BxB,CAAC;AA8BH;;;;GAIG;AACH,wBAAsB,oBAAoB,CACxC,GAAG,EAAE,WAAW,CAAC,GAAG,CAAC,EACrB,WAAW,EAAE,MAAM,EACnB,OAAO,EAAE,cAAc,GACtB,OAAO,CAAC,eAAe,CAAC,CA+hB1B"}
+{"version":3,"file":"storage.d.ts","sourceRoot":"","sources":["../../src/server/storage.ts"],"names":[],"mappings":"AAYA,OAAO,KAAK,EAAE,kBAAkB,IAAI,WAAW,EAAE,MAAM,eAAe,CAAC;AAIvE,OAAO,KAAK,EAAkB,eAAe,EAAE,cAAc,EAAE,eAAe,EAAE,MAAM,kBAAkB,CAAC;AAmRzG;;;GAGG;AACH,eAAO,MAAM,aAAa,kHAOzB,CAAC;AA+DF;;;;GAIG;AACH,wBAAsB,oBAAoB,CACxC,GAAG,EAAE,WAAW,CAAC,GAAG,CAAC,EACrB,WAAW,EAAE,MAAM,EACnB,OAAO,EAAE,cAAc,GACtB,OAAO,CAAC,eAAe,CAAC,CA+hB1B"}

package/dist/{server-DozOFGfO.js → server-B8RSiOmu.js}

@@ -1022,6 +1022,152 @@ async function handleObservationalMemoryOperation(ctx, convexTable, request) {
 	}
 }
 //#endregion
+//#region src/server/run-fencing.ts
+const RUN_FENCE_CONFLICT_CODE = "STORAGE_RUN_FENCE_CONFLICT";
+const RUN_CLAIM_OPS = /* @__PURE__ */ new Set([
+	"claimRunOwnership",
+	"renewRunOwnership",
+	"releaseRunOwnership",
+	"getRunOwnership",
+	"raiseRunFence",
+	"retireRunFence",
+	"fenced"
+]);
+const RUN_CLAIM_TABLES = /* @__PURE__ */ new Set(["mastra_workflow_run_owners", "mastra_memory_run_fences"]);
+const CLAIMS_TABLE = "mastra_documents";
+function isRunClaimRequest(request) {
+	return RUN_CLAIM_OPS.has(request.op);
+}
+async function findClaim(ctx, table, runId) {
+	return ctx.db.query(CLAIMS_TABLE).withIndex("by_table_primary", (q) => q.eq("table", table).eq("primaryKey", runId)).unique();
+}
+async function writeClaim(ctx, table, doc, claim) {
+	const record = {
+		id: claim.runId,
+		...claim
+	};
+	if (doc) await ctx.db.patch(doc._id, { record });
+	else await ctx.db.insert(CLAIMS_TABLE, {
+		table,
+		primaryKey: claim.runId,
+		record
+	});
+	return claim;
+}
+function holds(claim, fence) {
+	return claim !== null && claim.generation === fence.generation && claim.ownerId === fence.ownerId;
+}
+function reply(applied, claim, now) {
+	return {
+		ok: true,
+		result: {
+			applied,
+			claim,
+			now
+		}
+	};
+}
+/**
+* Runs a run fencing operation. Each reads and writes the run's claim in the
+* calling mutation, so Convex serializes it against every other claim
+* operation and fenced write on the same run. Leases are measured on the
+* mutation's clock.
+*
+* `route` runs the request a fenced operation wraps.
+*/
+async function handleRunClaimOperation(ctx, request, route) {
+	if (!RUN_CLAIM_TABLES.has(request.tableName)) return {
+		ok: false,
+		error: `Unsupported run claim table ${request.tableName}`
+	};
+	const now = Date.now();
+	switch (request.op) {
+		case "claimRunOwnership": {
+			const { tableName, runId, ownerId, leaseMs, force, expectedGeneration } = request;
+			const doc = await findClaim(ctx, tableName, runId);
+			const current = doc?.record ?? null;
+			const leaseExpiresAt = now + leaseMs;
+			if (!current) {
+				if (expectedGeneration !== void 0 && expectedGeneration !== 0) return reply(false, null, now);
+				return reply(true, await writeClaim(ctx, tableName, doc, {
+					runId,
+					generation: Math.max(1, now),
+					ownerId,
+					leaseExpiresAt,
+					updatedAt: now
+				}), now);
+			}
+			if (expectedGeneration !== void 0 && current.generation !== expectedGeneration) return reply(false, current, now);
+			if (!force && current.leaseExpiresAt !== null && current.leaseExpiresAt > now) return reply(false, current, now);
+			return reply(true, await writeClaim(ctx, tableName, doc, {
+				runId,
+				generation: current.generation + 1,
+				ownerId,
+				leaseExpiresAt,
+				updatedAt: now
+			}), now);
+		}
+		case "renewRunOwnership": {
+			const { tableName, fence, leaseMs } = request;
+			const doc = await findClaim(ctx, tableName, fence.runId);
+			const current = doc?.record ?? null;
+			if (!holds(current, fence) || current.leaseExpiresAt === null) return reply(false, current, now);
+			return reply(true, await writeClaim(ctx, tableName, doc, {
+				...current,
+				leaseExpiresAt: now + leaseMs,
+				updatedAt: now
+			}), now);
+		}
+		case "releaseRunOwnership": {
+			const { tableName, fence } = request;
+			const doc = await findClaim(ctx, tableName, fence.runId);
+			const current = doc?.record ?? null;
+			if (!holds(current, fence)) return reply(false, current, now);
+			return reply(true, await writeClaim(ctx, tableName, doc, {
+				...current,
+				leaseExpiresAt: null,
+				updatedAt: now
+			}), now);
+		}
+		case "getRunOwnership": return reply(true, (await findClaim(ctx, request.tableName, request.runId))?.record ?? null, now);
+		case "raiseRunFence": {
+			const { tableName, fence } = request;
+			const doc = await findClaim(ctx, tableName, fence.runId);
+			const current = doc?.record ?? null;
+			if (!current || current.generation < fence.generation || holds(current, fence)) return reply(true, await writeClaim(ctx, tableName, doc, {
+				...fence,
+				leaseExpiresAt: null,
+				retiredAt: null
+			}), now);
+			return reply(false, current, now);
+		}
+		case "retireRunFence": {
+			const { tableName, fence } = request;
+			const doc = await findClaim(ctx, tableName, fence.runId);
+			const current = doc?.record ?? null;
+			if (!holds(current, fence)) return reply(false, current, now);
+			return reply(true, await writeClaim(ctx, tableName, doc, {
+				...current,
+				retiredAt: now
+			}), now);
+		}
+		case "fenced": {
+			const { tableName, fence, request: inner } = request;
+			if (!holds((await findClaim(ctx, tableName, fence.runId))?.record ?? null, fence)) return {
+				ok: false,
+				error: `Run ${fence.runId} is no longer claimed by generation ${fence.generation} of ${fence.ownerId}`,
+				code: RUN_FENCE_CONFLICT_CODE
+			};
+			if (!inner) return { ok: true };
+			if (RUN_CLAIM_OPS.has(inner.op)) return {
+				ok: false,
+				error: `Unsupported fenced operation ${inner.op}`
+			};
+			return route(ctx, inner);
+		}
+	}
+}
+//#endregion
 //#region src/server/workflow-snapshot.ts
 const PENDING_MARKER_KEY = "__mastra_pending__";
 function isPendingMarker(val) {
@@ -1303,6 +1449,10 @@ function resolveTable(tableName) {
 * Routes operations to the appropriate typed table.
 */
 const mastraStorage = mutationGeneric(async (ctx, request) => {
+	if (isRunClaimRequest(request)) return handleRunClaimOperation(ctx, request, routeStorageRequest);
+	return routeStorageRequest(ctx, request);
+});
+async function routeStorageRequest(ctx, request) {
 	try {
 		const { convexTable, isTyped } = resolveTable(request.tableName);
 		if (request.tableName.startsWith(VECTOR_TABLE_PREFIX) && request.tableName !== TABLE_VECTOR_INDEXES) return await handleVectorOperation(ctx, request);
@@ -1322,7 +1472,7 @@ const mastraStorage = mutationGeneric(async (ctx, request) => {
 			error: error.message
 		};
 	}
-});
+}
 function parseStoredSnapshot(stored, runId) {
 	if (typeof stored === "string") return JSON.parse(stored);
 	return JSON.parse(JSON.stringify(stored ?? createEmptyWorkflowSnapshot(runId)));
@@ -2143,4 +2293,4 @@ const mastraNativeVectorMutation = mutationGeneric({
 //#endregion
 export { mastraCache as a, mastraStorage as i, mastraNativeVectorMutation as n, mastraNativeVectorQuery as r, mastraNativeVectorAction as t };
 
-//# sourceMappingURL=server-DozOFGfO.js.map
+//# sourceMappingURL=server-B8RSiOmu.js.map