Add user selector in the gateway usage page by TomeHirata · Pull Request #20944 · mlflow/mlflow

TomeHirata · 2026-02-18T03:26:36Z

🥞 Stacked PR

Use this link to review incremental changes.

stack/gateway-user/user-selector-metrics [Files changed]

Related Issues/PRs

n/a

What changes are proposed in this pull request?

This PR introduces a way to filter gateway usage metrics by user.

How is this PR tested?

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

Does this PR require documentation update?

Does this PR require updating the MLflow Skills repository?

No. You can skip the rest of this section.
Yes. Please link the corresponding PR or explain how you plan to update it.

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

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)

github-actions · 2026-02-18T03:38:15Z

🛠 DevTools 🛠

Install mlflow from this PR

# mlflow
pip install git+https://github.com/mlflow/mlflow.git@refs/pull/20944/merge
# mlflow-skinny
pip install git+https://github.com/mlflow/mlflow.git@refs/pull/20944/merge#subdirectory=libs/skinny

For Databricks, use the following command:

%sh curl -LsSf https://raw.githubusercontent.com/mlflow/mlflow/HEAD/dev/install-skinny.sh | sh -s pull/20944/merge

Copilot

Pull request overview

Adds an admin-only “User” selector to the AI Gateway Usage page and plumbs trace-metrics filter expressions through the overview chart stack so charts can be scoped to a selected authenticated user.

Changes:

Add /mlflow/users/list (REST + ajax) auth endpoint and an integration test covering admin vs non-admin behavior.
Introduce a useUsersQuery hook + a user dropdown in GatewayUsagePage, emitting a trace-metadata filter on mlflow.auth.userId.
Extend overview chart context and trace chart data hooks to accept and forward filters to trace-metrics queries.

Reviewed changes

Copilot reviewed 15 out of 15 changed files in this pull request and generated 5 comments.

Show a summary per file

File	Description
tests/server/auth/test_auth.py	Adds integration coverage for the new list-users auth endpoint.
mlflow/server/auth/routes.py	Declares REST + AJAX route constants for listing users.
mlflow/server/auth/init.py	Registers the list-users routes, validator mapping, and handler implementation.
mlflow/server/js/src/gateway/hooks/useUsersQuery.ts	Adds a React Query hook for fetching users from the auth ajax endpoint.
mlflow/server/js/src/gateway/pages/GatewayUsagePage.tsx	Adds user selection UI and constructs trace-metadata filters passed into charts.
mlflow/server/js/src/gateway/components/GatewayChartsPanel.tsx	Wires optional `filters` prop through to the overview charts provider/query layer.
mlflow/server/js/src/experiment-tracking/pages/experiment-overview/OverviewChartContext.tsx	Adds `filters` to shared chart context.
mlflow/server/js/src/experiment-tracking/pages/experiment-overview/hooks/useTrace*.ts	Plumbs `filters` into trace-metrics queries for requests/latency/errors/tokens/cost.
mlflow/server/js/src/shared/web-shared/model-trace-explorer/TraceMetrics.types.ts	Adds helper for building metadata filter expressions + constant for auth user-id metadata key.

mlflow/server/js/src/shared/web-shared/model-trace-explorer/TraceMetrics.types.ts

tests/server/auth/test_auth.py

mlflow/server/auth/__init__.py

mlflow/server/js/src/gateway/pages/GatewayUsagePage.tsx

mlflow/server/js/src/gateway/hooks/useUsersQuery.ts

github-actions · 2026-02-18T03:48:43Z

Documentation preview for b81302a is available at:

https://pr-20944--mlflow-docs-preview.netlify.app/docs/latest/

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.

TomeHirata · 2026-02-18T08:23:42Z

mlflow/server/js/src/gateway/pages/GatewayUsagePage.tsx

