Official agent skill

Databricks AI Functions

by databricks in databricks/databricks-agent-skills

Use Databricks built-in AI Functions (aiclassify, aiextract, aisummarize, aimask, aitranslate, aifixgrammar, aigen, aianalyzesentiment, aisimilarity, aiparsedocument, aiprepsearch, aiquery…

OfficialCustom licenceAuto-check passedDocuments & Office

Install Databricks AI Functions

skills CLI
$ npx skills add databricks/databricks-agent-skills --skill databricks-ai-functions -a claude-code

Project install by default; add -g for ~/.claude/skills/.

GitHub CLI
$ gh skill install databricks/databricks-agent-skills databricks-ai-functions --agent claude-code

Project scope by default; add --scope user for a personal install. Needs GitHub CLI 2.90.0 or later (public preview).

Manual copy
$ git clone --depth 1 https://github.com/databricks/databricks-agent-skills.git skills-src && mkdir -p .claude/skills && cp -r skills-src/skills/databricks-ai-functions .claude/skills/databricks-ai-functions && rm -rf skills-src

Use ~/.claude/skills/ instead of .claude/skills for a personal install. The folder must contain SKILL.md.

Claude Code skills documentation · loads skills from .claude/skills/

Facts

Skill name
databricks-ai-functions
GitHub stars
345
Token cost
~3.9k tokens
SKILL.md length
1,108 words
Files
7 (incl. references, assets)
Skills in repo
32
Repo updated
First seen
Licence
Custom licence

At a glance

Use Databricks built-in AI Functions (aiclassify, aiextract, aisummarize, aimask, aitranslate, aifixgrammar, aigen, aianalyzesentiment, aisimilarity, aiparsedocument, aiprepsearch, aiquery…

  • Tasks that involve Document parsing
  • SKILL.md covers Overview, Patterns, Document Processing Pipeline and Reference Files, plus 1 more section
  • Instructions only: no scripts, shell commands, URLs or credentials in SKILL.md
  • Tasks that involve Copy editing and proofreading

What it does

Databricks AI Functions is an agent skill from databricks/databricks-agent-skills, published by the product's own GitHub organization. Use Databricks built-in AI Functions (aiclassify, aiextract, aisummarize, aimask, aitranslate, aifixgrammar, aigen, aianalyzesentiment, aisimilarity, aiparsedocument, aiprepsearch, aiquery, aiforecast) to add AI capabilities directly to SQL and PySpark pipelines without managing model endpoints. Also covers document parsing and building custom RAG pipelines (parse → prepsearch → index → query).

Its SKILL.md is about 3.9k tokens, which your agent loads only when the skill is triggered. The skill folder holds 9 other files, including reference files and assets (for example `agents/openai.yaml`, `references/1-task-functions.md` and `references/2-ai-query.md`). Compatibility notes: Requires databricks CLI (= v1.0.0)

It sits in Documents & Office, covering Document parsing, Copy editing and proofreading and SQL. It works with Databricks, SQL and Apache Spark. The repository describes itself as: Databricks AI Tools: skills and plugins for building on Databricks with Claude Code, Cursor, Codex, GitHub Copilot, and other AI coding agents.

When your agent uses it

  • Tasks that involve Document parsing
  • Tasks that involve Copy editing and proofreading
  • Tasks that involve SQL

Example prompts

  • “/databricks-ai-functions”

Requirements

  • Python 3
  • Compatibility (from SKILL.md): Requires databricks CLI (>= v1.0.0)

What it can do on your machine

Read from SKILL.md and the folder at commit f4fcec5. It shows what the files ask for, not the result of running them.

  • Tool permissions

    Pre-approves nothing: there is no allowed-tools line, so your agent's usual permission prompts apply.

    From allowed-tools in the SKILL.md frontmatter.

  • Runs code

    No scripts in the folder and no shell commands in SKILL.md (its code samples are sql).

    From the folder's file list and the shell code blocks in SKILL.md.

  • Network

    Links to these hosts (documentation or services it may open):

    • docs.databricks.com
    • github.com

    From URLs in SKILL.md, links to its own repository left out.

  • Credentials

    Names no API keys, tokens, secrets or passwords.

    From names ending in _API_KEY, _TOKEN, _SECRET, _KEY or _PASSWORD in SKILL.md.

  • Compatibility

    Requires databricks CLI (>= v1.0.0)

    From compatibility in the SKILL.md frontmatter.

Context cost

Databricks AI Functions loads about 3.9k tokens when it runs, and up to ~11k if it reads all its reference files. Until then it costs about 110 tokens; SKILL.md has 1,108 words of instructions outside code blocks.

Always · name and description, kept in context so the agent knows when to use it
~110
When it runs · the whole SKILL.md, loaded when a task matches
~3.9k
With references · SKILL.md plus every file in references/, read only if the agent opens them
~11k

Estimates: characters ÷ 4, the usual rule of thumb; real counts depend on the model's tokenizer. Scripts and assets cost tokens only if the agent reads them.

Safety

Auto-check passed

The automated check found no risky patterns in SKILL.md.

Automated static check — not a guarantee. Review scripts before installing. It scans the text of SKILL.md for risky patterns (piping downloads into a shell, reading credential files, hidden Unicode, destructive commands); files beside SKILL.md are not scanned.

SKILL.md

Its licence (Custom licence) doesn't allow us to republish the file, so here is its outline and opening line. It has 1,108 words (~3,924 tokens).

“Databricks AI Functions are built-in SQL and PySpark functions that call Foundation Model APIs directly from your data pipelines — no model endpoint setup, no API keys, no boilerplate. They operate on table columns as naturally as UPPER() or LENGTH()…”

— opening of SKILL.md by databricks, Custom licence
name
databricks-ai-functions
compatibility
Requires databricks CLI (>= v1.0.0)
metadata.version
0.2.0
parent
databricks-core

Read the full SKILL.md on GitHub

Files

SKILL.md and 6 other files (references, assets) in skills/databricks-ai-functions of databricks/databricks-agent-skills.

  • SKILL.md
  • agents/openai.yaml
  • assets/databricks.png
  • assets/databricks.svg
  • references/1-task-functions.md
  • references/2-ai-query.md
  • references/3-ai-forecast.md

Open the folder on GitHubat commit f4fcec5

Compare with similar skills

Databricks AI Functions next to the 5 skills that share the most tags, products or categories with it. Stars are the repository's; “used in” counts other GitHub owners with a copy.

Databricks AI Functions compared with similar skills
SkillStarsUsed inTokensAuto-checkLicenceRepo updated
Databricks AI Functions this skilldatabricks/databricks-agent-skills345—~3.9kAutomated safety check: PassCustom licence
Optimizing Databricks SQLAltimateAI/data-engineering-skills127—~6.7kAutomated safety check: PassMIT
Mendix Odata Pushdownmendixlabs/mxcli128—~2.5kAutomated safety check: PassApache-2.0
Duckdb EnaAAaqwq/AGI-Super-Team1051 repos~1.6kAutomated safety check: PassMIT
SQL Correctnessdatabricks-solutions/ai-dev-kit1.9k—~438Automated safety check: PassCustom licence
Databricksrocky-data/rocky304—~2kAutomated safety check: PassApache-2.0

Similar skills

  • Optimizing Databricks SQL

    AltimateAI/data-engineering-skills

    Analyze DBSQL queries, including SQL embedded in notebooks (spark.sql(...), %sql cells), for anti-patterns, lint issues, and performance problems, using Databricks-specific dialect and platform…

    127 GitHub stars~6.7k tokensUpdated 6 days ago
    DatabasesAuto-check passed
  • Mendix Odata Pushdown

    mendixlabs/mxcli

    Push OData query options into the SQL of a Mendix resource served by a read microflow, so $filter, $orderby, $top, $skip, $count and the key lookup reach the database instead of being silently…

    128 GitHub stars~2.5k tokensUpdated today
    DatabasesAuto-check passed
  • Duckdb En

    aAAaqwq/AGI-Super-Team

    DuckDB CLI specialist for SQL analysis, data processing and file conversion.

    105 GitHub starsUsed in 1 repo~1.6k tokens
    DatabasesAuto-check passed
  • SQL Correctness

    databricks-solutions/ai-dev-kit

    SQL evaluation criteria for Databricks. An agent skill from databricks-solutions/ai-dev-kit.

    1.9k GitHub stars~438 tokensUpdated 1 mo ago
    DatabasesAuto-check passed
  • Databricks

    rocky-data/rocky

    Databricks REST API and SQL reference for Rocky's warehouse adapter.

    304 GitHub stars~2k tokensUpdated today
    Backend & APIsAuto-check passed
  • Gaik Toolkit

    GAIK-project/gaik-toolkit

    GAIK toolkit overview and reference. An agent skill from GAIK-project/gaik-toolkit.

    100 GitHub stars~5.7k tokensUpdated yesterday
    Documents & OfficeAuto-check passed

More from databricks/databricks-agent-skills

All 32 skills in this repo
  • Databricks Dbsql

    databricks/databricks-agent-skills

    Official

    Databricks SQL (DBSQL) advanced features and SQL warehouse capabilities.

    345 GitHub starsUsed in 1 repo~2.8k tokens
    Auto-check passed
  • Databricks Synthetic Data Gen

    databricks/databricks-agent-skills

    Official

    Generate realistic synthetic data using Spark + Faker (strongly recommended).

    345 GitHub stars~3.2k tokensUpdated yesterday
    Auto-check passed
  • Databricks Model Serving

    databricks/databricks-agent-skills

    Official

    Databricks Model Serving endpoint lifecycle and ops. An agent skill from databricks/databricks-agent-skills.

    345 GitHub starsUsed in 1 repo~3.3k tokens
    Auto-check passed
  • Databricks Python SDK

    databricks/databricks-agent-skills

    Official

    Databricks development guidance including Python SDK, Databricks Connect, CLI, and REST API.

    345 GitHub starsUsed in 1 repo~4.6k tokens
    Auto-check passed
  • Databricks Spark Structured Streaming

    databricks/databricks-agent-skills

    Official

    Comprehensive guide to Spark Structured Streaming for production workloads.

    345 GitHub starsUsed in 1 repo~956 tokens
    Auto-check passed
  • Databricks App Design

    databricks/databricks-agent-skills

    Official

    Design the UX of custom-code Databricks Apps (AppKit/React) data screens — KPI/overview pages, reports, charts, tables, and Genie/chat data assistants — mapped to concrete AppKit components.

    345 GitHub stars~1.7k tokensUpdated yesterday
    Auto-check passed

Questions about Databricks AI Functions

What does Databricks AI Functions do?

Use Databricks built-in AI Functions (aiclassify, aiextract, aisummarize, aimask, aitranslate, aifixgrammar, aigen, aianalyzesentiment, aisimilarity, aiparsedocument, aiprepsearch, aiquery…. Databricks AI Functions is an agent skill from databricks/databricks-agent-skills, published by the product's own GitHub organization. Use Databricks built-in AI Functions (aiclassify, aiextract, aisummarize, aimask, aitranslate, aifixgrammar, aigen, aianalyzesentiment, aisimilarity, aiparsedocument, aiprepsearch, aiquery, aiforecast) to add AI capabilities directly to SQL and PySpark pipelines without managing model endpoints.

When should I use Databricks AI Functions?

Databricks AI Functions fits situations like: tasks that involve Document parsing; tasks that involve Copy editing and proofreading; tasks that involve SQL.

How do I install Databricks AI Functions in Claude Code?

Run `npx skills add databricks/databricks-agent-skills --skill databricks-ai-functions -a claude-code`. Or copy the skill folder (skills/databricks-ai-functions in databricks/databricks-agent-skills) into .claude/skills/databricks-ai-functions in your project. Claude Code loads it when a task matches its description.

How do I install Databricks AI Functions in Codex?

Run `npx skills add databricks/databricks-agent-skills --skill databricks-ai-functions -a codex`. Or copy the skill folder (skills/databricks-ai-functions in databricks/databricks-agent-skills) into .agents/skills/databricks-ai-functions in your project. Codex loads it when a task matches its description.

Can I use Databricks AI Functions in Cursor, Gemini CLI or GitHub Copilot?

Cursor, Gemini CLI, GitHub Copilot and OpenCode also load SKILL.md folders. With the skills CLI, run `npx skills add databricks/databricks-agent-skills --skill databricks-ai-functions -a cursor` (or -a gemini-cli, github-copilot or opencode for the others). To copy it by hand, put the folder in .cursor/skills/databricks-ai-functions, .gemini/skills/databricks-ai-functions, .github/skills/databricks-ai-functions and .opencode/skills/databricks-ai-functions in your project.

What does Databricks AI Functions need to run?

SKILL.md names no scripts, command-line tools or credentials: Databricks AI Functions is instructions for the agent only. Our summary lists: Python 3. Compatibility (from SKILL.md): Requires databricks CLI (>= v1.0.0).

Does Databricks AI Functions access the network?

SKILL.md names 2 domains. As links in the text: docs.databricks.com and github.com. This is read from the text; nothing was executed.

Is Databricks AI Functions safe to install?

Our automated static check of SKILL.md found no risky patterns, such as piping downloads into a shell, reading credential files or hidden Unicode. It is not a guarantee. Review the folder before installing.

What licence does Databricks AI Functions use?

Databricks AI Functions has a licence file (the repository's licence) that doesn't match a standard licence. Read it on GitHub before reusing the skill.

How many tokens does Databricks AI Functions use?

About 3.9k tokens (SKILL.md is roughly 16k characters). Agents keep only the skill's name and description in context until a task matches; then they load SKILL.md in full. Its references folder adds about 6.9k tokens, read only when the agent opens those files.

What are the alternatives to Databricks AI Functions?

Skills that share tags, products or a category with Databricks AI Functions: Optimizing Databricks SQL (AltimateAI/data-engineering-skills, 127 stars), Mendix Odata Pushdown (mendixlabs/mxcli, 128 stars), Duckdb En (aAAaqwq/AGI-Super-Team, 105 stars) and SQL Correctness (databricks-solutions/ai-dev-kit, 1.9k stars). The comparison table on this page puts their stars, adoption, token cost, safety result and licence side by side.

Who maintains Databricks AI Functions?

databricks (a GitHub organization, an official publisher) maintains it in databricks/databricks-agent-skills, which has 345 GitHub stars. The repository holds 32 skills in this directory. The repository was last updated on October 6, 2026.

Source: databricks/databricks-agent-skills on GitHub. Facts on this page come from the repository at the commit we read; the author's words are quoted as theirs.