feat:Add telemetry headers

The change adds custom telemetry headers (x-goog-api-client, user-agent) to BigQuery and BigQuery Storage API calls made by the plugin, including the ADK and Java versions.

PiperOrigin-RevId: 901368290

Author: google-genai-bot

core/src/main/java/com/google/adk/plugins/agentanalytics/BigQueryAgentAnalyticsPlugin.java

@@ -17,6 +17,7 @@
 package com.google.adk.plugins.agentanalytics;
 
 import static com.google.adk.plugins.agentanalytics.BigQueryUtils.createAnalyticsViews;
+import static com.google.adk.plugins.agentanalytics.BigQueryUtils.getVersionHeaderValue;
 import static com.google.adk.plugins.agentanalytics.BigQueryUtils.maybeUpgradeSchema;
 import static com.google.adk.plugins.agentanalytics.JsonFormatter.convertToJsonNode;
 import static com.google.adk.plugins.agentanalytics.JsonFormatter.smartTruncate;
@@ -40,6 +41,7 @@
 import com.google.adk.tools.ToolContext;
 import com.google.adk.tools.mcp.AbstractMcpTool;
 import com.google.adk.utils.AgentEnums.AgentOrigin;
+import com.google.api.gax.rpc.FixedHeaderProvider;
 import com.google.auth.oauth2.GoogleCredentials;
 import com.google.cloud.bigquery.BigQuery;
 import com.google.cloud.bigquery.BigQueryException;
@@ -110,6 +112,13 @@ public BigQueryAgentAnalyticsPlugin(BigQueryLoggerConfig config, BigQuery bigQue
 
   private static BigQuery createBigQuery(BigQueryLoggerConfig config) throws IOException {
     BigQueryOptions.Builder builder = BigQueryOptions.newBuilder();
+    builder.setHeaderProvider(
+        FixedHeaderProvider.create(
+            ImmutableMap.of(
+                "x-goog-api-client",
+                getVersionHeaderValue(),
+                "user-agent",
+                getVersionHeaderValue())));
     if (config.credentials() != null) {
       builder.setCredentials(config.credentials());
     } else {

core/src/main/java/com/google/adk/plugins/agentanalytics/BigQueryUtils.java

@@ -21,6 +21,9 @@
 import static com.google.common.collect.ImmutableList.toImmutableList;
 import static com.google.common.collect.ImmutableMap.toImmutableMap;
 import static java.util.stream.Collectors.toCollection;
+import static com.google.common.base.StandardSystemProperty.JAVA_VERSION;
+
+import com.google.adk.Version;
 
 import com.google.cloud.bigquery.BigQuery;
 import com.google.cloud.bigquery.BigQueryException;
@@ -136,6 +139,13 @@ final class BigQueryUtils {
                   "JSON_QUERY(content, '$.args') AS tool_args"))
           .buildOrThrow();
 
+  /** Returns the telemetry header value. */
+  static String getVersionHeaderValue() {
+    String frameworkLabel = "google-adk-bq-logger/" + Version.JAVA_ADK_VERSION;
+    String languageLabel = "gl-java/" + JAVA_VERSION.value();
+    return String.format("%s %s", frameworkLabel, languageLabel);
+  }
+
   /** Creates and/or replaces the analytics views in BigQuery. */
   static void createAnalyticsViews(BigQuery bigQuery, BigQueryLoggerConfig config) {
     for (Map.Entry<String, ImmutableList<String>> entry : EVENT_VIEW_DEFS.entrySet()) {

core/src/main/java/com/google/adk/plugins/agentanalytics/PluginState.java

@@ -1,16 +1,19 @@
 package com.google.adk.plugins.agentanalytics;
 
+import static com.google.adk.plugins.agentanalytics.BigQueryUtils.getVersionHeaderValue;
 import static java.util.concurrent.TimeUnit.MILLISECONDS;
 
 import com.google.api.gax.core.FixedCredentialsProvider;
 import com.google.api.gax.retrying.RetrySettings;
+import com.google.api.gax.rpc.FixedHeaderProvider;
 import com.google.cloud.bigquery.storage.v1.BigQueryWriteClient;
 import com.google.cloud.bigquery.storage.v1.BigQueryWriteSettings;
 import com.google.cloud.bigquery.storage.v1.StreamWriter;
 import com.google.common.annotations.VisibleForTesting;
 import com.google.common.base.VerifyException;
 import com.google.common.cache.Cache;
 import com.google.common.cache.CacheBuilder;
+import com.google.common.collect.ImmutableMap;
 import java.io.IOException;
 import java.util.Collection;
 import java.util.concurrent.ConcurrentHashMap;
@@ -67,13 +70,20 @@ void markProcessed(String invocationId) {
   }
 
   protected BigQueryWriteClient createWriteClient(BigQueryLoggerConfig config) throws IOException {
+    BigQueryWriteSettings.Builder settingsBuilder =
+        BigQueryWriteSettings.newBuilder()
+            .setHeaderProvider(
+                FixedHeaderProvider.create(
+                    ImmutableMap.of(
+                        "x-goog-api-client",
+                        getVersionHeaderValue(),
+                        "user-agent",
+                        getVersionHeaderValue())));
+
     if (config.credentials() != null) {
-      return BigQueryWriteClient.create(
-          BigQueryWriteSettings.newBuilder()
-              .setCredentialsProvider(FixedCredentialsProvider.create(config.credentials()))
-              .build());
+      settingsBuilder.setCredentialsProvider(FixedCredentialsProvider.create(config.credentials()));
     }
-    return BigQueryWriteClient.create();
+    return BigQueryWriteClient.create(settingsBuilder.build());
   }
 
   protected StreamWriter createWriter() {