+        </SimpleSelect>
+        {isLoadingEndpoints && <Spinner size="small" />}
+      </div>
+      {!usersError && (


Main change is here

Signed-off-by: Tomu Hirata <tomu.hirata@gmail.com>

mlflow/server/js/src/gateway/hooks/useUsersQuery.ts

daniellok-db

lgtm!

Signed-off-by: Tomu Hirata <tomu.hirata@gmail.com>

TomeHirata mentioned this pull request Feb 18, 2026

Add an endpoint for list users #20943

Merged

31 tasks

TomeHirata marked this pull request as ready for review February 18, 2026 03:38

Copilot AI review requested due to automatic review settings February 18, 2026 03:38

github-actions bot added rn/feature Mention under Features in Changelogs. size/L Large PR (200-499 LoC) labels Feb 18, 2026

Copilot started reviewing on behalf of TomeHirata February 18, 2026 03:38 View session

TomeHirata force-pushed the stack/gateway-user/user-selector-metrics branch from 17bd52c to f4294de Compare February 18, 2026 03:38

github-actions bot added the v3.10.0 label Feb 18, 2026

TomeHirata added the team-review Trigger a team review request label Feb 18, 2026

github-actions bot requested review from WeichenXu123 and daniellok-db February 18, 2026 03:40

Copilot AI reviewed Feb 18, 2026

View reviewed changes

TomeHirata force-pushed the stack/gateway-user/user-selector-metrics branch 2 times, most recently from 5010ace to 46f718b Compare February 18, 2026 08:09

TomeHirata requested a review from harupy February 18, 2026 08:22

TomeHirata commented Feb 18, 2026

View reviewed changes

TomeHirata added 2 commits February 18, 2026 23:36

Add user selector in the gateway usage page

fd4b1f7

Signed-off-by: Tomu Hirata <tomu.hirata@gmail.com>

fix test

9ea099b

Signed-off-by: Tomu Hirata <tomu.hirata@gmail.com>

TomeHirata force-pushed the stack/gateway-user/user-selector-metrics branch from 9857234 to 9ea099b Compare February 18, 2026 14:36

daniellok-db reviewed Feb 19, 2026

View reviewed changes

mlflow/server/js/src/gateway/hooks/useUsersQuery.ts Outdated Show resolved Hide resolved

daniellok-db approved these changes Feb 19, 2026

View reviewed changes

github-actions bot assigned daniellok-db Feb 19, 2026

refactor

b81302a

Signed-off-by: Tomu Hirata <tomu.hirata@gmail.com>

TomeHirata enabled auto-merge February 19, 2026 05:30

TomeHirata added this pull request to the merge queue Feb 19, 2026

Merged via the queue into mlflow:master with commit 6478047 Feb 19, 2026
14 checks passed

TomeHirata deleted the stack/gateway-user/user-selector-metrics branch February 19, 2026 05:45

daniellok-db pushed a commit to daniellok-db/mlflow that referenced this pull request Feb 20, 2026

Add user selector in the gateway usage page (mlflow#20944)

1a94619

Signed-off-by: Tomu Hirata <tomu.hirata@gmail.com>

daniellok-db pushed a commit that referenced this pull request Feb 20, 2026

Add user selector in the gateway usage page (#20944)

1a2588a

Signed-off-by: Tomu Hirata <tomu.hirata@gmail.com>

Provide feedback

Saved searches

Use saved searches to filter your results more quickly

Add user selector in the gateway usage page#20944

Add user selector in the gateway usage page#20944
TomeHirata merged 3 commits intomlflow:masterfrom
TomeHirata:stack/gateway-user/user-selector-metrics

TomeHirata commented Feb 18, 2026 •

edited

Loading

Uh oh!

github-actions bot commented Feb 18, 2026

Install mlflow from this PR

Uh oh!

Copilot AI left a comment

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

github-actions bot commented Feb 18, 2026 •

edited

Loading

Uh oh!

TomeHirata Feb 18, 2026

Uh oh!

Uh oh!

daniellok-db left a comment

Uh oh!

Uh oh!

Reviewers

Assignees

Labels

Projects

Milestone

Development

Uh oh!

3 participants

Conversation

TomeHirata commented Feb 18, 2026 • edited Loading Uh oh! There was an error while loading. Please reload this page.

Uh oh!

🥞 Stacked PR

Related Issues/PRs

What changes are proposed in this pull request?

How is this PR tested?

Does this PR require documentation update?

Does this PR require updating the MLflow Skills repository?

Release Notes

Is this a user-facing change?

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

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

Should this PR be included in the next patch release?

Uh oh!

github-actions bot commented Feb 18, 2026

Install mlflow from this PR

Uh oh!

Copilot AI left a comment

Choose a reason for hiding this comment

Pull request overview

Reviewed changes

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

github-actions bot commented Feb 18, 2026 • edited Loading Uh oh! There was an error while loading. Please reload this page.

Uh oh!

Uh oh!

TomeHirata Feb 18, 2026

Choose a reason for hiding this comment

Uh oh!

Uh oh!

daniellok-db left a comment

Choose a reason for hiding this comment

Uh oh!

Uh oh!

Reviewers

Assignees

Labels

Projects

Milestone

Development

Uh oh!

3 participants

TomeHirata commented Feb 18, 2026 •

edited

Loading

github-actions bot commented Feb 18, 2026 •

edited

Loading