databricks-model-serving

Deploy and query Databricks model serving endpoints for MLflow and PyFunc models.

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

SYSTEM DOCUMENTATION & REQUIREMENTS

What problem does it solve?

Organizations need a streamlined way to deploy MLflow models, custom PyFunc models, and GenAI agents as scalable, low‑latency REST endpoints on Databricks, while handling registration, tooling, and endpoint management.

Core Features & Use Cases

  • Deploy classical ML models with autolog and manual logging.
  • Register and serve custom Python (PyFunc) models.
  • Build and serve GenAI agents using ResponsesAgent and LangGraph.
  • Integrate Unity Catalog functions and Vector Search tools.
  • Query, monitor, and update serving endpoints via SDK, REST, or MCP tools.

Quick Start

Use the databricks-model-serving skill to deploy a model and query its endpoint in a single workflow.

Frequently Asked Questions about databricks-model-serving

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

FAQPage Schema
How do I deploy MLflow models as REST API endpoints on Databricks?

Deploy MLflow models on Databricks by creating model serving endpoints, which packages your trained model into a scalable, low-latency REST API. This workflow supports both autolog and manual logging for classical ML models.

Can I serve custom Python models and GenAI agents using Databricks model serving?

Yes, Databricks model serving supports custom PyFunc models and GenAI agents. You can build and serve agents using frameworks like ResponsesAgent and LangGraph through scalable REST endpoints.

Does Databricks model serving work with Unity Catalog functions and Vector Search?

Databricks model serving integrates directly with Unity Catalog functions and Vector Search tools, allowing you to register models, manage endpoints, and connect external tools within your Databricks workspace.

What is the best way to query and monitor Databricks serving endpoints?

Query and monitor Databricks serving endpoints using the SDK, REST APIs, or MCP tools. This allows you to programmatically update endpoints and retrieve predictions from your deployed MLflow or GenAI models.

Do I need Unity Catalog registration to serve models on Databricks?

Yes, Unity Catalog registration is a core requirement for serving models on Databricks. It handles the governance and registration of MLflow models, custom PyFunc models, and GenAI agents before endpoint deployment.