migrate-from-model-serving

Migrate MLflow ResponsesAgent from Databricks Model Serving to Databricks Apps.

Updated May 10, 2026
One-click install
npx skills add https://github.com/keqingli1129/agent-langgraph-one --skill migrate-from-model-serving-keqingli1129
Or copy as Structured Prompt for Agent▼
Please help me install this Agent Skill.
Skill: migrate-from-model-serving
Source: https://github.com/keqingli1129/agent-langgraph-one/tree/main/.claude/skills/migrate-from-model-serving
Command: npx skills add https://github.com/keqingli1129/agent-langgraph-one --skill migrate-from-model-serving-keqingli1129

SYSTEM DOCUMENTATION & REQUIREMENTS

What problem does it solve?

This Skill solves the challenge of moving an MLflow ResponsesAgent from Databricks Model Serving to Databricks Apps, where the deployment model changes from predict()/predict_stream() methods to @invoke/@stream functions.

Core Features & Use Cases

  • End-to-end migration workflow: guides you through downloading the original serving artifacts, analyzing the agent code, and transforming it into Databricks Apps server entry points.
  • Async or sync code paths: supports migrating to fully async for better concurrency or keeping synchronous logic with minimal changes.
  • Deployment-ready bundle configuration: instructs how to update databricks.yml using resources parsed from the original MLmodel, then validate/deploy/run the app with Databricks Asset Bundles.
  • Stateful agent considerations: covers checkpointer/store migration patterns using Lakebase and mapping custom_inputs fields like thread_id or user_id.

Quick Start

Tell the AI to migrate your MLflow ResponsesAgent endpoint to a new Databricks App, and specify your Databricks CLI profile, the target app name, and whether you want an async migration.

Frequently Asked Questions about migrate-from-model-serving

High-intent search queries and answers about installing and using this skill.

FAQPage Schema
How do I migrate an MLflow ResponsesAgent from Model Serving to Databricks Apps?▼

To migrate an MLflow ResponsesAgent to Databricks Apps, download the original serving artifacts, refactor predict() and predict_stream() methods into @invoke and @stream decorated functions, update databricks.yml resources, and deploy using Databricks Asset Bundles.

Can I keep synchronous agent logic when moving to Databricks Apps?▼

You can keep synchronous logic with minimal changes during migration to Databricks Apps, or choose a fully async code path for better concurrency depending on your agent's original implementation.

What do I need to migrate a stateful agent with checkpointer memory to Databricks Apps?▼

Migrating a stateful agent requires handling checkpointer and store migration patterns using Lakebase, and mapping custom_inputs fields like thread_id or user_id to preserve conversation memory in the new app.

Does migrating to Databricks Apps require updating databricks.yml configurations?▼

Yes, migration requires updating databricks.yml using resources parsed from the original MLmodel file to configure the deployment bundle before validating, deploying, and running the migrated app.

How do I handle artifacts and code from the original MLflow model bundle during migration?▼

During migration, you must download the model artifacts, analyze the agent code, and copy the /code directory and associated artifacts from the original MLflow model bundle into the new Databricks Apps implementation.

Do I need Databricks CLI authentication to migrate an agent endpoint to a new app?▼

Yes, valid Databricks CLI authentication is required, along with specifying your CLI profile, target app name, and async migration preference to successfully deploy and run the migrated app.