@mastra/cloudflare 1.3.1 → 1.3.2-alpha.0

package/CHANGELOG.md

@@ -1,5 +1,14 @@
 # @mastra/cloudflare
 
+## 1.3.2-alpha.0
+
+### Patch Changes
+
+- Add `BackgroundTasksStorage` domain implementation so `@mastra/core` background task execution works with any storage adapter. ([#15307](https://github.com/mastra-ai/mastra/pull/15307))
+
+- Updated dependencies [[`d63ffdb`](https://github.com/mastra-ai/mastra/commit/d63ffdbb2c11e76fe5ea45faab44bc15460f010c)]:
+  - @mastra/core@1.25.1-alpha.0
+
 ## 1.3.1
 
 ### Patch Changes

package/dist/{chunk-ZBYNKKG6.cjs → chunk-3SMRLX2Z.cjs}

@@ -3,8 +3,8 @@
 var error = require('@mastra/core/error');
 var storage = require('@mastra/core/storage');
 var Cloudflare = require('cloudflare');
-var agent = require('@mastra/core/agent');
 var base = require('@mastra/core/base');
+var agent = require('@mastra/core/agent');
 var evals = require('@mastra/core/evals');
 
 function _interopDefault (e) { return e && e.__esModule ? e : { default: e }; }
@@ -564,7 +564,131 @@ var CloudflareKVDB = class extends base.MastraBase {
   }
 };
 
-// src/kv/storage/domains/memory/index.ts
+// 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,
+    retry_count: task.retryCount,
+    max_retries: task.maxRetries,
+    timeout_ms: task.timeoutMs,
+    createdAt: task.createdAt.toISOString(),
+    startedAt: task.startedAt?.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,
+    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,
+    completedAt: record.completedAt ? new Date(record.completedAt) : void 0
+  };
+}
+var BackgroundTasksStorageCloudflare = class extends storage.BackgroundTasksStorage {
+  #db;
+  constructor(config) {
+    super();
+    this.#db = new CloudflareKVDB(resolveCloudflareConfig(config));
+  }
+  async dangerouslyClearAll() {
+    await this.#db.clearTable({ tableName: storage.TABLE_BACKGROUND_TASKS });
+  }
+  async createTask(task) {
+    await this.#db.putKV({ tableName: 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 ("retryCount" in update) merged.retryCount = update.retryCount;
+    if ("startedAt" in update) merged.startedAt = update.startedAt;
+    if ("completedAt" in update) merged.completedAt = update.completedAt;
+    await this.#db.putKV({ tableName: storage.TABLE_BACKGROUND_TASKS, key: taskId, value: toRecord(merged) });
+  }
+  async getTask(taskId) {
+    const data = await this.#db.getKV(storage.TABLE_BACKGROUND_TASKS, taskId);
+    return data ? fromRecord(data) : null;
+  }
+  async listTasks(filter) {
+    const keys = await this.#db.listKV(storage.TABLE_BACKGROUND_TASKS);
+    if (keys.length === 0) return { tasks: [], total: 0 };
+    const records = await Promise.all(keys.map((k) => this.#db.getKV(storage.TABLE_BACKGROUND_TASKS, k.name)));
+    let tasks = records.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.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(storage.TABLE_BACKGROUND_TASKS, taskId);
+  }
+  async deleteTasks(filter) {
+    const { tasks } = await this.listTasks(filter);
+    await Promise.all(tasks.map((t) => this.#db.deleteKV(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;
+  }
+};
 var MemoryStorageCloudflare = class extends storage.MemoryStorage {
   #db;
   constructor(config) {
@@ -2192,7 +2316,13 @@ var CloudflareKVStorage = class extends storage.MastraCompositeStore {
     if (!config.bindings) {
       throw new Error("KV bindings are required when using Workers Binding API");
     }
-    const requiredTables = [storage.TABLE_THREADS, storage.TABLE_MESSAGES, storage.TABLE_WORKFLOW_SNAPSHOT, storage.TABLE_SCORERS];
+    const requiredTables = [
+      storage.TABLE_THREADS,
+      storage.TABLE_MESSAGES,
+      storage.TABLE_WORKFLOW_SNAPSHOT,
+      storage.TABLE_SCORERS,
+      storage.TABLE_BACKGROUND_TASKS
+    ];
     for (const table of requiredTables) {
       if (!(table in config.bindings)) {
         throw new Error(`Missing KV binding for table: ${table}`);
@@ -2216,6 +2346,7 @@ var CloudflareKVStorage = class extends storage.MastraCompositeStore {
       let workflows;
       let memory;
       let scores;
+      let backgroundTasks;
       if (isWorkersConfig(config)) {
         this.validateWorkersConfig(config);
         this.bindings = config.bindings;
@@ -2228,6 +2359,7 @@ var CloudflareKVStorage = class extends storage.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();
@@ -2244,11 +2376,13 @@ var CloudflareKVStorage = class extends storage.MastraCompositeStore {
         workflows = new WorkflowsStorageCloudflare(domainConfig);
         memory = new MemoryStorageCloudflare(domainConfig);
         scores = new ScoresStorageCloudflare(domainConfig);
+        backgroundTasks = new BackgroundTasksStorageCloudflare(domainConfig);
       }
       this.stores = {
         workflows,
         memory,
-        scores
+        scores,
+        backgroundTasks
       };
     } catch (error$1) {
       throw new error.MastraError(
@@ -2266,10 +2400,11 @@ var CloudflareKVStorage = class extends storage.MastraCompositeStore {
 };
 var CloudflareStore = CloudflareKVStorage;
 
+exports.BackgroundTasksStorageCloudflare = BackgroundTasksStorageCloudflare;
 exports.CloudflareKVStorage = CloudflareKVStorage;
 exports.CloudflareStore = CloudflareStore;
 exports.MemoryStorageCloudflare = MemoryStorageCloudflare;
 exports.ScoresStorageCloudflare = ScoresStorageCloudflare;
 exports.WorkflowsStorageCloudflare = WorkflowsStorageCloudflare;
-//# sourceMappingURL=chunk-ZBYNKKG6.cjs.map
-//# sourceMappingURL=chunk-ZBYNKKG6.cjs.map
+//# sourceMappingURL=chunk-3SMRLX2Z.cjs.map
+//# sourceMappingURL=chunk-3SMRLX2Z.cjs.map