Skip to content

Commit 1e37d4b

Browse files
committed
INTPYTHON-648 Add driver info
1 parent a844933 commit 1e37d4b

File tree

1 file changed

+6
-2
lines changed

1 file changed

+6
-2
lines changed

pymongo_voyageai_multimodal/client.py

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -3,6 +3,7 @@
33
import io
44
import logging
55
from collections.abc import Mapping, Sequence
6+
from importlib.metadata import version
67
from time import monotonic, sleep
78
from typing import Any
89

@@ -14,6 +15,7 @@
1415
from langchain_mongodb.vectorstores import DEFAULT_INSERT_BATCH_SIZE
1516
from PIL import Image
1617
from pymongo import MongoClient, ReplaceOne
18+
from pymongo.driver_info import DriverInfo
1719
from voyageai.client import Client
1820

1921
from .document import Document, DocumentType, ImageDocument, StoredDocument, TextDocument
@@ -171,8 +173,10 @@ def __init__(
171173
"""
172174
self._dimensions = dimensions # the size of the VoyageAI model.
173175
self._vo = voyageai_client or Client(api_key=voyageai_api_key)
174-
# TODO: driver=DriverInfo(name="Langchain", version=version("langchain-mongodb")),
175-
self._mo = mongo_client or MongoClient(mongo_connection_string)
176+
driver = (
177+
DriverInfo(name="multimodal-search", version=version("pymongo-voyageai-multimodal")),
178+
)
179+
self._mo = mongo_client or MongoClient(mongo_connection_string, driver=driver)
176180
self._index_name = index_name
177181
self._embedding_key = embedding_key
178182
self._relevance_score_fn = relevance_score_fn

0 commit comments

Comments
 (0)