@mastra/cloudflare 1.6.2 → 1.6.3-alpha.1

package/dist/{kv-DRnLtFWJ.js → kv-FPgTjsj1.js}

@@ -1,8 +1,8 @@
 import { ErrorCategory, ErrorDomain, MastraError } from "@mastra/core/error";
-import { BackgroundTasksStorage, MastraCompositeStore, MemoryStorage, ScoresStorage, TABLE_BACKGROUND_TASKS, TABLE_MESSAGES, TABLE_RESOURCES, TABLE_SCORERS, TABLE_THREADS, TABLE_TRACES, TABLE_WORKFLOW_SNAPSHOT, WorkflowsStorage, calculatePagination, createStorageErrorId, ensureDate, filterByDateRange, normalizePerPage, serializeDate, storageMessageMatchesMetadataFilter, transformScoreRow, validateStorageMetadataFilter } from "@mastra/core/storage";
+import { MastraCompositeStore, MemoryStorage, ScoresStorage, TABLE_MESSAGES, TABLE_RESOURCES, TABLE_SCORERS, TABLE_THREADS, TABLE_TRACES, TABLE_WORKFLOW_SNAPSHOT, WorkflowsStorage, calculatePagination, createStorageErrorId, ensureDate, filterByDateRange, normalizePerPage, serializeDate, storageMessageMatchesMetadataFilter, transformScoreRow, validateStorageMetadataFilter } from "@mastra/core/storage";
 import Cloudflare from "cloudflare";
-import { MastraBase } from "@mastra/core/base";
 import { MessageList } from "@mastra/core/agent";
