databricks-model-serving

Deploy and manage MLflow models and AI agents on Databricks Model Serving endpoints.

38|12|Updated Apr 24, 2026
One-click install
npx skills add https://github.com/databrickslabs/coding-agents-databricks-apps --skill databricks-model-serving-databrickslabs
Or copy as Structured Prompt for Agent
Please help me install this Agent Skill.
Skill: databricks-model-serving
Source: https://github.com/databrickslabs/coding-agents-databricks-apps/tree/main/.claude/skills/databricks-model-serving
Command: npx skills add https://github.com/databrickslabs/coding-agents-databricks-apps --skill databricks-model-serving-databrickslabs

SYSTEM DOCUMENTATION & REQUIREMENTS

What problem does it solve?

This Skill eliminates the complexity of manually configuring infrastructure to host MLflow models and AI agents on Databricks, providing a standardized, governed workflow for deploying scalable serving endpoints.

Core Features & Use Cases

  • Multi-Model Deployment: Supports classical ML models (sklearn, xgboost, pytorch), custom PyFunc models with preprocessing logic, and GenAI agents built with ResponsesAgent and LangGraph.
  • Tool Integration: Enables agents to use Unity Catalog Functions and Vector Search indexes for retrieval-augmented generation and external tool calls.
  • End-to-End Management: Covers model logging, Unity Catalog registration, async deployment, endpoint querying, and status monitoring via MCP tools, SDK, or REST API. For example, a data scientist can deploy a customer churn prediction model or a support chatbot agent to a production endpoint in minutes, with built-in authentication and governance.

Quick Start

Use the databricks-model-serving skill to deploy your trained customer churn prediction model to a scalable Databricks serving endpoint and start querying it for real-time predictions.

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 to Databricks serving endpoints without manual infrastructure configuration?

Deploy MLflow models to Databricks serving endpoints by logging the model, registering it in Unity Catalog, and triggering an async job-based deployment. This workflow eliminates manual infrastructure setup and provides scalable, governed model hosting with built-in authentication.

Can I deploy GenAI agents built with LangGraph to Databricks Model Serving?

Yes, Databricks Model Serving supports deploying GenAI agents built with LangGraph and ResponsesAgent. The deployment workflow handles agents that integrate with Unity Catalog Functions and Vector Search indexes for retrieval-augmented generation and external tool calls.

Does Databricks Model Serving support custom PyFunc models with preprocessing logic?

Yes, Databricks Model Serving supports custom PyFunc models containing preprocessing logic alongside classical ML frameworks like sklearn, xgboost, and pytorch. You can package custom transformations and prediction logic into a single deployable endpoint.

What is the best way to monitor Databricks endpoint status after model deployment?

Monitor Databricks endpoint status after model deployment using MCP tools, the Databricks SDK, or the REST API. These interfaces enable you to check deployment progress, query endpoints for real-time predictions, and verify endpoint health programmatically.

How do I query a deployed ML model on a Databricks endpoint for real-time predictions?

Query a deployed ML model on a Databricks endpoint for real-time predictions by sending requests through the Databricks SDK or REST API. The endpoint handles authentication and routes requests to your registered MLflow model or AI agent for immediate inference.

Do I need Unity Catalog to register MLflow models before serving them on Databricks?

Yes, registering MLflow models in Unity Catalog is part of the standardized deployment workflow on Databricks. Unity Catalog provides the governance and authentication layer required before models or AI agents can be deployed to scalable serving endpoints.