@mastra/convex 1.7.2-alpha.0 → 1.7.3-alpha.0

package/dist/index.js

@@ -1,4 +1,4 @@
-import { a as mastraCache, i as mastraStorage, n as mastraNativeVectorMutation, r as mastraNativeVectorQuery, t as mastraNativeVectorAction } from "./server-D3SG5LQf.js";
+import { a as mastraCache, i as mastraStorage, n as mastraNativeVectorMutation, r as mastraNativeVectorQuery, t as mastraNativeVectorAction } from "./server-DozOFGfO.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";
 import { MastraServerCache } from "@mastra/core/cache";
 import { BackgroundTasksStorage, ChannelsStorage, MastraCompositeStore, MemoryStorage, SchedulesStorage, ScoresStorage, TABLE_BACKGROUND_TASKS as TABLE_BACKGROUND_TASKS$1, TABLE_CHANNEL_CONFIG as TABLE_CHANNEL_CONFIG$1, TABLE_CHANNEL_INSTALLATIONS as TABLE_CHANNEL_INSTALLATIONS$1, TABLE_MESSAGES as TABLE_MESSAGES$1, TABLE_RESOURCES as TABLE_RESOURCES$1, TABLE_SCHEDULES as TABLE_SCHEDULES$1, TABLE_SCHEDULE_TRIGGERS as TABLE_SCHEDULE_TRIGGERS$1, TABLE_SCORERS as TABLE_SCORERS$1, TABLE_THREADS as TABLE_THREADS$1, TABLE_WORKFLOW_SNAPSHOT as TABLE_WORKFLOW_SNAPSHOT$1, WorkflowsStorage, calculatePagination, createStorageErrorId, createVectorErrorId, filterByDateRange, normalizePerPage, normalizeScheduleTarget, safelyParseJSON, storageMessageMatchesMetadataFilter, validateStorageMetadataFilter } from "@mastra/core/storage";
@@ -487,15 +487,43 @@ var ConvexDB = class extends MastraBase {
 			...args
 		});
 	}
