mirror of
https://github.com/googleapis/genai-toolbox.git
synced 2026-05-02 03:00:36 -04:00
4abf0c39e7
This change allows the agent developer to control the maxium number of rows returned from tools running BigQuery SQL query. Using this feature the agent developer could limit how large output is presented to LLM in an agentic user journey. ## Description > Should include a concise description of the changes (bug or feature), it's > impact, along with a summary of the solution ## PR Checklist > Thank you for opening a Pull Request! Before submitting your PR, there are a > few things you can do to make sure it goes smoothly: - [x] Make sure you reviewed [CONTRIBUTING.md](https://github.com/googleapis/genai-toolbox/blob/main/CONTRIBUTING.md) - [ ] Make sure to open an issue https://github.com/googleapis/genai-toolbox/issues/2261 before writing your code! That way we can discuss the change, evaluate designs, and agree on the general idea - [x] Ensure the tests and linter pass - [x] Code coverage does not decrease (if any source code was changed) - [x] Appropriate docs were updated (if necessary) - [ ] Make sure to add `!` if this involve a breaking change 🛠️ Fixes #2261 2261
84 lines
2.5 KiB
YAML
84 lines
2.5 KiB
YAML
# 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.
|
|
|
|
sources:
|
|
bigquery-source:
|
|
kind: "bigquery"
|
|
project: ${BIGQUERY_PROJECT}
|
|
location: ${BIGQUERY_LOCATION:}
|
|
useClientOAuth: ${BIGQUERY_USE_CLIENT_OAUTH:false}
|
|
scopes: ${BIGQUERY_SCOPES:}
|
|
maxQueryResultRows: ${BIGQUERY_MAX_QUERY_RESULT_ROWS:50}
|
|
|
|
tools:
|
|
analyze_contribution:
|
|
kind: bigquery-analyze-contribution
|
|
source: bigquery-source
|
|
description: Use this tool to analyze the contribution about changes to key metrics in multi-dimensional data.
|
|
|
|
ask_data_insights:
|
|
kind: 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.
|
|
|
|
execute_sql:
|
|
kind: bigquery-execute-sql
|
|
source: bigquery-source
|
|
description: Use this tool to execute sql statement.
|
|
|
|
forecast:
|
|
kind: bigquery-forecast
|
|
source: bigquery-source
|
|
description: Use this tool to forecast time series data.
|
|
|
|
get_dataset_info:
|
|
kind: bigquery-get-dataset-info
|
|
source: bigquery-source
|
|
description: Use this tool to get dataset metadata.
|
|
|
|
get_table_info:
|
|
kind: bigquery-get-table-info
|
|
source: bigquery-source
|
|
description: Use this tool to get table metadata.
|
|
|
|
list_dataset_ids:
|
|
kind: bigquery-list-dataset-ids
|
|
source: bigquery-source
|
|
description: Use this tool to list datasets.
|
|
|
|
list_table_ids:
|
|
kind: bigquery-list-table-ids
|
|
source: bigquery-source
|
|
description: Use this tool to list tables.
|
|
|
|
search_catalog:
|
|
kind: bigquery-search-catalog
|
|
source: bigquery-source
|
|
description: Use this tool to find tables, views, models, routines or connections.
|
|
|
|
toolsets:
|
|
bigquery_database_tools:
|
|
- analyze_contribution
|
|
- ask_data_insights
|
|
- execute_sql
|
|
- forecast
|
|
- get_dataset_info
|
|
- get_table_info
|
|
- list_dataset_ids
|
|
- list_table_ids
|
|
- search_catalog
|