Add extra info to client version

singlestore-labs · Feb 20, 2025 · 2f8375e · 2f8375e
1 parent 6719dfe
commit 2f8375e
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/singlestoredb/http/connection.py b/singlestoredb/http/connection.py
@@ -972,6 +972,10 @@ class Connection(connection.Connection):
 
     def __init__(self, **kwargs: Any):
         from .. import __version__ as client_version
+
+        if 'SINGLESTOREDB_WORKLOAD_TYPE' in os.environ:
+            client_version += '+' + os.environ['SINGLESTOREDB_WORKLOAD_TYPE']
+
         connection.Connection.__init__(self, **kwargs)
 
         host = kwargs.get('host', get_option('host'))

diff --git a/singlestoredb/mysql/connection.py b/singlestoredb/mysql/connection.py
@@ -625,6 +625,9 @@ def _config(key, arg):
 
         from .. import __version__ as VERSION_STRING
 
+        if 'SINGLESTOREDB_WORKLOAD_TYPE' in os.environ:
+            VERSION_STRING += '+' + os.environ['SINGLESTOREDB_WORKLOAD_TYPE']
+
         self._connect_attrs = {
             '_os': str(sys.platform),
             '_pid': str(os.getpid()),