-	async omUpdateActive(args) {
+	async omInitialize(record) {
+		return this.client.callStorage({
+			op: "omInitialize",
+			tableName: TABLE_OBSERVATIONAL_MEMORY$1,
+			record
+		});
+	}
+	async omCreateReflectionGeneration(args) {
+		return this.client.callStorage({
+			op: "omCreateReflectionGeneration",
+			tableName: TABLE_OBSERVATIONAL_MEMORY$1,
+			...args
+		});
+	}
+	async omSetPendingMessageTokens(args) {
+		await this.client.callStorage({
+			op: "omSetPendingMessageTokens",
+			tableName: TABLE_OBSERVATIONAL_MEMORY$1,
+			...args
+		});
+	}
+	async omSetBufferingObservationFlag(args) {
 		await this.client.callStorage({
+			op: "omSetBufferingObservationFlag",
+			tableName: TABLE_OBSERVATIONAL_MEMORY$1,
+			...args
+		});
+	}
+	async omUpdateActive(args) {
+		return this.client.callStorage({
 			op: "omUpdateActive",
 			tableName: TABLE_OBSERVATIONAL_MEMORY$1,
 			...args
 		});
 	}
 	async omAppendBufferedChunk(args) {
-		await this.client.callStorage({
+		return this.client.callStorage({
 			op: "omAppendBufferedChunk",
 			tableName: TABLE_OBSERVATIONAL_MEMORY$1,
 			...args
@@ -900,6 +928,17 @@ var ChannelsConvex = class extends ChannelsStorage {
 	}
 };
 //#endregion
+//#region ../../packages/_internal-core/dist/memory/index.js
+/**
+* Backs `updateActiveObservations` in adapters that implement `commitActiveObservations`.
+* Callers of `updateActiveObservations` (which resolves `void`) can't see a commit result, so a
+* write that was not applied throws instead of resolving as if it had committed.
+*/
+function assertActiveObservationsApplied(result, id) {
+	if (result.applied) return;
+	throw new Error(result.reason === "retired" ? `Observational memory record ${id} was superseded by a newer generation; observations were not written` : `Observational memory record ${id} changed since the observations were composed; observations were not written`);
+}
+//#endregion
 //#region src/storage/domains/memory/index.ts
 function parseStoredThread(row) {
 	return {
@@ -975,7 +1014,8 @@ function parseStoredOMRecord(doc) {
 		config: config ?? {},
 		metadata: metadata ?? void 0,
 		observedMessageIds: doc.observedMessageIds || void 0,
-		observedTimezone: doc.observedTimezone || void 0
+		observedTimezone: doc.observedTimezone || void 0,
+		supersededBy: doc.supersededBy ?? null
 	};
 }
 var MemoryConvex = class extends MemoryStorage {
@@ -1463,9 +1503,9 @@ var MemoryConvex = class extends MemoryStorage {
 		})).map((doc) => parseStoredOMRecord(doc));
 	}
 	async initializeObservationalMemory(input) {
-		const id = crypto.randomUUID();
 		const now = /* @__PURE__ */ new Date();
 		const lookupKey = this.getOMKey(input.threadId, input.resourceId);
+		const id = crypto.randomUUID();
 		const record = {
 			id,
 			scope: input.scope,
@@ -1487,38 +1527,36 @@ var MemoryConvex = class extends MemoryStorage {
 			lastBufferedAtTokens: 0,
 			lastBufferedAtTime: null,
 			config: input.config,
-			observedTimezone: input.observedTimezone
+			observedTimezone: input.observedTimezone,
+			supersededBy: null
 		};
-		await this.#db.insert({
-			tableName: TABLE_OBSERVATIONAL_MEMORY$1,
-			record: {
-				id,
-				lookupKey,
-				scope: input.scope,
-				resourceId: input.resourceId,
-				threadId: input.threadId || null,
-				activeObservations: "",
-				activeObservationsPendingUpdate: null,
-				originType: "initial",
-				config: JSON.stringify(input.config ?? {}),
-				generationCount: 0,
-				lastObservedAt: null,
-				lastReflectionAt: null,
-				pendingMessageTokens: 0,
-				totalTokensObserved: 0,
-				observationTokenCount: 0,
-				isObserving: false,
-				isReflecting: false,
-				isBufferingObservation: false,
-				isBufferingReflection: false,
-				lastBufferedAtTokens: 0,
-				lastBufferedAtTime: null,
-				observedTimezone: input.observedTimezone || null,
-				createdAt: now.toISOString(),
-				updatedAt: now.toISOString()
-			}
+		const doc = await this.#db.omInitialize({
+			id,
+			lookupKey,
+			scope: input.scope,
+			resourceId: input.resourceId,
+			threadId: input.threadId || null,
+			activeObservations: "",
+			activeObservationsPendingUpdate: null,
+			originType: "initial",
+			config: JSON.stringify(input.config ?? {}),
+			generationCount: 0,
+			lastObservedAt: null,
+			lastReflectionAt: null,
+			pendingMessageTokens: 0,
+			totalTokensObserved: 0,
+			observationTokenCount: 0,
+			isObserving: false,
+			isReflecting: false,
+			isBufferingObservation: false,
+			isBufferingReflection: false,
+			lastBufferedAtTokens: 0,
+			lastBufferedAtTime: null,
+			observedTimezone: input.observedTimezone || null,
+			createdAt: now.toISOString(),
+			updatedAt: now.toISOString()
 		});
-		return record;
+		return doc ? parseStoredOMRecord(doc) : record;
 	}
 	async insertObservationalMemoryRecord(record) {
 		const lookupKey = this.getOMKey(record.threadId, record.resourceId);
@@ -1554,22 +1592,30 @@ var MemoryConvex = class extends MemoryStorage {
 				lastBufferedAtTime: record.lastBufferedAtTime ? toISO(record.lastBufferedAtTime) : null,
 				observedTimezone: record.observedTimezone || null,
 				metadata: record.metadata ? JSON.stringify(record.metadata) : null,
+				supersededBy: record.supersededBy ?? null,
 				createdAt: toISO(record.createdAt),
 				updatedAt: toISO(record.updatedAt)
 			}
 		});
 	}
 	async updateActiveObservations(input) {
-		await this.#db.omUpdateActive({
+		assertActiveObservationsApplied(await this.commitActiveObservations(input), input.id);
+	}
+	async commitActiveObservations(input) {
+		return await this.#db.omUpdateActive({
 			id: input.id,
 			observations: input.observations,
 			tokenCount: input.tokenCount,
 			lastObservedAt: toISO(input.lastObservedAt),
 			observedMessageIds: input.observedMessageIds ?? null,
-			updatedAt: (/* @__PURE__ */ new Date()).toISOString()
-		});
+			updatedAt: (/* @__PURE__ */ new Date()).toISOString(),
+			...input.expectedActiveObservations !== void 0 ? { expectedActiveObservations: input.expectedActiveObservations } : {}
+		}) ?? { applied: true };
 	}
 	async updateBufferedObservations(input) {
+		await this.appendBufferedObservations(input);
+	}
+	async appendBufferedObservations(input) {
 		const chunk = {
 			id: `ombuf-${crypto.randomUUID()}`,
 			cycleId: input.chunk.cycleId,
@@ -1585,12 +1631,15 @@ var MemoryConvex = class extends MemoryStorage {
 			extractedValues: input.chunk.extractedValues,
 			extractionFailures: input.chunk.extractionFailures
 		};
-		await this.#db.omAppendBufferedChunk({
+		return await this.#db.omAppendBufferedChunk({
 			id: input.id,
 			chunk,
 			lastBufferedAtTime: input.lastBufferedAtTime ? toISO(input.lastBufferedAtTime) : void 0,
 			updatedAt: (/* @__PURE__ */ new Date()).toISOString()
-		});
+		}) ?? {
+			persisted: true,
+			recordId: input.id
+		};
 	}
 	async swapBufferedToActive(input) {
 		return this.#db.omSwapBuffered({
@@ -1604,65 +1653,33 @@ var MemoryConvex = class extends MemoryStorage {
 			now: (/* @__PURE__ */ new Date()).toISOString()
 		});
 	}
