encounter/adk-python
0
0
Fork 0
mirror of https://github.com/encounter/adk-python.git synced 2026-07-09 18:19:28 -07:00
Code Issues Packages Projects Releases Wiki Activity

chore: Move framework filled params to the end in the BigQuery forecast tool

Browse Source 
This is for better readability of the tool, specially in relation to the examples in the docstring.

PiperOrigin-RevId: 806460056
This commit is contained in:
Google Team Member
2025-09-12 16:03:57 -07:00
committed by Copybara-Service
parent c53c02f022
commit 4f07228f57
1 changed files with 6 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/google/adk/tools/bigquery/query_tool.py
+6 -5
View File
@@ -604,11 +604,12 @@ def forecast(
history_data: str,
timestamp_col: str,
data_col: str,
horizon: int = 10,
id_cols: Optional[list[str]] = None,
*,
credentials: Credentials,
settings: BigQueryToolConfig,
tool_context: ToolContext,
horizon: int = 10,
id_cols: Optional[list[str]] = None,
) -> dict:
"""Run a BigQuery AI time series forecast using AI.FORECAST.
@@ -622,14 +623,14 @@ def forecast(
each data point.
data_col (str): The name of the column containing the numerical values to
be forecasted.
credentials (Credentials): The credentials to use for the request.
settings (BigQueryToolConfig): The settings for the tool.
tool_context (ToolContext): The context for the tool.
horizon (int, optional): The number of time steps to forecast into the
future. Defaults to 10.
id_cols (list, optional): The column names of the id columns to indicate
each time series when there are multiple time series in the table. All
elements must be strings. Defaults to None.
credentials (Credentials): The credentials to use for the request.
settings (BigQueryToolConfig): The settings for the tool.
tool_context (ToolContext): The context for the tool.
Returns:
dict: Dictionary representing the result of the forecast. The result