databricks-model-serving

Deploy and query Databricks model-serving endpoints as scalable REST APIs.

4|4|Updated Jan 5, 2026
One-click install
npx skills add https://github.com/RamVegiraju/databricks-samples --skill databricks-model-serving-ramvegiraju
Or copy as Structured Prompt for Agent
Please help me install this Agent Skill.
Skill: databricks-model-serving
Source: https://github.com/RamVegiraju/databricks-samples/tree/main/.claude/skills/databricks-model-serving
Command: npx skills add https://github.com/RamVegiraju/databricks-samples --skill databricks-model-serving-ramvegiraju

SYSTEM DOCUMENTATION & REQUIREMENTS

What problem does it solve?

Databricks Model Serving deployments require coordinating MLflow logging, Unity Catalog permissions, endpoint configuration, and monitoring; this skill provides a cohesive workflow to deploy and query endpoints for both traditional models and GenAI agents.

Core Features & Use Cases

  • End-to-end deployment of MLflow-logged models to scalable endpoints.
  • GenAI agent support with ResponsesAgent/LangGraph patterns and tool integrations.
  • Vector Search & UC function tool integration for retrieval-augmented workflows and accessible APIs.

Quick Start

Install the required packages, log a model or agent with MLflow, deploy a serving endpoint, and start querying it.

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 APIs using Databricks model serving?

To deploy MLflow models as scalable REST APIs, log the model with MLflow, configure the endpoint, and deploy it. This workflow exposes traditional ML, custom PyFunc models, and GenAI agents as accessible REST APIs.

Can I deploy GenAI agents with vector search and Unity Catalog tools on Databricks?

Yes, you can deploy GenAI agents on Databricks using ResponsesAgent or LangGraph patterns. This workflow supports integrating Unity Catalog function tools and vector search for retrieval-augmented workflows exposed as accessible REST APIs.

What prerequisites are needed to set up Databricks model serving endpoints?

Setting up Databricks model serving endpoints requires Unity Catalog, Model Serving enabled, and compatible MLflow or foundation model APIs. You also need specific endpoint names from the SKILL.md file to execute the deployment workflow.

Does this workflow handle both classical ML and GenAI agent deployment?

Yes, this workflow handles both classical ML and GenAI agent deployment. It provides a cohesive process to configure endpoints and expose models as scalable REST APIs while coordinating MLflow logging and Unity Catalog permissions.

What is the best way to coordinate MLflow logging and Unity Catalog permissions for model serving?

The best way to coordinate MLflow logging and Unity Catalog permissions is using a cohesive deployment workflow. This approach manages endpoint configuration and monitoring while exposing ML models and GenAI agents as scalable REST APIs.