databricks-model-serving

Deploy and query ML models and AI agents on Databricks Model Serving.

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

SYSTEM DOCUMENTATION & REQUIREMENTS

💡 This Skill includes scripts (resource) and references (resource) components.

What problem does it solve?

This Skill streamlines the deployment and querying of machine learning models and AI agents on Databricks, enabling scalable REST API access for various applications.

Core Features & Use Cases

  • Model Deployment: Deploy classical ML models (sklearn, xgboost), custom Python functions (pyfunc), and GenAI agents (ResponsesAgent, LangGraph) to scalable endpoints.
  • Agent Integration: Seamlessly integrate Unity Catalog Functions and Vector Search as tools for AI agents.
  • Endpoint Querying: Query deployed models and agents using Python SDK, REST API, or MCP tools.
  • Use Case: Deploy a customer churn prediction model for real-time scoring or deploy a conversational AI agent to answer user queries about company products.

Quick Start

Use the databricks-model-serving skill to deploy the 'main.models.my_classifier' model version 1 to a serving endpoint named 'my-classifier-endpoint'.

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 machine learning models to a Databricks serving endpoint?

To deploy machine learning models to a Databricks serving endpoint, you use the deployment functionality to host classical ML, custom pyfunc, and GenAI agents. This creates scalable REST API access for real-time application querying.

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

Yes, Databricks Model Serving supports deploying GenAI agents built with LangGraph and ResponsesAgent. It allows seamless integration of Unity Catalog Functions and Vector Search as tools for these deployed AI agents.

What's the best way to query a deployed model on Databricks using a REST API?

The best way to query a deployed model on Databricks using a REST API is through the provided Python SDK or MCP tools. This endpoint querying capability enables scalable access for both classical ML models and GenAI agents.

Does Databricks Model Serving support custom Python functions and pyfunc models?

Yes, Databricks Model Serving fully supports custom Python functions and pyfunc models. You can deploy them alongside classical ML frameworks like sklearn and xgboost to scalable endpoints for real-time scoring.

How do I integrate Unity Catalog Functions as tools for AI agents on Databricks?

You integrate Unity Catalog Functions as tools for AI agents on Databricks during the model serving deployment process. This allows GenAI agents to seamlessly query these functions alongside Vector Search tools.

What types of ML models can I deploy using Databricks Model Serving endpoints?

You can deploy classical ML models like sklearn and xgboost, custom Python functions, and GenAI agents using Databricks Model Serving endpoints. This supports applications ranging from customer churn prediction to conversational AI.