From 54db3d4434e0706b83a589fa2499d11d439a6e4e Mon Sep 17 00:00:00 2001 From: Google Team Member Date: Tue, 4 Nov 2025 14:04:56 -0800 Subject: [PATCH] fix: update the contribution analysis tool to use original write mode PiperOrigin-RevId: 828126582 --- src/google/adk/tools/bigquery/query_tool.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/google/adk/tools/bigquery/query_tool.py b/src/google/adk/tools/bigquery/query_tool.py index cf2f23f8..4a4913d3 100644 --- a/src/google/adk/tools/bigquery/query_tool.py +++ b/src/google/adk/tools/bigquery/query_tool.py @@ -1057,7 +1057,7 @@ def analyze_contribution( # Create a session and run the create model query. original_write_mode = settings.write_mode try: - if settings.write_mode == WriteMode.BLOCKED: + if original_write_mode == WriteMode.BLOCKED: raise ValueError("analyze_contribution is not allowed in this session.") elif original_write_mode != WriteMode.PROTECTED: # Running create temp model requires a session. So we set the write mode