databricks-model-serving

Deploy MLflow models and LangGraph AI agents to Databricks Model Serving endpoints.

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

SYSTEM DOCUMENTATION & REQUIREMENTS

💡 This Skill requires mlflow, databricks-langchain, langgraph, databricks-agents, pydantic, and includes scripts (resource) and references (resource) and assets (resource) components.

What problem does it solve?

This Skill enables the deployment and querying of Databricks Model Serving endpoints, allowing users to deploy MLflow models and AI agents to scalable REST API endpoints and interact with them programmatically.

Core Features & Use Cases

  • Deploy MLflow Models: Deploy models trained with MLflow to Databricks Model Serving endpoints.
  • Deploy AI Agents: Deploy AI agents (GenAI) using the LangGraph framework to interact with Model Serving endpoints.
  • Query Endpoints: Send requests to deployed endpoints and retrieve predictions or other responses.

Quick Start

Deploy a GenAI agent by installing the necessary packages, creating the agent file, uploading it to the workspace, testing it, logging the model, and finally deploying the agent.

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 using this functionality to upload your model files, log them, and create scalable REST API endpoints for programmatic interaction and predictions.

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

Yes, you can deploy GenAI agents built with the LangGraph framework to Databricks Model Serving endpoints, enabling programmatic access and interaction with your deployed AI agents.

What packages do I need to deploy and query Databricks Model Serving endpoints?

To deploy and query Databricks Model Serving endpoints, you need to install and configure MLflow, databricks-langchain, langgraph, databricks-agents, and pydantic packages in your environment.

Does deploying custom pyfunc models to Databricks Model Serving require special configuration?

Deploying custom pyfunc models to Databricks Model Serving uses the same deployment workflow as traditional ML models, allowing you to package custom logic into scalable REST API endpoints without special configuration.

How do I query predictions from a deployed Databricks Model Serving endpoint?

You query Databricks Model Serving endpoints by sending programmatic REST API requests to the deployed endpoint, retrieving predictions or other responses generated by your MLflow models or AI agents.

What is the best way to deploy GenAI agents to Databricks Model Serving endpoints?

The best way to deploy GenAI agents to Databricks Model Serving is to create the agent file, upload it to your workspace, test it, log the model with MLflow, and finally deploy the agent to the endpoint.