From c06729a042cbf4346eb0f61a19eb65326344edc6 Mon Sep 17 00:00:00 2001
From: Jeffrey Phillips <jephilli@redhat.com>
Date: Wed, 3 Apr 2024 08:01:20 -0400
Subject: [PATCH] [RHOAIENG-4859] Fix for showing correct models in project
 overview

---
 .../serverModels/deployedModels/DeployedModelsCard.tsx | 10 +++-------
 .../deployedModels/DeployedModelsSection.tsx           |  5 ++++-
 2 files changed, 7 insertions(+), 8 deletions(-)

diff --git a/frontend/src/pages/projects/screens/detail/overview/serverModels/deployedModels/DeployedModelsCard.tsx b/frontend/src/pages/projects/screens/detail/overview/serverModels/deployedModels/DeployedModelsCard.tsx
index 95f894d5c3..16271abe3c 100644
--- a/frontend/src/pages/projects/screens/detail/overview/serverModels/deployedModels/DeployedModelsCard.tsx
+++ b/frontend/src/pages/projects/screens/detail/overview/serverModels/deployedModels/DeployedModelsCard.tsx
@@ -17,7 +17,6 @@ import {
 } from '@patternfly/react-core';
 import { ExclamationCircleIcon } from '@patternfly/react-icons';
 import HeaderIcon from '~/concepts/design/HeaderIcon';
-import { ProjectDetailsContext } from '~/pages/projects/ProjectDetailsContext';
 import ModelServingContextProvider from '~/pages/modelServing/ModelServingContext';
 import { ProjectObjectType } from '~/concepts/design/utils';
 import TypeBorderedCard from '~/concepts/design/TypeBorderedCard';
@@ -30,10 +29,10 @@ enum FilterStates {
 
 interface DeployedModelsCardProps {
   isMultiPlatform: boolean;
+  namespace?: string;
 }
 
-const DeployedModelsCard: React.FC<DeployedModelsCardProps> = ({ isMultiPlatform }) => {
-  const { currentProject } = React.useContext(ProjectDetailsContext);
+const DeployedModelsCard: React.FC<DeployedModelsCardProps> = ({ isMultiPlatform, namespace }) => {
   const [filteredState, setFilteredState] = React.useState<FilterStates | undefined>();
 
   const renderError = (message?: string): React.ReactElement => (
@@ -92,10 +91,7 @@ const DeployedModelsCard: React.FC<DeployedModelsCardProps> = ({ isMultiPlatform
         </Flex>
       </CardHeader>
       <CardBody>
-        <ModelServingContextProvider
-          namespace={currentProject.metadata.namespace}
-          getErrorComponent={renderError}
-        >
+        <ModelServingContextProvider namespace={namespace} getErrorComponent={renderError}>
           <DeployedModelsGallery
             showSuccessful={!filteredState || filteredState === FilterStates.success}
             showFailed={!filteredState || filteredState === FilterStates.failed}
diff --git a/frontend/src/pages/projects/screens/detail/overview/serverModels/deployedModels/DeployedModelsSection.tsx b/frontend/src/pages/projects/screens/detail/overview/serverModels/deployedModels/DeployedModelsSection.tsx
index 5148acbe10..7c5476c51c 100644
--- a/frontend/src/pages/projects/screens/detail/overview/serverModels/deployedModels/DeployedModelsSection.tsx
+++ b/frontend/src/pages/projects/screens/detail/overview/serverModels/deployedModels/DeployedModelsSection.tsx
@@ -61,7 +61,10 @@ const DeployedModelsSection: React.FC<DeployedModelsSectionProps> = ({ isMultiPl
 
   return (
     <CollapsibleSection title="Serve models" data-testid="model-server-section">
-      <DeployedModelsCard isMultiPlatform={isMultiPlatform} />
+      <DeployedModelsCard
+        isMultiPlatform={isMultiPlatform}
+        namespace={currentProject.metadata.name}
+      />
     </CollapsibleSection>
   );
 };