databricks-model-serving

Deploy MLflow models and GenAI agents to Databricks model-serving endpoints.

Updated Jan 30, 2026
One-click install
npx skills add https://github.com/teegin-g/Slopcast --skill databricks-model-serving-teegin-g
Or copy as Structured Prompt for Agent
Please help me install this Agent Skill.
Skill: databricks-model-serving
Source: https://github.com/teegin-g/Slopcast/tree/main/.agents/skills/databricks-model-serving
Command: npx skills add https://github.com/teegin-g/Slopcast --skill databricks-model-serving-teegin-g

SYSTEM DOCUMENTATION & REQUIREMENTS

What problem does it solve?

Deploying ML models and AI agents to scalable Databricks model-serving endpoints can be complex, requiring careful orchestration of training, logging, deployment, and runtime querying.

Core Features & Use Cases

  • Deploy traditional ML models with MLflow autolog and register to Unity Catalog endpoints.
  • Deploy Custom PyFunc models with signatures and artifacts to dedicated endpoints.
  • Build and deploy GenAI agents (ResponsesAgent / LangGraph) and query them via REST/SDK; integrate UC Functions and Vector Search as tools.
  • Query endpoints, check statuses, and manage deployments from a single workflow.
  • Use cases include hosting ML models for inference, agent-based workflows, and tool-enabled agents in production.

Quick Start

Install the required packages and follow the steps to deploy and query a Databricks model-serving 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 MLflow models to Databricks model-serving endpoints?

To deploy MLflow models to Databricks model-serving endpoints, you register the model and apply it to dedicated endpoints. This workflow supports traditional ML models, custom PyFunc models with signatures, and artifacts for scalable inference.

Can I deploy GenAI agents with Unity Catalog functions and Vector Search?

Yes, you can deploy GenAI agents like ResponsesAgent or LangGraph to Databricks endpoints. This process integrates Unity Catalog functions and Vector Search as tools, allowing you to build and query tool-enabled agents via REST or SDK.

What do I need to set up before serving models on Databricks?

Before serving models on Databricks, you need a Unity Catalog-enabled workspace with model-serving enabled. You also need access to defined foundation model endpoints to successfully deploy and query your ML or GenAI agents.

How do I query endpoints and check deployment statuses in Databricks?

You can query endpoints and check statuses in Databricks from a single workflow. This allows you to manage deployments, send inference requests to traditional ML models or GenAI agents, and monitor their operational state continuously.

Does this approach support both traditional ML and custom PyFunc models?

Yes, this approach supports deploying both traditional ML models using MLflow autolog and custom PyFunc models. You can register both types with their respective signatures and artifacts to dedicated Databricks model-serving endpoints.

What is the best way to host ML models and agent-based workflows in production?

The best way to host ML models and agent-based workflows in production is deploying them to scalable Databricks model-serving endpoints. This orchestrates training, logging, deployment, and runtime querying for both traditional ML and GenAI agents.