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.