Skip to content

Commit bead607

Browse files
google-genai-botcopybara-github
authored andcommitted
chore: Hide the ask_data_insights tool until the API is publicly available
PiperOrigin-RevId: 789806535
1 parent 041f04e commit bead607

File tree

3 files changed

+1
-14
lines changed

3 files changed

+1
-14
lines changed

contributing/samples/bigquery/README.md

Lines changed: 0 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -25,16 +25,6 @@ distributed via the `google.adk.tools.bigquery` module. These tools include:
2525

2626
Runs a SQL query in BigQuery.
2727

28-
1. `ask_data_insights`
29-
30-
Natural language-in, natural language-out tool that answers questions
31-
about structured data in BigQuery. Provides a one-stop solution for generating
32-
insights from data.
33-
34-
**Note**: This tool requires additional setup in your project. Please refer to
35-
the official [Conversational Analytics API documentation](https://cloud.google.com/gemini/docs/conversational-analytics-api/overview)
36-
for instructions.
37-
3828
## How to use
3929

4030
Set up environment variables in your `.env` file for using

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

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -21,7 +21,6 @@
2121
from google.adk.agents.readonly_context import ReadonlyContext
2222
from typing_extensions import override
2323

24-
from . import data_insights_tool
2524
from . import metadata_tool
2625
from . import query_tool
2726
from ...tools.base_tool import BaseTool
@@ -79,7 +78,6 @@ async def get_tools(
7978
metadata_tool.list_dataset_ids,
8079
metadata_tool.list_table_ids,
8180
query_tool.get_execute_sql(self._tool_config),
82-
data_insights_tool.ask_data_insights,
8381
]
8482
]
8583

tests/unittests/tools/bigquery/test_bigquery_toolset.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -34,7 +34,7 @@ async def test_bigquery_toolset_tools_default():
3434
tools = await toolset.get_tools()
3535
assert tools is not None
3636

37-
assert len(tools) == 6
37+
assert len(tools) == 5
3838
assert all([isinstance(tool, BigQueryTool) for tool in tools])
3939

4040
expected_tool_names = set([
@@ -43,7 +43,6 @@ async def test_bigquery_toolset_tools_default():
4343
"list_table_ids",
4444
"get_table_info",
4545
"execute_sql",
46-
"ask_data_insights",
4746
])
4847
actual_tool_names = set([tool.name for tool in tools])
4948
assert actual_tool_names == expected_tool_names

0 commit comments

Comments
 (0)