databricks-ai-functions
Installation
SKILL.md
Databricks AI Functions
Official Docs: https://docs.databricks.com/aws/en/large-language-models/ai-functions Individual function reference: https://docs.databricks.com/aws/en/sql/language-manual/functions/
Overview
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(), and are optimized for batch inference at scale.
There are three categories:
| Category | Functions | Use when |
|---|---|---|
| Task-specific | ai_analyze_sentiment, ai_classify, ai_extract, ai_fix_grammar, ai_gen, ai_mask, ai_similarity, ai_summarize, ai_translate, ai_parse_document |
The task is well-defined — prefer these always |
| General-purpose | ai_query |
Complex nested JSON, custom endpoints, multimodal — last resort only |
| Table-valued | ai_forecast |
Time series forecasting |
Function selection rule — always prefer a task-specific function over ai_query: