Class MemoryStorageMongoDB

stores/mongodb/src/storage/domains/memory/index.ts:57–2399

import { formatDateForMongoDB } from '../utils';

export class MemoryStorageMongoDB extends MemoryStorage {
  readonly supportsObservationalMemory = true;

  #connector: MongoDBConnector;
  #skipDefaultIndexes?: boolean;
  #indexes?: MongoDBIndexConfig[];

  /** Collections managed by this domain */
  static readonly MANAGED_COLLECTIONS = [TABLE_THREADS, TABLE_MESSAGES, TABLE_RESOURCES, OM_TABLE] as const;

  /**
   * Retention-eligible collections. The observational-memory collection is
   * excluded: it has no timestamp anchor to age on. All anchors are BSON dates.
   */
  static override readonly retentionTables: RetentionTablesDescriptor = {
    messages: { table: TABLE_MESSAGES, column: 'createdAt', indexed: true },
    resources: { table: TABLE_RESOURCES, column: 'createdAt', indexed: true },
    threads: { table: TABLE_THREADS, column: 'createdAt', indexed: true },
  };

  constructor(config: MongoDBDomainConfig) {
    super();
    this.#connector = resolveMongoDBConfig(config);
    this.#skipDefaultIndexes = config.skipDefaultIndexes;
    // Filter indexes to only those for collections managed by this domain
    this.#indexes = config.indexes?.filter(idx =>
      (MemoryStorageMongoDB.MANAGED_COLLECTIONS as readonly string[]).includes(idx.collection),
    );
  }

  private async getCollection(name: string) {
    return this.#connector.getCollection(name);
  }

  async init(): Promise<void> {
    await this.createDefaultIndexes();
    await this.createCustomIndexes();
  }

  /**
   * Delete memory rows older than each table's `maxAge`, batched. Order is
   * messages → resources → threads so child rows never outlive the delete of
   * their thread. Like `deleteThread()`, this does not sweep vector-store
   * embeddings — semantic-recall vectors live in a separate vector store the
   * memory domain cannot reach; cleaning those up is the operator's concern.
   */
  async prune(policies: Record<string, TableRetentionPolicy>, options?: PruneOptions): Promise<PruneResult[]> {
    const targets = resolveTargets({
      policies,
      descriptor: MemoryStorageMongoDB.retentionTables,
      order: ['messages', 'resources', 'threads'],
    });
    return runPrune({ connector: this.#connector, domain: 'memory', targets, options, logger: this.logger });
  }

  /**
   * Returns default index definitions for the memory domain collections.
   */