encounter/adk-python
0
0
Fork 0
mirror of https://github.com/encounter/adk-python.git synced 2026-07-09 18:19:28 -07:00
Code Issues Packages Projects Releases Wiki Activity

refactor(bigquery): Remove dataset_id from logging schema

Browse Source 
Removes the dataset_id field from the BigQuery table schema and from each log entry created by the BigQueryAgentAnalyticsPlugin. This field is redundant, as all rows logged to a specific table belong to the same dataset.

To ensure the plugin can still target the correct dataset, dataset_id is now a required argument in the BigQueryAgentAnalyticsPlugin constructor, and its default value has been removed.

The BigQuery client user_agent is also updated with plugin version info to help identify traffic originating from this plugin. Unit tests have been updated to reflect the removal of dataset_id from log entries.

PiperOrigin-RevId: 826596499
This commit is contained in:
Google Team Member
2025-10-31 12:57:54 -07:00
committed by Copybara-Service
parent 156d235479
commit 0b56f2287b
2 changed files with 2 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/unittests/plugins/test_bigquery_logging_plugin.py
-1
View File
@@ -127,7 +127,6 @@ class TestBigQueryAgentAnalyticsPlugin(PluginTestBase):
return rows[0]
def _assert_common_fields(self, log_entry, event_type):
assert log_entry["dataset_id"] == self.dataset_id
assert log_entry["event_type"] == event_type
assert log_entry["agent"] == "MyTestAgent"
assert log_entry["session_id"] == "session-123"