- {getEntityIcon(reference.type)}
+ {searchClassBase.getEntityIcon(reference.type)}
{reference.type}
diff --git a/openmetadata-ui/src/main/resources/ui/src/components/Entity/EntityLineage/AppPipelineModel/AddPipeLineModal.tsx b/openmetadata-ui/src/main/resources/ui/src/components/Entity/EntityLineage/AppPipelineModel/AddPipeLineModal.tsx
index ae2fa3d83c7..50406275c4c 100644
--- a/openmetadata-ui/src/main/resources/ui/src/components/Entity/EntityLineage/AppPipelineModel/AddPipeLineModal.tsx
+++ b/openmetadata-ui/src/main/resources/ui/src/components/Entity/EntityLineage/AppPipelineModel/AddPipeLineModal.tsx
@@ -29,7 +29,7 @@ import {
getEntityReferenceFromEntity,
} from '../../../../utils/EntityUtils';
import Fqn from '../../../../utils/Fqn';
-import { getEntityIcon } from '../../../../utils/TableUtils';
+import searchClassBase from '../../../../utils/SearchClassBase';
import { showErrorToast } from '../../../../utils/ToastUtils';
import '../../../ActivityFeed/FeedEditor/feed-editor.less';
import ErrorPlaceHolder from '../../../common/ErrorWithPlaceholder/ErrorPlaceHolder';
@@ -157,7 +157,7 @@ const AddPipeLineModal = ({
{edgeOptions.map((item) => {
- const icon = getEntityIcon(item.type);
+ const icon = searchClassBase.getEntityIcon(item.type);
const breadcrumb = Fqn.split(item.fullyQualifiedName ?? '').join('/');
return (
diff --git a/openmetadata-ui/src/main/resources/ui/src/components/Entity/EntityLineage/EntityLineageSidebar.component.tsx b/openmetadata-ui/src/main/resources/ui/src/components/Entity/EntityLineage/EntityLineageSidebar.component.tsx
index 31a0d5864e9..d4d83a67641 100644
--- a/openmetadata-ui/src/main/resources/ui/src/components/Entity/EntityLineage/EntityLineageSidebar.component.tsx
+++ b/openmetadata-ui/src/main/resources/ui/src/components/Entity/EntityLineage/EntityLineageSidebar.component.tsx
@@ -20,7 +20,7 @@ import { Node } from 'reactflow';
import { ReactComponent as DragIconDotted } from '../../../assets/svg/dots-six-bold.svg';
import { entityData } from '../../../constants/Lineage.constants';
import { useApplicationStore } from '../../../hooks/useApplicationStore';
-import { getEntityIcon } from '../../../utils/TableUtils';
+import searchClassBase from '../../../utils/SearchClassBase';
import './entity-lineage-sidebar.less';
interface SidebarProps extends HTMLAttributes
{
@@ -56,7 +56,7 @@ const EntityNode: FC = ({ type, label, draggable }) => {
e.preventDefault();
e.stopPropagation();
}}>
- {getEntityIcon(type || '')}
+ {searchClassBase.getEntityIcon(type ?? '')}
{
@@ -47,7 +47,7 @@ const LineageLayers = () => {
onClick={() => onButtonClick(LineageLayerView.COLUMN)}>
- {getEntityIcon(EntityType.TABLE)}
+ {searchClassBase.getEntityIcon(EntityType.TABLE)}
{t('label.column')}
diff --git a/openmetadata-ui/src/main/resources/ui/src/components/Entity/EntityLineage/LineageSearchSelect/LineageSearchSelect.tsx b/openmetadata-ui/src/main/resources/ui/src/components/Entity/EntityLineage/LineageSearchSelect/LineageSearchSelect.tsx
index 240471f8c4b..7b99c456887 100644
--- a/openmetadata-ui/src/main/resources/ui/src/components/Entity/EntityLineage/LineageSearchSelect/LineageSearchSelect.tsx
+++ b/openmetadata-ui/src/main/resources/ui/src/components/Entity/EntityLineage/LineageSearchSelect/LineageSearchSelect.tsx
@@ -22,8 +22,8 @@ import { useLineageProvider } from '../../../../context/LineageProvider/LineageP
import { Column } from '../../../../generated/entity/data/table';
import { getEntityChildrenAndLabel } from '../../../../utils/EntityLineageUtils';
import { getEntityName } from '../../../../utils/EntityUtils';
+import searchClassBase from '../../../../utils/SearchClassBase';
import serviceUtilClassBase from '../../../../utils/ServiceUtilClassBase';
-import { getEntityIcon } from '../../../../utils/TableUtils';
const LineageSearchSelect = () => {
const { t } = useTranslation();
@@ -85,7 +85,7 @@ const LineageSearchSelect = () => {
- {getEntityIcon(node.entityType ?? '')}
+ {searchClassBase.getEntityIcon(node.entityType ?? '')}
{getEntityName(column)}
diff --git a/openmetadata-ui/src/main/resources/ui/src/components/Entity/EntityLineage/NodeChildren/NodeChildren.component.tsx b/openmetadata-ui/src/main/resources/ui/src/components/Entity/EntityLineage/NodeChildren/NodeChildren.component.tsx
index bdacf4163d5..1ac8f1ce1d4 100644
--- a/openmetadata-ui/src/main/resources/ui/src/components/Entity/EntityLineage/NodeChildren/NodeChildren.component.tsx
+++ b/openmetadata-ui/src/main/resources/ui/src/components/Entity/EntityLineage/NodeChildren/NodeChildren.component.tsx
@@ -26,7 +26,7 @@ import { EntityType } from '../../../../enums/entity.enum';
import { Column, Table } from '../../../../generated/entity/data/table';
import { getEntityChildrenAndLabel } from '../../../../utils/EntityLineageUtils';
import { getEntityName } from '../../../../utils/EntityUtils';
-import { getEntityIcon } from '../../../../utils/TableUtils';
+import searchClassBase from '../../../../utils/SearchClassBase';
import { getColumnContent } from '../CustomNode.utils';
import TestSuiteSummaryWidget from '../TestSuiteSummaryWidget/TestSuiteSummaryWidget.component';
import { EntityChildren, NodeChildrenProps } from './NodeChildren.interface';
@@ -200,7 +200,7 @@ const NodeChildren = ({ node, isConnectable }: NodeChildrenProps) => {
}}>
- {getEntityIcon(node.entityType ?? '')}
+ {searchClassBase.getEntityIcon(node.entityType ?? '')}
{childrenHeading}
{isExpanded ? (
diff --git a/openmetadata-ui/src/main/resources/ui/src/components/Entity/EntityList/EntityList.tsx b/openmetadata-ui/src/main/resources/ui/src/components/Entity/EntityList/EntityList.tsx
index 561a2f28a8a..00f4ad933f9 100644
--- a/openmetadata-ui/src/main/resources/ui/src/components/Entity/EntityList/EntityList.tsx
+++ b/openmetadata-ui/src/main/resources/ui/src/components/Entity/EntityList/EntityList.tsx
@@ -18,7 +18,7 @@ import { Link } from 'react-router-dom';
import { EntityReference } from '../../../generated/entity/type';
import entityUtilClassBase from '../../../utils/EntityUtilClassBase';
import { getEntityName } from '../../../utils/EntityUtils';
-import { getEntityIcon } from '../../../utils/TableUtils';
+import searchClassBase from '../../../utils/SearchClassBase';
import EntityListSkeleton from '../../common/Skeleton/MyData/EntityListSkeleton/EntityListSkeleton.component';
import './entity.less';
@@ -75,7 +75,7 @@ export const EntityListWithV1: FunctionComponent = ({
className="entity-button flex-center p-0 m--ml-1"
icon={
- {getEntityIcon(item.type || '')}
+ {searchClassBase.getEntityIcon(item.type || '')}
}
title={getEntityName(
diff --git a/openmetadata-ui/src/main/resources/ui/src/components/Explore/ExploreTree/ExploreTree.tsx b/openmetadata-ui/src/main/resources/ui/src/components/Explore/ExploreTree/ExploreTree.tsx
index 044969a93a4..1afbf9a5e47 100644
--- a/openmetadata-ui/src/main/resources/ui/src/components/Explore/ExploreTree/ExploreTree.tsx
+++ b/openmetadata-ui/src/main/resources/ui/src/components/Explore/ExploreTree/ExploreTree.tsx
@@ -33,7 +33,6 @@ import {
} from '../../../utils/ExploreUtils';
import searchClassBase from '../../../utils/SearchClassBase';
import serviceUtilClassBase from '../../../utils/ServiceUtilClassBase';
-import { getEntityIcon } from '../../../utils/TableUtils';
import { showErrorToast } from '../../../utils/ToastUtils';
import { UrlParams } from '../ExplorePage.interface';
import {
@@ -136,7 +135,10 @@ const ExploreTree = ({ onFieldValueSelect }: ExploreTreeProps) => {
let logo = undefined;
if (isEntityType) {
- logo = getEntityIcon(bucket.key, 'service-icon w-4 h-4') ?? <>>;
+ logo = searchClassBase.getEntityIcon(
+ bucket.key,
+ 'service-icon w-4 h-4'
+ ) ?? <>>;
} else if (isServiceType) {
const serviceIcon = serviceUtilClassBase.getServiceLogo(bucket.key);
logo = (
diff --git a/openmetadata-ui/src/main/resources/ui/src/components/ExploreV1/ExploreSearchCard/ExploreSearchCard.tsx b/openmetadata-ui/src/main/resources/ui/src/components/ExploreV1/ExploreSearchCard/ExploreSearchCard.tsx
index d505a1bf097..a62129188f7 100644
--- a/openmetadata-ui/src/main/resources/ui/src/components/ExploreV1/ExploreSearchCard/ExploreSearchCard.tsx
+++ b/openmetadata-ui/src/main/resources/ui/src/components/ExploreV1/ExploreSearchCard/ExploreSearchCard.tsx
@@ -33,7 +33,7 @@ import { getEntityId, getEntityName } from '../../../utils/EntityUtils';
import { getDomainPath } from '../../../utils/RouterUtils';
import searchClassBase from '../../../utils/SearchClassBase';
import { stringToHTML } from '../../../utils/StringsUtils';
-import { getEntityIcon, getUsagePercentile } from '../../../utils/TableUtils';
+import { getUsagePercentile } from '../../../utils/TableUtils';
import TitleBreadcrumb from '../../common/TitleBreadcrumb/TitleBreadcrumb.component';
import TableDataCardBody from '../../Database/TableDataCardBody/TableDataCardBody';
import { GlossaryStatusBadge } from '../../Glossary/GlossaryStatusBadge/GlossaryStatusBadge.component';
@@ -166,13 +166,13 @@ const ExploreSearchCard: React.FC = forwardRef<
return (
- {getEntityIcon(source.entityType ?? '')}
+ {searchClassBase.getEntityIcon(source.entityType ?? '')}
);
}
return;
- }, [source, showEntityIcon, getEntityIcon]);
+ }, [source, showEntityIcon]);
const entityLink = useMemo(
() => searchClassBase.getEntityLink(source),
diff --git a/openmetadata-ui/src/main/resources/ui/src/components/MyData/MyDataWidget/MyDataWidget.component.tsx b/openmetadata-ui/src/main/resources/ui/src/components/MyData/MyDataWidget/MyDataWidget.component.tsx
index e659fecc3ae..5779dc6bf05 100644
--- a/openmetadata-ui/src/main/resources/ui/src/components/MyData/MyDataWidget/MyDataWidget.component.tsx
+++ b/openmetadata-ui/src/main/resources/ui/src/components/MyData/MyDataWidget/MyDataWidget.component.tsx
@@ -30,9 +30,9 @@ import { searchData } from '../../../rest/miscAPI';
import { Transi18next } from '../../../utils/CommonUtils';
import entityUtilClassBase from '../../../utils/EntityUtilClassBase';
import { getEntityName } from '../../../utils/EntityUtils';
-import { getEntityIcon } from '../../../utils/TableUtils';
import { useApplicationStore } from '../../../hooks/useApplicationStore';
+import searchClassBase from '../../../utils/SearchClassBase';
import ErrorPlaceHolder from '../../common/ErrorWithPlaceholder/ErrorPlaceHolder';
import EntityListSkeleton from '../../common/Skeleton/MyData/EntityListSkeleton/EntityListSkeleton.component';
import { SourceType } from '../../SearchedData/SearchedData.interface';
@@ -172,7 +172,9 @@ const MyDataWidgetInternal = ({
className="entity-button flex-center p-0 m--ml-1"
icon={
- {getEntityIcon(item.entityType ?? '')}
+ {searchClassBase.getEntityIcon(
+ item.entityType ?? ''
+ )}
}
type="text">
diff --git a/openmetadata-ui/src/main/resources/ui/src/components/MyData/MyDataWidget/MyDataWidget.test.tsx b/openmetadata-ui/src/main/resources/ui/src/components/MyData/MyDataWidget/MyDataWidget.test.tsx
index 7eb0b7da924..1090e8ea9c0 100644
--- a/openmetadata-ui/src/main/resources/ui/src/components/MyData/MyDataWidget/MyDataWidget.test.tsx
+++ b/openmetadata-ui/src/main/resources/ui/src/components/MyData/MyDataWidget/MyDataWidget.test.tsx
@@ -63,7 +63,7 @@ jest.mock('../../../utils/EntityUtils', () => ({
getEntityName: jest.fn().mockImplementation((obj) => obj.name),
}));
-jest.mock('../../../utils/TableUtils', () => ({
+jest.mock('../../../utils/SearchClassBase', () => ({
getEntityIcon: jest.fn().mockImplementation((obj) => obj.name),
}));
diff --git a/openmetadata-ui/src/main/resources/ui/src/components/MyData/Widgets/RecentlyViewed/RecentlyViewed.tsx b/openmetadata-ui/src/main/resources/ui/src/components/MyData/Widgets/RecentlyViewed/RecentlyViewed.tsx
index ec6f32d951b..81b79866a27 100644
--- a/openmetadata-ui/src/main/resources/ui/src/components/MyData/Widgets/RecentlyViewed/RecentlyViewed.tsx
+++ b/openmetadata-ui/src/main/resources/ui/src/components/MyData/Widgets/RecentlyViewed/RecentlyViewed.tsx
@@ -28,7 +28,7 @@ import {
} from '../../../../utils/CommonUtils';
import entityUtilClassBase from '../../../../utils/EntityUtilClassBase';
import { getEntityName } from '../../../../utils/EntityUtils';
-import { getEntityIcon } from '../../../../utils/TableUtils';
+import searchClassBase from '../../../../utils/SearchClassBase';
import ErrorPlaceHolder from '../../../common/ErrorWithPlaceholder/ErrorPlaceHolder';
import EntityListSkeleton from '../../../common/Skeleton/MyData/EntityListSkeleton/EntityListSkeleton.component';
import './recently-viewed.less';
@@ -143,7 +143,7 @@ const RecentlyViewed = ({
className="entity-button flex-center p-0 m--ml-1"
icon={
- {getEntityIcon(item.type || '')}
+ {searchClassBase.getEntityIcon(item.type ?? '')}
}
title={getEntityName(
diff --git a/openmetadata-ui/src/main/resources/ui/src/components/common/CustomPropertyTable/PropertyValue.tsx b/openmetadata-ui/src/main/resources/ui/src/components/common/CustomPropertyTable/PropertyValue.tsx
index 54b25fdf250..9c4e54d84db 100644
--- a/openmetadata-ui/src/main/resources/ui/src/components/common/CustomPropertyTable/PropertyValue.tsx
+++ b/openmetadata-ui/src/main/resources/ui/src/components/common/CustomPropertyTable/PropertyValue.tsx
@@ -50,7 +50,7 @@ import { EntityReference } from '../../../generated/entity/type';
import { EnumConfig } from '../../../generated/type/customProperty';
import entityUtilClassBase from '../../../utils/EntityUtilClassBase';
import { getEntityName } from '../../../utils/EntityUtils';
-import { getEntityIcon } from '../../../utils/TableUtils';
+import searchClassBase from '../../../utils/SearchClassBase';
import { showErrorToast } from '../../../utils/ToastUtils';
import DataAssetAsyncSelectList from '../../DataAssets/DataAssetAsyncSelectList/DataAssetAsyncSelectList';
import { DataAssetOption } from '../../DataAssets/DataAssetAsyncSelectList/DataAssetAsyncSelectList.interface';
@@ -702,7 +702,7 @@ export const PropertyValue: FC = ({
width="18"
/>
) : (
- getEntityIcon(item.type)
+ searchClassBase.getEntityIcon(item.type)
)}
}
@@ -753,7 +753,7 @@ export const PropertyValue: FC
= ({
width="18"
/>
) : (
- getEntityIcon(item.type)
+ searchClassBase.getEntityIcon(item.type)
)}
}
diff --git a/openmetadata-ui/src/main/resources/ui/src/constants/Assets.constants.ts b/openmetadata-ui/src/main/resources/ui/src/constants/Assets.constants.ts
index 7a432865ef2..046c1a404a0 100644
--- a/openmetadata-ui/src/main/resources/ui/src/constants/Assets.constants.ts
+++ b/openmetadata-ui/src/main/resources/ui/src/constants/Assets.constants.ts
@@ -178,4 +178,5 @@ export const NON_SERVICE_TYPE_ASSETS = [
EntityType.PERSONA,
EntityType.ROLE,
EntityType.POLICY,
+ EntityType.KPI,
];
diff --git a/openmetadata-ui/src/main/resources/ui/src/pages/AlertDetailsPage/AlertDetailsPage.tsx b/openmetadata-ui/src/main/resources/ui/src/pages/AlertDetailsPage/AlertDetailsPage.tsx
index 6bcf4adb3a3..36f34d4f6e3 100644
--- a/openmetadata-ui/src/main/resources/ui/src/pages/AlertDetailsPage/AlertDetailsPage.tsx
+++ b/openmetadata-ui/src/main/resources/ui/src/pages/AlertDetailsPage/AlertDetailsPage.tsx
@@ -55,7 +55,7 @@ import {
getObservabilityAlertsEditPath,
getSettingPath,
} from '../../utils/RouterUtils';
-import { getEntityIcon } from '../../utils/TableUtils';
+import searchClassBase from '../../utils/SearchClassBase';
import { AlertDetailsPageProps } from './AlertDetailsPage.interface';
function AlertDetailsPage({
@@ -366,7 +366,7 @@ function AlertDetailsPage({
subHeading={t('message.alerts-source-description')}>