From b6af240e97d58dd2ed32f0a89ede621ec9b02b0c Mon Sep 17 00:00:00 2001
From: ryota-cloud <ryota.egashira@acryl.io>
Date: Mon, 31 Mar 2025 18:16:20 -0700
Subject: [PATCH] fix(ui) remove SQL from TaskIcon for Vertex AI Pipeline task 
 (#13024)

Co-authored-by: Andrew Sikowitz <andrew.sikowitz@acryl.io>
---
 .../src/app/entityV2/shared/components/subtypes.tsx          | 5 +++++
 datahub-web-react/src/images/vertexai-pipeline-task.svg      | 3 +++
 2 files changed, 8 insertions(+)
 create mode 100644 datahub-web-react/src/images/vertexai-pipeline-task.svg

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 <LineChartOutlined className={TYPE_ICON_CLASS_NAME} />;
     }
+    if (lowerSubType === SubType.VertexAIPipelineTask.toLowerCase()) {
+        return <Icon component={VertexAIPipelineTaskLogo} className={TYPE_ICON_CLASS_NAME} />;
+    }
     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 @@
+<svg width="1em" height="1em" viewBox="0 0 18 18" fit="" preserveAspectRatio="xMidYMid meet">
+    <path fill="currentColor" d="M2.8125,6.49498705 L8.18741924,9.56954253 L8.18741924,15.75 L2.8125,12.6691394 L2.8125,6.49498705 Z M14.9625,6.48040046 L14.9625,12.6720259 L9.59178711,15.75 L9.59178711,9.55502167 L14.9625,6.48040046 Z M8.94505902,2.25 L14.254385,5.33084726 L8.94505902,8.37364036 L3.52352226,5.28994599 L8.94505902,2.25 Z"/>
+</svg>