@yielded/agent-storage-memory 0.1.0-beta.172 → 0.1.0-beta.174

package/dist/MemoryThreadStore.mjs

@@ -33,19 +33,19 @@ const isBatchBoundary = (thread, sequence) => {
 	return record !== void 0 && thread.batches.get(record.batchId)?.result.lastSequence === sequence;
 };
 const storedBytes = (thread, sequence) => thread.chunks.get(Math.floor((sequence - 1) / RECORD_CHUNK_SIZE))?.bytes[(sequence - 1) % RECORD_CHUNK_SIZE];
-const storedPage = (thread, after, limit) => {
+const storedPage = Effect.fnUntraced(function* (thread, after, limit) {
 	const page = [];
 	let bytes = 0;
 	for (let sequence = after + 1; sequence <= Math.min(thread.tailSequence, after + limit); sequence++) {
 		const record = storedRecord(thread, sequence);
-		if (record === void 0) throw new Error("Incomplete canonical chunk table");
+		if (record === void 0) return yield* storeError("read", "Incomplete canonical chunk table");
 		const size = storedBytes(thread, sequence) ?? 0;
 		if (bytes + size > 33554432) break;
 		bytes += size;
 		page.push(record);
 	}
 	return page;
-};
+});
 const retainRecords = (thread, records) => {
 	for (const entry of records) {
 		const chunkId = Math.floor((entry.sequence - 1) / RECORD_CHUNK_SIZE);
@@ -747,7 +747,7 @@ const makeThreadStore = Effect.gen(function* () {
 		const index = yield* readyIndex(request.threadId, thread.tailSequence).pipe(Effect.mapError((cause) => storeError("countPeerMessages", cause.message)));
 		return Math.min(index.foreign.peer + (deliveryTransfer?.peerCount(request.threadId, request.nowMillis, request.limit) ?? 0), request.limit);
 	});
-	const readSnapshot = (threadId, afterSequence, limit) => Ref.get(state).pipe(Effect.flatMap((current) => findThread(current, threadId)), Effect.map((thread) => {
+	const readSnapshot = (threadId, afterSequence, limit) => Ref.get(state).pipe(Effect.flatMap((current) => findThread(current, threadId)), Effect.flatMap((thread) => {
 		return storedPage(thread, afterSequence ?? ZERO_CANONICAL_SEQUENCE, limit);
 	}));
 	const selectedStream = (sources, after, through, maximum) => Stream.paginate({
@@ -783,7 +783,7 @@ const makeThreadStore = Effect.gen(function* () {
 			after: request.afterSequence ?? ZERO_CANONICAL_SEQUENCE,
 			count: 0
 		}, (cursor) => Effect.gen(function* () {
-			const page = storedPage(thread, cursor.after, Math.min(READ_PAGE_SIZE, request.limit - cursor.count));
+			const page = yield* storedPage(thread, cursor.after, Math.min(READ_PAGE_SIZE, request.limit - cursor.count));
 			const next = {
 				after: page.at(-1)?.sequence ?? cursor.after,
 				count: cursor.count + page.length
@@ -1048,7 +1048,7 @@ const makeThreadStore = Effect.gen(function* () {
 	const verificationReader = makeExportReader(false);
 	const exportThread = (request) => exportThreadPage(request).pipe(Effect.provideService(ThreadExporterReader, exportReader));
 	const verificationPage = (request) => exportThreadPage(request).pipe(Effect.provideService(ThreadExporterReader, verificationReader));
-	const exactRunRecords = (thread, runId) => Effect.gen(function* () {
+	const exactRunRecords = Effect.fnUntraced(function* (thread, runId) {
 		const records = [];
 		let bytes = 0;
 		const facts = thread.runRecords.get(runId) ?? [];
@@ -1480,20 +1480,20 @@ const makeThreadStore = Effect.gen(function* () {
 		initializeWork: (threadId) => {
 			if (!MutableRef.get(state.ref).threads.has(threadId) && !MutableRef.get(workState.ref).has(threadId)) publishWork(threadId, emptyWorkIndex());
 		},
-		registerWorkOwner: (kind, owner) => withMutation(Effect.sync(() => {
-			if (workOwners.has(kind)) throw new Error(`MemoryThreadStore already has a ${kind} work owner`);
+		registerWorkOwner: (kind, owner) => withMutation(Effect.gen(function* () {
+			if (workOwners.has(kind)) return yield* Effect.die(storeError("initialize", `MemoryThreadStore already has a ${kind} work owner`));
 			workOwners.set(kind, owner);
 		})),
-		registerDeliveryLookup: (lookup) => withMutation(Effect.sync(() => {
-			if (deliveryLookup !== void 0) throw new Error("MemoryThreadStore already has a delivery lookup");
+		registerDeliveryLookup: (lookup) => withMutation(Effect.gen(function* () {
+			if (deliveryLookup !== void 0) return yield* Effect.die(storeError("initialize", "MemoryThreadStore already has a delivery lookup"));
 			deliveryLookup = lookup;
 		})),
-		registerLedgerTransfer: (transfer) => withMutation(Effect.sync(() => {
-			if (ledgerTransfer !== void 0) throw new Error("MemoryThreadStore already has a paired SubmissionLedger");
+		registerLedgerTransfer: (transfer) => withMutation(Effect.gen(function* () {
+			if (ledgerTransfer !== void 0) return yield* Effect.die(storeError("initialize", "MemoryThreadStore already has a paired SubmissionLedger"));
 			ledgerTransfer = transfer;
 		})),
-		registerDeliveryTransfer: (transfer) => withMutation(Effect.sync(() => {
-			if (deliveryTransfer !== void 0) throw new Error("MemoryThreadStore already has a paired delivery store");
+		registerDeliveryTransfer: (transfer) => withMutation(Effect.gen(function* () {
+			if (deliveryTransfer !== void 0) return yield* Effect.die(storeError("initialize", "MemoryThreadStore already has a paired delivery store"));
 			deliveryTransfer = transfer;
 		})),
 		prepareAppend,