databricks-ai-functions

Run Databricks built-in AI Functions directly from SQL and PySpark.

11|3|Updated Jun 10, 2025
One-click install
npx skills add https://github.com/Paldom/databricks-apps-fastapi-starter --skill databricks-ai-functions-paldom
Or copy as Structured Prompt for Agent
Please help me install this Agent Skill.
Skill: databricks-ai-functions
Source: https://github.com/Paldom/databricks-apps-fastapi-starter/tree/main/.gemini/skills/databricks-ai-functions
Command: npx skills add https://github.com/Paldom/databricks-apps-fastapi-starter --skill databricks-ai-functions-paldom

SYSTEM DOCUMENTATION & REQUIREMENTS

What problem does it solve?

This Skill enables you to run Databricks built-in AI Functions directly from SQL and PySpark, eliminating the need to manage external model endpoints and complex integrations.

Core Features & Use Cases

  • Task-specific AI functions for classification, extraction, summarization, sentiment analysis, translation, and document parsing, enabling rich data enrichment in pipelines.
  • Free-form and structured outputs via ai_query as a last-resort option for complex JSON and multimodal inputs.
  • Quick-startable patterns to ingest and transform data from raw text or binary documents into actionable insights.

Quick Start

Execute a sample query that demonstrates ai_classify, ai_extract, and ai_summarize directly in SQL.

Frequently Asked Questions about databricks-ai-functions

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

FAQPage Schema
How do I use Databricks AI functions for text classification and extraction in SQL?

You can use Databricks built-in AI functions like ai_classify and ai_extract directly within SQL queries. This enables task-specific text analysis and data enrichment without managing external model endpoints.

Can I parse PDF documents and extract insights directly in PySpark data pipelines?

Yes, Databricks AI functions support PDF and document parsing directly from PySpark. You can ingest and transform raw text or binary documents into actionable insights without external integrations.

What is the best way to summarize large text fields in a Databricks data pipeline?

The best way to summarize large text fields is using the ai_summarize function directly in SQL or PySpark. This provides built-in summarization without needing to configure or manage separate external model endpoints.

When should I use ai_query instead of task-specific AI functions for data enrichment?

You should use ai_query as a last-resort option for complex scenarios involving multimodal inputs or structured JSON outputs. Task-specific functions like ai_classify and ai_summarize are preferred for standard analysis and extraction.

Do I need to manage external model endpoints to run AI functions in Databricks SQL?

No, you do not need to manage external model endpoints. Databricks built-in AI functions handle analysis, extraction, and translation directly within SQL and PySpark using safe defaults and centralized configuration.

How do I handle batch versus near-real-time AI processing in Databricks data pipelines?

You handle batch versus near-real-time processing by following documented best practices and safe defaults. A centralized config.yml standardizes prompts and model names to manage error handling across both processing modes.