diff --git a/src/google/adk/tools/bigquery/query_tool.py b/src/google/adk/tools/bigquery/query_tool.py index f4a82871..e66ea3ac 100644 --- a/src/google/adk/tools/bigquery/query_tool.py +++ b/src/google/adk/tools/bigquery/query_tool.py @@ -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