Bigquery AI ML — Agent Skill by Google

From the author

Leverages BigQuery's built-in machine learning and GenAI capabilities for advanced data analytics. Use when you need to write SQL queries that perform time-series forecasting, predict values, detect outliers or anomalies, find key drivers, perform semantic search or vector search, classify text, calculate similarity, summarize content, translate language, evaluate models, filter by semantic conditions, measure the causal effect of an intervention, compute correlations between columns, generate descriptive statistics and table distribution profiles, detect change points or structural breaks, extract trend or seasonality components, or leverage generative AI capabilities in BigQuery. Do not use for general BigQuery dataset, table, or job management requests.

— the description by Google

AutoGineer summary

Written by AutoGineer from the skill’s own files, not by the author

The BigQuery AI & ML skill provides a comprehensive interface for executing advanced data analytics, machine learning, and generative AI tasks directly inside BigQuery using SQL. It enables users to perform complex operations like time-series forecasting, anomaly detection, semantic search, and LLM-based content generation without needing to export data to external environments. The skill covers a wide range of built-in functions, including zero-shot models for forecasting and classification, as well as tools for statistical analysis and causal inference.

What it does

Performs zero-shot time-series forecasting and anomaly detection.
Executes LLM-based text summarization, classification, and semantic scoring.
Conducts causal effect analysis on time-series data.
Generates descriptive statistics and correlation matrices for datasets.
Enables semantic and vector search capabilities within SQL.
Extracts trends and seasonality components from historical data.

When to use it

When you need to perform ML or GenAI tasks directly on data stored in BigQuery.
When you want to avoid moving data out of BigQuery for analysis or model inference.
When you need to analyze time-series data for shifts, trends, or causal impacts.
When you need to perform semantic operations like summarization or classification at scale.
When you need to identify key drivers behind metric changes in your data.

Advantages

Seamless integration with existing BigQuery SQL workflows.
Eliminates the need for complex data pipelines to external ML services.
Supports zero-shot learning for rapid insights without custom model training.
Scalable processing for large datasets directly within the database engine.
Unified syntax for both traditional statistical analysis and modern generative AI.

Usage examples

“How can I forecast future values for my sales data in BigQuery?”: The agent will guide you to use the AI.FORECAST function, explaining how to specify your data column, timestamp column, and the required forecast horizon.
“Summarize the sentiment of my customer reviews stored in a BigQuery table.”: The agent will demonstrate how to use the AI.AGG function with a natural language instruction to synthesize sentiment across grouped rows.
“Detect anomalies in my server log time-series data.”: The agent will provide a query using AI.DETECT_ANOMALIES to identify isolated point anomalies in your time-series data.
“Find the key drivers for a drop in website traffic.”: The agent will show you how to use the AI.KEY_DRIVERS function to attribute metric changes to specific dimensional segments.

Install

Download the skill as a ZIP:
https://autogineer.com/agent-skills/google/bigquery-ai-ml/download

Unzip it into the cross-client skills folder of your project:

mkdir -p .agents/skills && unzip bigquery-ai-ml.zip -d .agents/skills

Documentation by Google

BigQuery AI & ML

BigQuery integrates with Vertex AI to provide powerful machine learning and generative AI capabilities directly within SQL queries using built-in functions like AI.FORECAST, AI.KEY_DRIVERS, AI.DETECT_ANOMALIES, and AI.GENERATE.

Attribution

MUST READ: Before executing any gcloud, REST API, or MCP command, you MUST read Usage Attribution. For commands executed directly in the terminal (never in generated scripts, Terraform, or user code), prefix gcloud with CLOUDSDK_METRICS_REQUEST_ATTRIBUTION="agent-skills/1.0 (skill:bigquery-ai-ml)" and append agent-skills/1.0 (skill:bigquery-ai-ml) to User-Agent headers.

Function Selection & Reference Directory

