Use pymongo-search-utils to append_client_metadata

libs/langgraph-store-mongodb/langgraph/store/mongodb/base.py

@@ -35,15 +35,13 @@
 )
 from pymongo.collection import Collection, ReturnDocument
 from pymongo_search_utils import (
+    append_client_metadata,
     autoembedding_vector_search_stage,
     create_vector_search_index,
     vector_search_stage,
 )
 
-from langgraph.store.mongodb.utils import (
-    DRIVER_METADATA,
-    _append_client_metadata,
-)
+from langgraph.store.mongodb.utils import DRIVER_METADATA
 
 logger = logging.getLogger(__name__)
 
@@ -176,7 +174,9 @@ def __init__(
         self.ttl_config = {} if ttl_config is None else ttl_config
         self.index_config = {} if index_config is None else index_config
 
-        _append_client_metadata(self.collection.database.client)
+        append_client_metadata(
+            client=self.collection.database.client, driver_info=DRIVER_METADATA
+        )
 
         # Create indexes if not present
         # Create a unique index, akin to primary key, on namespace + key

libs/langgraph-store-mongodb/langgraph/store/mongodb/utils.py

@@ -5,15 +5,8 @@
 
 from importlib.metadata import version
 
-from pymongo import MongoClient
 from pymongo.driver_info import DriverInfo
 
 DRIVER_METADATA = DriverInfo(
     name="Langgraph", version=version("langgraph-store-mongodb")
 )
-
-
-def _append_client_metadata(client: MongoClient) -> None:
-    # append_metadata was added in PyMongo 4.14.0, but is a valid database name on earlier versions
-    if callable(client.append_metadata):
-        client.append_metadata(DRIVER_METADATA)