@mastra/mongodb 1.7.1 → 1.7.2-alpha.1

package/dist/index.js

@@ -1,5 +1,5 @@
 import { MastraError, ErrorCategory, ErrorDomain } from '@mastra/core/error';
-import { createVectorErrorId, AgentsStorage, TABLE_AGENTS, TABLE_AGENT_VERSIONS, createStorageErrorId, normalizePerPage, calculatePagination, BlobStore, TABLE_SKILL_BLOBS, DatasetsStorage, TABLE_DATASETS, TABLE_DATASET_ITEMS, TABLE_DATASET_VERSIONS, ensureDate, safelyParseJSON, TABLE_EXPERIMENTS, TABLE_EXPERIMENT_RESULTS, ExperimentsStorage, MCPClientsStorage, TABLE_MCP_CLIENTS, TABLE_MCP_CLIENT_VERSIONS, MCPServersStorage, TABLE_MCP_SERVERS, TABLE_MCP_SERVER_VERSIONS, MemoryStorage, TABLE_THREADS, TABLE_MESSAGES, TABLE_RESOURCES, ObservabilityStorage, TABLE_SPANS, listTracesArgsSchema, toTraceSpans, PromptBlocksStorage, TABLE_PROMPT_BLOCKS, TABLE_PROMPT_BLOCK_VERSIONS, ScorerDefinitionsStorage, TABLE_SCORER_DEFINITIONS, TABLE_SCORER_DEFINITION_VERSIONS, ScoresStorage, TABLE_SCORERS, SkillsStorage, TABLE_SKILLS, TABLE_SKILL_VERSIONS, WorkflowsStorage, TABLE_WORKFLOW_SNAPSHOT, WorkspacesStorage, TABLE_WORKSPACES, TABLE_WORKSPACE_VERSIONS, MastraCompositeStore, TraceStatus, transformScoreRow as transformScoreRow$1 } from '@mastra/core/storage';
+import { createVectorErrorId, AgentsStorage, TABLE_AGENTS, TABLE_AGENT_VERSIONS, createStorageErrorId, normalizePerPage, calculatePagination, BackgroundTasksStorage, TABLE_BACKGROUND_TASKS, BlobStore, TABLE_SKILL_BLOBS, DatasetsStorage, TABLE_DATASETS, TABLE_DATASET_ITEMS, TABLE_DATASET_VERSIONS, ensureDate, safelyParseJSON, TABLE_EXPERIMENTS, TABLE_EXPERIMENT_RESULTS, ExperimentsStorage, MCPClientsStorage, TABLE_MCP_CLIENTS, TABLE_MCP_CLIENT_VERSIONS, MCPServersStorage, TABLE_MCP_SERVERS, TABLE_MCP_SERVER_VERSIONS, MemoryStorage, TABLE_THREADS, TABLE_MESSAGES, TABLE_RESOURCES, ObservabilityStorage, TABLE_SPANS, listTracesArgsSchema, toTraceSpans, PromptBlocksStorage, TABLE_PROMPT_BLOCKS, TABLE_PROMPT_BLOCK_VERSIONS, ScorerDefinitionsStorage, TABLE_SCORER_DEFINITIONS, TABLE_SCORER_DEFINITION_VERSIONS, ScoresStorage, TABLE_SCORERS, SkillsStorage, TABLE_SKILLS, TABLE_SKILL_VERSIONS, WorkflowsStorage, TABLE_WORKFLOW_SNAPSHOT, WorkspacesStorage, TABLE_WORKSPACES, TABLE_WORKSPACE_VERSIONS, MastraCompositeStore, TraceStatus, transformScoreRow as transformScoreRow$1 } from '@mastra/core/storage';
 import { MastraVector, validateUpsertInput, validateVectorValues } from '@mastra/core/vector';
 import { MongoClient } from 'mongodb';
 import { v4 } from 'uuid';