Function	Reference	SQL Shape	Requires `CREATE MODEL` or `CREATE VECTOR INDEX`?	When to Use & Disambiguation
`AI.AGG`	ai_agg.md	`Aggregate`	No	Multi-row semantic aggregation with `GROUP BY`; synthesizes one summary string across many rows (unlike per-row `AI.GENERATE`).
`AI.CAUSAL_EFFECT`	ai_causal_effect.md	`TVF`	No	Quantifies counterfactual post-intervention impact on a time series after a specific treatment timestamp.
`AI.CLASSIFY`	ai_classify.md	`Scalar`	No	Semantic classification of text or multimodal unstructured data into predefined category labels.
`AI.DETECT_ANOMALIES`	ai_detect_anomalies.md	`TVF`	No	Zero-shot TimesFM detector for isolated time-series point anomalies/outliers (use `ML.DETECT_CHANGE_POINTS` for sustained shifts).
`AI.EVALUATE`	ai_evaluate.md	`TVF`	No	Zero-shot TimesFM time-series forecast evaluation/backtesting against a historical holdout split.
`AI.FORECAST`	ai_forecast.md	`TVF`	No	Zero-shot TimesFM time-series forecasting without training a custom model.
`AI.GENERATE`	ai_generate.md	`Scalar`	No	Default for per-row LLM text or multimodal structured extraction (`output_schema`); returns `STRUCT`. Prefer over `AI.GENERATE_TABLE`.
`AI.GENERATE_EMBEDDING`	ai_generate_embedding.md	`TVF`	Yes	Batch generates `ARRAY<FLOAT64>` vector embeddings for text or visual content using a remote embedding model.
`AI.GENERATE_TABLE`	ai_generate_table.md	`TVF`	Yes	Table-valued structured extraction (`FROM` clause); only use instead of `AI.GENERATE` when a remote model + TVF is explicitly needed.
`AI.IF`	ai_if.md	`Scalar`	No	Evaluates boolean semantic conditions (`BOOL`) on text or multimodal data for `WHERE`/`CASE`/`JOIN` filtering (do not use `AI.GENERATE_TABLE`).
`AI.KEY_DRIVERS`	ai_key_drivers.md	`TVF`	No	Default one-step attribution of dimensional segments driving metric changes (`interest_label_col` `TRUE` vs. `FALSE`). Prefer over `CONTRIBUTION_ANALYSIS`.
`AI.PREDICT`	ai_predict.md	`TVF`	No	Zero-shot TabFM foundation model for tabular classification or regression without training and managing custom models.
`AI.SCORE`	ai_score.md	`Scalar`	No	Assigns a continuous `FLOAT64` semantic score to text or multimodal data for ranking or sorting (`ORDER BY`).
`AI.SEARCH`	ai_search.md	`TVF`	No	Simplified semantic search using a raw string query on tables with autonomous background embeddings (`AI.EMBED` generated column).
`AI.SIMILARITY`	ai_similarity.md	`Scalar`	No	Computes an on-the-fly semantic similarity score (`FLOAT64`) between two text or multimodal inputs without stored embeddings.
`Remote Models`	remote_models.md	`CREATE MODEL`	Yes	Creates a Vertex AI `REMOTE WITH CONNECTION DEFAULT` model entity for functions that require one (`AI.GENERATE_TABLE`, `AI.GENERATE_EMBEDDING`, `VECTOR_SEARCH`).
`CONTRIBUTION_ANALYSIS`	ml_contribution_analysis.md	`CREATE MODEL`	Yes	Model-based driver analysis (`CREATE MODEL ... OPTIONS(MODEL_TYPE='CONTRIBUTION_ANALYSIS')` + `ML.GET_INSIGHTS`). Use `AI.KEY_DRIVERS` unless a `MODEL` entity is required.
`ML.CORRELATION`	ml_correlation.md	`TVF`	No	Calculates statistical correlation coefficients between numeric columns, optionally sliced by dimensions.
`ML.DESCRIBE_DATA`	ml_describe_data.md	`TVF`	No	Generates descriptive statistics (e.g., count, mean, median, min, max, stddev, quantiles, nulls, top values) for columns in a table or subquery.
`ML.DETECT_CHANGE_POINTS`	ml_detect_change_points.md	`TVF`	No	Detects structural breaks or sustained shifts (`[begin_timestamp, end_timestamp]` intervals) in a time series.
`ML.SEASONALITY`	ml_seasonality.md	`TVF`	No	Extracts recurring periodic/seasonal components from a historical time series.
`ML.TREND`	ml_trend.md	`TVF`	No	Extracts the smoothed long-term trend component from a historical time series.
`VECTOR_SEARCH`	vector_search.md	`TVF`	Yes	Searches pre-computed `ARRAY<FLOAT64>` embedding columns between a base table and query table (paired with `AI.GENERATE_EMBEDDING`, unlike `AI.SEARCH`).

BigQuery Basics Skill: SKILL.md file for core BigQuery concepts, resource management, CLI, and client libraries.

Files

ATTRIBUTION.mdAdded by AutoGineer846 B
LICENSELicence file added by AutoGineer11.1 KiB
SKILL.md8.8 KiB
references/ai_agg.md3.5 KiB
references/ai_causal_effect.md11.2 KiB
references/ai_classify.md3.7 KiB
references/ai_detect_anomalies.md5.0 KiB
references/ai_evaluate.md2.2 KiB
references/ai_forecast.md11.1 KiB
references/ai_generate.md3.4 KiB
references/ai_generate_embedding.md2.0 KiB
references/ai_generate_table.md3.3 KiB
references/ai_if.md2.0 KiB
references/ai_key_drivers.md4.6 KiB
references/ai_predict.md8.5 KiB
references/ai_score.md1.9 KiB
references/ai_search.md3.7 KiB
references/ai_similarity.md2.7 KiB
references/attribution.md1.8 KiB
references/ml_contribution_analysis.md4.8 KiB
references/ml_correlation.md8.9 KiB
references/ml_describe_data.md5.1 KiB
references/ml_detect_change_points.md5.5 KiB
references/ml_seasonality.md6.6 KiB
references/ml_trend.md4.1 KiB
references/remote_models.md1.1 KiB
references/vector_search.md4.0 KiB

