diff --git a/src/google/adk/tools/bigquery/client.py b/src/google/adk/tools/bigquery/client.py index ea1bebc7..23f1befc 100644 --- a/src/google/adk/tools/bigquery/client.py +++ b/src/google/adk/tools/bigquery/client.py @@ -18,7 +18,9 @@ import google.api_core.client_info 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 index 612dddd6..e8b37341 100644 --- 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, + )