@mastra/mongodb 1.9.0 → 1.9.1

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, 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 { 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, NotificationsStorage, TABLE_NOTIFICATIONS, 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.9.0"};
+  version: "1.9.1"};
 var MongoDBFilterTranslator = class extends BaseFilterTranslator {
   getSupportedOperators() {
     return {
@@ -6285,6 +6285,294 @@ ${unreflectedContent}` : bufferedReflection;
     }
   }
 };
+var statusTimestamp = (status, now) => {
+  if (status === "delivered") return { deliveredAt: now };
+  if (status === "seen") return { seenAt: now };
+  if (status === "dismissed") return { dismissedAt: now };
+  if (status === "archived") return { archivedAt: now };
+  if (status === "discarded") return { discardedAt: now };
+  return {};
+};
+var cloneDate = (value) => value ? new Date(value) : void 0;
+var cloneValue = (value) => value === void 0 ? void 0 : structuredClone(value);
+var cloneRecord = (record) => ({
+  ...record,
+  createdAt: new Date(record.createdAt),
+  updatedAt: new Date(record.updatedAt),
+  deliveredAt: cloneDate(record.deliveredAt),
+  seenAt: cloneDate(record.seenAt),
+  dismissedAt: cloneDate(record.dismissedAt),
+  archivedAt: cloneDate(record.archivedAt),
+  discardedAt: cloneDate(record.discardedAt),
+  deliverAt: cloneDate(record.deliverAt),
+  summaryAt: cloneDate(record.summaryAt),
+  lastDeliveryAttemptAt: cloneDate(record.lastDeliveryAttemptAt),
+  payload: cloneValue(record.payload),
+  attributes: cloneValue(record.attributes),
+  metadata: cloneValue(record.metadata)
+});
+function rowToNotification(row) {
+  return {
+    id: row.id,
+    threadId: row.threadId,
+    source: row.source,
+    kind: row.kind,
+    priority: row.priority,
+    status: row.status,
+    summary: row.summary,
+    payload: row.payload ?? void 0,
+    resourceId: row.resourceId ?? void 0,
+    agentId: row.agentId ?? void 0,
+    sourceId: row.sourceId ?? void 0,
+    dedupeKey: row.dedupeKey ?? void 0,
+    coalesceKey: row.coalesceKey ?? void 0,
+    coalescedCount: Number(row.coalescedCount ?? 1),
+    attributes: row.attributes,
+    createdAt: row.createdAt instanceof Date ? row.createdAt : new Date(row.createdAt),
+    updatedAt: row.updatedAt instanceof Date ? row.updatedAt : new Date(row.updatedAt),
+    deliveredAt: row.deliveredAt ? row.deliveredAt instanceof Date ? row.deliveredAt : new Date(row.deliveredAt) : void 0,
+    seenAt: row.seenAt ? row.seenAt instanceof Date ? row.seenAt : new Date(row.seenAt) : void 0,
+    dismissedAt: row.dismissedAt ? row.dismissedAt instanceof Date ? row.dismissedAt : new Date(row.dismissedAt) : void 0,
+    archivedAt: row.archivedAt ? row.archivedAt instanceof Date ? row.archivedAt : new Date(row.archivedAt) : void 0,
+    discardedAt: row.discardedAt ? row.discardedAt instanceof Date ? row.discardedAt : new Date(row.discardedAt) : void 0,
+    deliverAt: row.deliverAt ? row.deliverAt instanceof Date ? row.deliverAt : new Date(row.deliverAt) : void 0,
+    summaryAt: row.summaryAt ? row.summaryAt instanceof Date ? row.summaryAt : new Date(row.summaryAt) : void 0,
+    deliveryReason: row.deliveryReason ?? void 0,
+    deliveryAttempts: Number(row.deliveryAttempts ?? 0),
+    lastDeliveryAttemptAt: row.lastDeliveryAttemptAt ? row.lastDeliveryAttemptAt instanceof Date ? row.lastDeliveryAttemptAt : new Date(row.lastDeliveryAttemptAt) : void 0,
+    lastDeliveryError: row.lastDeliveryError ?? void 0,
+    deliveredSignalId: row.deliveredSignalId ?? void 0,
+    summarySignalId: row.summarySignalId ?? void 0,
+    metadata: row.metadata ?? void 0
+  };
+}
+function recordToDocument(record) {
+  return {
+    ...record,
+    payload: record.payload ?? null,
+    resourceId: record.resourceId ?? null,
+    agentId: record.agentId ?? null,
+    sourceId: record.sourceId ?? null,
+    dedupeKey: record.dedupeKey ?? null,
+    coalesceKey: record.coalesceKey ?? null,
+    attributes: record.attributes ?? null,
+    deliveredAt: record.deliveredAt ?? null,
+    seenAt: record.seenAt ?? null,
+    dismissedAt: record.dismissedAt ?? null,
+    archivedAt: record.archivedAt ?? null,
+    discardedAt: record.discardedAt ?? null,
+    deliverAt: record.deliverAt ?? null,
+    summaryAt: record.summaryAt ?? null,
+    deliveryReason: record.deliveryReason ?? null,
+    lastDeliveryAttemptAt: record.lastDeliveryAttemptAt ?? null,
+    lastDeliveryError: record.lastDeliveryError ?? null,
+    deliveredSignalId: record.deliveredSignalId ?? null,
+    summarySignalId: record.summarySignalId ?? null,
+    metadata: record.metadata ?? null
+  };
+}
+var NotificationsMongoDB = class _NotificationsMongoDB extends NotificationsStorage {
+  #connector;
+  #skipDefaultIndexes;
+  #indexes;
+  static MANAGED_COLLECTIONS = [TABLE_NOTIFICATIONS];
+  constructor(config) {
+    super();
+    this.#connector = resolveMongoDBConfig(config);
+    this.#skipDefaultIndexes = config.skipDefaultIndexes;
+    this.#indexes = config.indexes?.filter(
+      (idx) => _NotificationsMongoDB.MANAGED_COLLECTIONS.includes(idx.collection)
+    );
+  }
+  async getCollection() {
+    return this.#connector.getCollection(TABLE_NOTIFICATIONS);
+  }
+  async init() {
+    await this.createDefaultIndexes();
+    await this.createCustomIndexes();
+  }
+  getDefaultIndexDefinitions() {
+    return [
+      { collection: TABLE_NOTIFICATIONS, keys: { threadId: 1, id: 1 }, options: { unique: true } },
+      { collection: TABLE_NOTIFICATIONS, keys: { threadId: 1, status: 1, updatedAt: -1 } },
+      {
+        collection: TABLE_NOTIFICATIONS,
+        keys: { threadId: 1, source: 1, kind: 1, status: 1, agentId: 1, resourceId: 1, dedupeKey: 1, coalesceKey: 1 }
+      },
+      { collection: TABLE_NOTIFICATIONS, keys: { status: 1, deliverAt: 1, summaryAt: 1 } }
+    ];
+  }
+  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 dangerouslyClearAll() {
+    const collection = await this.getCollection();
+    await collection.deleteMany({});
+  }
+  async createNotification(input) {
+    const existing = await this.findCoalescable(input);
+    const collection = await this.getCollection();
+    if (existing) {
+      const now2 = /* @__PURE__ */ new Date();
+      const attributes = input.attributes ? { ...cloneValue(existing.attributes), ...cloneValue(input.attributes) } : cloneValue(existing.attributes);
+      const metadata = input.metadata ? { ...cloneValue(existing.metadata), ...cloneValue(input.metadata) } : cloneValue(existing.metadata);
+      await collection.updateOne(
+        { threadId: existing.threadId, id: existing.id },
+        {
+          $set: {
+            summary: input.summary,
+            payload: cloneValue(input.payload ?? existing.payload) ?? null,
+            priority: input.priority ?? existing.priority,
+            attributes: attributes ?? null,
+            updatedAt: now2,
+            deliverAt: input.deliverAt ?? existing.deliverAt ?? null,
+            summaryAt: input.summaryAt ?? existing.summaryAt ?? null,
+            deliveryReason: input.deliveryReason ?? existing.deliveryReason ?? null,
+            metadata: metadata ?? null
+          },
+          $inc: { coalescedCount: 1 }
+        }
+      );
+      const updated = await this.getNotification({ threadId: existing.threadId, id: existing.id });
+      if (!updated) throw new Error(`Notification ${existing.id} was not found for thread ${existing.threadId}`);
+      return updated;
+    }
+    const now = input.createdAt ?? /* @__PURE__ */ new Date();
+    const record = {
+      id: input.id ?? randomUUID(),
+      threadId: input.threadId,
+      source: input.source,
+      kind: input.kind,
+      priority: input.priority ?? "medium",
+      status: "pending",
+      summary: input.summary,
+      payload: cloneValue(input.payload),
+      resourceId: input.resourceId,
+      agentId: input.agentId,
+      sourceId: input.sourceId,
+      dedupeKey: input.dedupeKey,
+      coalesceKey: input.coalesceKey,
+      coalescedCount: 1,
+      attributes: cloneValue(input.attributes),
+      createdAt: now,
+      updatedAt: now,
+      deliverAt: input.deliverAt,
+      summaryAt: input.summaryAt,
+      deliveryReason: input.deliveryReason,
+      deliveryAttempts: 0,
+      metadata: cloneValue(input.metadata)
+    };
+    await collection.insertOne(recordToDocument(record));
+    return cloneRecord(record);
+  }
+  async listNotifications(input) {
+    const filter = { threadId: input.threadId };
+    if (input.status) filter.status = Array.isArray(input.status) ? { $in: input.status } : input.status;
+    if (input.priority) filter.priority = Array.isArray(input.priority) ? { $in: input.priority } : input.priority;
+    if (input.source) filter.source = input.source;
+    if (input.resourceId) filter.resourceId = input.resourceId;
+    if (input.agentId) filter.agentId = input.agentId;
+    if (input.search) {
+      const search = input.search.replace(/[.*+?^${}()|[\]\\]/g, "\\$&");
+      filter.$or = [{ summary: { $regex: search, $options: "i" } }, { kind: { $regex: search, $options: "i" } }];
+    }
+    const collection = await this.getCollection();
+    const cursor = collection.find(filter).sort({ updatedAt: -1 });
+    if (input.limit) cursor.limit(input.limit);
+    const rows = await cursor.toArray();
+    return rows.map((row) => rowToNotification(row));
+  }
+  async listDueNotifications(input) {
+    const filter = {
+      status: "pending",
+      $or: [{ deliverAt: { $ne: null, $lte: input.now } }, { summaryAt: { $ne: null, $lte: input.now } }]
+    };
+    if (input.agentId) filter.agentId = input.agentId;
+    if (input.resourceId) filter.resourceId = input.resourceId;
+    const collection = await this.getCollection();
+    const rows = await collection.find(filter).sort({ updatedAt: 1 }).toArray();
+    const due = rows.map((row) => rowToNotification(row)).sort((a, b) => dueTime(a) - dueTime(b) || a.updatedAt.getTime() - b.updatedAt.getTime());
+    return due.slice(0, input.limit ?? due.length).map(cloneRecord);
+  }
+  async getNotification(input) {
+    const collection = await this.getCollection();
+    const row = await collection.findOne({ threadId: input.threadId, id: input.id });
+    return row ? rowToNotification(row) : null;
+  }
+  async updateNotification(input) {
+    const existing = await this.getNotification({ threadId: input.threadId, id: input.id });
+    if (!existing) {
+      throw new Error(`Notification ${input.id} was not found for thread ${input.threadId}`);
+    }
+    const now = /* @__PURE__ */ new Date();
+    const set = {
+      ...input.status ? { status: input.status, ...statusTimestamp(input.status, now) } : {},
+      ...input.summary !== void 0 ? { summary: input.summary } : {},
+      ...input.payload !== void 0 ? { payload: cloneValue(input.payload) } : {},
+      ...input.attributes !== void 0 ? { attributes: cloneValue(input.attributes) } : {},
+      ...input.metadata !== void 0 ? { metadata: cloneValue(input.metadata) } : {},
+      ...input.deliverAt !== void 0 ? { deliverAt: input.deliverAt } : {},
+      ...input.summaryAt !== void 0 ? { summaryAt: input.summaryAt } : {},
+      ...input.deliveryReason !== void 0 ? { deliveryReason: input.deliveryReason } : {},
+      ...input.deliveryAttempts !== void 0 ? { deliveryAttempts: input.deliveryAttempts } : {},
+      ...input.lastDeliveryAttemptAt !== void 0 ? { lastDeliveryAttemptAt: input.lastDeliveryAttemptAt } : {},
+      ...input.lastDeliveryError !== void 0 ? { lastDeliveryError: input.lastDeliveryError } : {},
+      ...input.deliveredSignalId !== void 0 ? { deliveredSignalId: input.deliveredSignalId } : {},
+      ...input.summarySignalId !== void 0 ? { summarySignalId: input.summarySignalId } : {},
+      updatedAt: now
+    };
+    const collection = await this.getCollection();
+    await collection.updateOne({ threadId: input.threadId, id: input.id }, { $set: set });
+    const updated = await this.getNotification({ threadId: input.threadId, id: input.id });
+    if (!updated) throw new Error(`Notification ${input.id} was not found for thread ${input.threadId}`);
+    return updated;
+  }
+  async findCoalescable(input) {
+    if (!input.dedupeKey && !input.coalesceKey) return void 0;
+    const collection = await this.getCollection();
+    const row = await collection.findOne(
+      {
+        threadId: input.threadId,
+        source: input.source,
+        kind: input.kind,
+        status: "pending",
+        agentId: input.agentId ?? null,
+        resourceId: input.resourceId ?? null,
+        $or: [
+          ...input.dedupeKey ? [{ dedupeKey: input.dedupeKey }] : [],
+          ...input.coalesceKey ? [{ coalesceKey: input.coalesceKey }] : []
+        ]
+      },
+      { sort: { updatedAt: -1 } }
+    );
+    return row ? rowToNotification(row) : void 0;
+  }
+};
+var dueTime = (record) => {
+  const deliverAt = record.deliverAt?.getTime();
+  const summaryAt = record.summaryAt?.getTime();
+  if (deliverAt !== void 0 && summaryAt !== void 0) return Math.min(deliverAt, summaryAt);
+  return deliverAt ?? summaryAt ?? Number.POSITIVE_INFINITY;
+};
 var ObservabilityMongoDB = class _ObservabilityMongoDB extends ObservabilityStorage {
   #connector;
   #skipDefaultIndexes;
@@ -10349,6 +10637,7 @@ var MongoDBStore = class extends MastraCompositeStore {
       indexes: config.indexes
     };
     const memory = new MemoryStorageMongoDB(domainConfig);
+    const notifications = new NotificationsMongoDB(domainConfig);
     const scores = new ScoresStorageMongoDB(domainConfig);
     const workflows = new WorkflowsStorageMongoDB(domainConfig);
     const observability = new ObservabilityMongoDB(domainConfig);
@@ -10366,6 +10655,7 @@ var MongoDBStore = class extends MastraCompositeStore {
     const schedules = new SchedulesMongoDB(domainConfig);
     this.stores = {
       memory,
+      notifications,
       scores,
       workflows,
       observability,
@@ -10499,6 +10789,6 @@ Example Complex Query:
   ]
 }`;
 
-export { BackgroundTasksStorageMongoDB, MONGODB_PROMPT, MemoryStorageMongoDB, MongoDBAgentsStorage, MongoDBBlobStore, MongoDBDatasetsStorage, MongoDBExperimentsStorage, MongoDBMCPClientsStorage, MongoDBMCPServersStorage, MongoDBPromptBlocksStorage, MongoDBScorerDefinitionsStorage, MongoDBSkillsStorage, MongoDBStore, MongoDBVector, MongoDBWorkspacesStorage, ObservabilityMongoDB, SchedulesMongoDB, ScoresStorageMongoDB, WorkflowsStorageMongoDB };
+export { BackgroundTasksStorageMongoDB, MONGODB_PROMPT, MemoryStorageMongoDB, MongoDBAgentsStorage, MongoDBBlobStore, MongoDBDatasetsStorage, MongoDBExperimentsStorage, MongoDBMCPClientsStorage, MongoDBMCPServersStorage, MongoDBPromptBlocksStorage, MongoDBScorerDefinitionsStorage, MongoDBSkillsStorage, MongoDBStore, MongoDBVector, MongoDBWorkspacesStorage, NotificationsMongoDB, ObservabilityMongoDB, SchedulesMongoDB, ScoresStorageMongoDB, WorkflowsStorageMongoDB };
 //# sourceMappingURL=index.js.map
 //# sourceMappingURL=index.js.map