@mastra/cloudflare 1.6.2 → 1.6.3-alpha.1

package/dist/{kv-Dxgei0aW.cjs → kv-CVZicPPC.cjs}

@@ -24,8 +24,8 @@ let _mastra_core_error = require("@mastra/core/error");
 let _mastra_core_storage = require("@mastra/core/storage");
 let cloudflare = require("cloudflare");
 cloudflare = __toESM(cloudflare, 1);
-let _mastra_core_base = require("@mastra/core/base");
 let _mastra_core_agent = require("@mastra/core/agent");
+let _mastra_core_base = require("@mastra/core/base");
 let _mastra_core_evals = require("@mastra/core/evals");
 //#region src/kv/storage/db/index.ts
 /**
@@ -371,7 +371,7 @@ var CloudflareKVDB = class extends _mastra_core_base.MastraBase {
 				await this.client.kv.namespaces.values.update(namespaceId, key, {
 					account_id: this.accountId,
 					value: serializedValue,
-					metadata: serializedMetadata
+					...serializedMetadata ? { metadata: serializedMetadata } : {}
 				});
 			}
 		} catch (error) {
@@ -444,18 +444,35 @@ var CloudflareKVDB = class extends _mastra_core_base.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 _mastra_core_error.MastraError({
 				id: (0, _mastra_core_storage.createStorageErrorId)("CLOUDFLARE", "LIST_NAMESPACE_KEYS", "FAILED"),
@@ -490,149 +507,6 @@ var CloudflareKVDB = class extends _mastra_core_base.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 _mastra_core_storage.BackgroundTasksStorage {
-	#db;
-	constructor(config) {
-		super();
-		this.#db = new CloudflareKVDB(resolveCloudflareConfig(config));
-	}
-	async dangerouslyClearAll() {
-		await this.#db.clearTable({ tableName: _mastra_core_storage.TABLE_BACKGROUND_TASKS });
-	}
-	async createTask(task) {
-		await this.#db.putKV({
-			tableName: _mastra_core_storage.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: _mastra_core_storage.TABLE_BACKGROUND_TASKS,
-			key: taskId,
-			value: toRecord(merged)
-		});
-	}
-	async getTask(taskId) {
-		const data = await this.#db.getKV(_mastra_core_storage.TABLE_BACKGROUND_TASKS, taskId);
-		return data ? fromRecord(data) : null;
-	}
-	async listTasks(filter) {
-		const keys = await this.#db.listKV(_mastra_core_storage.TABLE_BACKGROUND_TASKS);
-		if (keys.length === 0) return {
-			tasks: [],
-			total: 0
-		};
-		let tasks = (await Promise.all(keys.map((k) => this.#db.getKV(_mastra_core_storage.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(_mastra_core_storage.TABLE_BACKGROUND_TASKS, taskId);
-	}
-	async deleteTasks(filter) {
-		const { tasks } = await this.listTasks(filter);
-		await Promise.all(tasks.map((t) => this.#db.deleteKV(_mastra_core_storage.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 _mastra_core_storage.MemoryStorage {
 	supportsPartialThreadUpdate = true;
@@ -825,7 +699,8 @@ var MemoryStorageCloudflare = class extends _mastra_core_storage.MemoryStorage {
 	async deleteThread({ threadId }) {
 		try {
 			if (!await this.getThreadById({ threadId })) throw new Error(`Thread ${threadId} not found`);
-			const threadMessageKeys = (await this.#db.listKV(_mastra_core_storage.TABLE_MESSAGES)).filter((key) => key.name.includes(`${_mastra_core_storage.TABLE_MESSAGES}:${threadId}:`));
+			const prefix = this.#db.namespacePrefix ? `${this.#db.namespacePrefix}:` : "";
+			const threadMessageKeys = (await this.#db.listKV(_mastra_core_storage.TABLE_MESSAGES, { prefix: `${prefix}${_mastra_core_storage.TABLE_MESSAGES}:${threadId}:` })).filter((key) => key.name.includes(`${_mastra_core_storage.TABLE_MESSAGES}:${threadId}:`));
 			await Promise.all([
 				this.#db.deleteKV(_mastra_core_storage.TABLE_MESSAGES, this.getThreadMessagesKey(threadId)),
 				...threadMessageKeys.map((key) => this.#db.deleteKV(_mastra_core_storage.TABLE_MESSAGES, key.name)),
@@ -2106,8 +1981,7 @@ var CloudflareKVStorage = class extends _mastra_core_storage.MastraCompositeStor
 			_mastra_core_storage.TABLE_THREADS,
 			_mastra_core_storage.TABLE_MESSAGES,
 			_mastra_core_storage.TABLE_WORKFLOW_SNAPSHOT,
-			_mastra_core_storage.TABLE_SCORERS,
-			_mastra_core_storage.TABLE_BACKGROUND_TASKS
+			_mastra_core_storage.TABLE_SCORERS
 		];
 		for (const table of requiredTables) if (!(table in config.bindings)) throw new Error(`Missing KV binding for table: ${table}`);
 	}
@@ -2126,7 +2000,6 @@ var CloudflareKVStorage = class extends _mastra_core_storage.MastraCompositeStor
 			let workflows;
 			let memory;
 			let scores;
-			let backgroundTasks;
 			if (isWorkersConfig(config)) {
 				this.validateWorkersConfig(config);
 				this.bindings = config.bindings;
@@ -2139,7 +2012,6 @@ var CloudflareKVStorage = class extends _mastra_core_storage.MastraCompositeStor
 				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();
@@ -2154,13 +2026,11 @@ var CloudflareKVStorage = class extends _mastra_core_storage.MastraCompositeStor
 				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 _mastra_core_error.MastraError({
@@ -2177,12 +2047,6 @@ var CloudflareKVStorage = class extends _mastra_core_storage.MastraCompositeStor
 */
 const CloudflareStore = CloudflareKVStorage;
 //#endregion
-Object.defineProperty(exports, "BackgroundTasksStorageCloudflare", {
-	enumerable: true,
-	get: function() {
-		return BackgroundTasksStorageCloudflare;
-	}
-});
 Object.defineProperty(exports, "CloudflareKVStorage", {
 	enumerable: true,
 	get: function() {
@@ -2214,4 +2078,4 @@ Object.defineProperty(exports, "WorkflowsStorageCloudflare", {
 	}
 });
 
-//# sourceMappingURL=kv-Dxgei0aW.cjs.map
+//# sourceMappingURL=kv-CVZicPPC.cjs.map