-	async createReflectionGeneration(input) {
-		const id = crypto.randomUUID();
-		const now = /* @__PURE__ */ new Date();
-		const lookupKey = this.getOMKey(input.currentRecord.threadId, input.currentRecord.resourceId);
-		const record = {
-			id,
-			scope: input.currentRecord.scope,
-			threadId: input.currentRecord.threadId,
-			resourceId: input.currentRecord.resourceId,
-			createdAt: now,
-			updatedAt: now,
-			lastObservedAt: input.currentRecord.lastObservedAt,
-			originType: "reflection",
-			generationCount: input.currentRecord.generationCount + 1,
-			activeObservations: input.reflection,
-			totalTokensObserved: input.currentRecord.totalTokensObserved,
-			observationTokenCount: input.tokenCount,
-			pendingMessageTokens: 0,
-			isReflecting: false,
-			isObserving: false,
-			isBufferingObservation: false,
-			isBufferingReflection: false,
-			lastBufferedAtTokens: 0,
-			lastBufferedAtTime: null,
-			config: input.currentRecord.config,
-			metadata: input.currentRecord.metadata,
-			observedTimezone: input.currentRecord.observedTimezone
+	/** The caller's snapshot as the server's rollover ops expect it. */
+	#serializeCurrentRecord(currentRecord) {
+		return {
+			id: currentRecord.id,
+			lookupKey: this.getOMKey(currentRecord.threadId, currentRecord.resourceId),
+			scope: currentRecord.scope,
+			threadId: currentRecord.threadId || null,
+			resourceId: currentRecord.resourceId,
+			config: JSON.stringify(currentRecord.config ?? {}),
+			metadata: currentRecord.metadata ? JSON.stringify(currentRecord.metadata) : null,
+			observedTimezone: currentRecord.observedTimezone || null,
+			lastObservedAt: currentRecord.lastObservedAt ? toISO(currentRecord.lastObservedAt) : null,
+			totalTokensObserved: currentRecord.totalTokensObserved,
+			generationCount: currentRecord.generationCount,
+			activeObservations: currentRecord.activeObservations ?? "",
+			observationTokenCount: currentRecord.observationTokenCount ?? 0
 		};
-		await this.#db.insert({
-			tableName: TABLE_OBSERVATIONAL_MEMORY$1,
-			record: {
-				id,
-				lookupKey,
-				scope: record.scope,
-				resourceId: record.resourceId,
-				threadId: record.threadId || null,
-				activeObservations: input.reflection,
-				activeObservationsPendingUpdate: null,
-				originType: "reflection",
-				config: JSON.stringify(record.config ?? {}),
-				generationCount: record.generationCount,
-				lastObservedAt: record.lastObservedAt ? toISO(record.lastObservedAt) : null,
-				lastReflectionAt: now.toISOString(),
-				pendingMessageTokens: 0,
-				totalTokensObserved: record.totalTokensObserved,
-				observationTokenCount: record.observationTokenCount,
-				isObserving: false,
-				isReflecting: false,
-				isBufferingObservation: false,
-				isBufferingReflection: false,
-				lastBufferedAtTokens: 0,
-				lastBufferedAtTime: null,
-				observedTimezone: record.observedTimezone || null,
-				metadata: record.metadata ? JSON.stringify(record.metadata) : null,
-				createdAt: now.toISOString(),
-				updatedAt: now.toISOString()
-			}
+	}
+	async createReflectionGeneration(input) {
+		const doc = await this.#db.omCreateReflectionGeneration({
+			currentRecord: this.#serializeCurrentRecord(input.currentRecord),
+			newId: input.newRecordId ?? crypto.randomUUID(),
+			reflection: input.reflection,
+			tokenCount: input.tokenCount,
+			now: (/* @__PURE__ */ new Date()).toISOString()
 		});
