JDSS/mem0
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add DriverInfo metadata to MongoDB vector store (#3648)

Browse Source
This commit is contained in:
Noah Stapp
2026-01-12 21:07:42 -08:00
committed by GitHub
parent 90a7b1afa0
commit a1db0f6362
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
mem0/vector_stores/mongodb.py
+4 -1
View File
@@ -1,10 +1,12 @@
import logging import logging
from importlib.metadata import version
from typing import Any, Dict, List, Optional from typing import Any, Dict, List, Optional
from pydantic import BaseModel from pydantic import BaseModel
try: try:
from pymongo import MongoClient from pymongo import MongoClient
from pymongo.driver_info import DriverInfo
from pymongo.errors import PyMongoError from pymongo.errors import PyMongoError
from pymongo.operations import SearchIndexModel from pymongo.operations import SearchIndexModel
except ImportError: except ImportError:
@@ -15,6 +17,7 @@ from mem0.vector_stores.base import VectorStoreBase
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)
logging.basicConfig(level=logging.INFO) logging.basicConfig(level=logging.INFO)
_DRIVER_METADATA = DriverInfo(name="Mem0", version=version("mem0"))
class OutputData(BaseModel): class OutputData(BaseModel):
id: Optional[str] id: Optional[str]
@@ -40,7 +43,7 @@ class MongoDB(VectorStoreBase):
self.embedding_model_dims = embedding_model_dims self.embedding_model_dims = embedding_model_dims
self.db_name = db_name self.db_name = db_name
self.client = MongoClient(mongo_uri) self.client = MongoClient(mongo_uri, driver=_DRIVER_METADATA)
self.db = self.client[db_name] self.db = self.client[db_name]
self.collection = self.create_col() self.collection = self.create_col()