+import { MastraBase } from "@mastra/core/base";
 import { saveScorePayloadSchema } from "@mastra/core/evals";
 //#region src/kv/storage/db/index.ts
 /**
@@ -348,7 +348,7 @@ var CloudflareKVDB = class extends MastraBase {
 				await this.client.kv.namespaces.values.update(namespaceId, key, {
 					account_id: this.accountId,
 					value: serializedValue,
-					metadata: serializedMetadata
+					...serializedMetadata ? { metadata: serializedMetadata } : {}
 				});
 			}
 		} catch (error) {
@@ -421,18 +421,35 @@ var CloudflareKVDB = class extends MastraBase {
 	}
 	async listNamespaceKeys(tableName, options) {
 		try {
-			if (this.bindings) return (await this.getBinding(tableName).list({
-				limit: options?.limit || 1e3,
-				prefix: options?.prefix
-			})).keys;
-			else {
+			const pageSize = options?.limit || 1e3;
+			const keys = [];
+			if (this.bindings) {
+				const binding = this.getBinding(tableName);
+				let cursor;
+				do {
+					const response = await binding.list({
+						limit: pageSize,
+						prefix: options?.prefix,
+						cursor
+					});
+					keys.push(...response.keys);
+					cursor = response.list_complete ? void 0 : response.cursor;
+				} while (cursor);
+			} else {
 				const namespaceId = await this.getNamespaceId(tableName);
-				return (await this.client.kv.namespaces.keys.list(namespaceId, {
-					account_id: this.accountId,
-					limit: options?.limit || 1e3,
-					prefix: options?.prefix
-				})).result;
+				let cursor;
+				do {
+					const page = await this.client.kv.namespaces.keys.list(namespaceId, {
+						account_id: this.accountId,
+						limit: pageSize,
+						prefix: options?.prefix,
+						...cursor ? { cursor } : {}
+					});
+					keys.push(...page.result);
+					cursor = page.result_info?.cursor || void 0;
+				} while (cursor);
 			}
+			return keys;
 		} catch (error) {
 			throw new MastraError({
 				id: createStorageErrorId("CLOUDFLARE", "LIST_NAMESPACE_KEYS", "FAILED"),
@@ -467,149 +484,6 @@ var CloudflareKVDB = class extends MastraBase {
 	}
 };
 //#endregion
-//#region src/kv/storage/domains/background-tasks/index.ts
-function toRecord(task) {
-	return {
-		id: task.id,
-		tool_call_id: task.toolCallId,
-		tool_name: task.toolName,
-		agent_id: task.agentId,
-		thread_id: task.threadId ?? null,
-		resource_id: task.resourceId ?? null,
-		run_id: task.runId,
-		status: task.status,
-		args: task.args,
-		result: task.result ?? null,
-		error: task.error ?? null,
-		suspend_payload: task.suspendPayload ?? null,
-		retry_count: task.retryCount,
-		max_retries: task.maxRetries,
-		timeout_ms: task.timeoutMs,
-		createdAt: task.createdAt.toISOString(),
-		startedAt: task.startedAt?.toISOString() ?? null,
-		suspendedAt: task.suspendedAt?.toISOString() ?? null,
-		completedAt: task.completedAt?.toISOString() ?? null
-	};
-}
-function fromRecord(record) {
-	return {
-		id: record.id,
-		status: record.status,
-		toolName: record.tool_name,
-		toolCallId: record.tool_call_id,
-		args: record.args ?? {},
-		agentId: record.agent_id,
-		threadId: record.thread_id ?? void 0,
-		resourceId: record.resource_id ?? void 0,
-		runId: record.run_id ?? "",
-		result: record.result ?? void 0,
-		error: record.error ?? void 0,
-		suspendPayload: record.suspend_payload ?? void 0,
-		retryCount: Number(record.retry_count ?? 0),
-		maxRetries: Number(record.max_retries ?? 0),
-		timeoutMs: Number(record.timeout_ms ?? 3e5),
-		createdAt: new Date(record.createdAt),
-		startedAt: record.startedAt ? new Date(record.startedAt) : void 0,
-		suspendedAt: record.suspendedAt ? new Date(record.suspendedAt) : void 0,
-		completedAt: record.completedAt ? new Date(record.completedAt) : void 0
-	};
-}
-var BackgroundTasksStorageCloudflare = class extends BackgroundTasksStorage {
-	#db;
-	constructor(config) {
-		super();
-		this.#db = new CloudflareKVDB(resolveCloudflareConfig(config));
-	}
-	async dangerouslyClearAll() {
-		await this.#db.clearTable({ tableName: TABLE_BACKGROUND_TASKS });
-	}
-	async createTask(task) {
-		await this.#db.putKV({
-			tableName: TABLE_BACKGROUND_TASKS,
-			key: task.id,
-			value: toRecord(task)
-		});
-	}
-	async updateTask(taskId, update) {
-		const existing = await this.getTask(taskId);
-		if (!existing) return;
-		const merged = { ...existing };
-		if ("status" in update) merged.status = update.status;
-		if ("result" in update) merged.result = update.result;
-		if ("error" in update) merged.error = update.error;
-		if ("suspendPayload" in update) merged.suspendPayload = update.suspendPayload;
-		if ("retryCount" in update) merged.retryCount = update.retryCount;
-		if ("startedAt" in update) merged.startedAt = update.startedAt;
-		if ("suspendedAt" in update) merged.suspendedAt = update.suspendedAt;
-		if ("completedAt" in update) merged.completedAt = update.completedAt;
-		await this.#db.putKV({
-			tableName: TABLE_BACKGROUND_TASKS,
-			key: taskId,
-			value: toRecord(merged)
-		});
-	}
-	async getTask(taskId) {
-		const data = await this.#db.getKV(TABLE_BACKGROUND_TASKS, taskId);
-		return data ? fromRecord(data) : null;
-	}
-	async listTasks(filter) {
-		const keys = await this.#db.listKV(TABLE_BACKGROUND_TASKS);
-		if (keys.length === 0) return {
-			tasks: [],
-			total: 0
-		};
-		let tasks = (await Promise.all(keys.map((k) => this.#db.getKV(TABLE_BACKGROUND_TASKS, k.name)))).filter(Boolean).map((r) => fromRecord(r));
-		if (filter.status) {
-			const s = Array.isArray(filter.status) ? filter.status : [filter.status];
-			tasks = tasks.filter((t) => s.includes(t.status));
-		}
-		if (filter.agentId) tasks = tasks.filter((t) => t.agentId === filter.agentId);
-		if (filter.threadId) tasks = tasks.filter((t) => t.threadId === filter.threadId);
-		if (filter.toolName) tasks = tasks.filter((t) => t.toolName === filter.toolName);
-		if (filter.toolCallId) tasks = tasks.filter((t) => t.toolCallId === filter.toolCallId);
-		if (filter.runId) tasks = tasks.filter((t) => t.runId === filter.runId);
-		const dateCol = filter.dateFilterBy ?? "createdAt";
-		if (filter.fromDate) tasks = tasks.filter((t) => {
-			const val = t[dateCol];
-			return val != null && val >= filter.fromDate;
-		});
-		if (filter.toDate) tasks = tasks.filter((t) => {
-			const val = t[dateCol];
-			return val != null && val < filter.toDate;
-		});
-		const orderBy = filter.orderBy ?? "createdAt";
-		const dir = filter.orderDirection === "desc" ? -1 : 1;
-		tasks.sort((a, b) => ((a[orderBy]?.getTime() ?? 0) - (b[orderBy]?.getTime() ?? 0)) * dir);
-		const total = tasks.length;
-		if (filter.page != null && filter.perPage != null) {
-			const start = filter.page * filter.perPage;
-			tasks = tasks.slice(start, start + filter.perPage);
-		} else if (filter.perPage != null) tasks = tasks.slice(0, filter.perPage);
-		return {
-			tasks,
-			total
-		};
-	}
-	async deleteTask(taskId) {
-		await this.#db.deleteKV(TABLE_BACKGROUND_TASKS, taskId);
-	}
-	async deleteTasks(filter) {
-		const { tasks } = await this.listTasks(filter);
-		await Promise.all(tasks.map((t) => this.#db.deleteKV(TABLE_BACKGROUND_TASKS, t.id)));
-	}
-	async getRunningCount() {
-		const { total } = await this.listTasks({ status: "running" });
-		return total;
-	}
-	async getRunningCountByAgent(agentId) {
-		const { total } = await this.listTasks({
-			status: "running",
-			agentId
-		});
-		return total;
-	}
-};
-//#endregion
 //#region src/kv/storage/domains/memory/index.ts
 var MemoryStorageCloudflare = class extends MemoryStorage {
 	supportsPartialThreadUpdate = true;
@@ -802,7 +676,8 @@ var MemoryStorageCloudflare = class extends MemoryStorage {
 	async deleteThread({ threadId }) {
 		try {
 			if (!await this.getThreadById({ threadId })) throw new Error(`Thread ${threadId} not found`);
-			const threadMessageKeys = (await this.#db.listKV(TABLE_MESSAGES)).filter((key) => key.name.includes(`${TABLE_MESSAGES}:${threadId}:`));
+			const prefix = this.#db.namespacePrefix ? `${this.#db.namespacePrefix}:` : "";
+			const threadMessageKeys = (await this.#db.listKV(TABLE_MESSAGES, { prefix: `${prefix}${TABLE_MESSAGES}:${threadId}:` })).filter((key) => key.name.includes(`${TABLE_MESSAGES}:${threadId}:`));
 			await Promise.all([
 				this.#db.deleteKV(TABLE_MESSAGES, this.getThreadMessagesKey(threadId)),
 				...threadMessageKeys.map((key) => this.#db.deleteKV(TABLE_MESSAGES, key.name)),
@@ -2083,8 +1958,7 @@ var CloudflareKVStorage = class extends MastraCompositeStore {
 			TABLE_THREADS,
 			TABLE_MESSAGES,
 			TABLE_WORKFLOW_SNAPSHOT,
-			TABLE_SCORERS,
-			TABLE_BACKGROUND_TASKS
+			TABLE_SCORERS
 		];
 		for (const table of requiredTables) if (!(table in config.bindings)) throw new Error(`Missing KV binding for table: ${table}`);
 	}
@@ -2103,7 +1977,6 @@ var CloudflareKVStorage = class extends MastraCompositeStore {
 			let workflows;
 			let memory;
 			let scores;
-			let backgroundTasks;
 			if (isWorkersConfig(config)) {
 				this.validateWorkersConfig(config);
 				this.bindings = config.bindings;
@@ -2116,7 +1989,6 @@ var CloudflareKVStorage = class extends MastraCompositeStore {
 				workflows = new WorkflowsStorageCloudflare(domainConfig);
 				memory = new MemoryStorageCloudflare(domainConfig);
 				scores = new ScoresStorageCloudflare(domainConfig);
-				backgroundTasks = new BackgroundTasksStorageCloudflare(domainConfig);
 			} else {
 				this.validateRestConfig(config);
 				this.accountId = config.accountId.trim();
@@ -2131,13 +2003,11 @@ var CloudflareKVStorage = class extends MastraCompositeStore {
 				workflows = new WorkflowsStorageCloudflare(domainConfig);
 				memory = new MemoryStorageCloudflare(domainConfig);
 				scores = new ScoresStorageCloudflare(domainConfig);
-				backgroundTasks = new BackgroundTasksStorageCloudflare(domainConfig);
 			}
 			this.stores = {
 				workflows,
 				memory,
-				scores,
-				backgroundTasks
+				scores
 			};
 		} catch (error) {
 			throw new MastraError({
@@ -2154,6 +2024,6 @@ var CloudflareKVStorage = class extends MastraCompositeStore {
 */
 const CloudflareStore = CloudflareKVStorage;
 //#endregion
-export { MemoryStorageCloudflare as a, ScoresStorageCloudflare as i, CloudflareStore as n, BackgroundTasksStorageCloudflare as o, WorkflowsStorageCloudflare as r, CloudflareKVStorage as t };
+export { MemoryStorageCloudflare as a, ScoresStorageCloudflare as i, CloudflareStore as n, WorkflowsStorageCloudflare as r, CloudflareKVStorage as t };
 
-//# sourceMappingURL=kv-DRnLtFWJ.js.map
+//# sourceMappingURL=kv-FPgTjsj1.js.map