-		return record;
+		return doc ? parseStoredOMRecord(doc) : input.currentRecord;
 	}
 	async updateBufferedReflection(input) {
 		await this.#db.omUpdateBufferedReflection({
@@ -1675,26 +1692,13 @@ var MemoryConvex = class extends MemoryStorage {
 		});
 	}
 	async swapBufferedReflectionToActive(input) {
-		const { currentRecord } = input;
-		const serializedCurrentRecord = {
-			id: currentRecord.id,
-			lookupKey: this.getOMKey(currentRecord.threadId, currentRecord.resourceId),
-			scope: currentRecord.scope,
-			threadId: currentRecord.threadId || null,
-			resourceId: currentRecord.resourceId,
-			config: JSON.stringify(currentRecord.config ?? {}),
-			metadata: currentRecord.metadata ? JSON.stringify(currentRecord.metadata) : null,
-			observedTimezone: currentRecord.observedTimezone || null,
-			lastObservedAt: currentRecord.lastObservedAt ? toISO(currentRecord.lastObservedAt) : null,
-			totalTokensObserved: currentRecord.totalTokensObserved,
-			generationCount: currentRecord.generationCount
-		};
-		return parseStoredOMRecord(await this.#db.omSwapBufferedReflection({
-			currentRecord: serializedCurrentRecord,
-			newId: crypto.randomUUID(),
+		const doc = await this.#db.omSwapBufferedReflection({
+			currentRecord: this.#serializeCurrentRecord(input.currentRecord),
+			newId: input.newRecordId ?? crypto.randomUUID(),
 			tokenCount: input.tokenCount,
 			now: (/* @__PURE__ */ new Date()).toISOString()
-		}));
+		});
+		return doc ? parseStoredOMRecord(doc) : input.currentRecord;
 	}
 	async setReflectingFlag(id, isReflecting) {
 		if (!await this.#db.patch({
@@ -1717,15 +1721,12 @@ var MemoryConvex = class extends MemoryStorage {
 		})) throw this.omRecordNotFound("SET_OBSERVING_FLAG", id);
 	}
 	async setBufferingObservationFlag(id, isBuffering, lastBufferedAtTokens) {
-		if (!await this.#db.patch({
-			tableName: "mastra_observational_memory",
+		await this.#db.omSetBufferingObservationFlag({
 			id,
-			record: {
-				isBufferingObservation: isBuffering,
-				...lastBufferedAtTokens !== void 0 ? { lastBufferedAtTokens } : {},
-				updatedAt: /* @__PURE__ */ new Date()
-			}
-		})) throw this.omRecordNotFound("SET_BUFFERING_OBSERVATION_FLAG", id);
+			isBuffering,
+			...lastBufferedAtTokens !== void 0 ? { lastBufferedAtTokens } : {},
+			updatedAt: (/* @__PURE__ */ new Date()).toISOString()
+		});
 	}
 	async setBufferingReflectionFlag(id, isBuffering) {
 		if (!await this.#db.patch({
@@ -1748,14 +1749,11 @@ var MemoryConvex = class extends MemoryStorage {
 				tokenCount
 			}
 		});
-		if (!await this.#db.patch({
-			tableName: "mastra_observational_memory",
+		await this.#db.omSetPendingMessageTokens({
 			id,
-			record: {
-				pendingMessageTokens: tokenCount,
-				updatedAt: /* @__PURE__ */ new Date()
-			}
-		})) throw this.omRecordNotFound("SET_PENDING_MESSAGE_TOKENS", id);
+			tokenCount,
+			updatedAt: (/* @__PURE__ */ new Date()).toISOString()
+		});
 	}
 	async clearObservationalMemory(threadId, resourceId) {
 		const lookupKey = this.getOMKey(threadId, resourceId);