t

langchain-ai · Aug 14, 2023 · 1dc11fc · 1dc11fc
1 parent 3f601b5
commit 1dc11fc
Show file tree

Hide file tree

Showing 3 changed files with 19 additions and 0 deletions.
diff --git a/libs/langchain/langchain/document_loaders/rocksetdb.py b/libs/langchain/langchain/document_loaders/rocksetdb.py
@@ -93,6 +93,12 @@ def __init__(
         self.paginator = QueryPaginator
         self.request_model = QueryRequestSql
 
+        try:
+            self.client.set_application('langchain')
+        except AttributeError:
+            # ignore
+            pass
+
     def load(self) -> List[Document]:
         return list(self.lazy_load())
 

diff --git a/libs/langchain/langchain/memory/chat_message_histories/rocksetdb.py b/libs/langchain/langchain/memory/chat_message_histories/rocksetdb.py
@@ -197,6 +197,13 @@ def __init__(
         self.message_uuid_method = message_uuid_method
         self.sync = sync
 
+        try:
+            self.client.set_application('langchain')
+        except AttributeError:
+            # ignore
+            pass
+
+
         if not self._collection_exists():
             self._create_collection()
             self._wait_until_collection_created()

diff --git a/libs/langchain/langchain/vectorstores/rocksetdb.py b/libs/langchain/langchain/vectorstores/rocksetdb.py
@@ -84,6 +84,12 @@ def __init__(
         self._embedding_key = embedding_key
         self._workspace = workspace
 
+        try:
+            self._client.set_application('langchain')
+        except AttributeError:
+            # ignore
+            pass
+
     @property
     def embeddings(self) -> Embeddings:
         return self._embeddings