chore: add google-adk/{version} to bigquery user agent

google-genai-bot · copybara-github · commit f9fa7841df81 · 2025-06-17T18:31:34.000-07:00
PiperOrigin-RevId: 772703504
diff --git a/src/google/adk/tools/bigquery/client.py b/src/google/adk/tools/bigquery/client.py
@@ -18,7 +18,9 @@
 from google.cloud import bigquery
 from google.oauth2.credentials import Credentials
 
-USER_AGENT = "adk-bigquery-tool"
+from ... import version
+
+USER_AGENT = f"adk-bigquery-tool google-adk/{version.__version__}"
 
 
 def get_bigquery_client(
diff --git a/tests/unittests/tools/bigquery/test_bigquery_client.py b/tests/unittests/tools/bigquery/test_bigquery_client.py
@@ -14,6 +14,7 @@
 from __future__ import annotations
 
 import os
+import re
 from unittest import mock
 
 from google.adk.tools.bigquery.client import get_bigquery_client
@@ -122,4 +123,7 @@ def test_bigquery_client_user_agent():
     # Verify that the tracking user agent was set
     client_info_arg = mock_connection.call_args[1].get("client_info")
     assert client_info_arg is not None
-    assert client_info_arg.user_agent == "adk-bigquery-tool"
+    assert re.search(
+        r"adk-bigquery-tool google-adk/([0-9A-Za-z._\-+/]+)",
+        client_info_arg.user_agent,
+    )
