@mastra/cloudflare 1.3.1 → 1.3.2-alpha.0

package/dist/{chunk-O57GFJSB.js → chunk-HJZCD37D.js}

@@ -1,8 +1,8 @@
 import { MastraError, ErrorCategory, ErrorDomain } from '@mastra/core/error';
-import { MemoryStorage, TABLE_MESSAGES, TABLE_THREADS, TABLE_RESOURCES, ensureDate, createStorageErrorId, normalizePerPage, calculatePagination, serializeDate, filterByDateRange, ScoresStorage, TABLE_SCORERS, WorkflowsStorage, TABLE_WORKFLOW_SNAPSHOT, MastraCompositeStore, TABLE_TRACES, transformScoreRow as transformScoreRow$1 } from '@mastra/core/storage';
+import { BackgroundTasksStorage, TABLE_BACKGROUND_TASKS, MemoryStorage, TABLE_MESSAGES, TABLE_THREADS, TABLE_RESOURCES, ensureDate, createStorageErrorId, normalizePerPage, calculatePagination, serializeDate, filterByDateRange, ScoresStorage, TABLE_SCORERS, WorkflowsStorage, TABLE_WORKFLOW_SNAPSHOT, MastraCompositeStore, TABLE_TRACES, transformScoreRow as transformScoreRow$1 } from '@mastra/core/storage';
 import Cloudflare from 'cloudflare';
-import { MessageList } from '@mastra/core/agent';
 import { MastraBase } from '@mastra/core/base';
+import { MessageList } from '@mastra/core/agent';
 import { saveScorePayloadSchema } from '@mastra/core/evals';
 
 // src/kv/index.ts
@@ -558,7 +558,131 @@ var CloudflareKVDB = class extends 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 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 ("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: 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 };
+    const records = await Promise.all(keys.map((k) => this.#db.getKV(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(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;
+  }
+};
 var MemoryStorageCloudflare = class extends MemoryStorage {
   #db;
   constructor(config) {
@@ -2186,7 +2310,13 @@ var CloudflareKVStorage = class extends MastraCompositeStore {
     if (!config.bindings) {
       throw new Error("KV bindings are required when using Workers Binding API");
     }
-    const requiredTables = [TABLE_THREADS, TABLE_MESSAGES, TABLE_WORKFLOW_SNAPSHOT, TABLE_SCORERS];
+    const requiredTables = [
+      TABLE_THREADS,
+      TABLE_MESSAGES,
+      TABLE_WORKFLOW_SNAPSHOT,
+      TABLE_SCORERS,
+      TABLE_BACKGROUND_TASKS
+    ];
     for (const table of requiredTables) {
       if (!(table in config.bindings)) {
         throw new Error(`Missing KV binding for table: ${table}`);
@@ -2210,6 +2340,7 @@ var CloudflareKVStorage = class extends MastraCompositeStore {
       let workflows;
       let memory;
       let scores;
+      let backgroundTasks;
       if (isWorkersConfig(config)) {
         this.validateWorkersConfig(config);
         this.bindings = config.bindings;
@@ -2222,6 +2353,7 @@ 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();
@@ -2238,11 +2370,13 @@ 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
+        scores,
+        backgroundTasks
       };
     } catch (error) {
       throw new MastraError(
@@ -2260,6 +2394,6 @@ var CloudflareKVStorage = class extends MastraCompositeStore {
 };
 var CloudflareStore = CloudflareKVStorage;
 
-export { CloudflareKVStorage, CloudflareStore, MemoryStorageCloudflare, ScoresStorageCloudflare, WorkflowsStorageCloudflare };
-//# sourceMappingURL=chunk-O57GFJSB.js.map
-//# sourceMappingURL=chunk-O57GFJSB.js.map
+export { BackgroundTasksStorageCloudflare, CloudflareKVStorage, CloudflareStore, MemoryStorageCloudflare, ScoresStorageCloudflare, WorkflowsStorageCloudflare };
+//# sourceMappingURL=chunk-HJZCD37D.js.map
+//# sourceMappingURL=chunk-HJZCD37D.js.map