@mastra/cloudflare 1.3.1 → 1.3.2-alpha.0

package/dist/{chunk-NSFHQATX.js → chunk-7THHWTAZ.js}

@@ -1,8 +1,8 @@
 import { MastraError, ErrorCategory, ErrorDomain } from '@mastra/core/error';
-import { createStorageErrorId, getSqlType, getDefaultValue, MemoryStorage, TABLE_SCHEMAS, TABLE_THREADS, TABLE_MESSAGES, TABLE_RESOURCES, ensureDate, normalizePerPage, calculatePagination, serializeDate, ScoresStorage, TABLE_SCORERS, WorkflowsStorage, TABLE_WORKFLOW_SNAPSHOT, MastraCompositeStore, transformScoreRow as transformScoreRow$1 } from '@mastra/core/storage';
-import { MessageList } from '@mastra/core/agent';
+import { createStorageErrorId, getSqlType, getDefaultValue, BackgroundTasksStorage, TABLE_SCHEMAS, TABLE_BACKGROUND_TASKS, MemoryStorage, TABLE_THREADS, TABLE_MESSAGES, TABLE_RESOURCES, ensureDate, normalizePerPage, calculatePagination, serializeDate, ScoresStorage, TABLE_SCORERS, WorkflowsStorage, TABLE_WORKFLOW_SNAPSHOT, MastraCompositeStore, transformScoreRow as transformScoreRow$1 } from '@mastra/core/storage';
 import { MastraBase } from '@mastra/core/base';
 import { parseSqlIdentifier } from '@mastra/core/utils';
+import { MessageList } from '@mastra/core/agent';
 import { saveScorePayloadSchema } from '@mastra/core/evals';
 
 // src/do/index.ts
@@ -657,7 +657,275 @@ var DODB = class extends MastraBase {
   }
 };
 