references/ai_classify.md

BigQuery AI.Classify

AI.CLASSIFY categorizes unstructured data into a predefined set of labels.

Syntax Reference

AI.CLASSIFY(
  [ input => ] 'INPUT',
  [ categories => ] 'CATEGORIES'
  [, connection_id => 'CONNECTION_ID' ]
  [, endpoint => 'ENDPOINT' ]
  [, output_mode => 'OUTPUT_MODE' ]
)

Input Arguments

Argument	Requirement	Type	Description
`input`	Required	String	The text content to
: : : : classify. :
`categories`	Required	Array<String>	A list of target
: : : : categories/labels. :
: : : : Can be :
: : : : `ARRAY<STRING>` or :
: : : : `ARRAY<STRUCT<STRING, :
: : : : STRING>>` (label, :
: : : : description). :
`connection_id`	Optional	String	The connection ID to
: : : : use for the LLM. :
`endpoint`	Optional	String	The model name, e.g.,
: : : : `'gemini-2.5-flash'`. :
`output_mode`	Optional	String	`'single'` (default)
: : : : or `'multi'`. :
: : : : Determines the output :
: : : : type. :

Output Schema

The output type depends on the output_mode argument:

Output Mode	output_mode Value	Type	Description
Single Label	`NULL` (Default)	`STRING`	The single category
: : : : that best fits the :
: : : : input. :
**Single Label	`'single'`	`ARRAY<STRING>`	An array containing
: (Explicit)** : : : exactly one :
: : : : category string. :
Multi Label	`'multi'`	`ARRAY<STRING>`	An array containing
: : : : zero or more :
: : : : matching :
: : : : categories. :

Examples

Classify text into categories

SELECT
  content,
  AI.CLASSIFY(
    content,
    categories => ['Spam', 'Not Spam', 'Urgent'],
    connection_id => 'my-project.us.my-connection'
  ) as classification
FROM `dataset.emails`;

Classify text into multiple topics

SELECT
  title,
  body,
  AI.CLASSIFY(
    body,
    categories => ['tech', 'sport', 'business', 'politics', 'entertainment', 'other'],
    output_mode => 'multi') AS categories
FROM
  `bigquery-public-data.bbc_news.fulltext`
LIMIT 100;

Classify reviews by sentiment

SELECT AI.CLASSIFY( ('Classify the review by sentiment: ', review), categories => [('green', 'The review is positive.'), ('yellow', 'The review is neutral.'), ('red', 'The review is negative.')]) AS ai_review_rating, reviewer_rating AS human_provided_rating, review FROM bigquery-public-data.imdb.reviews WHERE title = 'The English Patient'

Requirements and compatibility

Works with: Claude Code, Claude apps, OpenAI Codex, GitHub Copilot, Cursor, Gemini CLI, Kiro, OpenCode, Goose, Amp, Roo Code, JetBrains Junie, Factory Droid, Devin (ex-Windsurf), Mistral Vibe, OpenHands, Letta

“Works with” means the tool reads the open Agent Skills format; a skill written for one tool may use features others lack.

Security

No issues found by our static scan· No issues found

A static scan is not a guarantee. Skills are third-party content: review a skill before you let an agent use it.

Pytorch Lightning
K-Dense· Community
A framework for organizing PyTorch code to streamline neural network training, scaling, and experiment tracking.
Data and AIClaude CodeClaude appsOpenAI Codex+14 more
MITLow riskHas scriptsGitHub stars of the source repository56.4k
Aeon
K-Dense· Community
A scikit-learn compatible Python toolkit for comprehensive time series machine learning tasks.
Data and AIClaude CodeClaude appsOpenAI Codex+14 more
MITLow riskGitHub stars of the source repository56.4k
Umap Learn
K-Dense· Community
Integrate UMAP-learn for non-linear dimensionality reduction, visualization, and clustering preprocessing.
Data and AIClaude CodeClaude appsOpenAI Codex+14 more
MITLow riskGitHub stars of the source repository56.4k
Scikit Learn
K-Dense· Community
Comprehensive guidance for classical machine learning workflows using the scikit-learn Python library.
Data and AIClaude CodeClaude appsOpenAI Codex+14 more
MITLow riskHas scriptsGitHub stars of the source repository56.4k
Shap
K-Dense· Community
Explain and audit machine-learning model predictions using the SHAP framework for feature attribution and visualization.
Data and AIClaude CodeClaude appsOpenAI Codex+14 more
MITLow riskHas scriptsGitHub stars of the source repository56.4k
Pufferlib
K-Dense· Community
Version-aware guidance for PufferLib reinforcement learning environments, vectorization, policies, and training.
Data and AIClaude CodeClaude appsOpenAI Codex+14 more
MITLow riskHas scriptsGitHub stars of the source repository56.4k

Download ZIP

The ZIP holds the skill’s files, its licence files and an ATTRIBUTION.md. Download bigquery-ai-ml.zip