# Copyright 2025 Google LLC
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

kind: source
name: bigquery-source
type: bigquery
project: ${BIGQUERY_PROJECT}
location: ${BIGQUERY_LOCATION:}
useClientOAuth: ${BIGQUERY_USE_CLIENT_OAUTH:false}
scopes: ${BIGQUERY_SCOPES:}
maxQueryResultRows: ${BIGQUERY_MAX_QUERY_RESULT_ROWS:50}
impersonateServiceAccount: ${BIGQUERY_IMPERSONATE_SERVICE_ACCOUNT:}
maximumBytesBilled: ${BIGQUERY_MAXIMUM_BYTES_BILLED:0}
---
kind: tool
name: analyze_contribution
type: bigquery-analyze-contribution
source: bigquery-source
description: Use this tool to analyze the contribution about changes to key metrics in multi-dimensional data.
---
kind: tool
name: ask_data_insights
type: bigquery-conversational-analytics
source: bigquery-source
description: |
  Use this tool to perform data analysis, get insights,
  or answer complex questions about the contents of specific
  BigQuery tables.
---
kind: tool
name: execute_sql
type: bigquery-execute-sql
source: bigquery-source
description: Use this tool to execute sql statement.
---
kind: tool
name: forecast
type: bigquery-forecast
source: bigquery-source
description: Use this tool to forecast time series data.
---
kind: tool
name: get_dataset_info
type: bigquery-get-dataset-info
source: bigquery-source
description: Use this tool to get dataset metadata.
---
kind: tool
name: get_table_info
type: bigquery-get-table-info
source: bigquery-source
description: Use this tool to get table metadata.
---
kind: tool
name: list_dataset_ids
type: bigquery-list-dataset-ids
source: bigquery-source
description: Use this tool to list datasets.
---
kind: tool
name: list_table_ids
type: bigquery-list-table-ids
source: bigquery-source
description: Use this tool to list tables.
---
kind: tool
name: search_catalog
type: bigquery-search-catalog
source: bigquery-source
description: Use this tool to find tables, views, models, routines or connections.
---
kind: toolset
name: data
tools:
- execute_sql
- list_dataset_ids
- list_table_ids
- get_dataset_info
- get_table_info
- search_catalog
---
kind: toolset
name: analytics
tools:
- analyze_contribution
- ask_data_insights
- forecast
- search_catalog