chore: Hide the ask_data_insights tool until the API is publicly available

PiperOrigin-RevId: 789806535
This commit is contained in:
Google Team Member
2025-08-01 10:17:13 -07:00
committed by Copybara-Service
parent 041f04e89c
commit bead607364
3 changed files with 1 additions and 14 deletions
-10
View File
@@ -25,16 +25,6 @@ distributed via the `google.adk.tools.bigquery` module. These tools include:
Runs a SQL query in BigQuery.
1. `ask_data_insights`
Natural language-in, natural language-out tool that answers questions
about structured data in BigQuery. Provides a one-stop solution for generating
insights from data.
**Note**: This tool requires additional setup in your project. Please refer to
the official [Conversational Analytics API documentation](https://cloud.google.com/gemini/docs/conversational-analytics-api/overview)
for instructions.
## How to use
Set up environment variables in your `.env` file for using
@@ -21,7 +21,6 @@ from typing import Union
from google.adk.agents.readonly_context import ReadonlyContext
from typing_extensions import override
from . import data_insights_tool
from . import metadata_tool
from . import query_tool
from ...tools.base_tool import BaseTool
@@ -79,7 +78,6 @@ class BigQueryToolset(BaseToolset):
metadata_tool.list_dataset_ids,
metadata_tool.list_table_ids,
query_tool.get_execute_sql(self._tool_config),
data_insights_tool.ask_data_insights,
]
]
@@ -34,7 +34,7 @@ async def test_bigquery_toolset_tools_default():
tools = await toolset.get_tools()
assert tools is not None
assert len(tools) == 6
assert len(tools) == 5
assert all([isinstance(tool, BigQueryTool) for tool in tools])
expected_tool_names = set([
@@ -43,7 +43,6 @@ 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