File tree Expand file tree Collapse file tree 3 files changed +1
-14
lines changed
contributing/samples/bigquery
src/google/adk/tools/bigquery
tests/unittests/tools/bigquery Expand file tree Collapse file tree 3 files changed +1
-14
lines changed Original file line number Diff line number Diff line change @@ -25,16 +25,6 @@ distributed via the `google.adk.tools.bigquery` module. These tools include:
25
25
26
26
Runs a SQL query in BigQuery.
27
27
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
-
38
28
## How to use
39
29
40
30
Set up environment variables in your ` .env ` file for using
Original file line number Diff line number Diff line change 21
21
from google .adk .agents .readonly_context import ReadonlyContext
22
22
from typing_extensions import override
23
23
24
- from . import data_insights_tool
25
24
from . import metadata_tool
26
25
from . import query_tool
27
26
from ...tools .base_tool import BaseTool
@@ -79,7 +78,6 @@ async def get_tools(
79
78
metadata_tool .list_dataset_ids ,
80
79
metadata_tool .list_table_ids ,
81
80
query_tool .get_execute_sql (self ._tool_config ),
82
- data_insights_tool .ask_data_insights ,
83
81
]
84
82
]
85
83
Original file line number Diff line number Diff line change @@ -34,7 +34,7 @@ async def test_bigquery_toolset_tools_default():
34
34
tools = await toolset .get_tools ()
35
35
assert tools is not None
36
36
37
- assert len (tools ) == 6
37
+ assert len (tools ) == 5
38
38
assert all ([isinstance (tool , BigQueryTool ) for tool in tools ])
39
39
40
40
expected_tool_names = set ([
@@ -43,7 +43,6 @@ async def test_bigquery_toolset_tools_default():
43
43
"list_table_ids" ,
44
44
"get_table_info" ,
45
45
"execute_sql" ,
46
- "ask_data_insights" ,
47
46
])
48
47
actual_tool_names = set ([tool .name for tool in tools ])
49
48
assert actual_tool_names == expected_tool_names
You can’t perform that action at this time.
0 commit comments