databricks-model-serving

Deploys MLflow models and GenAI agents as Databricks Model Serving endpoints.

1|Updated Apr 24, 2026
One-click install
npx skills add https://github.com/itsadijmbt/SecureMCP-Servers --skill databricks-model-serving-itsadijmbt
Or copy as Structured Prompt for Agent
Please help me install this Agent Skill.
Skill: databricks-model-serving
Source: https://github.com/itsadijmbt/SecureMCP-Servers/tree/main/TEST_SERVERS/PORTED_TO_SECUREMCP/databrickslab-mcp/ai-dev-kit/databricks-skills/databricks-model-serving
Command: npx skills add https://github.com/itsadijmbt/SecureMCP-Servers --skill databricks-model-serving-itsadijmbt

SYSTEM DOCUMENTATION & REQUIREMENTS

What problem does it solve?

This Skill removes the friction of turning trained MLflow models and GenAI agents into production-ready Databricks Model Serving endpoints that you can query reliably.

Core Features & Use Cases

  • Classical ML deployment: Use MLflow autolog (e.g., sklearn/xgboost/lightgbm/pytorch/tensorflow/spark) to log models into Unity Catalog and serve them.
  • Custom PyFunc deployment: Package preprocessing, signatures, and custom inference logic with MLflow pyfunc for serving.
  • GenAI agent serving: Deploy ResponsesAgent/LangGraph agents to scalable endpoints and query them using OpenAI-compatible message formats, including tool calling via UC Functions and Vector Search.

Quick Start

Deploy a ResponsesAgent by logging it to Unity Catalog, running a job-based deployment, and then querying the resulting endpoint with a messages payload using the manage_serving_endpoint tool.

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 Model Serving endpoints?

You can deploy MLflow models to Databricks Model Serving by logging them into Unity Catalog via autolog or custom pyfunc, then running a job-based deployment to create and query the endpoint.

Can I serve GenAI agents built with LangGraph on Databricks?

Yes, you can serve ResponsesAgent or LangGraph agents on Databricks by logging them to Unity Catalog, deploying them as scalable endpoints, and querying them using OpenAI-compatible message formats.

Does Databricks Model Serving support tool calling with Unity Catalog functions?

Yes, Databricks Model Serving supports tool calling by integrating Unity Catalog Functions and Vector Search with your deployed GenAI agents for actionable inference.

How do I package custom preprocessing logic for MLflow model serving?

You package custom preprocessing logic for MLflow model serving by using MLflow pyfunc to bundle your inference logic and signatures alongside the model before registering it in Unity Catalog.

What is the correct input format for querying a Databricks serving endpoint?

The correct input format for querying a Databricks serving endpoint depends on the model type: GenAI agents use OpenAI-compatible messages payloads, while classical ML models require correctly formatted inputs matching their signatures.

Why does querying a Databricks model endpoint fail immediately after deployment?

Querying a Databricks model endpoint fails immediately after deployment because the endpoint requires readiness handling before inference, meaning you must check endpoint status to ensure it is fully provisioned before querying.