feat: set per-tool user agent in BQ calls and tool label in BQ jobs

This will help per tool usage for BigQuery tools.

PiperOrigin-RevId: 829142106

Author: google-genai-bot

src/google/adk/tools/bigquery/client.py

@@ -25,12 +25,16 @@
 USER_AGENT = f"adk-bigquery-tool google-adk/{version.__version__}"
 
 
+from typing import List
+from typing import Union
+
+
 def get_bigquery_client(
     *,
     project: Optional[str],
     credentials: Credentials,
     location: Optional[str] = None,
-    user_agent: Optional[str] = None,
+    user_agent: Optional[Union[str, List[str]]] = None,
 ) -> bigquery.Client:
   """Get a BigQuery client.
 
@@ -44,9 +48,16 @@ def get_bigquery_client(
     A BigQuery client.
   """
 
-  user_agent = f"{USER_AGENT} {user_agent}" if user_agent else USER_AGENT
+  user_agents = [USER_AGENT]
+  if user_agent:
+    if isinstance(user_agent, str):
+      user_agents.append(user_agent)
+    else:
+      user_agents.extend([ua for ua in user_agent if ua])
 
-  client_info = google.api_core.client_info.ClientInfo(user_agent=user_agent)
+  client_info = google.api_core.client_info.ClientInfo(
+      user_agent=" ".join(user_agents)
+  )
 
   bigquery_client = bigquery.Client(
       project=project,

src/google/adk/tools/bigquery/metadata_tool.py

@@ -51,7 +51,7 @@ def list_dataset_ids(
         project=project_id,
         credentials=credentials,
         location=settings.location,
-        user_agent=settings.application_name,
+        user_agent=[settings.application_name, "list_dataset_ids"],
     )
 
     datasets = []
@@ -123,7 +123,7 @@ def get_dataset_info(
         project=project_id,
         credentials=credentials,
         location=settings.location,
-        user_agent=settings.application_name,
+        user_agent=[settings.application_name, "get_dataset_info"],
     )
     dataset = bq_client.get_dataset(
         bigquery.DatasetReference(project_id, dataset_id)
@@ -162,7 +162,7 @@ def list_table_ids(
         project=project_id,
         credentials=credentials,
         location=settings.location,
-        user_agent=settings.application_name,
+        user_agent=[settings.application_name, "list_table_ids"],
     )
 
     tables = []
@@ -285,7 +285,7 @@ def get_table_info(
         project=project_id,
         credentials=credentials,
         location=settings.location,
-        user_agent=settings.application_name,
+        user_agent=[settings.application_name, "get_table_info"],
     )
     return bq_client.get_table(
         bigquery.TableReference(
@@ -579,8 +579,9 @@ def get_job_info(
         project=project_id,
         credentials=credentials,
         location=settings.location,
-        user_agent=settings.application_name,
+        user_agent=[settings.application_name, "get_job_info"],
     )
+
     job = bq_client.get_job(job_id)
     # We need to use _properties to get the job info because it contains all
     # the job info.