Skip to content

Commit 54db3d4

Browse files
google-genai-botcopybara-github
authored andcommitted
fix: update the contribution analysis tool to use original write mode
PiperOrigin-RevId: 828126582
1 parent 6e5c0eb commit 54db3d4

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

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

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1057,7 +1057,7 @@ def analyze_contribution(
10571057
# Create a session and run the create model query.
10581058
original_write_mode = settings.write_mode
10591059
try:
1060-
if settings.write_mode == WriteMode.BLOCKED:
1060+
if original_write_mode == WriteMode.BLOCKED:
10611061
raise ValueError("analyze_contribution is not allowed in this session.")
10621062
elif original_write_mode != WriteMode.PROTECTED:
10631063
# Running create temp model requires a session. So we set the write mode

0 commit comments

Comments
 (0)