diff --git a/libs/langgraph-store-mongodb/langgraph/store/mongodb/base.py b/libs/langgraph-store-mongodb/langgraph/store/mongodb/base.py index 4b87c748..154da611 100644 --- a/libs/langgraph-store-mongodb/langgraph/store/mongodb/base.py +++ b/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 diff --git a/libs/langgraph-store-mongodb/langgraph/store/mongodb/utils.py b/libs/langgraph-store-mongodb/langgraph/store/mongodb/utils.py index 1338d9bd..a4df9eb8 100644 --- a/libs/langgraph-store-mongodb/langgraph/store/mongodb/utils.py +++ b/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)