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

SinaChavoshi · web-flow · commit d50d26d18b89 · 2021-05-06T09:09:15.000-04:00
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 = {

Original file line number	Diff line number	Diff line change
`@@ -75,3 +75,6 @@`
`75`	`75`	`"MOBILE_JETSON_LOW_LATENCY_1",`
`76`	`76`	`},`
`77`	`77`	`}`
	`78`	`+`
	`79`	`+# Used in constructing the requests user_agent header for metrics reporting.`
	`80`	`+USER_AGENT_PRODUCT = "model-builder"`
Original file line number	Diff line number	Diff line change
`@@ -259,7 +259,8 @@ def create_client(`
`259`	`259`	`"google-cloud-aiplatform",`
`260`	`260`	`).version`
`261`	`261`	`client_info = gapic_v1.client_info.ClientInfo(`
`262`		`- gapic_version=gapic_version, user_agent=f"model-builder/{gapic_version}"`
	`262`	`+ gapic_version=gapic_version,`
	`263`	`+ user_agent=f"{constants.USER_AGENT_PRODUCT}/{gapic_version}",`
`263`	`264`	`)`
`264`	`265`
`265`	`266`	`kwargs = {`