chore: Update user-agent header

sararob · copybara-github · commit e48389c6b76d · 2026-04-07T13:05:52.000-07:00
PiperOrigin-RevId: 896025555
diff --git a/vertexai/_genai/client.py b/vertexai/_genai/client.py
@@ -41,26 +41,29 @@
 
 _GENAI_MODULES_TELEMETRY_HEADER = "vertex-genai-modules"
 
-genai_append_method = _api_client.append_library_version_headers
+# genai_append_method = _api_client.append_library_version_headers
 
 
-def _add_tracking_headers(headers: dict[str, str]) -> None:
-    """Adds Vertex Gen AI tracking headers."""
-    is_vertex = headers.get(
-        "x-goog-vertex-sdk"
-    ) == "true" or "vertex-genai-modules" in headers.get("user-agent", "")
-    if is_vertex:
-        tracking_label = f"{_GENAI_MODULES_TELEMETRY_HEADER}/{aip_version.__version__}"
+# def _add_tracking_headers(headers: dict[str, str]) -> None:
+#     """Adds Vertex Gen AI tracking headers."""
+#     is_vertex = headers.get(
+#         "x-goog-vertex-sdk"
+#     ) == "true" or "vertex-genai-modules" in headers.get("user-agent", "")
+#     if is_vertex:
+#         tracking_label = (
+#             f"{_GENAI_MODULES_TELEMETRY_HEADER}/{aip_version.__version__} "
+#             f"gl-python/{aip_version.__version__}"
+#         )
 
-        headers["user-agent"] = tracking_label
-        headers["x-goog-api-client"] = tracking_label
+#         headers["user-agent"] = tracking_label
+#         headers["x-goog-api-client"] = tracking_label
 
-        headers.pop("x-goog-vertex-sdk", None)
-    else:
-        genai_append_method(headers)
+#         headers.pop("x-goog-vertex-sdk", None)
+#     else:
+#         genai_append_method(headers)
 
 
-_api_client.append_library_version_headers = _add_tracking_headers
+# _api_client.append_library_version_headers = _add_tracking_headers
 
 
 class AsyncClient:
@@ -229,7 +232,13 @@ def __init__(
             http_options = types.HttpOptions()
         if http_options.headers is None:
             http_options.headers = {}
-        http_options.headers["x-goog-vertex-sdk"] = "true"
+
+        tracking_label = (
+            f"{_GENAI_MODULES_TELEMETRY_HEADER}/{aip_version.__version__}"
+        )
+
+        http_options.headers["user-agent"] = tracking_label
+        http_options.headers["x-goog-api-client"] = tracking_label
 
         self._api_client = genai_client.Client._get_api_client(
             vertexai=True,
