From d50d26d18b892767dcf2e37d30d85b2b405e2708 Mon Sep 17 00:00:00 2001
From: sina chavoshi <sina.chavoshi@gmail.com>
Date: Thu, 6 May 2021 06:09:15 -0700
Subject: [PATCH] feat(aiplatform): Add support for setting User agent header
 (#364)

---
 google/cloud/aiplatform/constants.py   | 3 +++
 google/cloud/aiplatform/initializer.py | 3 ++-
 2 files changed, 5 insertions(+), 1 deletion(-)

diff --git a/google/cloud/aiplatform/constants.py b/google/cloud/aiplatform/constants.py
index 9e71427d0f..a7d81084cd 100644
--- 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
index fa00fa3387..41c698b6d3 100644
--- 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 = {