@@ -12,7 +12,7 @@ import { saveScorePayloadSchema } from '@mastra/core/evals';
 
 // package.json
 var package_default = {
-  version: "1.7.1"};
+  version: "1.7.2-alpha.1"};
 var MongoDBFilterTranslator = class extends BaseFilterTranslator {
   getSupportedOperators() {
     return {
@@ -1503,6 +1503,191 @@ var MongoDBAgentsStorage = class _MongoDBAgentsStorage extends AgentsStorage {
     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 BackgroundTasksStorage {
+  #connector;
+  #skipDefaultIndexes;
+  #indexes;
+  static MANAGED_COLLECTIONS = [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(TABLE_BACKGROUND_TASKS);
+  }
+  getDefaultIndexDefinitions() {
+    return [
+      { collection: TABLE_BACKGROUND_TASKS, keys: { id: 1 }, options: { unique: true } },
+      { collection: TABLE_BACKGROUND_TASKS, keys: { status: 1, createdAt: 1 } },
+      { collection: TABLE_BACKGROUND_TASKS, keys: { agent_id: 1, status: 1 } },
+      { collection: TABLE_BACKGROUND_TASKS, keys: { thread_id: 1, createdAt: 1 } },
+      { collection: 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 ${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 ${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 BlobStore {
   #connector;
   #skipDefaultIndexes;
@@ -6463,6 +6648,44 @@ Note: This migration may take some time for large collections.
       );
     }
   }
+  async getTraceLight(args) {
+    const { traceId } = args;
+    try {
+      const collection = await this.getCollection(TABLE_SPANS);
+      const spans = await collection.find(
+        { traceId },
+        {
+          projection: {
+            input: 0,
+            output: 0,
+            attributes: 0,
+            metadata: 0,
+            tags: 0,
+            links: 0
+          }
+        }
+      ).sort({ startedAt: 1 }).toArray();
+      if (!spans || spans.length === 0) {
+        return null;
+      }
+      return {
+        traceId,
+        spans: spans.map((span) => this.transformSpanFromMongo(span))
+      };
+    } catch (error) {
+      throw new MastraError(
+        {
+          id: createStorageErrorId("MONGODB", "GET_TRACE_LIGHT", "FAILED"),
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.USER,
+          details: {
+            traceId
+          }
+        },
+        error
+      );
+    }
+  }
   async updateSpan(args) {
     const { traceId, spanId, updates } = args;
     try {
@@ -9881,6 +10104,7 @@ var MongoDBStore = class extends 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,
@@ -9894,6 +10118,7 @@ var MongoDBStore = class extends MastraCompositeStore {
       workspaces,
       skills,
       blobs,
+      backgroundTasks,
       datasets,
       experiments
     };
@@ -10014,6 +10239,6 @@ Example Complex Query:
   ]
 }`;
 
-export { MONGODB_PROMPT, MemoryStorageMongoDB, MongoDBAgentsStorage, MongoDBBlobStore, MongoDBDatasetsStorage, MongoDBExperimentsStorage, MongoDBMCPClientsStorage, MongoDBMCPServersStorage, MongoDBPromptBlocksStorage, MongoDBScorerDefinitionsStorage, MongoDBSkillsStorage, MongoDBStore, MongoDBVector, MongoDBWorkspacesStorage, ObservabilityMongoDB, ScoresStorageMongoDB, WorkflowsStorageMongoDB };
+export { BackgroundTasksStorageMongoDB, MONGODB_PROMPT, MemoryStorageMongoDB, MongoDBAgentsStorage, MongoDBBlobStore, MongoDBDatasetsStorage, MongoDBExperimentsStorage, MongoDBMCPClientsStorage, MongoDBMCPServersStorage, MongoDBPromptBlocksStorage, MongoDBScorerDefinitionsStorage, MongoDBSkillsStorage, MongoDBStore, MongoDBVector, MongoDBWorkspacesStorage, ObservabilityMongoDB, ScoresStorageMongoDB, WorkflowsStorageMongoDB };
 //# sourceMappingURL=index.js.map
 //# sourceMappingURL=index.js.map