feat: add chat first-party tool

This tool answers questions about structured data in BigQuery using natural language.

PiperOrigin-RevId: 789000987

Author: google-genai-bot

contributing/samples/bigquery/README.md

@@ -25,6 +25,16 @@ 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

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

@@ -65,7 +65,9 @@ def __init__(
         if credentials_config
         else None
     )
-    self._tool_config = bigquery_tool_config
+    self._tool_config = (
+        bigquery_tool_config if bigquery_tool_config else BigQueryToolConfig()
+    )
 
   @override
   async def run_async(

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

@@ -21,6 +21,7 @@
 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
@@ -78,6 +79,7 @@ async def get_tools(
             metadata_tool.list_dataset_ids,
             metadata_tool.list_table_ids,
             query_tool.get_execute_sql(self._tool_config),
+            data_insights_tool.ask_data_insights,
         ]
     ]
 

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

@@ -14,6 +14,8 @@
 
 from __future__ import annotations
 
+from typing import Optional
+
 import google.api_core.client_info
 from google.auth.credentials import Credentials
 from google.cloud import bigquery
@@ -24,7 +26,7 @@
 
 
 def get_bigquery_client(
-    *, project: str, credentials: Credentials
+    *, project: Optional[str], credentials: Credentials
 ) -> bigquery.Client:
   """Get a BigQuery client."""
 

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

@@ -54,3 +54,8 @@ class BigQueryToolConfig(BaseModel):
   By default, the tool will allow only read operations. This behaviour may
   change in future versions.
   """
+
+  max_query_result_rows: int = 50
+  """Maximum number of rows to return from a query.
+
+  By default, the query result will be limited to 50 rows."""