databricks-ai-functions

Call Databricks AI functions from SQL and PySpark pipelines.

Updated Mar 25, 2026
One-click install
npx skills add https://github.com/JingyiWng/databricks_ai_dev_kit_price_watcher --skill databricks-ai-functions-jingyiwng
Or copy as Structured Prompt for Agent▼
Please help me install this Agent Skill.
Skill: databricks-ai-functions
Source: https://github.com/JingyiWng/databricks_ai_dev_kit_price_watcher/tree/main/.claude/skills/databricks-ai-functions
Command: npx skills add https://github.com/JingyiWng/databricks_ai_dev_kit_price_watcher --skill databricks-ai-functions-jingyiwng

SYSTEM DOCUMENTATION & REQUIREMENTS

What problem does it solve?

Built-in AI functions let you call Databricks AI APIs from SQL and PySpark without managing external model endpoints or API keys, enabling seamless, scalable AI-powered data pipelines.

Core Features & Use Cases

  • Task-specific AI operations (classify, extract, summarize, translate, analyze sentiment, and more) integrated directly into SQL.
  • Document processing and parsing capabilities (ai_parse_document) to ingest and structure text from PDFs and other formats.
  • End-to-end data enrichment and analysis patterns, including nested JSON handling with ai_query as a last resort for complex scenarios.

Quick Start

Run a simple enrichment by applying ai_classify and ai_extract to the text column to produce a labeled category and structured fields.

Frequently Asked Questions about databricks-ai-functions

High-intent search queries and answers about installing and using this skill.

FAQPage Schema
How do I run AI text analysis directly inside SQL and PySpark pipelines?▼

You can run AI text analysis directly inside SQL and PySpark pipelines by using Databricks AI Functions, which expose built-in APIs for tasks like classify, extract, and summarize without managing external model endpoints.

Can I parse PDF documents and extract structured data using SQL?▼

Yes, you can parse PDF documents and extract structured data using SQL by applying the ai_parse_document function, which ingests and structures text from PDFs and other document formats directly within your data pipelines.

Do I need to manage API keys to use Databricks AI Functions for batch workloads?▼

No, you do not need to manage API keys to use Databricks AI Functions for batch workloads, because these built-in APIs handle authentication internally and enable seamless, scalable AI-powered data transformations.

What is the best way to handle complex nested JSON from AI queries in PySpark?▼

The best way to handle complex nested JSON from AI queries in PySpark is to use ai_query as a last resort, layering task-specific functions like ai_classify and ai_extract first for structured data enrichment.

Does Databricks support built-in functions for sentiment analysis and grammar correction?▼

Yes, Databricks supports built-in functions for sentiment analysis and grammar correction through its AI Functions, allowing you to perform these text analysis tasks seamlessly within SQL and PySpark transformations.

Can I apply text masking and similarity scoring in streaming workloads?▼

Yes, you can apply text masking and similarity scoring in streaming workloads by layering Databricks AI Functions inside PySpark transformations, enabling pipeline-friendly AI tasks for both batch and streaming data.