Skip to content

[ Gateway 9.5/10 ] Refactor the provider display for better UX#19503

Merged
BenWilson2 merged 1 commit intomlflow:masterfrom
BenWilson2:stack/gateway-ui/provider-optimization
Dec 26, 2025
Merged

[ Gateway 9.5/10 ] Refactor the provider display for better UX#19503
BenWilson2 merged 1 commit intomlflow:masterfrom
BenWilson2:stack/gateway-ui/provider-optimization

Conversation

@BenWilson2
Copy link
Member

@BenWilson2 BenWilson2 commented Dec 19, 2025

🥞 Stacked PR

Use this link to review incremental changes.


Related Issues/PRs

#xxx

What changes are proposed in this pull request?

Create a compound single-entity provider listing that combines Vertex provider listings, OpenAI / Azure OpenAI, and separates the LiteLLM-backed providers to a separate listing section (with navigation to move between the two main listings).

Screenshot 2025-12-19 at 7 33 28 PM

How is this PR tested?

  • Existing unit/integration tests
  • New unit/integration tests
  • Manual tests

Does this PR require documentation update?

  • No. You can skip the rest of this section.
  • Yes. I've updated:
    • Examples
    • API references
    • Instructions

Release Notes

Is this a user-facing change?

  • No. You can skip the rest of this section.
  • Yes. Give a description of this change to be included in the release notes for MLflow users.

What component(s), interfaces, languages, and integrations does this PR affect?

Components

  • area/tracking: Tracking Service, tracking client APIs, autologging
  • area/models: MLmodel format, model serialization/deserialization, flavors
  • area/model-registry: Model Registry service, APIs, and the fluent client calls for Model Registry
  • area/scoring: MLflow Model server, model deployment tools, Spark UDFs
  • area/evaluation: MLflow model evaluation features, evaluation metrics, and evaluation workflows
  • area/gateway: MLflow AI Gateway client APIs, server, and third-party integrations
  • area/prompts: MLflow prompt engineering features, prompt templates, and prompt management
  • area/tracing: MLflow Tracing features, tracing APIs, and LLM tracing functionality
  • area/projects: MLproject format, project running backends
  • area/uiux: Front-end, user experience, plotting, JavaScript, JavaScript dev server
  • area/build: Build and test infrastructure for MLflow
  • area/docs: MLflow documentation pages

How should the PR be classified in the release notes? Choose one:

  • rn/none - No description will be included. The PR will be mentioned only by the PR number in the "Small Bugfixes and Documentation Updates" section
  • rn/breaking-change - The PR will be mentioned in the "Breaking Changes" section
  • rn/feature - A new user-facing feature worth mentioning in the release notes
  • rn/bug-fix - A user-facing bug fix worth mentioning in the release notes
  • rn/documentation - A user-facing documentation change worth mentioning in the release notes

Should this PR be included in the next patch release?

Yes should be selected for bug fixes, documentation updates, and other small changes. No should be selected for new features and larger changes. If you're unsure about the release classification of this PR, leave this unchecked to let the maintainers decide.

What is a minor/patch release?
  • Minor release: a release that increments the second part of the version number (e.g., 1.2.0 -> 1.3.0).
    Bug fixes, doc updates and new features usually go into minor releases.
  • Patch release: a release that increments the third part of the version number (e.g., 1.2.0 -> 1.2.1).
    Bug fixes and doc updates usually go into patch releases.
  • Yes (this PR will be cherry-picked and included in the next patch release)
  • No (this PR will be included in the next minor release)

Copy link
Contributor

Copilot AI left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Pull request overview

This PR refactors the AI Gateway provider display to improve UX by:

  • Creating compound provider listings that combine related providers (OpenAI/Azure OpenAI, Vertex AI variants)
  • Separating LiteLLM-backed providers into a distinct section with navigation
  • Adding comprehensive UI components for API key management, endpoints, and model configuration
  • Enhancing the Python backend with additional authentication modes and model metadata fields

Reviewed changes

Copilot reviewed 112 out of 115 changed files in this pull request and generated no comments.

Show a summary per file
File Description
mlflow/utils/providers.py Adds bedrock API key auth mode, sagemaker provider config, excludes bedrock_converse, adds model metadata fields
mlflow/server/js/src/lang/default/en.json Adds 200+ internationalization strings for new UI features
mlflow/server/js/src/gateway/utils/*.ts New utility modules for provider grouping, formatting, error handling, and date conversion
mlflow/server/js/src/gateway/hooks/*.ts Comprehensive React Query hooks for data fetching and mutations
mlflow/server/js/src/gateway/components/*.tsx New UI components for side navigation, secrets, API keys, and endpoints
mlflow/server/js/src/gateway/pages/*.tsx Refactored main gateway page and new pages for API keys and endpoint management
mlflow/server/js/src/gateway/types.ts Updated TypeScript type definitions with renamed interfaces and new fields

The changes are well-structured with good separation of concerns, comprehensive test coverage for utility functions, and proper TypeScript typing throughout.


💡 Add Copilot custom instructions for smarter, more guided reviews. Learn how to get started.

@github-actions
Copy link
Contributor

github-actions bot commented Dec 20, 2025

Documentation preview for 8f42321 is available at:

More info
  • Ignore this comment if this PR does not change the documentation.
  • The preview is updated when a new commit is pushed to this PR.
  • This comment was created by this workflow run.
  • The documentation was built by this workflow run.

@BenWilson2 BenWilson2 force-pushed the stack/gateway-ui/provider-optimization branch from b370741 to 8916ec0 Compare December 20, 2025 02:24
@BenWilson2 BenWilson2 force-pushed the stack/gateway-ui/provider-optimization branch 6 times, most recently from 449716c to 6d2416a Compare December 23, 2025 00:27
@BenWilson2 BenWilson2 force-pushed the stack/gateway-ui/provider-optimization branch 8 times, most recently from ab3bf85 to 1200acc Compare December 23, 2025 22:10
Copy link
Collaborator

@TomeHirata TomeHirata left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM

@BenWilson2 BenWilson2 force-pushed the stack/gateway-ui/provider-optimization branch 2 times, most recently from b24ac74 to ccb8f35 Compare December 26, 2025 18:54
Signed-off-by: Ben Wilson <benjamin.wilson@databricks.com>
@BenWilson2 BenWilson2 force-pushed the stack/gateway-ui/provider-optimization branch from ccb8f35 to 8f42321 Compare December 26, 2025 19:30
@BenWilson2 BenWilson2 added this pull request to the merge queue Dec 26, 2025
Merged via the queue into mlflow:master with commit 2cfb8ff Dec 26, 2025
72 of 74 checks passed
@BenWilson2 BenWilson2 deleted the stack/gateway-ui/provider-optimization branch December 26, 2025 20:47
omarfarhoud pushed a commit to omarfarhoud/mlflow that referenced this pull request Jan 20, 2026
…w#19503)

Signed-off-by: Ben Wilson <benjamin.wilson@databricks.com>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

area/uiux Front-end, user experience, plotting, JavaScript, JavaScript dev server rn/feature Mention under Features in Changelogs.

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants