diff --git a/datahub-web-react/src/app/entityV2/shared/components/subtypes.tsx b/datahub-web-react/src/app/entityV2/shared/components/subtypes.tsx
index 7f74d52077..dffe59727f 100644
--- a/datahub-web-react/src/app/entityV2/shared/components/subtypes.tsx
+++ b/datahub-web-react/src/app/entityV2/shared/components/subtypes.tsx
@@ -10,6 +10,7 @@ import ViewComfyOutlinedIcon from '@mui/icons-material/ViewComfyOutlined';
import TableauWorkbookLogo from '../../../../images/tableau-workbook.svg?react';
import TableauEmbeddedDataSourceLogo from '../../../../images/tableau-embedded-data-source.svg?react';
import TableauPublishedDataSourceLogo from '../../../../images/tableau-published-data-source.svg?react';
+import VertexAIPipelineTaskLogo from '../../../../images/vertexai-pipeline-task.svg?react';
export enum SubType {
Database = 'Database',
@@ -25,6 +26,7 @@ export enum SubType {
LookerExplore = 'Explore',
Looker = 'Look',
DbtSource = 'Source',
+ VertexAIPipelineTask = 'Pipeline Task',
}
export const TYPE_ICON_CLASS_NAME = 'typeIcon';
@@ -62,5 +64,8 @@ export function getSubTypeIcon(subType?: string): JSX.Element | undefined {
if (lowerSubType === SubType.TableauWorksheet.toLowerCase()) {
return ;
}
+ if (lowerSubType === SubType.VertexAIPipelineTask.toLowerCase()) {
+ return ;
+ }
return undefined;
}
diff --git a/datahub-web-react/src/images/vertexai-pipeline-task.svg b/datahub-web-react/src/images/vertexai-pipeline-task.svg
new file mode 100644
index 0000000000..660e0db5df
--- /dev/null
+++ b/datahub-web-react/src/images/vertexai-pipeline-task.svg
@@ -0,0 +1,3 @@
+