@mastra/mongodb 1.7.1 → 1.7.2-alpha.0

package/CHANGELOG.md

@@ -1,5 +1,14 @@
 # @mastra/mongodb
 
+## 1.7.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.7.1
 
 ### Patch Changes

package/dist/docs/SKILL.md

@@ -3,7 +3,7 @@ name: mastra-mongodb
 description: Documentation for @mastra/mongodb. Use when working with @mastra/mongodb APIs, configuration, or implementation.
 metadata:
   package: "@mastra/mongodb"
-  version: "1.7.1"
+  version: "1.7.2-alpha.0"
 ---
 
 ## When to use

package/dist/docs/assets/SOURCE_MAP.json

@@ -1,5 +1,5 @@
 {
-  "version": "1.7.1",
+  "version": "1.7.2-alpha.0",
   "package": "@mastra/mongodb",
   "exports": {},
   "modules": {}

package/dist/index.cjs

@@ -14,7 +14,7 @@ var evals = require('@mastra/core/evals');
 
 // package.json
 var package_default = {
-  version: "1.7.1"};
+  version: "1.7.2-alpha.0"};
 var MongoDBFilterTranslator = class extends filter.BaseFilterTranslator {
   getSupportedOperators() {
     return {
@@ -1505,6 +1505,191 @@ var MongoDBAgentsStorage = class _MongoDBAgentsStorage extends storage.AgentsSto
     return result;
   }
 };
+function toDoc(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 fromDoc(doc) {
+  return {
+    id: doc.id,
+    status: doc.status,
+    toolName: doc.tool_name,
+    toolCallId: doc.tool_call_id,
+    args: doc.args ?? {},
+    agentId: doc.agent_id,
+    threadId: doc.thread_id ?? void 0,
+    resourceId: doc.resource_id ?? void 0,
+    runId: doc.run_id ?? "",
+    result: doc.result ?? void 0,
+    error: doc.error ?? void 0,
+    retryCount: Number(doc.retry_count ?? 0),
+    maxRetries: Number(doc.max_retries ?? 0),
+    timeoutMs: Number(doc.timeout_ms ?? 3e5),
+    createdAt: new Date(doc.createdAt),
+    startedAt: doc.startedAt ? new Date(doc.startedAt) : void 0,
+    completedAt: doc.completedAt ? new Date(doc.completedAt) : void 0
+  };
+}
+var BackgroundTasksStorageMongoDB = class _BackgroundTasksStorageMongoDB extends storage.BackgroundTasksStorage {
+  #connector;
+  #skipDefaultIndexes;
+  #indexes;
+  static MANAGED_COLLECTIONS = [storage.TABLE_BACKGROUND_TASKS];
+  constructor(config) {
+    super();
+    this.#connector = resolveMongoDBConfig(config);
+    this.#skipDefaultIndexes = config.skipDefaultIndexes;
+    this.#indexes = config.indexes?.filter(
+      (idx) => _BackgroundTasksStorageMongoDB.MANAGED_COLLECTIONS.includes(idx.collection)
+    );
+  }
+  async getCollection() {
+    return this.#connector.getCollection(storage.TABLE_BACKGROUND_TASKS);
+  }
+  getDefaultIndexDefinitions() {
+    return [
+      { collection: storage.TABLE_BACKGROUND_TASKS, keys: { id: 1 }, options: { unique: true } },
+      { collection: storage.TABLE_BACKGROUND_TASKS, keys: { status: 1, createdAt: 1 } },
+      { collection: storage.TABLE_BACKGROUND_TASKS, keys: { agent_id: 1, status: 1 } },
+      { collection: storage.TABLE_BACKGROUND_TASKS, keys: { thread_id: 1, createdAt: 1 } },
+      { collection: storage.TABLE_BACKGROUND_TASKS, keys: { tool_call_id: 1 } }
+    ];
+  }
+  async createDefaultIndexes() {
+    if (this.#skipDefaultIndexes) return;
+    for (const indexDef of this.getDefaultIndexDefinitions()) {
+      try {
+        const collection = await this.getCollection();
+        await collection.createIndex(indexDef.keys, indexDef.options);
+      } catch (error) {
+        this.logger?.warn?.(`Failed to create index on ${storage.TABLE_BACKGROUND_TASKS}:`, error);
+      }
+    }
+  }
+  async createCustomIndexes() {
+    if (!this.#indexes || this.#indexes.length === 0) return;
+    for (const indexDef of this.#indexes) {
+      try {
+        const collection = await this.getCollection();
+        await collection.createIndex(indexDef.keys, indexDef.options);
+      } catch (error) {
+        this.logger?.warn?.(`Failed to create custom index on ${storage.TABLE_BACKGROUND_TASKS}:`, error);
+      }
+    }
+  }
+  async init() {
+    await this.createDefaultIndexes();
+    await this.createCustomIndexes();
+  }
+  async dangerouslyClearAll() {
+    const collection = await this.getCollection();
+    await collection.deleteMany({});
+  }
+  async createTask(task) {
+    const collection = await this.getCollection();
+    await collection.insertOne(toDoc(task));
+  }
+  async updateTask(taskId, update) {
+    const $set = {};
+    if ("status" in update) $set.status = update.status;
+    if ("result" in update) $set.result = update.result ?? null;
+    if ("error" in update) $set.error = update.error ?? null;
+    if ("retryCount" in update) $set.retry_count = update.retryCount;
+    if ("startedAt" in update) $set.startedAt = update.startedAt?.toISOString() ?? null;
+    if ("completedAt" in update) $set.completedAt = update.completedAt?.toISOString() ?? null;
+    if (Object.keys($set).length === 0) return;
+    const collection = await this.getCollection();
+    await collection.updateOne({ id: taskId }, { $set });
+  }
+  async getTask(taskId) {
+    const collection = await this.getCollection();
+    const doc = await collection.findOne({ id: taskId });
+    return doc ? fromDoc(doc) : null;
+  }
+  async listTasks(filter) {
+    const query = {};
+    if (filter.status) {
+      const statuses = Array.isArray(filter.status) ? filter.status : [filter.status];
+      query.status = statuses.length === 1 ? statuses[0] : { $in: statuses };
+    }
+    if (filter.agentId) query.agent_id = filter.agentId;
+    if (filter.threadId) query.thread_id = filter.threadId;
+    if (filter.resourceId) query.resource_id = filter.resourceId;
+    if (filter.runId) query.run_id = filter.runId;
+    if (filter.toolName) query.tool_name = filter.toolName;
+    const dateCol = filter.dateFilterBy === "startedAt" ? "startedAt" : filter.dateFilterBy === "completedAt" ? "completedAt" : "createdAt";
+    if (filter.fromDate) {
+      query[dateCol] = { ...query[dateCol] || {}, $gte: filter.fromDate.toISOString() };
+    }
+    if (filter.toDate) {
+      query[dateCol] = { ...query[dateCol] || {}, $lt: filter.toDate.toISOString() };
+    }
+    const orderCol = filter.orderBy === "startedAt" ? "startedAt" : filter.orderBy === "completedAt" ? "completedAt" : "createdAt";
+    const direction = filter.orderDirection === "desc" ? -1 : 1;
+    const collection = await this.getCollection();
+    const total = await collection.countDocuments(query);
+    let cursor = collection.find(query).sort({ [orderCol]: direction });
+    if (filter.perPage != null) {
+      if (filter.page != null) {
+        cursor = cursor.skip(filter.page * filter.perPage);
+      }
+      cursor = cursor.limit(filter.perPage);
+    }
+    const docs = await cursor.toArray();
+    return { tasks: docs.map(fromDoc), total };
+  }
+  async deleteTask(taskId) {
+    const collection = await this.getCollection();
+    await collection.deleteOne({ id: taskId });
+  }
+  async deleteTasks(filter) {
+    const query = {};
+    if (filter.status) {
+      const statuses = Array.isArray(filter.status) ? filter.status : [filter.status];
+      query.status = statuses.length === 1 ? statuses[0] : { $in: statuses };
+    }
+    const dateCol = filter.dateFilterBy === "startedAt" ? "startedAt" : filter.dateFilterBy === "completedAt" ? "completedAt" : "createdAt";
+    if (filter.fromDate) {
+      query[dateCol] = { ...query[dateCol] || {}, $gte: filter.fromDate.toISOString() };
+    }
+    if (filter.toDate) {
+      query[dateCol] = { ...query[dateCol] || {}, $lt: filter.toDate.toISOString() };
+    }
+    if (filter.agentId) query.agent_id = filter.agentId;
+    if (filter.threadId) query.thread_id = filter.threadId;
+    if (filter.resourceId) query.resource_id = filter.resourceId;
+    if (filter.runId) query.run_id = filter.runId;
+    if (filter.toolName) query.tool_name = filter.toolName;
+    if (Object.keys(query).length === 0) return;
+    const collection = await this.getCollection();
+    await collection.deleteMany(query);
+  }
+  async getRunningCount() {
+    const collection = await this.getCollection();
+    return collection.countDocuments({ status: "running" });
+  }
+  async getRunningCountByAgent(agentId) {
+    const collection = await this.getCollection();
+    return collection.countDocuments({ status: "running", agent_id: agentId });
+  }
+};
 var MongoDBBlobStore = class _MongoDBBlobStore extends storage.BlobStore {
   #connector;
   #skipDefaultIndexes;
@@ -9883,6 +10068,7 @@ var MongoDBStore = class extends storage.MastraCompositeStore {
     const blobs = new MongoDBBlobStore(domainConfig);
     const datasets = new MongoDBDatasetsStorage(domainConfig);
     const experiments = new MongoDBExperimentsStorage(domainConfig);
+    const backgroundTasks = new BackgroundTasksStorageMongoDB(domainConfig);
     this.stores = {
       memory,
       scores,
@@ -9896,6 +10082,7 @@ var MongoDBStore = class extends storage.MastraCompositeStore {
       workspaces,
       skills,
       blobs,
+      backgroundTasks,
       datasets,
       experiments
     };
@@ -10016,6 +10203,7 @@ Example Complex Query:
   ]
 }`;
 
+exports.BackgroundTasksStorageMongoDB = BackgroundTasksStorageMongoDB;
 exports.MONGODB_PROMPT = MONGODB_PROMPT;
 exports.MemoryStorageMongoDB = MemoryStorageMongoDB;
 exports.MongoDBAgentsStorage = MongoDBAgentsStorage;