feat: Add the ask_data_insights tool for natural language queries on BigQuery data

PiperOrigin-RevId: 799267061
This commit is contained in:
Google Team Member
2025-08-25 14:23:10 -07:00
committed by Copybara-Service
parent 6806deaf88
commit 47b88d2b06
5 changed files with 34 additions and 7 deletions
@@ -41,7 +41,7 @@ async def test_bigquery_toolset_tools_default():
tools = await toolset.get_tools()
assert tools is not None
assert len(tools) == 5
assert len(tools) == 6
assert all([isinstance(tool, GoogleTool) for tool in tools])
expected_tool_names = set([
@@ -50,6 +50,7 @@ async def test_bigquery_toolset_tools_default():
"list_table_ids",
"get_table_info",
"execute_sql",
"ask_data_insights",
])
actual_tool_names = set([tool.name for tool in tools])
assert actual_tool_names == expected_tool_names