fix(MR): disable archive for mr/mv if has deploys

Signed-off-by: gitdallas <[email protected]>
opendatahub-io · Oct 2, 2024 · 980ca3b · 980ca3b
1 parent 330dfdf
commit 980ca3b
Show file tree

Hide file tree

Showing 8 changed files with 106 additions and 38 deletions.
diff --git a/frontend/src/pages/modelRegistry/screens/ModelVersionDetails/ModelVersionDetails.tsx b/frontend/src/pages/modelRegistry/screens/ModelVersionDetails/ModelVersionDetails.tsx
@@ -118,7 +118,11 @@ const ModelVersionsDetails: React.FC<ModelVersionsDetailProps> = ({ tab, ...page
               />
             </FlexItem>
             <FlexItem>
-              <ModelVersionsDetailsHeaderActions mv={mv} refresh={refresh} />
+              <ModelVersionsDetailsHeaderActions
+                mv={mv}
+                hasDeployment={inferenceServices.data.length > 0}
+                refresh={refresh}
+              />
             </FlexItem>
           </Flex>
         )

diff --git a/.../src/pages/modelRegistry/screens/ModelVersionDetails/ModelVersionDetailsHeaderActions.tsx b/.../src/pages/modelRegistry/screens/ModelVersionDetails/ModelVersionDetailsHeaderActions.tsx
@@ -13,11 +13,13 @@ import DeployRegisteredModelModal from '~/pages/modelRegistry/screens/components
 
 interface ModelVersionsDetailsHeaderActionsProps {
   mv: ModelVersion;
+  hasDeployment?: boolean;
   refresh: () => void;
 }
 
 const ModelVersionsDetailsHeaderActions: React.FC<ModelVersionsDetailsHeaderActionsProps> = ({
   mv,
+  hasDeployment = false,
   refresh,
 }) => {
   const { apiState } = React.useContext(ModelRegistryContext);
@@ -60,10 +62,14 @@ const ModelVersionsDetailsHeaderActions: React.FC<ModelVersionsDetailsHeaderActi
             Deploy
           </DropdownItem>
           <DropdownItem
+            isAriaDisabled={hasDeployment}
             id="archive-version-button"
             aria-label="Archive version"
             key="archive-version-button"
             onClick={() => setIsArchiveModalOpen(true)}
+            tooltipProps={
+              hasDeployment ? { content: 'Deployed versions cannot be archived' } : undefined
+            }
             ref={tooltipRef}
           >
             Archive version

diff --git a/frontend/src/pages/modelRegistry/screens/ModelVersions/ModelVersions.tsx b/frontend/src/pages/modelRegistry/screens/ModelVersions/ModelVersions.tsx
@@ -9,6 +9,8 @@ import { ModelRegistrySelectorContext } from '~/concepts/modelRegistry/context/M
 import { filterLiveVersions } from '~/concepts/modelRegistry/utils';
 import { ModelState } from '~/concepts/modelRegistry/types';
 import { registeredModelArchiveDetailsUrl } from '~/pages/modelRegistry/screens/routeUtils';
+import { useMakeFetchObject } from '~/utilities/useMakeFetchObject';
+import useInferenceServices from '~/pages/modelServing/useInferenceServices';
 import ModelVersionsTabs from './ModelVersionsTabs';
 import ModelVersionsHeaderActions from './ModelVersionsHeaderActions';
 import { ModelVersionsTab } from './const';
@@ -28,6 +30,7 @@ const ModelVersions: React.FC<ModelVersionsProps> = ({ tab, ...pageProps }) => {
   const loadError = mvLoadError || rmLoadError;
   const loaded = mvLoaded && rmLoaded;
   const navigate = useNavigate();
+  const inferenceServices = useMakeFetchObject(useInferenceServices(undefined, rmId));
 
   useEffect(() => {
     if (rm?.state === ModelState.ARCHIVED) {
@@ -53,7 +56,11 @@ const ModelVersions: React.FC<ModelVersionsProps> = ({ tab, ...pageProps }) => {
         </Breadcrumb>
       }
       title={rm?.name}
-      headerAction={rm && <ModelVersionsHeaderActions rm={rm} />}
+      headerAction={
+        rm && (
+          <ModelVersionsHeaderActions hasDeployments={!!inferenceServices.data.length} rm={rm} />
+        )
+      }
       description={<Truncate content={rm?.description || ''} />}
       loadError={loadError}
       loaded={loaded}

diff --git a/frontend/src/pages/modelRegistry/screens/ModelVersions/ModelVersionsHeaderActions.tsx b/frontend/src/pages/modelRegistry/screens/ModelVersions/ModelVersionsHeaderActions.tsx
@@ -16,9 +16,13 @@ import { RegisteredModel, ModelState } from '~/concepts/modelRegistry/types';
 
 interface ModelVersionsHeaderActionsProps {
   rm: RegisteredModel;
+  hasDeployments?: boolean;
 }
 
-const ModelVersionsHeaderActions: React.FC<ModelVersionsHeaderActionsProps> = ({ rm }) => {
+const ModelVersionsHeaderActions: React.FC<ModelVersionsHeaderActionsProps> = ({
+  rm,
+  hasDeployments = false,
+}) => {
   const { apiState } = React.useContext(ModelRegistryContext);
   const { preferredModelRegistry } = React.useContext(ModelRegistrySelectorContext);
 
@@ -56,6 +60,12 @@ const ModelVersionsHeaderActions: React.FC<ModelVersionsHeaderActionsProps> = ({
                 key="archive-model-button"
                 onClick={() => setIsArchiveModalOpen(true)}
                 ref={tooltipRef}
+                isAriaDisabled={hasDeployments}
+                tooltipProps={
+                  hasDeployments
+                    ? { content: 'Models with deployed versions cannot be archived.' }
+                    : undefined
+                }
               >
                 Archive model
               </DropdownItem>

diff --git a/frontend/src/pages/modelRegistry/screens/ModelVersions/ModelVersionsTable.tsx b/frontend/src/pages/modelRegistry/screens/ModelVersions/ModelVersionsTable.tsx
@@ -2,8 +2,11 @@ import * as React from 'react';
 import { Table } from '~/components/table';
 import { ModelVersion } from '~/concepts/modelRegistry/types';
 import DashboardEmptyTableView from '~/concepts/dashboard/DashboardEmptyTableView';
-import ModelVersionsTableRow from './ModelVersionsTableRow';
+import useInferenceServices from '~/pages/modelServing/useInferenceServices';
+import { useMakeFetchObject } from '~/utilities/useMakeFetchObject';
+import { KnownLabels } from '~/k8sTypes';
 import { mvColumns } from './ModelVersionsTableColumns';
+import ModelVersionsTableRow from './ModelVersionsTableRow';
 
 type ModelVersionsTableProps = {
   clearFilters: () => void;
@@ -18,24 +21,35 @@ const ModelVersionsTable: React.FC<ModelVersionsTableProps> = ({
   toolbarContent,
   isArchiveModel,
   refresh,
-}) => (
-  <Table
-    data-testid="model-versions-table"
-    data={modelVersions}
-    columns={mvColumns}
-    toolbarContent={toolbarContent}
-    defaultSortColumn={3}
-    enablePagination
-    emptyTableView={<DashboardEmptyTableView onClearFilters={clearFilters} />}
-    rowRenderer={(mv) => (
-      <ModelVersionsTableRow
-        key={mv.name}
-        modelVersion={mv}
-        isArchiveModel={isArchiveModel}
-        refresh={refresh}
-      />
-    )}
-  />
-);
+}) => {
+  const inferenceServices = useMakeFetchObject(
+    useInferenceServices(undefined, modelVersions[0].registeredModelId),
+  );
+  return (
+    <Table
+      data-testid="model-versions-table"
+      data={modelVersions}
+      columns={mvColumns}
+      toolbarContent={toolbarContent}
+      defaultSortColumn={3}
+      enablePagination
+      emptyTableView={<DashboardEmptyTableView onClearFilters={clearFilters} />}
+      rowRenderer={(mv) => {
+        const hasDeployment = !!inferenceServices.data.some(
+          (s) => s.metadata.labels?.[KnownLabels.MODEL_VERSION_ID] === mv.id,
+        );
+        return (
+          <ModelVersionsTableRow
+            hasDeployment={hasDeployment}
+            key={mv.name}
+            modelVersion={mv}
+            isArchiveModel={isArchiveModel}
+            refresh={refresh}
+          />
+        );
+      }}
+    />
+  );
+};
 
 export default ModelVersionsTable;
diff --git a/frontend/src/pages/modelRegistry/screens/ModelVersions/ModelVersionsTableRow.tsx b/frontend/src/pages/modelRegistry/screens/ModelVersions/ModelVersionsTableRow.tsx
@@ -21,13 +21,15 @@ type ModelVersionsTableRowProps = {
   modelVersion: ModelVersion;
   isArchiveRow?: boolean;
   isArchiveModel?: boolean;
+  hasDeployment?: boolean;
   refresh: () => void;
 };
 
 const ModelVersionsTableRow: React.FC<ModelVersionsTableRowProps> = ({
   modelVersion: mv,
   isArchiveRow,
   isArchiveModel,
+  hasDeployment = false,
   refresh,
 }) => {
   const navigate = useNavigate();
@@ -52,6 +54,10 @@ const ModelVersionsTableRow: React.FC<ModelVersionsTableRowProps> = ({
         {
           title: 'Archive model version',
           onClick: () => setIsArchiveModalOpen(true),
+          isAriaDisabled: hasDeployment,
+          tooltipProps: hasDeployment
+            ? { content: 'Deployed versions cannot be archived' }
+            : undefined,
         },
       ];
 

diff --git a/frontend/src/pages/modelRegistry/screens/RegisteredModels/RegisteredModelTable.tsx b/frontend/src/pages/modelRegistry/screens/RegisteredModels/RegisteredModelTable.tsx
@@ -2,8 +2,11 @@ import * as React from 'react';
 import { Table } from '~/components/table';
 import { RegisteredModel } from '~/concepts/modelRegistry/types';
 import DashboardEmptyTableView from '~/concepts/dashboard/DashboardEmptyTableView';
-import { rmColumns } from './RegisteredModelsTableColumns';
+import useInferenceServices from '~/pages/modelServing/useInferenceServices';
+import { useMakeFetchObject } from '~/utilities/useMakeFetchObject';
+import { KnownLabels } from '~/k8sTypes';
 import RegisteredModelTableRow from './RegisteredModelTableRow';
+import { rmColumns } from './RegisteredModelsTableColumns';
 
 type RegisteredModelTableProps = {
   clearFilters: () => void;
@@ -16,19 +19,31 @@ const RegisteredModelTable: React.FC<RegisteredModelTableProps> = ({
   registeredModels,
   toolbarContent,
   refresh,
-}) => (
-  <Table
-    data-testid="registered-model-table"
-    data={registeredModels}
-    columns={rmColumns}
-    toolbarContent={toolbarContent}
-    defaultSortColumn={2}
-    enablePagination
-    emptyTableView={<DashboardEmptyTableView onClearFilters={clearFilters} />}
-    rowRenderer={(rm) => (
-      <RegisteredModelTableRow key={rm.name} registeredModel={rm} refresh={refresh} />
-    )}
-  />
-);
+}) => {
+  const inferenceServices = useMakeFetchObject(useInferenceServices());
+  const hasDeploys = (rmId: string) =>
+    !!inferenceServices.data.some(
+      (s) => s.metadata.labels?.[KnownLabels.REGISTERED_MODEL_ID] === rmId,
+    );
+  return (
+    <Table
+      data-testid="registered-model-table"
+      data={registeredModels}
+      columns={rmColumns}
+      toolbarContent={toolbarContent}
+      defaultSortColumn={2}
+      enablePagination
+      emptyTableView={<DashboardEmptyTableView onClearFilters={clearFilters} />}
+      rowRenderer={(rm) => (
+        <RegisteredModelTableRow
+          key={rm.name}
+          hasDeploys={hasDeploys(rm.id)}
+          registeredModel={rm}
+          refresh={refresh}
+        />
+      )}
+    />
+  );
+};
 
 export default RegisteredModelTable;
diff --git a/frontend/src/pages/modelRegistry/screens/RegisteredModels/RegisteredModelTableRow.tsx b/frontend/src/pages/modelRegistry/screens/RegisteredModels/RegisteredModelTableRow.tsx
@@ -19,12 +19,14 @@ import { ModelVersionsTab } from '~/pages/modelRegistry/screens/ModelVersions/co
 type RegisteredModelTableRowProps = {
   registeredModel: RegisteredModel;
   isArchiveRow?: boolean;
+  hasDeploys?: boolean;
   refresh: () => void;
 };
 
 const RegisteredModelTableRow: React.FC<RegisteredModelTableRowProps> = ({
   registeredModel: rm,
   isArchiveRow,
+  hasDeploys = false,
   refresh,
 }) => {
   const { apiState } = React.useContext(ModelRegistryContext);
@@ -57,6 +59,10 @@ const RegisteredModelTableRow: React.FC<RegisteredModelTableRowProps> = ({
       : {
           title: 'Archive model',
           onClick: () => setIsArchiveModalOpen(true),
+          isAriaDisabled: hasDeploys,
+          tooltipProps: hasDeploys
+            ? { content: 'Models with deployed versions cannot be archived.' }
+            : undefined,
         },
   ];