Bump mlflow from 1.13.1 to 2.20.0rc0 in /environments_setup/mlops-composer-mlflow

[dependabot]

Bumps mlflow from 1.13.1 to 2.20.0rc0.

Release notes

Sourced from mlflow's releases.

MLflow 2.20.0rc0

Release Candidate

MLflow 2.20.0rc0 is a release candidate for 2.20.0. To install, run the following command:

pip install mlflow==2.20.0rc0

Please try it out and report any issues on the issue tracker!

Major New Features

• 💡Type Hint-Based Model Signature: Define your model's signature in the most Pythonic way. MLflow now supports defining a model signature based on the type hints in your PythonModel's predict function, and validating input data payloads against it. (#14182, #14168, #14130, #14100, #14099, @​serena-ruan)

• 🧠 Bedrock / Groq Tracing Support: MLflow Tracing now offers a one-line auto-tracing experience for Amazon Bedrock and Groq LLMs. Track LLM invocation within your model by simply adding mlflow.bedrock.tracing or mlflow.groq.tracing call to the code. (#14018, @​B-Step62, #14006, @​anumita0203)

• 🗒️ Inline Trace Rendering in Jupyter Notebook: MLflow now supports rendering a trace UI within the notebook where you are running models. This eliminates the need to frequently switch between the notebook and browser, creating a seamless local model debugging experience. (#13955, @​daniellok-db)

• ⚡️Faster Model Validation with uv Package Manager: MLflow has adopted uv, a new Rust-based, super-fast Python package manager. This release adds support for the new package manager in the mlflow.models.predict API, enabling faster model environment validation. Stay tuned for more updates! (#13824, @​serena-ruan)

• 🖥️ New Chat Panel in Trace UI: THe MLflow Trace UI now shows a unified chat panel for LLM invocations. The update allows you to view chat messages and function calls in a rich and consistent UI across LLM providers, as well as inspect the raw input and output payloads. (#14211, @​TomuHirata)

Other Features:

• Introduced ChatAgent base class for defining custom python agent (#13797, @​bbqiu)
• Supported Tool Calling in DSPy Tracing (#14196, @​B-Step62)
• Added support for invokers rights in Databricks Resources (#14212, @​aravind-segu)
• Applied timeout override to within-request local scoring server for Spark UDF inference (#14202, @​BenWilson2)
• Supported dictionary type for inference params (#14091, @​serena-ruan)
• Make context parameter optional for calling PythonModel instance (#14059, @​serena-ruan)
• Set default task for ChatModel (#14068, @​stevenchen-db)

MLflow 2.19.0

We are excited to announce the release of MLflow 2.19.0! This release includes a number of significant features, enhancements, and bug fixes.

Major New Features

• ChatModel enhancements - ChatModel now adopts ChatCompletionRequest and ChatCompletionResponse as its new schema. The predict_stream interface uses ChatCompletionChunk to deliver true streaming responses. Additionally, the custom_inputs and custom_outputs fields in ChatModel now utilize AnyType, enabling support for a wider variety of data types. Note: In a future version of MLflow, ChatParams (and by extension, ChatCompletionRequest) will have the default values for n, temperature, and stream removed. (#13782, #13857, @​stevenchen-db)

• Tracing improvements - MLflow Tracing now supports both automatic and manual tracing for DSPy, LlamaIndex and Langchain flavors. Tracing feature is also auto-enabled for mlflow evaluation for all supported flavors. (#13790, #13793, #13795, #13897, @​B-Step62)

• New Tracing Integrations - MLflow Tracing now supports CrewAI and Anthropic, enabling a one-line, fully automated tracing experience. (#13903, @​TomeHirata, #13851, @​gabrielfu)

• Any Type in model signature - MLflow now supports AnyType in model signature. It can be used to host any data types that were not supported before. (#13766, @​serena-ruan)

Other Features:

• [Tracking] Add update_current_trace API for adding tags to an active trace. (#13828, @​B-Step62)
• [Deployments] Update databricks deployments to support AI gateway & additional update endpoints (#13513, @​djliden)
• [Models] Support uv in mlflow.models.predict (#13824, @​serena-ruan)
• [Models] Add type hints support including pydantic models (#13924, @​serena-ruan)
• [Tracking] Add the trace.search_spans() method for searching spans within traces (#13984, @​B-Step62)

... (truncated)

Changelog

Sourced from mlflow's changelog.

CHANGELOG

2.21.0 (2025-03-14)

We are excited to announce the release of MLflow 2.21.0! This release includes a number of significant features, enhancements, and bug fixes.

Major New Features

• 📚 Documentation Redesign: MLflow documentation is fully revamped with a new MDX-based website that provides better navigation and makes it easier to find the information you need! (#13645, @​daniellok-db)
• 🤖 Prompt Registry: MLflow Prompt Registry is a powerful tool that streamlines prompt engineering and management in your GenAI applications. It enables you to version, track, and reuse prompts across your organization. (#14795, #14834, #14936, @​B-Step62, #14960, #14984, @​daniellok-db, #14909, @​hubertzub-db)
• ⚡️ FastAPI Scoring Server: The MLflow inference serverhas been migrated from Flask to FastAPI, enabling ASGI-based scalable inference for improved performance and throughput. (#14307, @​TomeHirata)
• 🔍 Enhanced Tracing Capabilities: MLflow Tracing now supports synchronous/asynchronous generators and auto-tracing for Async OpenAI, providing more flexible and comprehensive tracing options. (#14459, #14400, #14793, @​14792, @​B-Step62)

Features:

• [Tracking] Support OpenAI Agent SDK Auto Tracing (#14987, @​B-Step62)
• [Sqlalchemy / Tracking] Support mysql ssl connections with client certs (#14839, @​aksylumoed)
• [Artifacts] Supports ADLS artifact repo (#14723, @​serena-ruan)
• [Tracking] Add import and docs for txtai integration (#14712, @​B-Step62)
• [Models] Introduce User Auth Policy for Pyfunc Models (#14538, @​aravind-segu)
• [Tracking] Support new Google GenAI SDK (#14576, @​TomeHirata)
• [Tracking] Support generating traces from DSPy built-in compilation and evaluation (#14400, @​B-Step62)
• [Tracking] Add mlflow.log_trace API (#14418, @​TomeHirata)
• [Models] ChatAgent LangGraph and LangChain Connectors (#14215, @​bbqiu)

Bug fixes:

• [Models] Fix infinite recursion error with warning handler module (#14954, @​BenWilson2)
• [Model Registry] Fix invalid type issue for ModelRegistry RestStore (#14980, @​B-Step62)
• [Tracking] Fix: ExperimentViewRunsControlsActionsSelectTags doesn't set loading state to false when set-tag request fails. (#14907, @​harupy)
• [Tracking] Fix a bug in tag creation where tag values containing ": " get truncated (#14896, @​harupy)
• [Tracking] Fix false alert from AMD GPU monitor (#14884, @​B-Step62)
• [Tracking] Fix mlflow.doctor to fall back to mlflow-skinny when mlflow is not found (#14782, @​harupy)
• [Models] Handle LangGraph breaking change (#14794, @​B-Step62)
• [Tracking] Fix DSPy tracing in serving (#14743, @​B-Step62)
• [Tracking] Add limit to the length of experiment artifact locations (#14416, @​daniellok-db)
• [Build] Fix build.py to restore specific files #14444 (#14448, @​arunkothari84)
• [Models] Fix false alert for ChatModel type hint (#14343, @​B-Step62)
• [Model Registry] use aes256 to talk to s3 (#14354, @​artjen)
• [Tracking] Fix LiteLLM autologging (#14340, @​B-Step62)
• [Models] Fix ChatCompletionResponse for model serving Pydantic 1.x (#14332, @​BenWilson2)

Documentation updates:

• [Tracking] Add guide about using MLflow tracing across thread (#14881, @​B-Step62)
• [Docs] Add guide for tracing deepseek (#14826, @​B-Step62)
• [Docs] Update llama Jupyter notebook source (#14754, @​emmanuel-ferdman)
• [Docs] Replace Databricks Community Edition with Lighthouse [1] (#14642, @​TomeHirata)
• [Docs] Update models from code guide and chat model guide to always recommend models from code (#14370, @​smurching)
• [Artifacts] [DOC-FIX #14183] Improve documentation for 'artifact_uri' in 'download_artifacts' (#14225, @​vinayakkgarg)

... (truncated)

Commits

• 3a66a82 Run python3 dev/update_mlflow_versions.py pre-release ... (#14240)
• 49d28d6 Run python3 dev/update_pypi_package_index.py (#14238)
• 8778eed Run python3 dev/update_ml_package_versions.py (#14237)
• 8436d0b Remove inefficient attribute lookup during setting DSPy span standard attribu...
• 1ff5a0d Fix chat tab transition issue (#14233)
• d81addf Fix read example (#14234)
• 146a228 Raise exception if pydantic object is used in infer_signature (#14228)
• 2d0c7df Add support for invokers rights in Databricks Resources (#14212)
• a63ded5 Raise exception for invalid type hint (#14227)
• 514000a Implement standard span attribute logic for DSPy (#14207)
• Additional commits viewable in compare view

Dependabot will resolve any conflicts with this PR as long as you don't alter it yourself. You can also trigger a rebase manually by commenting @dependabot rebase.

---

Dependabot commands and options

You can trigger Dependabot actions by commenting on this PR:

• @dependabot rebase will rebase this PR
• @dependabot recreate will recreate this PR, overwriting any edits that have been made to it
• @dependabot merge will merge this PR after your CI passes on it
• @dependabot squash and merge will squash and merge this PR after your CI passes on it
• @dependabot cancel merge will cancel a previously requested merge and block automerging
• @dependabot reopen will reopen this PR if it is closed
• @dependabot close will close this PR and stop Dependabot recreating it. You can achieve the same result by closing it manually
• @dependabot show <dependency name> ignore conditions will show all of the ignore conditions of the specified dependency
• @dependabot ignore this major version will close this PR and stop Dependabot creating any more for this major version (unless you reopen the PR or upgrade to it yourself)
• @dependabot ignore this minor version will close this PR and stop Dependabot creating any more for this minor version (unless you reopen the PR or upgrade to it yourself)
• @dependabot ignore this dependency will close this PR and stop Dependabot creating any more for this dependency (unless you reopen the PR or upgrade to it yourself)
  You can disable automated security fix PRs for this repo from the Security Alerts page.

[dependabot]

Superseded by #234.