@langchain/langgraph-checkpoint-mongodb 0.0.6 → 0.1.0

package/CHANGELOG.md

@@ -0,0 +1,15 @@
+# @langchain/langgraph-checkpoint-mongodb
+
+## 0.1.0
+
+### Minor Changes
+
+- ccbcbc1: Add delete thread method to checkpointers
+- Updated dependencies [773ec0d]
+  - @langchain/langgraph-checkpoint@0.1.0
+
+### Patch Changes
+
+- Updated dependencies [ccbcbc1]
+- Updated dependencies [10f292a]
+- Updated dependencies [3fd7f73]

package/dist/index.cjs

@@ -167,8 +167,10 @@ class MongoDBSaver extends langgraph_checkpoint_1.BaseCheckpointSaver {
         if (thread_id === undefined) {
             throw new Error(`The provided config must contain a configurable field with a "thread_id" field.`);
         }
-        const [checkpointType, serializedCheckpoint] = this.serde.dumpsTyped(checkpoint);
-        const [metadataType, serializedMetadata] = this.serde.dumpsTyped(metadata);
+        const [[checkpointType, serializedCheckpoint], [metadataType, serializedMetadata],] = await Promise.all([
+            this.serde.dumpsTyped(checkpoint),
+            this.serde.dumpsTyped(metadata),
+        ]);
         if (checkpointType !== metadataType) {
             throw new Error("Mismatched checkpoint and metadata types.");
         }
@@ -183,9 +185,9 @@ class MongoDBSaver extends langgraph_checkpoint_1.BaseCheckpointSaver {
             checkpoint_ns,
             checkpoint_id,
         };
-        await this.db.collection(this.checkpointCollectionName).updateOne(upsertQuery, {
-            $set: doc,
-        }, { upsert: true });
+        await this.db
+            .collection(this.checkpointCollectionName)
+            .updateOne(upsertQuery, { $set: doc }, { upsert: true });
         return {
             configurable: {
                 thread_id,
@@ -206,7 +208,7 @@ class MongoDBSaver extends langgraph_checkpoint_1.BaseCheckpointSaver {
             checkpoint_id === undefined) {
             throw new Error(`The provided config must contain a configurable field with "thread_id", "checkpoint_ns" and "checkpoint_id" fields.`);
         }
-        const operations = writes.map(([channel, value], idx) => {
+        const operations = await Promise.all(writes.map(async ([channel, value], idx) => {
             const upsertQuery = {
                 thread_id,
                 checkpoint_ns,
@@ -214,24 +216,27 @@ class MongoDBSaver extends langgraph_checkpoint_1.BaseCheckpointSaver {
                 task_id: taskId,
                 idx,
             };
-            const [type, serializedValue] = this.serde.dumpsTyped(value);
+            const [type, serializedValue] = await this.serde.dumpsTyped(value);
             return {
                 updateOne: {
                     filter: upsertQuery,
-                    update: {
-                        $set: {
-                            channel,
-                            type,
-                            value: serializedValue,
-                        },
-                    },
+                    update: { $set: { channel, type, value: serializedValue } },
                     upsert: true,
                 },
             };
-        });
+        }));
         await this.db
             .collection(this.checkpointWritesCollectionName)
             .bulkWrite(operations);
     }
+    async deleteThread(threadId) {
+        await this.db
+            .collection(this.checkpointCollectionName)
+            .deleteMany({ thread_id: threadId });
+        await this.db
+            .collection(this.checkpointWritesCollectionName)
+            .deleteMany({ thread_id: threadId });
+    }
 }
 exports.MongoDBSaver = MongoDBSaver;
+//# sourceMappingURL=index.js.map

package/dist/index.d.ts

@@ -38,4 +38,5 @@ export declare class MongoDBSaver extends BaseCheckpointSaver {
      * Saves intermediate writes associated with a checkpoint to the MongoDB database.
      */
     putWrites(config: RunnableConfig, writes: PendingWrite[], taskId: string): Promise<void>;
+    deleteThread(threadId: string): Promise<void>;
 }

package/dist/index.js

@@ -164,8 +164,10 @@ export class MongoDBSaver extends BaseCheckpointSaver {
         if (thread_id === undefined) {
             throw new Error(`The provided config must contain a configurable field with a "thread_id" field.`);
         }
-        const [checkpointType, serializedCheckpoint] = this.serde.dumpsTyped(checkpoint);
-        const [metadataType, serializedMetadata] = this.serde.dumpsTyped(metadata);
+        const [[checkpointType, serializedCheckpoint], [metadataType, serializedMetadata],] = await Promise.all([
+            this.serde.dumpsTyped(checkpoint),
+            this.serde.dumpsTyped(metadata),
+        ]);
         if (checkpointType !== metadataType) {
             throw new Error("Mismatched checkpoint and metadata types.");
         }
@@ -180,9 +182,9 @@ export class MongoDBSaver extends BaseCheckpointSaver {
             checkpoint_ns,
             checkpoint_id,
         };
-        await this.db.collection(this.checkpointCollectionName).updateOne(upsertQuery, {
-            $set: doc,
-        }, { upsert: true });
+        await this.db
+            .collection(this.checkpointCollectionName)
+            .updateOne(upsertQuery, { $set: doc }, { upsert: true });
         return {
             configurable: {
                 thread_id,
@@ -203,7 +205,7 @@ export class MongoDBSaver extends BaseCheckpointSaver {
             checkpoint_id === undefined) {
             throw new Error(`The provided config must contain a configurable field with "thread_id", "checkpoint_ns" and "checkpoint_id" fields.`);
         }
-        const operations = writes.map(([channel, value], idx) => {
+        const operations = await Promise.all(writes.map(async ([channel, value], idx) => {
             const upsertQuery = {
                 thread_id,
                 checkpoint_ns,
@@ -211,23 +213,26 @@ export class MongoDBSaver extends BaseCheckpointSaver {
                 task_id: taskId,
                 idx,
             };
-            const [type, serializedValue] = this.serde.dumpsTyped(value);
+            const [type, serializedValue] = await this.serde.dumpsTyped(value);
             return {
                 updateOne: {
                     filter: upsertQuery,
-                    update: {
-                        $set: {
-                            channel,
-                            type,
-                            value: serializedValue,
-                        },
-                    },
+                    update: { $set: { channel, type, value: serializedValue } },
                     upsert: true,
                 },
             };
-        });
+        }));
         await this.db
             .collection(this.checkpointWritesCollectionName)
             .bulkWrite(operations);
     }
+    async deleteThread(threadId) {
+        await this.db
+            .collection(this.checkpointCollectionName)
+            .deleteMany({ thread_id: threadId });
+        await this.db
+            .collection(this.checkpointWritesCollectionName)
+            .deleteMany({ thread_id: threadId });
+    }
 }
+//# sourceMappingURL=index.js.map

package/dist/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAEA,OAAO,EACL,mBAAmB,GAQpB,MAAM,iCAAiC,CAAC;AASzC;;GAEG;AACH,MAAM,OAAO,YAAa,SAAQ,mBAAmB;IASnD,YACE,EACE,MAAM,EACN,MAAM,EACN,wBAAwB,EACxB,8BAA8B,GACX,EACrB,KAA0B;QAE1B,KAAK,CAAC,KAAK,CAAC,CAAC;QAjBL;;;;;WAAoB;QAEpB;;;;;WAAkB;QAE5B;;;;mBAA2B,aAAa;WAAC;QAEzC;;;;mBAAiC,mBAAmB;WAAC;QAYnD,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,EAAE,GAAG,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC,MAAM,CAAC,CAAC;QACjC,IAAI,CAAC,wBAAwB;YAC3B,wBAAwB,IAAI,IAAI,CAAC,wBAAwB,CAAC;QAC5D,IAAI,CAAC,8BAA8B;YACjC,8BAA8B,IAAI,IAAI,CAAC,8BAA8B,CAAC;IAC1E,CAAC;IAED;;;;;OAKG;IACH,KAAK,CAAC,QAAQ,CAAC,MAAsB;QACnC,MAAM,EACJ,SAAS,EACT,aAAa,GAAG,EAAE,EAClB,aAAa,GACd,GAAG,MAAM,CAAC,YAAY,IAAI,EAAE,CAAC;QAC9B,IAAI,KAAK,CAAC;QACV,IAAI,aAAa,EAAE,CAAC;YAClB,KAAK,GAAG;gBACN,SAAS;gBACT,aAAa;gBACb,aAAa;aACd,CAAC;QACJ,CAAC;aAAM,CAAC;YACN,KAAK,GAAG,EAAE,SAAS,EAAE,aAAa,EAAE,CAAC;QACvC,CAAC;QACD,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,EAAE;aACzB,UAAU,CAAC,IAAI,CAAC,wBAAwB,CAAC;aACzC,IAAI,CAAC,KAAK,CAAC;aACX,IAAI,CAAC,eAAe,EAAE,CAAC,CAAC,CAAC;aACzB,KAAK,CAAC,CAAC,CAAC;aACR,OAAO,EAAE,CAAC;QACb,IAAI,MAAM,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACxB,OAAO,SAAS,CAAC;QACnB,CAAC;QACD,MAAM,GAAG,GAAG,MAAM,CAAC,CAAC,CAAC,CAAC;QACtB,MAAM,kBAAkB,GAAG;YACzB,SAAS;YACT,aAAa;YACb,aAAa,EAAE,GAAG,CAAC,aAAa;SACjC,CAAC;QACF,MAAM,UAAU,GAAG,CAAC,MAAM,IAAI,CAAC,KAAK,CAAC,UAAU,CAC7C,GAAG,CAAC,IAAI,EACR,GAAG,CAAC,UAAU,CAAC,KAAK,CAAC,MAAM,CAAC,CAC7B,CAAe,CAAC;QACjB,MAAM,gBAAgB,GAAG,MAAM,IAAI,CAAC,EAAE;aACnC,UAAU,CAAC,IAAI,CAAC,8BAA8B,CAAC;aAC/C,IAAI,CAAC,kBAAkB,CAAC;aACxB,OAAO,EAAE,CAAC;QACb,MAAM,aAAa,GAA6B,MAAM,OAAO,CAAC,GAAG,CAC/D,gBAAgB,CAAC,GAAG,CAAC,KAAK,EAAE,eAAe,EAAE,EAAE;YAC7C,OAAO;gBACL,eAAe,CAAC,OAAO;gBACvB,eAAe,CAAC,OAAO;gBACvB,MAAM,IAAI,CAAC,KAAK,CAAC,UAAU,CACzB,eAAe,CAAC,IAAI,EACpB,eAAe,CAAC,KAAK,CAAC,KAAK,CAAC,MAAM,CAAC,CACpC;aACwB,CAAC;QAC9B,CAAC,CAAC,CACH,CAAC;QACF,OAAO;YACL,MAAM,EAAE,EAAE,YAAY,EAAE,kBAAkB,EAAE;YAC5C,UAAU;YACV,aAAa;YACb,QAAQ,EAAE,CAAC,MAAM,IAAI,CAAC,KAAK,CAAC,UAAU,CACpC,GAAG,CAAC,IAAI,EACR,GAAG,CAAC,QAAQ,CAAC,KAAK,CAAC,MAAM,CAAC,CAC3B,CAAuB;YACxB,YAAY,EACV,GAAG,CAAC,oBAAoB,IAAI,IAAI;gBAC9B,CAAC,CAAC;oBACE,YAAY,EAAE;wBACZ,SAAS;wBACT,aAAa;wBACb,aAAa,EAAE,GAAG,CAAC,oBAAoB;qBACxC;iBACF;gBACH,CAAC,CAAC,SAAS;SAChB,CAAC;IACJ,CAAC;IAED;;;;OAIG;IACH,KAAK,CAAC,CAAC,IAAI,CACT,MAAsB,EACtB,OAA+B;QAE/B,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,MAAM,EAAE,GAAG,OAAO,IAAI,EAAE,CAAC;QAChD,MAAM,KAAK,GAA4B,EAAE,CAAC;QAE1C,IAAI,MAAM,EAAE,YAAY,EAAE,SAAS,EAAE,CAAC;YACpC,KAAK,CAAC,SAAS,GAAG,MAAM,CAAC,YAAY,CAAC,SAAS,CAAC;QAClD,CAAC;QAED,IACE,MAAM,EAAE,YAAY,EAAE,aAAa,KAAK,SAAS;YACjD,MAAM,EAAE,YAAY,EAAE,aAAa,KAAK,IAAI,EAC5C,CAAC;YACD,KAAK,CAAC,aAAa,GAAG,MAAM,CAAC,YAAY,CAAC,aAAa,CAAC;QAC1D,CAAC;QAED,IAAI,MAAM,EAAE,CAAC;YACX,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,GAAG,EAAE,KAAK,CAAC,EAAE,EAAE;gBAC9C,KAAK,CAAC,YAAY,GAAG,EAAE,CAAC,GAAG,KAAK,CAAC;YACnC,CAAC,CAAC,CAAC;QACL,CAAC;QAED,IAAI,MAAM,EAAE,CAAC;YACX,KAAK,CAAC,aAAa,GAAG,EAAE,GAAG,EAAE,MAAM,CAAC,YAAY,EAAE,aAAa,EAAE,CAAC;QACpE,CAAC;QAED,IAAI,MAAM,GAAG,IAAI,CAAC,EAAE;aACjB,UAAU,CAAC,IAAI,CAAC,wBAAwB,CAAC;aACzC,IAAI,CAAC,KAAK,CAAC;aACX,IAAI,CAAC,eAAe,EAAE,CAAC,CAAC,CAAC,CAAC;QAE7B,IAAI,KAAK,KAAK,SAAS,EAAE,CAAC;YACxB,MAAM,GAAG,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;QAC/B,CAAC;QAED,IAAI,KAAK,EAAE,MAAM,GAAG,IAAI,MAAM,EAAE,CAAC;YAC/B,MAAM,UAAU,GAAG,CAAC,MAAM,IAAI,CAAC,KAAK,CAAC,UAAU,CAC7C,GAAG,CAAC,IAAI,EACR,GAAG,CAAC,UAAU,CAAC,KAAK,CAAC,MAAM,CAAC,CAC7B,CAAe,CAAC;YACjB,MAAM,QAAQ,GAAG,CAAC,MAAM,IAAI,CAAC,KAAK,CAAC,UAAU,CAC3C,GAAG,CAAC,IAAI,EACR,GAAG,CAAC,QAAQ,CAAC,KAAK,CAAC,MAAM,CAAC,CAC3B,CAAuB,CAAC;YAEzB,MAAM;gBACJ,MAAM,EAAE;oBACN,YAAY,EAAE;wBACZ,SAAS,EAAE,GAAG,CAAC,SAAS;wBACxB,aAAa,EAAE,GAAG,CAAC,aAAa;wBAChC,aAAa,EAAE,GAAG,CAAC,aAAa;qBACjC;iBACF;gBACD,UAAU;gBACV,QAAQ;gBACR,YAAY,EAAE,GAAG,CAAC,oBAAoB;oBACpC,CAAC,CAAC;wBACE,YAAY,EAAE;4BACZ,SAAS,EAAE,GAAG,CAAC,SAAS;4BACxB,aAAa,EAAE,GAAG,CAAC,aAAa;4BAChC,aAAa,EAAE,GAAG,CAAC,oBAAoB;yBACxC;qBACF;oBACH,CAAC,CAAC,SAAS;aACd,CAAC;QACJ,CAAC;IACH,CAAC;IAED;;;OAGG;IACH,KAAK,CAAC,GAAG,CACP,MAAsB,EACtB,UAAsB,EACtB,QAA4B;QAE5B,MAAM,SAAS,GAAG,MAAM,CAAC,YAAY,EAAE,SAAS,CAAC;QACjD,MAAM,aAAa,GAAG,MAAM,CAAC,YAAY,EAAE,aAAa,IAAI,EAAE,CAAC;QAC/D,MAAM,aAAa,GAAG,UAAU,CAAC,EAAE,CAAC;QACpC,IAAI,SAAS,KAAK,SAAS,EAAE,CAAC;YAC5B,MAAM,IAAI,KAAK,CACb,iFAAiF,CAClF,CAAC;QACJ,CAAC;QACD,MAAM,CACJ,CAAC,cAAc,EAAE,oBAAoB,CAAC,EACtC,CAAC,YAAY,EAAE,kBAAkB,CAAC,EACnC,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC;YACpB,IAAI,CAAC,KAAK,CAAC,UAAU,CAAC,UAAU,CAAC;YACjC,IAAI,CAAC,KAAK,CAAC,UAAU,CAAC,QAAQ,CAAC;SAChC,CAAC,CAAC;QAEH,IAAI,cAAc,KAAK,YAAY,EAAE,CAAC;YACpC,MAAM,IAAI,KAAK,CAAC,2CAA2C,CAAC,CAAC;QAC/D,CAAC;QACD,MAAM,GAAG,GAAG;YACV,oBAAoB,EAAE,MAAM,CAAC,YAAY,EAAE,aAAa;YACxD,IAAI,EAAE,cAAc;YACpB,UAAU,EAAE,oBAAoB;YAChC,QAAQ,EAAE,kBAAkB;SAC7B,CAAC;QACF,MAAM,WAAW,GAAG;YAClB,SAAS;YACT,aAAa;YACb,aAAa;SACd,CAAC;QACF,MAAM,IAAI,CAAC,EAAE;aACV,UAAU,CAAC,IAAI,CAAC,wBAAwB,CAAC;aACzC,SAAS,CAAC,WAAW,EAAE,EAAE,IAAI,EAAE,GAAG,EAAE,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC;QAE3D,OAAO;YACL,YAAY,EAAE;gBACZ,SAAS;gBACT,aAAa;gBACb,aAAa;aACd;SACF,CAAC;IACJ,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,SAAS,CACb,MAAsB,EACtB,MAAsB,EACtB,MAAc;QAEd,MAAM,SAAS,GAAG,MAAM,CAAC,YAAY,EAAE,SAAS,CAAC;QACjD,MAAM,aAAa,GAAG,MAAM,CAAC,YAAY,EAAE,aAAa,CAAC;QACzD,MAAM,aAAa,GAAG,MAAM,CAAC,YAAY,EAAE,aAAa,CAAC;QACzD,IACE,SAAS,KAAK,SAAS;YACvB,aAAa,KAAK,SAAS;YAC3B,aAAa,KAAK,SAAS,EAC3B,CAAC;YACD,MAAM,IAAI,KAAK,CACb,qHAAqH,CACtH,CAAC;QACJ,CAAC;QAED,MAAM,UAAU,GAAG,MAAM,OAAO,CAAC,GAAG,CAClC,MAAM,CAAC,GAAG,CAAC,KAAK,EAAE,CAAC,OAAO,EAAE,KAAK,CAAC,EAAE,GAAG,EAAE,EAAE;YACzC,MAAM,WAAW,GAAG;gBAClB,SAAS;gBACT,aAAa;gBACb,aAAa;gBACb,OAAO,EAAE,MAAM;gBACf,GAAG;aACJ,CAAC;YAEF,MAAM,CAAC,IAAI,EAAE,eAAe,CAAC,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC;YAEnE,OAAO;gBACL,SAAS,EAAE;oBACT,MAAM,EAAE,WAAW;oBACnB,MAAM,EAAE,EAAE,IAAI,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE,KAAK,EAAE,eAAe,EAAE,EAAE;oBAC3D,MAAM,EAAE,IAAI;iBACb;aACF,CAAC;QACJ,CAAC,CAAC,CACH,CAAC;QAEF,MAAM,IAAI,CAAC,EAAE;aACV,UAAU,CAAC,IAAI,CAAC,8BAA8B,CAAC;aAC/C,SAAS,CAAC,UAAU,CAAC,CAAC;IAC3B,CAAC;IAED,KAAK,CAAC,YAAY,CAAC,QAAgB;QACjC,MAAM,IAAI,CAAC,EAAE;aACV,UAAU,CAAC,IAAI,CAAC,wBAAwB,CAAC;aACzC,UAAU,CAAC,EAAE,SAAS,EAAE,QAAQ,EAAE,CAAC,CAAC;QAEvC,MAAM,IAAI,CAAC,EAAE;aACV,UAAU,CAAC,IAAI,CAAC,8BAA8B,CAAC;aAC/C,UAAU,CAAC,EAAE,SAAS,EAAE,QAAQ,EAAE,CAAC,CAAC;IACzC,CAAC;CACF"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@langchain/langgraph-checkpoint-mongodb",
-  "version": "0.0.6",
+  "version": "0.1.0",
   "description": "LangGraph",
   "type": "module",
   "engines": {
@@ -21,10 +21,9 @@
     "lint": "yarn lint:eslint && yarn lint:dpdm",
     "lint:fix": "yarn lint:eslint --fix && yarn lint:dpdm",
     "prepack": "yarn build",
-    "test": "NODE_OPTIONS=--experimental-vm-modules jest --testPathIgnorePatterns=\\.int\\.test.ts --testTimeout 30000 --maxWorkers=50%",
-    "test:watch": "NODE_OPTIONS=--experimental-vm-modules jest --watch --testPathIgnorePatterns=\\.int\\.test.ts",
-    "test:single": "NODE_OPTIONS=--experimental-vm-modules yarn run jest --config jest.config.cjs --testTimeout 100000",
-    "test:int": "NODE_OPTIONS=--experimental-vm-modules jest --testPathPattern=\\.int\\.test.ts --testTimeout 100000 --maxWorkers=50%",
+    "test": "vitest run",
+    "test:watch": "vitest watch",
+    "test:int": "vitest run --mode int",
     "format": "prettier --config .prettierrc --write \"src\"",
     "format:check": "prettier --config .prettierrc --check \"src\""
   },
@@ -35,16 +34,13 @@
   },
   "peerDependencies": {
     "@langchain/core": ">=0.2.31 <0.4.0",
-    "@langchain/langgraph-checkpoint": "~0.0.15"
+    "@langchain/langgraph-checkpoint": "^0.1.0"
   },
   "devDependencies": {
-    "@jest/globals": "^29.5.0",
-    "@langchain/langgraph-checkpoint": "workspace:*",
+    "@langchain/langgraph-checkpoint": "0.1.0",
     "@langchain/scripts": ">=0.1.3 <0.2.0",
-    "@swc/core": "^1.3.90",
-    "@swc/jest": "^0.2.29",
     "@tsconfig/recommended": "^1.0.3",
-    "@types/better-sqlite3": "^7.6.9",
+    "@types/better-sqlite3": "^7.6.12",
     "@types/uuid": "^10",
     "@typescript-eslint/eslint-plugin": "^6.12.0",
     "@typescript-eslint/parser": "^6.12.0",
@@ -54,17 +50,14 @@
     "eslint-config-airbnb-base": "^15.0.0",
     "eslint-config-prettier": "^8.6.0",
     "eslint-plugin-import": "^2.29.1",
-    "eslint-plugin-jest": "^28.8.0",
     "eslint-plugin-no-instanceof": "^1.0.1",
     "eslint-plugin-prettier": "^4.2.1",
-    "jest": "^29.5.0",
-    "jest-environment-node": "^29.6.4",
     "prettier": "^2.8.3",
-    "release-it": "^17.6.0",
-    "rollup": "^4.34.7",
-    "ts-jest": "^29.1.0",
-    "tsx": "^4.7.0",
-    "typescript": "^4.9.5 || ^5.4.5"
+    "release-it": "^19.0.2",
+    "rollup": "^4.37.0",
+    "tsx": "^4.19.3",
+    "typescript": "^4.9.5 || ^5.4.5",
+    "vitest": "^3.1.2"
   },
   "publishConfig": {
     "access": "public",
@@ -89,4 +82,4 @@
     "index.d.ts",
     "index.d.cts"
   ]
-}
+}