chore: add user agent suffix (#842)

daniel-sanche · web-flow · commit 0b3606fd14e7 · 2023-08-17T11:00:21.000-07:00
diff --git a/google/cloud/bigtable/data/_async/client.py b/google/cloud/bigtable/data/_async/client.py
@@ -122,7 +122,7 @@ def __init__(
         BigtableClientMeta._transport_registry[transport_str] = transport
         # set up client info headers for veneer library
         client_info = DEFAULT_CLIENT_INFO
-        client_info.client_library_version = client_info.gapic_version
+        client_info.client_library_version = self._client_version()
         # parse client options
         if type(client_options) is dict:
             client_options = client_options_lib.from_dict(client_options)
@@ -163,6 +163,13 @@ def __init__(
                 stacklevel=2,
             )
 
+    @staticmethod
+    def _client_version() -> str:
+        """
+        Helper function to return the client version string for this client
+        """
+        return f"{google.cloud.bigtable.__version__}-data-async"
+
     def _start_background_channel_refresh(self) -> None:
         """
         Starts a background task to ping and warm each channel in the pool
diff --git a/tests/unit/data/_async/test_client.py b/tests/unit/data/_async/test_client.py
@@ -39,7 +39,7 @@
     from mock import AsyncMock  # type: ignore
 
 VENEER_HEADER_REGEX = re.compile(
-    r"gapic\/[0-9]+\.[\w.-]+ gax\/[0-9]+\.[\w.-]+ gccl\/[0-9]+\.[\w.-]+ gl-python\/[0-9]+\.[\w.-]+ grpc\/[0-9]+\.[\w.-]+"
+    r"gapic\/[0-9]+\.[\w.-]+ gax\/[0-9]+\.[\w.-]+ gccl\/[0-9]+\.[\w.-]+-data-async gl-python\/[0-9]+\.[\w.-]+ grpc\/[0-9]+\.[\w.-]+"
 )
 
 

Original file line number	Diff line number	Diff line change
`@@ -39,7 +39,7 @@`
`39`	`39`	`from mock import AsyncMock # type: ignore`
`40`	`40`
`41`	`41`	`VENEER_HEADER_REGEX = re.compile(`
`42`		`- r"gapic\/[0-9]+\.[\w.-]+ gax\/[0-9]+\.[\w.-]+ gccl\/[0-9]+\.[\w.-]+ gl-python\/[0-9]+\.[\w.-]+ grpc\/[0-9]+\.[\w.-]+"`
	`42`	`+ r"gapic\/[0-9]+\.[\w.-]+ gax\/[0-9]+\.[\w.-]+ gccl\/[0-9]+\.[\w.-]+-data-async gl-python\/[0-9]+\.[\w.-]+ grpc\/[0-9]+\.[\w.-]+"`
`43`	`43`	`)`
`44`	`44`
`45`	`45`