@mastra/mongodb 1.7.4 → 1.8.0-alpha.0

package/dist/index.js

@@ -1,6 +1,6 @@
 import { v4 } from '@lukeed/uuid';
 import { MastraError, ErrorCategory, ErrorDomain } from '@mastra/core/error';
-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 { 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, SchedulesStorage, TABLE_SCHEDULES, TABLE_SCHEDULE_TRIGGERS, 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 { BaseFilterTranslator } from '@mastra/core/vector/filter';
@@ -12,7 +12,7 @@ import { saveScorePayloadSchema } from '@mastra/core/evals';
 
 // package.json
 var package_default = {
-  version: "1.7.4"};
+  version: "1.8.0-alpha.0"};
 var MongoDBFilterTranslator = class extends BaseFilterTranslator {
   getSupportedOperators() {
     return {
@@ -7593,6 +7593,206 @@ var MongoDBPromptBlocksStorage = class _MongoDBPromptBlocksStorage extends Promp
     return result;
   }
 };
+function scheduleToDoc(schedule) {
+  return {
+    id: schedule.id,
+    target: schedule.target,
+    cron: schedule.cron,
+    timezone: schedule.timezone ?? null,
+    status: schedule.status,
+    next_fire_at: schedule.nextFireAt,
+    last_fire_at: schedule.lastFireAt ?? null,
+    last_run_id: schedule.lastRunId ?? null,
+    created_at: schedule.createdAt,
+    updated_at: schedule.updatedAt,
+    metadata: schedule.metadata ?? null
+  };
+}
+function docToSchedule(doc) {
+  const target = doc.target;
+  if (!target) {
+    throw new Error(`Schedule ${doc.id} has invalid target`);
+  }
+  const schedule = {
+    id: String(doc.id),
+    target,
+    cron: String(doc.cron),
+    status: String(doc.status),
+    nextFireAt: Number(doc.next_fire_at),
+    createdAt: Number(doc.created_at),
+    updatedAt: Number(doc.updated_at)
+  };
+  if (doc.timezone != null) schedule.timezone = String(doc.timezone);
+  if (doc.last_fire_at != null) schedule.lastFireAt = Number(doc.last_fire_at);
+  if (doc.last_run_id != null) schedule.lastRunId = String(doc.last_run_id);
+  if (doc.metadata != null) schedule.metadata = doc.metadata;
+  return schedule;
+}
+function triggerToDoc(trigger) {
+  return {
+    schedule_id: trigger.scheduleId,
+    run_id: trigger.runId,
+    scheduled_fire_at: trigger.scheduledFireAt,
+    actual_fire_at: trigger.actualFireAt,
+    status: trigger.status,
+    error: trigger.error ?? null
+  };
+}
+function docToTrigger(doc) {
+  const trigger = {
+    scheduleId: String(doc.schedule_id),
+    runId: String(doc.run_id),
+    scheduledFireAt: Number(doc.scheduled_fire_at),
+    actualFireAt: Number(doc.actual_fire_at),
+    status: String(doc.status)
+  };
+  if (doc.error != null) trigger.error = String(doc.error);
+  return trigger;
+}
+var SchedulesMongoDB = class _SchedulesMongoDB extends SchedulesStorage {
+  #connector;
+  #skipDefaultIndexes;
+  #indexes;
+  static MANAGED_COLLECTIONS = [TABLE_SCHEDULES, TABLE_SCHEDULE_TRIGGERS];
+  constructor(config) {
+    super();
+    this.#connector = resolveMongoDBConfig(config);
+    this.#skipDefaultIndexes = config.skipDefaultIndexes;
+    this.#indexes = config.indexes?.filter(
+      (idx) => _SchedulesMongoDB.MANAGED_COLLECTIONS.includes(idx.collection)
+    );
+  }
+  getSchedulesCollection() {
+    return this.#connector.getCollection(TABLE_SCHEDULES);
+  }
+  getTriggersCollection() {
+    return this.#connector.getCollection(TABLE_SCHEDULE_TRIGGERS);
+  }
+  getDefaultIndexDefinitions() {
+    return [
+      { collection: TABLE_SCHEDULES, keys: { id: 1 }, options: { unique: true } },
+      { collection: TABLE_SCHEDULES, keys: { status: 1, next_fire_at: 1 } },
+      { collection: TABLE_SCHEDULES, keys: { "target.workflowId": 1 } },
+      { collection: TABLE_SCHEDULE_TRIGGERS, keys: { schedule_id: 1, actual_fire_at: -1 } },
+      { collection: TABLE_SCHEDULE_TRIGGERS, keys: { run_id: 1 }, options: { unique: true } }
+    ];
+  }
+  async createDefaultIndexes() {
+    if (this.#skipDefaultIndexes) return;
+    for (const indexDef of this.getDefaultIndexDefinitions()) {
+      try {
+        const collection = await this.#connector.getCollection(indexDef.collection);
+        await collection.createIndex(indexDef.keys, indexDef.options);
+      } catch (error) {
+        this.logger?.warn?.(`Failed to create index on ${indexDef.collection}:`, error);
+      }
+    }
+  }
+  async createCustomIndexes() {
+    if (!this.#indexes || this.#indexes.length === 0) return;
+    for (const indexDef of this.#indexes) {
+      try {
+        const collection = await this.#connector.getCollection(indexDef.collection);
+        await collection.createIndex(indexDef.keys, indexDef.options);
+      } catch (error) {
+        this.logger?.warn?.(`Failed to create custom index on ${indexDef.collection}:`, error);
+      }
+    }
+  }
+  async init() {
+    await this.createDefaultIndexes();
+    await this.createCustomIndexes();
+  }
+  async dangerouslyClearAll() {
+    const triggers = await this.getTriggersCollection();
+    await triggers.deleteMany({});
+    const schedules = await this.getSchedulesCollection();
+    await schedules.deleteMany({});
+  }
+  async createSchedule(schedule) {
+    const collection = await this.getSchedulesCollection();
+    const existing = await collection.findOne({ id: schedule.id });
+    if (existing) {
+      throw new Error(`Schedule with id "${schedule.id}" already exists`);
+    }
+    await collection.insertOne(scheduleToDoc(schedule));
+    return schedule;
+  }
+  async getSchedule(id) {
+    const collection = await this.getSchedulesCollection();
+    const doc = await collection.findOne({ id });
+    return doc ? docToSchedule(doc) : null;
+  }
+  async listSchedules(filter) {
+    const query = {};
+    if (filter?.status) query.status = filter.status;
+    if (filter?.workflowId) query["target.workflowId"] = filter.workflowId;
+    const collection = await this.getSchedulesCollection();
+    const docs = await collection.find(query).sort({ created_at: 1 }).toArray();
+    return docs.map(docToSchedule);
+  }
+  async listDueSchedules(now, limit) {
+    const cap = limit ?? 100;
+    const collection = await this.getSchedulesCollection();
+    const docs = await collection.find({ status: "active", next_fire_at: { $lte: now } }).sort({ next_fire_at: 1 }).limit(cap).toArray();
+    return docs.map(docToSchedule);
+  }
+  async updateSchedule(id, patch) {
+    const $set = {};
+    if ("cron" in patch && patch.cron !== void 0) $set.cron = patch.cron;
+    if ("timezone" in patch) $set.timezone = patch.timezone ?? null;
+    if ("status" in patch && patch.status !== void 0) $set.status = patch.status;
+    if ("nextFireAt" in patch && patch.nextFireAt !== void 0) $set.next_fire_at = patch.nextFireAt;
+    if ("target" in patch && patch.target !== void 0) $set.target = patch.target;
+    if ("metadata" in patch) $set.metadata = patch.metadata ?? null;
+    $set.updated_at = Date.now();
+    const collection = await this.getSchedulesCollection();
+    const result = await collection.findOneAndUpdate({ id }, { $set }, { returnDocument: "after" });
+    if (!result) throw new Error(`Schedule ${id} not found`);
+    return docToSchedule(result);
+  }
+  async updateScheduleNextFire(id, expectedNextFireAt, newNextFireAt, lastFireAt, lastRunId) {
+    const collection = await this.getSchedulesCollection();
+    const result = await collection.updateOne(
+      { id, next_fire_at: expectedNextFireAt, status: "active" },
+      {
+        $set: {
+          next_fire_at: newNextFireAt,
+          last_fire_at: lastFireAt,
+          last_run_id: lastRunId,
+          updated_at: Date.now()
+        }
+      }
+    );
+    return result.matchedCount > 0;
+  }
+  async deleteSchedule(id) {
+    const triggers = await this.getTriggersCollection();
+    await triggers.deleteMany({ schedule_id: id });
+    const schedules = await this.getSchedulesCollection();
+    await schedules.deleteOne({ id });
+  }
+  async recordTrigger(trigger) {
+    const collection = await this.getTriggersCollection();
+    await collection.insertOne(triggerToDoc(trigger));
+  }
+  async listTriggers(scheduleId, opts) {
+    const query = { schedule_id: scheduleId };
+    if (opts?.fromActualFireAt != null || opts?.toActualFireAt != null) {
+      const range = {};
+      if (opts?.fromActualFireAt != null) range.$gte = opts.fromActualFireAt;
+      if (opts?.toActualFireAt != null) range.$lt = opts.toActualFireAt;
+      query.actual_fire_at = range;
+    }
+    const collection = await this.getTriggersCollection();
+    let cursor = collection.find(query).sort({ actual_fire_at: -1 });
+    if (opts?.limit != null) {
+      cursor = cursor.limit(Math.floor(opts.limit));
+    }
+    const docs = await cursor.toArray();
+    return docs.map(docToTrigger);
+  }
+};
 var SNAPSHOT_FIELDS5 = [
   "name",
   "description",
@@ -10105,6 +10305,7 @@ var MongoDBStore = class extends MastraCompositeStore {
     const datasets = new MongoDBDatasetsStorage(domainConfig);
     const experiments = new MongoDBExperimentsStorage(domainConfig);
     const backgroundTasks = new BackgroundTasksStorageMongoDB(domainConfig);
+    const schedules = new SchedulesMongoDB(domainConfig);
     this.stores = {
       memory,
       scores,
@@ -10120,7 +10321,8 @@ var MongoDBStore = class extends MastraCompositeStore {
       blobs,
       backgroundTasks,
       datasets,
-      experiments
+      experiments,
+      schedules
     };
   }
   /**
@@ -10239,6 +10441,6 @@ Example Complex Query:
   ]
 }`;
 
-export { BackgroundTasksStorageMongoDB, 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, SchedulesMongoDB, ScoresStorageMongoDB, WorkflowsStorageMongoDB };
 //# sourceMappingURL=index.js.map
 //# sourceMappingURL=index.js.map