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.