INTPYTHON-689 Remove Motor dependency (#170)

Author: blink1073

.github/workflows/_lint.yml

@@ -30,7 +30,7 @@ jobs:
         # so linting on fewer versions makes CI faster.
         python-version:
           - "3.9"
-          - "3.12"
+          - "3.13"
     steps:
       - uses: actions/checkout@v4
         with:

.github/workflows/_test.yml

@@ -19,7 +19,7 @@ jobs:
       matrix:
         python-version:
           - "3.9"
-          - "3.12"
+          - "3.13"
     name: "run test #${{ matrix.python-version }}"
     steps:
       - uses: actions/checkout@v4

libs/langchain-mongodb/langchain_mongodb/vectorstores.py

@@ -407,7 +407,7 @@ def get_by_ids(self, ids: Sequence[str], /) -> list[Document]:
             _id = doc.pop("_id")
             text = doc.pop("text")
             del doc["embedding"]
-            docs.append(Document(page_content=text, id=_id, metadata=doc))
+            docs.append(Document(page_content=text, id=oid_to_str(_id), metadata=doc))
         return docs
 
     def bulk_embed_and_insert_texts(

libs/langchain-mongodb/pyproject.toml

@@ -32,10 +32,10 @@ dev = [
     "mongomock>=4.2.0.post1",
     "pre-commit>=4.0",
     "mypy>=1.10",
-    "simsimd>=5.0.0",
+    "simsimd>=6.5.0",
     "langchain-ollama>=0.2.2",
     "langchain-openai>=0.2.14",
-    "langchain-community>=0.3.14",
+    "langchain-community>=0.3.27",
     "pypdf>=5.0.1",
     "langgraph>=0.2.72",
     "flaky>=3.8.1",

libs/langchain-mongodb/uv.lock

@@ -7,10 +7,9 @@
 from contextlib import asynccontextmanager
 from datetime import datetime
 from importlib.metadata import version
-from typing import Any, Optional
+from typing import Any, Optional, cast
 
 from langchain_core.runnables import RunnableConfig
-from motor.motor_asyncio import AsyncIOMotorClient, AsyncIOMotorDatabase
 from pymongo import UpdateOne
 from pymongo.asynchronous.database import AsyncDatabase
 from pymongo.asynchronous.mongo_client import AsyncMongoClient
@@ -73,12 +72,12 @@ class AsyncMongoDBSaver(BaseCheckpointSaver):
         input=3, output=4
     """
 
-    client: AsyncIOMotorClient | AsyncMongoClient
-    db: AsyncIOMotorDatabase | AsyncDatabase
+    client: AsyncMongoClient
+    db: AsyncDatabase
 
     def __init__(
         self,
-        client: AsyncIOMotorClient | AsyncMongoClient,
+        client: AsyncMongoClient,
         db_name: str = "checkpointing_db",
         checkpoint_collection_name: str = "checkpoints_aio",
         writes_collection_name: str = "checkpoint_writes_aio",
@@ -101,10 +100,10 @@ async def _setup(self) -> None:
         self._setup_future = asyncio.Future()
         if isinstance(self.client, AsyncMongoClient):
             num_indexes = len(
-                await (await self.checkpoint_collection.list_indexes()).to_list()  # type:ignore[misc]
+                await (await self.checkpoint_collection.list_indexes()).to_list()
             )
         else:
-            num_indexes = len(await self.checkpoint_collection.list_indexes().to_list())  # type:ignore[union-attr]
+            num_indexes = len(await self.checkpoint_collection.list_indexes().to_list())
         if num_indexes < 2:
             await self.checkpoint_collection.create_index(
                 keys=[("thread_id", 1), ("checkpoint_ns", 1), ("checkpoint_id", -1)],
@@ -117,10 +116,10 @@ async def _setup(self) -> None:
                 )
         if isinstance(self.client, AsyncMongoClient):
             num_indexes = len(
-                await (await self.writes_collection.list_indexes()).to_list()  # type:ignore[misc]
+                await (await self.writes_collection.list_indexes()).to_list()
             )
         else:
-            num_indexes = len(await self.writes_collection.list_indexes().to_list())  # type:ignore[union-attr]
+            num_indexes = len(await self.writes_collection.list_indexes().to_list())
         if num_indexes < 2:
             await self.writes_collection.create_index(
                 keys=[
@@ -154,9 +153,9 @@ async def from_conn_string(
 
         This includes creation of collections and indexes if they don't exist
         """
-        client: Optional[AsyncIOMotorClient] = None
+        client: Optional[AsyncMongoClient] = None
         try:
-            client = AsyncIOMotorClient(
+            client = AsyncMongoClient(
                 conn_string,
                 driver=DriverInfo(
                     name="Langgraph", version=version("langgraph-checkpoint-mongodb")
@@ -175,7 +174,7 @@ async def from_conn_string(
 
         finally:
             if client:
-                client.close()
+                await client.close()
 
     async def aget_tuple(self, config: RunnableConfig) -> Optional[CheckpointTuple]:
         """Get a checkpoint tuple from the database asynchronously.
@@ -479,7 +478,7 @@ def list(
         while True:
             try:
                 yield asyncio.run_coroutine_threadsafe(
-                    anext(aiter_),
+                    cast(Any, anext(aiter_)),
                     self.loop,
                 ).result()
             except StopAsyncIteration:

libs/langgraph-checkpoint-mongodb/langgraph/checkpoint/mongodb/aio.py

@@ -11,8 +11,7 @@ requires-python = ">=3.9"
 dependencies = [
     "langgraph-checkpoint>=2.0.23,<3.0.0",
     "langchain-mongodb>=0.6.1",
-    "pymongo>=4.10,<4.14",
-    "motor>3.6.0",
+    "pymongo>=4.12,<4.14",
 ]
 
 [dependency-groups]

libs/langgraph-checkpoint-mongodb/pyproject.toml

@@ -11,7 +11,6 @@
 from langgraph.checkpoint.memory import InMemorySaver
 from langgraph.checkpoint.mongodb import MongoDBSaver
 from langgraph.graph import END, StateGraph
-from langgraph.graph.graph import CompiledGraph
 
 # --- Configuration ---
 MONGODB_URI = os.environ.get("MONGODB_URI", "mongodb://localhost:27017")
@@ -77,16 +76,14 @@ def node_double(state: State) -> State:
 
     # --- Compile Graph (with Interruption) ---
     # Using sync for simplicity in this demo
-    graph: CompiledGraph = builder.compile(
-        checkpointer=checkpointer, interrupt_after=["increment"]
-    )
+    graph = builder.compile(checkpointer=checkpointer, interrupt_after=["increment"])
 
     # --- Configure  ---
     config: RunnableConfig = {"configurable": {"thread_id": "thread_#1"}}
     initial_input = {"value": 10, "step": 0}
 
     # --- 1st invoke, with Interruption
-    interrupted_state = graph.invoke(initial_input, config=config)
+    interrupted_state = graph.invoke(initial_input, config=config)  # type:ignore[arg-type]
     assert interrupted_state == {"value": 10 + 1, "step": 1}
     state_history = list(graph.get_state_history(config))
     assert len(state_history) == 3
@@ -103,7 +100,7 @@ def node_double(state: State) -> State:
 
     # --- 3rd invoke, but with an input ===> the CompiledGraph is restarted.
     new_input = {"value": 100, "step": -100}
-    third_state = graph.invoke(new_input, config=config)
+    third_state = graph.invoke(new_input, config=config)  # type:ignore[arg-type]
     assert third_state == {"value": 101, "step": -99}
 
     # The entire state history is preserved however

libs/langgraph-checkpoint-mongodb/tests/integration_tests/test_interrupts.py

@@ -3,7 +3,7 @@
 
 import pytest
 from bson.errors import InvalidDocument
-from motor.motor_asyncio import AsyncIOMotorClient
+from pymongo import AsyncMongoClient
 
 from langgraph.checkpoint.mongodb.aio import AsyncMongoDBSaver
 
@@ -14,7 +14,7 @@
 
 async def test_asearch(input_data: dict[str, Any]) -> None:
     # Clear collections if they exist
-    client: AsyncIOMotorClient = AsyncIOMotorClient(MONGODB_URI)
+    client: AsyncMongoClient = AsyncMongoClient(MONGODB_URI)
     db = client[DB_NAME]
 
     for clxn in await db.list_collection_names():

libs/langgraph-checkpoint-mongodb/tests/unit_tests/test_async.py

@@ -97,7 +97,7 @@ def test_null_chars(input_data: dict[str, Any]) -> None:
         )
         assert saver.get_tuple(null_value_cfg).metadata["my_key"] == null_str  # type: ignore
         assert (
-            list(saver.list(None, filter={"my_key": null_str}))[0].metadata["my_key"]  # type: ignore
+            list(saver.list(None, filter={"my_key": null_str}))[0].metadata["my_key"]
             == null_str
         )