feat(ui): register and display model_type from catalog and model registry by Taj010 · Pull Request #2545 · kubeflow/model-registry

Taj010 · 2026-04-03T19:12:24Z

Description

This PR adds a shared Model type field (Generative / Predictive) on register-from-catalog and register-from-registry flows.

Catalog registration prefills the field from catalog metadata when it is generative or predictive.
Register-from-registry requires a selection before submit.

Model overview and version details was also adjusted to show model type property.

How Has This Been Tested?

Manually Testing

Open register-from-catalog with a catalog model confirm dropdown prefills
Open MR Register model, confirm submit stays disabled until model type is chosen
Open model and version detail pages and confirm Model type displays as expected.

Automated Testing

Added tests to modelCatalogUtils.spec.tsand utils.spec.ts file and adjusted tests in registerAndStoreFields.cy.ts

Merge criteria:

All the commits have been signed-off (To pass the DCO check)

The commits have meaningful messages
Automated tests are provided as part of the PR for major new functionalities; testing instructions have been added in the PR body (for PRs involving changes that are not immediately obvious).
The developer has manually tested the changes and verified that the changes work.
Code changes follow the kubeflow contribution guidelines.
For first time contributors: Please reach out to the Reviewers to ensure all tests are being run, ensuring the label ok-to-test has been added to the PR.

If you have UI changes

The developer has added tests or explained why testing cannot be added.
Included any necessary screenshots or gifs if it was a UI change.
Verify that UI/UX changes conform the UX guidelines for Kubeflow.

Signed-off-by: Taj010 <rsheen003@gmail.com>

…o it Signed-off-by: Taj010 <rsheen003@gmail.com>

Signed-off-by: Taj010 <rsheen003@gmail.com>

YuliaKrimerman · 2026-04-08T18:32:47Z

...nts/ui/frontend/src/app/pages/modelRegistry/screens/RegisterModel/RegisterModelTypeField.tsx

+}) => {
+  const stored = getModelTypeStoredValueFromCustomProperties(modelCustomProperties);
+
+  const handleChange = (key: string, isPlaceholder: boolean) => {


The handleChange callback checks both isPlaceholder and key === MODEL_TYPE_PLACEHOLDER_KEY. The isPlaceholder check alone should be sufficient — checking the key as well is redundant but harmless.

YuliaKrimerman · 2026-04-08T18:33:28Z

clients/ui/frontend/src/app/pages/modelRegistry/screens/RegisterModel/registerModelTypeUtils.ts

+      string_value: next,
+    };
+  }
+  return result;


Please add unit tests for this utility.

Added registerModelTypeUtils.spec.ts

YuliaKrimerman · 2026-04-08T18:37:35Z

clients/ui/frontend/src/app/pages/modelRegistry/screens/RegisterModel/utils.ts

-  isSubmitDisabledForCommonFields(formData, namespaceHasAccess, isNamespaceAccessLoading) ||
-  !isNameValid(formData.modelName) ||
-  isModelNameExisting(formData.modelName, registeredModels);
+  options?: { requireModelType?: boolean },


We've changed the default behavior here which is fine, the only thing I would maybe comment here that if in the future a caller will forget to opt-out it might be surprising . Alternatively we can default to false with explicit opt-in( just safer) not a must, you're choice

Adjusted to have requireModelType now defaults to false and adjusted the utils.spec.ts too

YuliaKrimerman · 2026-04-08T18:38:46Z

clients/ui/frontend/src/app/pages/modelCatalog/utils/modelCatalogUtils.ts

