feat(aiplatform): Add support for setting User agent header (#364)

googleapis · May 6, 2021 · d50d26d · d50d26d
1 parent 6fda925
commit d50d26d
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/google/cloud/aiplatform/constants.py b/google/cloud/aiplatform/constants.py
@@ -75,3 +75,6 @@
         "MOBILE_JETSON_LOW_LATENCY_1",
     },
 }
+
+# Used in constructing the requests user_agent header for metrics reporting.
+USER_AGENT_PRODUCT = "model-builder"
diff --git a/google/cloud/aiplatform/initializer.py b/google/cloud/aiplatform/initializer.py
@@ -259,7 +259,8 @@ def create_client(
             "google-cloud-aiplatform",
         ).version
         client_info = gapic_v1.client_info.ClientInfo(
-            gapic_version=gapic_version, user_agent=f"model-builder/{gapic_version}"
+            gapic_version=gapic_version,
+            user_agent=f"{constants.USER_AGENT_PRODUCT}/{gapic_version}",
         )
 
         kwargs = {