-// src/do/storage/domains/memory/index.ts
+// src/do/storage/domains/background-tasks/index.ts
+function serializeJson(v) {
+  if (typeof v === "object" && v != null) return JSON.stringify(v);
+  return v ?? void 0;
+}
+function rowToTask(row) {
+  const parseJson = (v) => {
+    if (v == null) return void 0;
+    if (typeof v === "string") {
+      try {
+        return JSON.parse(v);
+      } catch {
+        return v;
+      }
+    }
+    return v;
+  };
+  const asDate = (v) => v ? new Date(String(v)) : void 0;
+  return {
+    id: String(row.id),
+    status: String(row.status),
+    toolName: String(row.tool_name),
+    toolCallId: String(row.tool_call_id),
+    args: parseJson(row.args) ?? {},
+    agentId: String(row.agent_id),
+    threadId: row.thread_id != null ? String(row.thread_id) : void 0,
+    resourceId: row.resource_id != null ? String(row.resource_id) : void 0,
+    runId: String(row.run_id),
+    result: parseJson(row.result),
+    error: parseJson(row.error),
+    retryCount: Number(row.retry_count ?? 0),
+    maxRetries: Number(row.max_retries ?? 0),
+    timeoutMs: Number(row.timeout_ms ?? 3e5),
+    createdAt: asDate(row.createdAt) ?? /* @__PURE__ */ new Date(),
+    startedAt: asDate(row.startedAt),
+    completedAt: asDate(row.completedAt)
+  };
+}
+function dateColumnName(col) {
+  return col;
+}
+var BackgroundTasksStorageDO = class extends BackgroundTasksStorage {
+  #db;
+  constructor(config) {
+    super();
+    this.#db = new DODB(config);
+  }
+  async init() {
+    await this.#db.createTable({
+      tableName: TABLE_BACKGROUND_TASKS,
+      schema: TABLE_SCHEMAS[TABLE_BACKGROUND_TASKS]
+    });
+  }
+  async dangerouslyClearAll() {
+    await this.#db.clearTable({ tableName: TABLE_BACKGROUND_TASKS });
+  }
+  async createTask(task) {
+    try {
+      await this.#db.insert({
+        tableName: TABLE_BACKGROUND_TASKS,
+        record: {
+          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: serializeJson(task.args),
+          result: serializeJson(task.result),
+          error: serializeJson(task.error),
+          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
+        }
+      });
+    } catch (error) {
+      throw new MastraError(
+        {
+          id: createStorageErrorId("CLOUDFLARE_DO", "BACKGROUND_TASKS_CREATE", "FAILED"),
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY
+        },
+        error
+      );
+    }
+  }
+  async updateTask(taskId, update) {
+    const columns = [];
+    const values = [];
+    if ("status" in update) {
+      columns.push("status");
+      values.push(update.status);
+    }
+    if ("result" in update) {
+      columns.push("result");
+      values.push(serializeJson(update.result));
+    }
+    if ("error" in update) {
+      columns.push("error");
+      values.push(serializeJson(update.error));
+    }
+    if ("retryCount" in update) {
+      columns.push("retry_count");
+      values.push(update.retryCount);
+    }
+    if ("startedAt" in update) {
+      columns.push("startedAt");
+      values.push(update.startedAt?.toISOString() ?? null);
+    }
+    if ("completedAt" in update) {
+      columns.push("completedAt");
+      values.push(update.completedAt?.toISOString() ?? null);
+    }
+    if (columns.length === 0) return;
+    try {
+      const fullTableName = this.#db.getTableName(TABLE_BACKGROUND_TASKS);
+      const query = createSqlBuilder().update(fullTableName, columns, values).where("id = ?", taskId);
+      const { sql, params } = query.build();
+      await this.#db.executeQuery({ sql, params });
+    } catch (error) {
+      throw new MastraError(
+        {
+          id: createStorageErrorId("CLOUDFLARE_DO", "BACKGROUND_TASKS_UPDATE", "FAILED"),
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY
+        },
+        error
+      );
+    }
+  }
+  async getTask(taskId) {
+    try {
+      const fullTableName = this.#db.getTableName(TABLE_BACKGROUND_TASKS);
+      const query = createSqlBuilder().select("*").from(fullTableName).where("id = ?", taskId);
+      const { sql, params } = query.build();
+      const result = await this.#db.executeQuery({ sql, params, first: true });
+      if (!result) return null;
+      const deserialized = {};
+      for (const [k, v] of Object.entries(result)) {
+        deserialized[k] = deserializeValue(v);
+      }
+      return rowToTask(deserialized);
+    } catch (error) {
+      throw new MastraError(
+        {
+          id: createStorageErrorId("CLOUDFLARE_DO", "BACKGROUND_TASKS_GET", "FAILED"),
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY
+        },
+        error
+      );
+    }
+  }
+  async listTasks(filter) {
+    try {
+      const fullTableName = this.#db.getTableName(TABLE_BACKGROUND_TASKS);
+      const applyConditions = (builder) => {
+        if (filter.status) {
+          const statuses = Array.isArray(filter.status) ? filter.status : [filter.status];
+          const placeholders = statuses.map(() => "?").join(", ");
+          builder.whereAnd(`status IN (${placeholders})`, ...statuses);
+        }
+        if (filter.agentId) builder.whereAnd("agent_id = ?", filter.agentId);
+        if (filter.threadId) builder.whereAnd("thread_id = ?", filter.threadId);
+        if (filter.resourceId) builder.whereAnd("resource_id = ?", filter.resourceId);
+        if (filter.toolName) builder.whereAnd("tool_name = ?", filter.toolName);
+        if (filter.runId) builder.whereAnd("run_id = ?", filter.runId);
+        const dateCol = dateColumnName(filter.dateFilterBy ?? "createdAt");
+        if (filter.fromDate) builder.whereAnd(`${dateCol} >= ?`, filter.fromDate.toISOString());
+        if (filter.toDate) builder.whereAnd(`${dateCol} < ?`, filter.toDate.toISOString());
+      };
+      const countQuery = createSqlBuilder().count().from(fullTableName);
+      applyConditions(countQuery);
+      const { sql: countSql, params: countParams } = countQuery.build();
+      const countResult = await this.#db.executeQuery({ sql: countSql, params: countParams, first: true });
+      const total = Number(countResult?.count ?? 0);
+      const dataQuery = createSqlBuilder().select("*").from(fullTableName);
+      applyConditions(dataQuery);
+      const orderBy = dateColumnName(filter.orderBy ?? "createdAt");
+      const direction = filter.orderDirection === "desc" ? "DESC" : "ASC";
+      dataQuery.orderBy(orderBy, direction);
+      if (filter.perPage != null) {
+        dataQuery.limit(filter.perPage);
+        if (filter.page != null) {
+          dataQuery.offset(filter.page * filter.perPage);
+        }
+      }
+      const { sql, params } = dataQuery.build();
+      const rows = await this.#db.executeQuery({ sql, params });
+      const tasks = (rows ?? []).map((row) => {
+        const deserialized = {};
+        for (const [k, v] of Object.entries(row)) {
+          deserialized[k] = deserializeValue(v);
+        }
+        return rowToTask(deserialized);
+      });
+      return { tasks, total };
+    } catch (error) {
+      throw new MastraError(
+        {
+          id: createStorageErrorId("CLOUDFLARE_DO", "BACKGROUND_TASKS_LIST", "FAILED"),
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY
+        },
+        error
+      );
+    }
+  }
+  async deleteTask(taskId) {
+    try {
+      const fullTableName = this.#db.getTableName(TABLE_BACKGROUND_TASKS);
+      const query = createSqlBuilder().delete(fullTableName).where("id = ?", taskId);
+      const { sql, params } = query.build();
+      await this.#db.executeQuery({ sql, params });
+    } catch (error) {
+      throw new MastraError(
+        {
+          id: createStorageErrorId("CLOUDFLARE_DO", "BACKGROUND_TASKS_DELETE", "FAILED"),
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY
+        },
+        error
+      );
+    }
+  }
+  async deleteTasks(filter) {
+    try {
+      const fullTableName = this.#db.getTableName(TABLE_BACKGROUND_TASKS);
+      const query = createSqlBuilder().delete(fullTableName);
+      if (filter.status) {
+        const statuses = Array.isArray(filter.status) ? filter.status : [filter.status];
+        const placeholders = statuses.map(() => "?").join(", ");
+        query.whereAnd(`status IN (${placeholders})`, ...statuses);
+      }
+      if (filter.agentId) query.whereAnd("agent_id = ?", filter.agentId);
+      if (filter.threadId) query.whereAnd("thread_id = ?", filter.threadId);
+      if (filter.resourceId) query.whereAnd("resource_id = ?", filter.resourceId);
+      if (filter.toolName) query.whereAnd("tool_name = ?", filter.toolName);
+      if (filter.runId) query.whereAnd("run_id = ?", filter.runId);
+      const dateCol = dateColumnName(filter.dateFilterBy ?? "createdAt");
+      if (filter.fromDate) query.whereAnd(`${dateCol} >= ?`, filter.fromDate.toISOString());
+      if (filter.toDate) query.whereAnd(`${dateCol} < ?`, filter.toDate.toISOString());
+      const { sql, params } = query.build();
+      await this.#db.executeQuery({ sql, params });
+    } catch (error) {
+      throw new MastraError(
+        {
+          id: createStorageErrorId("CLOUDFLARE_DO", "BACKGROUND_TASKS_DELETE_MANY", "FAILED"),
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY
+        },
+        error
+      );
+    }
+  }
+  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 MemoryStorageDO = class extends MemoryStorage {
   #db;
   constructor(config) {
@@ -2098,7 +2366,8 @@ var CloudflareDOStorage = class extends MastraCompositeStore {
       this.stores = {
         memory: new MemoryStorageDO(domainConfig),
         workflows: new WorkflowsStorageDO(domainConfig),
-        scores: new ScoresStorageDO(domainConfig)
+        scores: new ScoresStorageDO(domainConfig),
+        backgroundTasks: new BackgroundTasksStorageDO(domainConfig)
       };
       this.logger.info("Using Durable Objects SqlStorage");
     } catch (error) {
@@ -2123,6 +2392,6 @@ var CloudflareDOStorage = class extends MastraCompositeStore {
 };
 var DOStore = CloudflareDOStorage;
 
-export { CloudflareDOStorage, DODB, DOStore, MemoryStorageDO, ScoresStorageDO, WorkflowsStorageDO };
-//# sourceMappingURL=chunk-NSFHQATX.js.map
-//# sourceMappingURL=chunk-NSFHQATX.js.map
+export { BackgroundTasksStorageDO, CloudflareDOStorage, DODB, DOStore, MemoryStorageDO, ScoresStorageDO, WorkflowsStorageDO };
+//# sourceMappingURL=chunk-7THHWTAZ.js.map
+//# sourceMappingURL=chunk-7THHWTAZ.js.map