+ * Returns model registry customProperties entries to prefill `model_type` when registering
+ * from the catalog. Only generative/predictive are copied; unknown or missing values yield {}.
+ */
+export const getCatalogModelTypePropertyForRegistration = (


In here we can reuse reuse getModelTypeStoredValueFromCustomProperties from registerModelTypeUtils.ts , or even better - buildCustomPropertiesWithModelType

Reimplemented via getModelTypeStoredValueFromCustomProperties and buildCustomPropertiesWithModelType

Signed-off-by: Taj010 <rsheen003@gmail.com>

google-oss-prow · 2026-04-09T20:41:26Z

[APPROVALNOTIFIER] This PR is NOT APPROVED

This pull-request has been approved by:
Once this PR has been reviewed and has the lgtm label, please ask for approval from yuliakrimerman. For more information see the Kubernetes Code Review Process.

The full list of commands accepted by this bot can be found here.

Details

Needs approval from an approver in each of these files:

clients/ui/OWNERS

Approvers can indicate their approval by writing /approve in a comment
Approvers can cancel approval by writing /approve cancel in a comment

Signed-off-by: Taj010 <rsheen003@gmail.com>

mturley

Nice work! One thing to check: model_type is now displayed in a dedicated "Model type" field on the detail pages, but the getProperties() function in screens/utils.ts doesn't exclude model_type from the editable properties table. This means it will show up twice — once as the new read-only display field, and again as an editable row in the properties section. Should it be filtered out of the properties table (similar to how _lastModified and _registeredFrom are excluded)?

Also I actually think we may want a tech debt ticket to clean those two up, at least _registeredFrom isn't used anymore now that we have modelSource* properties, but that's out of scope.

mturley · 2026-04-09T21:25:17Z

clients/ui/frontend/src/app/pages/modelRegistry/screens/ModelVersions/ModelDetailsCard.tsx

      >
        <ModelTimestamp timeSinceEpoch={rm.createTimeSinceEpoch} />
      </DashboardDescriptionListGroup>
+      <DashboardDescriptionListGroup title="Model type">
+        <Content component="p" data-testid="registered-model-model-type">
+          {formatModelTypeDisplay(modelTypeRaw)}
+        </Content>


model_type will also appear in the editable properties table rendered by ModelPropertiesExpandableSection below, since getProperties() in screens/utils.ts doesn't filter it out. This means the user will see it twice — here as read-only, and again as an editable property row. Consider adding model_type to the exclusion list in getProperties().

Taj010 added 2 commits April 3, 2026 14:39

model type in register flow and adjusting the model details

a0d6444

Signed-off-by: Taj010 <rsheen003@gmail.com>

adjust registerAndStoreFields file failing tests and add Model Type t…

d707a9a

…o it Signed-off-by: Taj010 <rsheen003@gmail.com>

google-oss-prow bot requested review from chambridge and fege April 3, 2026 19:12

google-oss-prow bot added the size/L label Apr 3, 2026

github-actions bot added Area/UI and removed size/L labels Apr 3, 2026

fix prettier

9dfa838

Signed-off-by: Taj010 <rsheen003@gmail.com>

google-oss-prow bot added the size/L label Apr 3, 2026

YuliaKrimerman suggested changes Apr 8, 2026

View reviewed changes

google-oss-prow bot assigned YuliaKrimerman Apr 8, 2026

address comments

15e0f3f

Signed-off-by: Taj010 <rsheen003@gmail.com>

google-oss-prow bot added size/XL and removed size/L labels Apr 9, 2026

fix prettier

a13e1ce

Signed-off-by: Taj010 <rsheen003@gmail.com>

mturley suggested changes Apr 9, 2026

View reviewed changes

google-oss-prow bot assigned mturley Apr 9, 2026

Provide feedback

Saved searches

Use saved searches to filter your results more quickly

feat(ui): register and display model_type from catalog and model registry#2545

feat(ui): register and display model_type from catalog and model registry#2545
Taj010 wants to merge 5 commits intokubeflow:mainfrom
Taj010:register-model-type

Taj010 commented Apr 3, 2026

Uh oh!

YuliaKrimerman Apr 8, 2026

Uh oh!

YuliaKrimerman Apr 8, 2026

Uh oh!

Taj010 Apr 9, 2026

Uh oh!

YuliaKrimerman Apr 8, 2026

Uh oh!

Taj010 Apr 9, 2026

Uh oh!

YuliaKrimerman Apr 8, 2026

Uh oh!

Taj010 Apr 9, 2026

Uh oh!

google-oss-prow bot commented Apr 9, 2026

Uh oh!

mturley left a comment

Uh oh!

mturley Apr 9, 2026 •

edited

Loading

Uh oh!

Reviewers

Assignees

Labels

Projects

Milestone

Development

Uh oh!

3 participants

Conversation

Taj010 commented Apr 3, 2026

Description

How Has This Been Tested?

Merge criteria:

Uh oh!

YuliaKrimerman Apr 8, 2026

Choose a reason for hiding this comment

Uh oh!

YuliaKrimerman Apr 8, 2026

Choose a reason for hiding this comment

Uh oh!

Taj010 Apr 9, 2026

Choose a reason for hiding this comment

Uh oh!

YuliaKrimerman Apr 8, 2026

Choose a reason for hiding this comment

Uh oh!

Taj010 Apr 9, 2026

Choose a reason for hiding this comment

Uh oh!

YuliaKrimerman Apr 8, 2026

Choose a reason for hiding this comment

Uh oh!

Taj010 Apr 9, 2026

Choose a reason for hiding this comment

Uh oh!

google-oss-prow bot commented Apr 9, 2026

Uh oh!

mturley left a comment

Choose a reason for hiding this comment

Uh oh!

mturley Apr 9, 2026 • edited Loading Uh oh! There was an error while loading. Please reload this page.

Uh oh!

Choose a reason for hiding this comment

Uh oh!

Reviewers

Assignees

Labels

Projects

Milestone

Development

Uh oh!

3 participants

mturley Apr 9, 2026 •

edited

Loading